|
|
|
#include <RPCSeedOverlapper.h>
Public Member Functions | |
| void | configure (const edm::ParameterSet &iConfig) |
| RPCSeedOverlapper () | |
| void | run () |
| void | setEventSetup (const edm::EventSetup &iSetup) |
| void | setIO (std::vector< weightedTrajectorySeed > *goodweightedRef, std::vector< weightedTrajectorySeed > *candidateweightedRef) |
| void | unsetIO () |
| ~RPCSeedOverlapper () | |
Private Types | |
| typedef RPCSeedPattern::weightedTrajectorySeed | weightedTrajectorySeed |
Private Member Functions | |
| void | CheckOverlap (const edm::EventSetup &iSetup, std::vector< weightedTrajectorySeed > *SeedsRef) |
| bool | isShareHit (const edm::OwnVector< TrackingRecHit > &RecHits, const TrackingRecHit &hit, edm::ESHandle< RPCGeometry > rpcGeometry) |
Private Attributes | |
| std::vector < weightedTrajectorySeed > * | candidateweightedSeedsRef |
| const edm::EventSetup * | eSetup |
| std::vector < weightedTrajectorySeed > * | goodweightedSeedsRef |
| bool | isCheckcandidateOverlap |
| bool | isCheckgoodOverlap |
| bool | isConfigured |
| bool | isEventSetupset |
| bool | isIOset |
| unsigned int | ShareRecHitsNumberThreshold |
Definition at line 21 of file RPCSeedOverlapper.h.
Definition at line 23 of file RPCSeedOverlapper.h.
| RPCSeedOverlapper::RPCSeedOverlapper | ( | ) |
Definition at line 14 of file RPCSeedOverlapper.cc.
{
isConfigured = false;
isIOset = false;
isEventSetupset = false;
}
| RPCSeedOverlapper::~RPCSeedOverlapper | ( | ) |
Definition at line 21 of file RPCSeedOverlapper.cc.
{
}
| void RPCSeedOverlapper::CheckOverlap | ( | const edm::EventSetup & | iSetup, |
| std::vector< weightedTrajectorySeed > * | SeedsRef | ||
| ) | [private] |
Definition at line 63 of file RPCSeedOverlapper.cc.
References edm::OwnVector< T, P >::clear(), gather_cfg::cout, edm::EventSetup::get(), N, edm::OwnVector< T, P >::push_back(), and edm::OwnVector< T, P >::size().
{
std::vector<weightedTrajectorySeed> sortweightedSeeds;
std::vector<weightedTrajectorySeed> tempweightedSeeds;
edm::OwnVector<TrackingRecHit> tempRecHits;
edm::ESHandle<RPCGeometry> rpcGeometry;
iSetup.get<MuonGeometryRecord>().get(rpcGeometry);
while(weightedSeedsRef->size() != 0) {
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;
}
| void RPCSeedOverlapper::configure | ( | const edm::ParameterSet & | iConfig | ) |
Definition at line 25 of file RPCSeedOverlapper.cc.
References edm::ParameterSet::getParameter().
{
isCheckgoodOverlap = iConfig.getParameter<bool>("isCheckgoodOverlap");
isCheckcandidateOverlap = iConfig.getParameter<bool>("isCheckcandidateOverlap");
ShareRecHitsNumberThreshold = iConfig.getParameter<unsigned int>("ShareRecHitsNumberThreshold");
isConfigured = true;
}
| bool RPCSeedOverlapper::isShareHit | ( | const edm::OwnVector< TrackingRecHit > & | RecHits, |
| const TrackingRecHit & | hit, | ||
| edm::ESHandle< RPCGeometry > | rpcGeometry | ||
| ) | [private] |
Definition at line 168 of file RPCSeedOverlapper.cc.
References edm::OwnVector< T, P >::begin(), gather_cfg::cout, edm::OwnVector< T, P >::end(), TrackingRecHit::geographicalId(), TrackingRecHit::localPosition(), n, edm::OwnVector< T, P >::size(), GeomDet::toGlobal(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().
{
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;
}
| void RPCSeedOverlapper::run | ( | void | ) |
Definition at line 51 of file RPCSeedOverlapper.cc.
References gather_cfg::cout.
{
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::setEventSetup | ( | const edm::EventSetup & | iSetup | ) |
Definition at line 45 of file RPCSeedOverlapper.cc.
{
eSetup = &iSetup;
isEventSetupset = true;
}
| void RPCSeedOverlapper::setIO | ( | std::vector< weightedTrajectorySeed > * | goodweightedRef, |
| std::vector< weightedTrajectorySeed > * | candidateweightedRef | ||
| ) |
Definition at line 33 of file RPCSeedOverlapper.cc.
{
goodweightedSeedsRef = goodweightedRef;
candidateweightedSeedsRef = candidateweightedRef;
isIOset = true;
}
| void RPCSeedOverlapper::unsetIO | ( | ) |
Definition at line 40 of file RPCSeedOverlapper.cc.
{
isIOset = false;
}
std::vector<weightedTrajectorySeed>* RPCSeedOverlapper::candidateweightedSeedsRef [private] |
Definition at line 46 of file RPCSeedOverlapper.h.
const edm::EventSetup* RPCSeedOverlapper::eSetup [private] |
Definition at line 47 of file RPCSeedOverlapper.h.
std::vector<weightedTrajectorySeed>* RPCSeedOverlapper::goodweightedSeedsRef [private] |
Definition at line 45 of file RPCSeedOverlapper.h.
bool RPCSeedOverlapper::isCheckcandidateOverlap [private] |
Definition at line 42 of file RPCSeedOverlapper.h.
bool RPCSeedOverlapper::isCheckgoodOverlap [private] |
Definition at line 41 of file RPCSeedOverlapper.h.
bool RPCSeedOverlapper::isConfigured [private] |
Definition at line 37 of file RPCSeedOverlapper.h.
bool RPCSeedOverlapper::isEventSetupset [private] |
Definition at line 39 of file RPCSeedOverlapper.h.
bool RPCSeedOverlapper::isIOset [private] |
Definition at line 38 of file RPCSeedOverlapper.h.
unsigned int RPCSeedOverlapper::ShareRecHitsNumberThreshold [private] |
Definition at line 43 of file RPCSeedOverlapper.h.
1.7.3