|
|
|
#include <VertexAssociatorByTracks.h>
Public Member Functions | |
| reco::VertexRecoToSimCollection | associateRecoToSim (edm::Handle< edm::View< reco::Vertex > > &, edm::Handle< TrackingVertexCollection > &, const edm::Event &, reco::RecoToSimCollection &) const |
| reco::VertexSimToRecoCollection | associateSimToReco (edm::Handle< edm::View< reco::Vertex > > &, edm::Handle< TrackingVertexCollection > &, const edm::Event &, reco::SimToRecoCollection &) const |
| VertexAssociatorByTracks (const edm::ParameterSet &) | |
| ~VertexAssociatorByTracks () | |
Private Attributes | |
| const edm::ParameterSet & | config_ |
| double | R2SMatchedRecoRatio_ |
| double | R2SMatchedSimRatio_ |
| double | S2RMatchedRecoRatio_ |
| double | S2RMatchedSimRatio_ |
| TrackingParticleSelector | selector_ |
| reco::TrackBase::TrackQuality | trackQuality_ |
Definition at line 11 of file VertexAssociatorByTracks.h.
| VertexAssociatorByTracks::VertexAssociatorByTracks | ( | const edm::ParameterSet & | config | ) | [explicit] |
Definition at line 24 of file VertexAssociatorByTracks.cc.
References edm::ParameterSet::getParameter(), reco::TrackBase::qualityByName(), R2SMatchedRecoRatio_, R2SMatchedSimRatio_, S2RMatchedRecoRatio_, S2RMatchedSimRatio_, selector_, and trackQuality_.
: config_(config) { R2SMatchedSimRatio_ = config.getParameter<double>("R2SMatchedSimRatio"); R2SMatchedRecoRatio_ = config.getParameter<double>("R2SMatchedRecoRatio"); S2RMatchedSimRatio_ = config.getParameter<double>("S2RMatchedSimRatio"); S2RMatchedRecoRatio_ = config.getParameter<double>("S2RMatchedRecoRatio"); std::string trackQualityType = config.getParameter<std::string>("trackQuality"); trackQuality_ = reco::TrackBase::qualityByName(trackQualityType); edm::ParameterSet param = config.getParameter<edm::ParameterSet>("trackingParticleSelector"); selector_ = TrackingParticleSelector( param.getParameter<double>("ptMinTP"), param.getParameter<double>("minRapidityTP"), param.getParameter<double>("maxRapidityTP"), param.getParameter<double>("tipTP"), param.getParameter<double>("lipTP"), param.getParameter<int>("minHitTP"), param.getParameter<bool>("signalOnlyTP"), param.getParameter<bool>("chargedOnlyTP"), param.getParameter<bool>("stableOnlyTP"), param.getParameter<std::vector<int> >("pdgIdTP") ); }
| VertexAssociatorByTracks::~VertexAssociatorByTracks | ( | ) |
Definition at line 52 of file VertexAssociatorByTracks.cc.
{
//do cleanup here
}
| reco::VertexRecoToSimCollection VertexAssociatorByTracks::associateRecoToSim | ( | edm::Handle< edm::View< reco::Vertex > > & | recoVertexes, |
| edm::Handle< TrackingVertexCollection > & | trackingVertexes, | ||
| const edm::Event & | event, | ||
| reco::RecoToSimCollection & | associator | ||
| ) | const [virtual] |
Implements VertexAssociatorBase.
Definition at line 58 of file VertexAssociatorByTracks.cc.
References gather_cfg::cout, edm::AssociationMap< Tag >::insert(), Association::map, match(), edm::AssociationMap< Tag >::numberOfAssociations(), R2SMatchedRecoRatio_, R2SMatchedSimRatio_, selector_, edm::AssociationMap< Tag >::size(), and trackQuality_.
{
reco::VertexRecoToSimCollection outputCollection;
std::map<TrackingVertexRef,std::pair<double, std::size_t> > matches;
std::cout << "reco::VertexCollection size = " << recoVertexes->size();
std::cout << " ; TrackingVertexCollection size = " << trackingVertexes->size() << std::endl << std::endl;
// Loop over RecoVertex
for (std::size_t recoIndex = 0; recoIndex < recoVertexes->size(); ++recoIndex)
{
reco::VertexBaseRef recoVertex(recoVertexes, recoIndex);
matches.clear();
double recoDaughterWeight = 0.;
// Loop over daughter tracks of RecoVertex
for (
reco::Vertex::trackRef_iterator recoDaughter = recoVertex->tracks_begin();
recoDaughter != recoVertex->tracks_end();
++recoDaughter
)
{
// Check the quality of the RecoDoughters
if ( !(*recoDaughter)->quality(trackQuality_) ) continue;
// Check for association for the given RecoDaughter
if ( associator.numberOfAssociations(*recoDaughter) > 0 )
{
std::vector<std::pair<TrackingParticleRef,double> > associations = associator[*recoDaughter];
// Loop over TrackingParticles associated with RecoDaughter
for (
std::vector<std::pair<TrackingParticleRef,double> >::const_iterator association = associations.begin();
association != associations.end();
++association
)
{
// Get a reference to parent vertex of TrackingParticle associated to the RecoDaughter
TrackingVertexRef trackingVertex = association->first->parentVertex();
// Store matched RecoDaughter to the trackingVertex
matches[trackingVertex].first += recoVertex->trackWeight(*recoDaughter);
matches[trackingVertex].second++;
}
}
recoDaughterWeight += recoVertex->trackWeight(*recoDaughter);
} //loop over daughter tracks of RecoVertex
std::size_t assoIndex = 0;
// Loop over map between TrackingVertexes and matched RecoDaugther
for (
std::map<TrackingVertexRef,std::pair<double, std::size_t> >::const_iterator match = matches.begin();
match != matches.end();
++match
)
{
// Getting the TrackingVertex information
TrackingVertexRef trackingVertex = match->first;
double matchedDaughterWeight = match->second.first;
std::size_t matchedDaughterCounter = match->second.second;
// Count for only reconstructible SimDaughters
std::size_t simDaughterCounter = 0;
for (
TrackingVertex::tp_iterator simDaughter = trackingVertex->daughterTracks_begin();
simDaughter != trackingVertex->daughterTracks_end();
++simDaughter
)
if ( selector_(**simDaughter) ) simDaughterCounter++;
// Sanity condition in case that reconstructable condition is too tight
if ( simDaughterCounter < matchedDaughterCounter )
simDaughterCounter = matchedDaughterCounter;
// Condition over S2RMatchedSimRatio
if ( (double)matchedDaughterCounter/simDaughterCounter < R2SMatchedSimRatio_ ) continue;
double quality = (double)matchedDaughterWeight/recoDaughterWeight;
// Condition over R2SMatchedRecoRatio
if (quality < R2SMatchedRecoRatio_) continue;
outputCollection.insert(recoVertex, std::make_pair(trackingVertex, quality));
std::cout << "R2S: INDEX " << assoIndex;
std::cout << " ; SimDaughterCounter = " << simDaughterCounter;
std::cout << " ; RecoDaughterWeight = " << recoDaughterWeight;
std::cout << " ; MatchedDaughterCounter = " << matchedDaughterCounter;
std::cout << " ; MatchedDaughterWeight = " << matchedDaughterWeight;
std::cout << " ; quality = " << quality << std::endl;
assoIndex++;
}
} // Loop on RecoVertex
std::cout << std::endl;
std::cout << "RecoToSim OUTPUT COLLECTION: outputCollection.size() = " << outputCollection.size() << std::endl << std::endl;
return outputCollection;
}
| reco::VertexSimToRecoCollection VertexAssociatorByTracks::associateSimToReco | ( | edm::Handle< edm::View< reco::Vertex > > & | recoVertexes, |
| edm::Handle< TrackingVertexCollection > & | trackingVertexes, | ||
| const edm::Event & | event, | ||
| reco::SimToRecoCollection & | associator | ||
| ) | const [virtual] |
Implements VertexAssociatorBase.
Definition at line 171 of file VertexAssociatorByTracks.cc.
References gather_cfg::cout, edm::RefToBase< T >::id(), edm::AssociationMap< Tag >::insert(), edm::RefToBase< T >::key(), Association::map, match(), edm::AssociationMap< Tag >::numberOfAssociations(), S2RMatchedRecoRatio_, S2RMatchedSimRatio_, selector_, edm::AssociationMap< Tag >::size(), and trackQuality_.
{
reco::VertexSimToRecoCollection outputCollection;
// Loop over TrackingVertexes
std::map<std::size_t,std::pair<double, std::size_t> > matches;
// Loop over TrackingVertexes
for (std::size_t simIndex = 0; simIndex < trackingVertexes->size(); ++simIndex)
{
TrackingVertexRef trackingVertex(trackingVertexes, simIndex);
matches.clear();
std::size_t simDaughterCounter = 0;
// Loop over daughter tracks of TrackingVertex
for (
TrackingVertex::tp_iterator simDaughter = trackingVertex->daughterTracks_begin();
simDaughter != trackingVertex->daughterTracks_end();
++simDaughter
)
{
// Select only reconstructible SimDaughters
if ( !selector_(**simDaughter) ) continue;
// Check for association for the given RecoDaughter
if ( associator.numberOfAssociations(*simDaughter) > 0 )
{
std::vector<std::pair<reco::TrackBaseRef, double> > associations = associator[*simDaughter];
// Loop over RecoTracks associated with TrackingParticle
for (
std::vector<std::pair<reco::TrackBaseRef,double> >::const_iterator association = associations.begin();
association != associations.end();
++association
)
{
reco::TrackBaseRef recoTrack = association->first;
for (std::size_t recoIndex = 0; recoIndex < recoVertexes->size(); ++recoIndex)
{
reco::VertexBaseRef recoVertex(recoVertexes, recoIndex);
for (
reco::Vertex::trackRef_iterator recoDaughter = recoVertex->tracks_begin();
recoDaughter != recoVertex->tracks_end();
++recoDaughter
)
{
// Store matched RecoDaughter to the RecoVertex
if (
recoDaughter->id() == recoTrack.id() &&
recoDaughter->key() == recoTrack.key()
)
{
matches[recoIndex].first += recoVertex->trackWeight(*recoDaughter);
matches[recoIndex].second++;
}
}
}
} // loop a recotracks
}
simDaughterCounter++;
} // loop a simDaughter
std::size_t assoIndex = 0;
// Loop over map, set score, add to outputCollection
for (
std::map<std::size_t,std::pair<double,std::size_t> >::const_iterator match = matches.begin();
match != matches.end();
++match
)
{
// Getting the TrackingVertex information
reco::VertexBaseRef recoVertex(recoVertexes, match->first);
double matchedDaughterWeight = match->second.first;
std::size_t matchedDaughterCounter = match->second.second;
double recoDaughterWeight = 0.;
// Weighted count of those tracks with a given quality of the RecoDoughters
for (
reco::Vertex::trackRef_iterator recoDaughter = recoVertex->tracks_begin();
recoDaughter != recoVertex->tracks_end();
++recoDaughter
)
if ( (*recoDaughter)->quality(trackQuality_) )
recoDaughterWeight += recoVertex->trackWeight(*recoDaughter);
// Sanity condition in case that track quality condition is too tight
if ( recoDaughterWeight < matchedDaughterWeight )
recoDaughterWeight = matchedDaughterWeight;
// Condition over S2RMatchedRecoRatio
if ( matchedDaughterWeight/recoDaughterWeight < S2RMatchedRecoRatio_ ) continue;
double quality = (double)matchedDaughterCounter/simDaughterCounter;
// Condition over S2RMatchedSimRatio
if (quality < S2RMatchedSimRatio_) continue;
outputCollection.insert(trackingVertex, std::make_pair(recoVertex, quality));
std::cout << "R2S: INDEX " << assoIndex;
std::cout << " ; SimDaughterCounter = " << simDaughterCounter;
std::cout << " ; RecoDaughterWeight = " << recoDaughterWeight;
std::cout << " ; MatchedDaughterCounter = " << matchedDaughterCounter;
std::cout << " ; MatchedDaughterWeight = " << matchedDaughterWeight;
std::cout << " ; quality = " << quality << std::endl;
assoIndex++;
}
} // loop over TrackingVertexes
std::cout << "SimToReco OUTPUT COLLECTION: outputCollection.size() = " << outputCollection.size() << std::endl << std::endl;
return outputCollection;
}
const edm::ParameterSet& VertexAssociatorByTracks::config_ [private] |
Definition at line 40 of file VertexAssociatorByTracks.h.
double VertexAssociatorByTracks::R2SMatchedRecoRatio_ [private] |
Definition at line 43 of file VertexAssociatorByTracks.h.
Referenced by associateRecoToSim(), and VertexAssociatorByTracks().
double VertexAssociatorByTracks::R2SMatchedSimRatio_ [private] |
Definition at line 42 of file VertexAssociatorByTracks.h.
Referenced by associateRecoToSim(), and VertexAssociatorByTracks().
double VertexAssociatorByTracks::S2RMatchedRecoRatio_ [private] |
Definition at line 45 of file VertexAssociatorByTracks.h.
Referenced by associateSimToReco(), and VertexAssociatorByTracks().
double VertexAssociatorByTracks::S2RMatchedSimRatio_ [private] |
Definition at line 44 of file VertexAssociatorByTracks.h.
Referenced by associateSimToReco(), and VertexAssociatorByTracks().
Definition at line 47 of file VertexAssociatorByTracks.h.
Referenced by associateRecoToSim(), associateSimToReco(), and VertexAssociatorByTracks().
Definition at line 48 of file VertexAssociatorByTracks.h.
Referenced by associateRecoToSim(), associateSimToReco(), and VertexAssociatorByTracks().
1.7.3