#include <TrackAssociatorByPositionImpl.h>

Inheritance diagram for TrackAssociatorByPositionImpl:

Public Types
enum	Method { Method::chi2, Method::dist, Method::momdr, Method::posdr }

typedef std::vector< SimHitTPPair >	SimHitTPAssociationList

typedef std::pair< TrackingParticleRef, TrackPSimHitRef >	SimHitTPPair

Public Member Functions
reco::RecoToSimCollection	associateRecoToSim (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< TrackingParticleCollection > &) const override
	compare reco to sim the handle of reco::Track and TrackingParticle collections More...

reco::SimToRecoCollection	associateSimToReco (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< TrackingParticleCollection > &) const override
	compare reco to sim the handle of reco::Track and TrackingParticle collections More...

	TrackAssociatorByPositionImpl (const TrackingGeometry geo, const Propagator prop, const SimHitTPAssociationList *assocList, double qMinCut, double qCut, double positionMinimumDistance, Method method, bool minIfNoMatch, bool considerAllSimHits)

Public Member Functions inherited from reco::TrackToTrackingParticleAssociatorBaseImpl
virtual reco::RecoToSimCollection	associateRecoToSim (const edm::Handle< edm::View< reco::Track > > &tCH, const edm::Handle< TrackingParticleCollection > &tPCH) const
	compare reco to sim the handle of reco::Track and TrackingParticle collections More...

virtual reco::RecoToSimCollectionSeed	associateRecoToSim (const edm::Handle< edm::View< TrajectorySeed > > &, const edm::Handle< TrackingParticleCollection > &) const

virtual reco::RecoToSimCollectionTCandidate	associateRecoToSim (const edm::Handle< TrackCandidateCollection > &, const edm::Handle< TrackingParticleCollection > &) const

virtual reco::SimToRecoCollection	associateSimToReco (const edm::Handle< edm::View< reco::Track > > &tCH, const edm::Handle< TrackingParticleCollection > &tPCH) const
	compare reco to sim the handle of reco::Track and TrackingParticle collections More...

virtual reco::SimToRecoCollectionSeed	associateSimToReco (const edm::Handle< edm::View< TrajectorySeed > > &, const edm::Handle< TrackingParticleCollection > &) const

virtual reco::SimToRecoCollectionTCandidate	associateSimToReco (const edm::Handle< TrackCandidateCollection > &, const edm::Handle< TrackingParticleCollection > &) const

	TrackToTrackingParticleAssociatorBaseImpl ()
	Constructor. More...

virtual	~TrackToTrackingParticleAssociatorBaseImpl ()
	Destructor. More...

Private Member Functions
FreeTrajectoryState	getState (const reco::Track &) const

TrajectoryStateOnSurface	getState (const TrackingParticleRef &, const SimHitTPAssociationList &simHitsTPAssoc) const

double	quality (const TrajectoryStateOnSurface &, const TrajectoryStateOnSurface &) const

Private Attributes
bool	theConsiderAllSimHits

const TrackingGeometry *	theGeometry

Method	theMethod

bool	theMinIfNoMatch

double	thePositionMinimumDistance

const Propagator *	thePropagator

double	theQCut

double	theQminCut

const SimHitTPAssociationList *	theSimHitsTPAssoc

Detailed Description

Class that performs the association of reco::Tracks and TrackingParticles based on position in muon detector

Author: vlimant

Definition at line 30 of file TrackAssociatorByPositionImpl.h.

Member Typedef Documentation

typedef std::vector<SimHitTPPair> TrackAssociatorByPositionImpl::SimHitTPAssociationList

Definition at line 34 of file TrackAssociatorByPositionImpl.h.

typedef std::pair<TrackingParticleRef, TrackPSimHitRef> TrackAssociatorByPositionImpl::SimHitTPPair

Definition at line 33 of file TrackAssociatorByPositionImpl.h.

Member Enumeration Documentation

enum TrackAssociatorByPositionImpl::Method

strong

Enumerator
chi2
dist
momdr
posdr

Definition at line 35 of file TrackAssociatorByPositionImpl.h.

35 { chi2, dist, momdr, posdr};

vertices_cff.chi2

chi2

Definition: vertices_cff.py:33

Constructor & Destructor Documentation

TrackAssociatorByPositionImpl::TrackAssociatorByPositionImpl	(	const TrackingGeometry *	geo,
		const Propagator *	prop,
		const SimHitTPAssociationList *	assocList,
		double	qMinCut,
		double	qCut,
		double	positionMinimumDistance,
		Method	method,
		bool	minIfNoMatch,
		bool	considerAllSimHits
	)

inline

Definition at line 37 of file TrackAssociatorByPositionImpl.h.

References associateRecoToSim(), associateSimToReco(), and quality().

                                                         :
   theGeometry(geo),
     thePropagator(prop),
     theSimHitsTPAssoc(assocList),
     theQminCut(qMinCut),
     theQCut(qCut),
     thePositionMinimumDistance(positionMinimumDistance),
     theMethod(method),
     theMinIfNoMatch(minIfNoMatch),
     theConsiderAllSimHits(considerAllSimHits) {}

Member Function Documentation

RecoToSimCollection TrackAssociatorByPositionImpl::associateRecoToSim	(	const edm::RefToBaseVector< reco::Track > &	tCH,
		const edm::RefVector< TrackingParticleCollection > &	tPCH
	)		const

overridevirtual

compare reco to sim the handle of reco::Track and TrackingParticle collections

Implements reco::TrackToTrackingParticleAssociatorBaseImpl.

Definition at line 113 of file TrackAssociatorByPositionImpl.cc.

References edm::AssociationMap< Tag >::insert(), TrajectoryStateOnSurface::isValid(), edm::AssociationMap< Tag >::post_insert(), jets_cff::quality, edm::RefToBaseVector< T >::size(), edm::RefVector< C, T, F >::size(), and TrajectoryStateOnSurface::surface().

Referenced by TrackAssociatorByPositionImpl().

                                                                                                                                   {
   RecoToSimCollection  outputCollection;
   //for each reco track find a matching tracking particle
   std::pair<unsigned int,unsigned int> minPair;
   const double dQmin_default=1542543;
   double dQmin=dQmin_default;
 
   //cdj edm::Handle<SimHitTPAssociationProducer::SimHitTPAssociationList> simHitsTPAssoc;
 
   //warning: make sure the TP collection used in the map is the same used in the associator!
   //e->getByLabel(_simHitTpMapTag,simHitsTPAssoc);
 
   for (unsigned int Ti=0; Ti!=tCH.size();++Ti){
     //initial state (initial OR inner OR outter)
     FreeTrajectoryState iState = getState(*(tCH)[Ti]);
 
     bool atLeastOne=false;
     //    for each tracking particle, find a state position and the plane to propagate the track to.
     for (unsigned int TPi=0;TPi!=tPCH.size();++TPi) {
       //get a state in the muon system 
       TrajectoryStateOnSurface simReferenceState = getState((tPCH)[TPi],*theSimHitsTPAssoc);
       if (!simReferenceState.isValid()) continue;
 
       //propagate the TRACK to the surface
       TrajectoryStateOnSurface trackReferenceState = thePropagator->propagate(iState,simReferenceState.surface());
       if (!trackReferenceState.isValid()) continue; 
       
       //comparison
       double dQ= quality(trackReferenceState,simReferenceState);
       if (dQ < theQCut){
     atLeastOne=true;
     outputCollection.insert(tCH[Ti],
                 std::make_pair(tPCH[TPi],-dQ));//association map with quality, is order greater-first
     edm::LogVerbatim("TrackAssociatorByPositionImpl")<<"track number: "<<Ti
                              <<" associated with dQ: "<<dQ
                              <<" to TrackingParticle number: " <<TPi;}
       if (dQ < dQmin){
     dQmin=dQ;
     minPair = std::make_pair(Ti,TPi);}
     }//loop over tracking particles
     if (theMinIfNoMatch && !atLeastOne && dQmin!=dQmin_default){
       outputCollection.insert(tCH[minPair.first],
                   std::make_pair(tPCH[minPair.second],-dQmin));}
   }//loop over tracks
   outputCollection.post_insert();
   return outputCollection;
 }

SimToRecoCollection TrackAssociatorByPositionImpl::associateSimToReco	(	const edm::RefToBaseVector< reco::Track > &	tCH,
		const edm::RefVector< TrackingParticleCollection > &	tPCH
	)		const

overridevirtual

compare reco to sim the handle of reco::Track and TrackingParticle collections

Implements reco::TrackToTrackingParticleAssociatorBaseImpl.

Definition at line 164 of file TrackAssociatorByPositionImpl.cc.

References edm::AssociationMap< Tag >::insert(), TrajectoryStateOnSurface::isValid(), edm::AssociationMap< Tag >::post_insert(), jets_cff::quality, edm::RefToBaseVector< T >::size(), edm::RefVector< C, T, F >::size(), and TrajectoryStateOnSurface::surface().

Referenced by TrackAssociatorByPositionImpl().

                                                                                                                                   {
   SimToRecoCollection  outputCollection;
   //for each tracking particle, find matching tracks.
 
   std::pair<unsigned int,unsigned int> minPair;
   const double dQmin_default=1542543;
   double dQmin=dQmin_default;
 
   //edm::Handle<SimHitTPAssociationProducer::SimHitTPAssociationList> simHitsTPAssoc;
 
   //warning: make sure the TP collection used in the map is the same used in the associator!
   //e->getByLabel(_simHitTpMapTag,simHitsTPAssoc);
 
   for (unsigned int TPi=0;TPi!=tPCH.size();++TPi){
     //get a state in the muon system
     TrajectoryStateOnSurface simReferenceState= getState((tPCH)[TPi],*theSimHitsTPAssoc);
       
     if (!simReferenceState.isValid()) continue; 
     bool atLeastOne=false;
     //  propagate every track from any state (initial, inner, outter) to the surface 
     //  and make the position test
     for (unsigned int Ti=0; Ti!=tCH.size();++Ti){
       //initial state
       FreeTrajectoryState iState = getState(*(tCH)[Ti]);
     
       //propagation to surface
       TrajectoryStateOnSurface trackReferenceState = thePropagator->propagate(iState,simReferenceState.surface());
       if (!trackReferenceState.isValid()) continue;
     
       //comparison
       double dQ= quality(trackReferenceState, simReferenceState);
       if (dQ < theQCut){
     atLeastOne=true;
     outputCollection.insert(tPCH[TPi],
                 std::make_pair(tCH[Ti],-dQ));//association map with quality, is order greater-first
     edm::LogVerbatim("TrackAssociatorByPositionImpl")<<"TrackingParticle number: "<<TPi
                              <<" associated with dQ: "<<dQ
                              <<" to track number: "<<Ti;}
       if (dQ <dQmin){
     dQmin=dQ;
     minPair = std::make_pair(TPi,Ti);}
     }//loop over tracks
     if (theMinIfNoMatch && !atLeastOne && dQmin!=dQmin_default){
       outputCollection.insert(tPCH[minPair.first],
                   std::make_pair(tCH[minPair.second],-dQmin));}
   }//loop over tracking particles
   
   outputCollection.post_insert();
   return outputCollection;
 }

FreeTrajectoryState TrackAssociatorByPositionImpl::getState ( const reco::Track & track ) const

private

Definition at line 72 of file TrackAssociatorByPositionImpl.cc.

References trajectoryStateTransform::initialFreeState().

                                                                                        {
   //may be you want to do more than that if track does not go to IP
   return trajectoryStateTransform::initialFreeState(track,thePropagator->magneticField());
 }

TrajectoryStateOnSurface TrackAssociatorByPositionImpl::getState	(	const TrackingParticleRef &	st,
		const SimHitTPAssociationList &	simHitsTPAssoc
	)		const

private

Definition at line 14 of file TrackAssociatorByPositionImpl.cc.

References SurfaceSideDefinition::atCenterOfSurface, KineDebug3::count(), edmIntegrityCheck::d, createTree::dd, end, LogDebug, mag(), GeomDet::surface(), Surface::toGlobal(), and DetId::Tracker.

                                                                                                                                                  {
 
   using SimHitTPPair =  std::pair<TrackingParticleRef, TrackPSimHitRef>;
   SimHitTPPair clusterTPpairWithDummyTP(st,TrackPSimHitRef());//SimHit is dummy:
   // sorting only the cluster is needed
   auto range = std::equal_range(simHitsTPAssoc.begin(), simHitsTPAssoc.end(),
                 clusterTPpairWithDummyTP, [](const SimHitTPPair& iLHS, const SimHitTPPair& iRHS) -> bool {
                                   return iLHS.first.key()> iRHS.first.key();
                                 });
   
   //  TrackingParticle* simtrack = const_cast<TrackingParticle*>(&st);
   //loop over PSimHits
   const PSimHit * psimhit=nullptr;
   const BoundPlane * plane=nullptr;
   double dLim=thePositionMinimumDistance;
 
   //    look for the further most hit beyond a certain limit
   auto start=range.first;
   auto end=range.second;
   LogDebug("TrackAssociatorByPositionImpl")<<range.second-range.first<<" PSimHits.";
 
   unsigned int count=0;
   for (auto ip=start;ip!=end;++ip){    
 
     TrackPSimHitRef psit = ip->second;
 
     //get the detid
     DetId dd(psit->detUnitId());
 
     if (!theConsiderAllSimHits && dd.det()!=DetId::Tracker) continue; 
 
     LogDebug("TrackAssociatorByPositionImpl")<<count++<<"] PSimHit on: "<<dd.rawId();
     //get the surface from the global geometry
     const GeomDet * gd=theGeometry->idToDet(dd);
     if (!gd){edm::LogError("TrackAssociatorByPositionImpl")<<"no geomdet for: "<<dd.rawId()<<". will skip.";
       continue;}
     double d=gd->surface().toGlobal(psit->localPosition()).mag();
     if (d>dLim ){
       dLim=d;
       psimhit=&(*psit);
       plane=&gd->surface();}
   }
 
 
   if (psimhit && plane){
     //build a trajectorystate on this surface    
     SurfaceSideDefinition::SurfaceSide surfaceside = SurfaceSideDefinition::atCenterOfSurface;
     GlobalPoint initialPoint=plane->toGlobal(psimhit->localPosition());
     GlobalVector initialMomentum=plane->toGlobal(psimhit->momentumAtEntry());
     int initialCharge =  (psimhit->particleType()>0) ? -1:1;
     CartesianTrajectoryError initialCartesianErrors; //no error at initial state  
     const GlobalTrajectoryParameters initialParameters(initialPoint,initialMomentum,initialCharge,thePropagator->magneticField());
     return TrajectoryStateOnSurface(initialParameters,initialCartesianErrors,*plane,surfaceside);}
   else{
     //    edm::LogError("TrackAssociatorByPositionImpl")<<"no corresponding PSimHit for a tracking particle. will fail.";
     return TrajectoryStateOnSurface();}
 }

double TrackAssociatorByPositionImpl::quality	(	const TrajectoryStateOnSurface &	tr,
		const TrajectoryStateOnSurface &	sim
	)		const

private

Definition at line 77 of file TrackAssociatorByPositionImpl.cc.

References vertices_cff::chi2, PV3DBase< T, PVType, FrameType >::eta(), TrajectoryStateOnSurface::globalDirection(), TrajectoryStateOnSurface::globalPosition(), TrajectoryStateOnSurface::localError(), TrajectoryStateOnSurface::localParameters(), funct::m, mag(), LocalTrajectoryError::matrix(), PV3DBase< T, PVType, FrameType >::phi(), findQualityFiles::v, and LocalTrajectoryParameters::vector().

Referenced by TrackAssociatorByPositionImpl().

                                                                                                                              {
   switch(theMethod){
   case Method::chi2:
     {
       AlgebraicVector5 v(tr.localParameters().vector() - sim.localParameters().vector());
       AlgebraicSymMatrix55 m(tr.localError().matrix());
       int ierr = ! m.Invert();
       if (ierr!=0) edm::LogInfo("TrackAssociatorByPositionImpl")<<"error inverting the error matrix:\n"<<m;
       double est = ROOT::Math::Similarity(v,m);
       return est;
       break;
     }
   case Method::dist:
     {
       return (tr.globalPosition() - sim.globalPosition()).mag();
       break;
     }
   case Method::momdr:
     {
       return  (deltaR<double>(tr.globalDirection().eta(),tr.globalDirection().phi(),
                   sim.globalDirection().eta(),sim.globalDirection().phi()));
       break;
     }
   case Method::posdr:
     {
       return (deltaR<double>(tr.globalPosition().eta(),tr.globalPosition().phi(),
                  sim.globalPosition().eta(),sim.globalPosition().phi()));
       break;
     }
   }
   //should never be reached
   edm::LogError("TrackAssociatorByPositionImpl")<<"option: "<<static_cast<int>(theMethod)<<" has not been recognized. association has no meaning.";
   return -1;
 }