RPCSeedOverlapper.cc

Go to the documentation of this file.

#include "RecoMuon/MuonSeedGenerator/src/RPCSeedOverlapper.h"
#include <Geometry/Records/interface/MuonGeometryRecord.h>
#include <Geometry/CommonDetUnit/interface/GeomDet.h>

using namespace std;
using namespace edm;

RPCSeedOverlapper::RPCSeedOverlapper() {
  isConfigured = false;
  isIOset = false;
  isEventSetupset = false;
}

RPCSeedOverlapper::~RPCSeedOverlapper() {}

void RPCSeedOverlapper::configure(const edm::ParameterSet &iConfig) {
  isCheckgoodOverlap = iConfig.getParameter<bool>("isCheckgoodOverlap");
  isCheckcandidateOverlap = iConfig.getParameter<bool>("isCheckcandidateOverlap");
  ShareRecHitsNumberThreshold = iConfig.getParameter<unsigned int>("ShareRecHitsNumberThreshold");
  isConfigured = true;
}

void RPCSeedOverlapper::setIO(std::vector<weightedTrajectorySeed> *goodweightedRef,
                              std::vector<weightedTrajectorySeed> *candidateweightedRef) {
  goodweightedSeedsRef = goodweightedRef;
  candidateweightedSeedsRef = candidateweightedRef;
  isIOset = true;
}

void RPCSeedOverlapper::unsetIO() { isIOset = false; }

void RPCSeedOverlapper::setEventSetup(const edm::EventSetup &iSetup) {
  eSetup = &iSetup;
  isEventSetupset = true;
}

void RPCSeedOverlapper::run() {
  if (isConfigured == false || isIOset == false || isEventSetupset == false) {
    cout << "Configuration or IO is not set yet" << endl;
    return;
  }
  if (isCheckgoodOverlap == true)
    CheckOverlap(*eSetup, goodweightedSeedsRef);
  if (isCheckcandidateOverlap == true)
    CheckOverlap(*eSetup, candidateweightedSeedsRef);
}

void RPCSeedOverlapper::CheckOverlap(const edm::EventSetup &iSetup,
                                     std::vector<weightedTrajectorySeed> *weightedSeedsRef) {
  std::vector<weightedTrajectorySeed> sortweightedSeeds;
  std::vector<weightedTrajectorySeed> tempweightedSeeds;
  edm::OwnVector<TrackingRecHit> tempRecHits;

  edm::ESHandle<RPCGeometry> rpcGeometry;
  iSetup.get<MuonGeometryRecord>().get(rpcGeometry);

  while (!weightedSeedsRef->empty()) {
    cout << "Finding the weighted seeds group from " << weightedSeedsRef->size() << " seeds which share some recHits"
         << endl;
    // Take 1st seed in SeedsRef as referrence and find a collection which always share some recHits with some other
    tempRecHits.clear();
    tempweightedSeeds.clear();
    int N = 0;
    for (vector<weightedTrajectorySeed>::iterator itweightedseed = weightedSeedsRef->begin();
         itweightedseed != weightedSeedsRef->end();
         N++) {
      TrajectorySeed::range RecHitsRange = itweightedseed->first.recHits();
      if (N == 0) {
        cout << "Always take the 1st weighted seed to be the referrence." << endl;
        for (TrajectorySeed::const_iterator it = RecHitsRange.first; it != RecHitsRange.second; it++) {
          cout << "Put its recHits to tempRecHits" << endl;
          tempRecHits.push_back(it->clone());
        }
        cout << "Put it to tempweightedSeeds" << endl;
        tempweightedSeeds.push_back(*itweightedseed);
        cout << "Then erase from weightedSeedsRef->" << endl;
        itweightedseed = weightedSeedsRef->erase(itweightedseed);
      } else {
        cout << "Come to other weighted seed for checking " << itweightedseed->first.nHits() << " recHits from "
             << tempRecHits.size() << " temp recHits" << endl;
        unsigned int ShareRecHitsNumber = 0;
        for (TrajectorySeed::const_iterator it = RecHitsRange.first; it != RecHitsRange.second; it++) {
          if (isShareHit(tempRecHits, *it, rpcGeometry))
            ShareRecHitsNumber++;
        }
        if (ShareRecHitsNumber >= ShareRecHitsNumberThreshold) {
          cout << "This seed is found to belong to current share group" << endl;
          for (TrajectorySeed::const_iterator it = RecHitsRange.first; it != RecHitsRange.second; it++) {
            if (!isShareHit(tempRecHits, *it, rpcGeometry)) {
              cout << "Put its extra recHits to tempRecHits" << endl;
              tempRecHits.push_back(it->clone());
            }
          }
          cout << "Put it to tempSeeds" << endl;
          tempweightedSeeds.push_back(*itweightedseed);
          cout << "Then erase from SeedsRef" << endl;
          itweightedseed = weightedSeedsRef->erase(itweightedseed);
        } else
          itweightedseed++;
      }
    }
    // Find the best weighted seed and kick out those share recHits with it
    // The best weighted seed save in sortweightedSeeds, those don't share recHits with it will be push back to weightedSeedsRef for next while loop
    weightedTrajectorySeed bestweightedSeed;
    vector<weightedTrajectorySeed>::iterator bestweightediter;
    // Find the min Spt wrt Pt as the best Seed
    double Quality = 1000000;
    unsigned NumberofHits = 0;
    cout << "Find " << tempweightedSeeds.size() << " seeds into one trajectory group" << endl;
    for (vector<weightedTrajectorySeed>::iterator itweightedseed = tempweightedSeeds.begin();
         itweightedseed != tempweightedSeeds.end();
         itweightedseed++) {
      unsigned int nHits = itweightedseed->first.nHits();
      //std::vector<float> seed_error = itweightedseed->first.startingState().errorMatrix();
      //double Spt = seed_error[1];
      double weightedQuality = itweightedseed->second;
      cout << "Find a weighted seed with quality " << weightedQuality << endl;
      if ((NumberofHits < nHits) || (NumberofHits == nHits && weightedQuality < Quality)) {
        NumberofHits = nHits;
        Quality = weightedQuality;
        bestweightedSeed = *itweightedseed;
        bestweightediter = itweightedseed;
      }
    }
    cout << "Best good temp seed's quality is " << Quality << endl;
    sortweightedSeeds.push_back(bestweightedSeed);
    tempweightedSeeds.erase(bestweightediter);
    tempRecHits.clear();

    for (TrajectorySeed::const_iterator it = bestweightedSeed.first.recHits().first;
         it != bestweightedSeed.first.recHits().second;
         it++)
      tempRecHits.push_back(it->clone());

    for (vector<weightedTrajectorySeed>::iterator itweightedseed = tempweightedSeeds.begin();
         itweightedseed != tempweightedSeeds.end();) {
      cout << "Checking the temp weighted seed's " << itweightedseed->first.nHits() << " hits to " << tempRecHits.size()
           << " temp recHits" << endl;
      TrajectorySeed::range RecHitsRange = itweightedseed->first.recHits();
      bool isShare = false;
      for (TrajectorySeed::const_iterator it = RecHitsRange.first; it != RecHitsRange.second; it++)
        if (isShareHit(tempRecHits, *it, rpcGeometry))
          isShare = true;

      if (isShare == true) {
        cout << "Find one temp seed share some recHits with best weighted seed" << endl;
        itweightedseed = tempweightedSeeds.erase(itweightedseed);
      } else {
        cout << "This seed has no relation with best weighted seed" << endl;
        weightedSeedsRef->push_back(*itweightedseed);
        itweightedseed = tempweightedSeeds.erase(itweightedseed);
      }
    }
  }
  // At the end exchange SeedsRef with sortSeeds
  weightedSeedsRef->clear();
  *weightedSeedsRef = sortweightedSeeds;
}

bool RPCSeedOverlapper::isShareHit(const edm::OwnVector<TrackingRecHit> &RecHits,
                                   const TrackingRecHit &hit,
                                   edm::ESHandle<RPCGeometry> rpcGeometry) {
  bool istheSame = false;
  unsigned int n = 1;
  cout << "Checking from " << RecHits.size() << " temp recHits" << endl;

  LocalPoint lpos1 = hit.localPosition();
  DetId RPCId1 = hit.geographicalId();
  const GeomDetUnit *rpcroll1 = rpcGeometry->idToDetUnit(RPCId1);
  GlobalPoint gpos1 = rpcroll1->toGlobal(lpos1);
  cout << "The hit's position: " << gpos1.x() << ", " << gpos1.y() << ", " << gpos1.z() << endl;
  for (edm::OwnVector<TrackingRecHit>::const_iterator it = RecHits.begin(); it != RecHits.end(); it++, n++) {
    cout << "Checking the " << n << " th recHit from tempRecHits" << endl;
    LocalPoint lpos2 = it->localPosition();
    DetId RPCId2 = it->geographicalId();
    const GeomDetUnit *rpcroll2 = rpcGeometry->idToDetUnit(RPCId2);
    GlobalPoint gpos2 = rpcroll2->toGlobal(lpos2);
    cout << "The temp hit's position: " << gpos2.x() << ", " << gpos2.y() << ", " << gpos2.z() << endl;

    if ((gpos1.x() == gpos2.x()) && (gpos1.y() == gpos2.y()) && (gpos1.z() == gpos2.z())) {
      cout << "This hit is found to be the same" << endl;
      istheSame = true;
    }
  }
  return istheSame;
}