#include <TrackGenAssociatorByChi2.h>

Inheritance diagram for TrackGenAssociatorByChi2:

Public Types
typedef std::map< double, SimTrack >	Chi2SimMap

typedef std::pair< reco::Track, Chi2SimMap >	RecoToSimPair

typedef std::vector < RecoToSimPair >	RecoToSimPairAssociation

Public Member Functions
reco::GenToRecoCollection	associateGenToReco (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< reco::GenParticleCollection > &, const edm::Event event=0, const edm::EventSetup setup=0) const
	Association Sim To Reco with Collections (Gen Particle version) More...

virtual reco::GenToRecoCollection	associateGenToReco (edm::Handle< edm::View< reco::Track > > &tCH, edm::Handle< reco::GenParticleCollection > &tPCH, const edm::Event event=0, const edm::EventSetup setup=0) const
	compare reco to sim the handle of reco::Track and GenParticle collections More...

reco::RecoToGenCollection	associateRecoToGen (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< reco::GenParticleCollection > &, const edm::Event event=0, const edm::EventSetup setup=0) const
	Association Sim To Reco with Collections (Gen Particle version) More...

virtual reco::RecoToGenCollection	associateRecoToGen (edm::Handle< edm::View< reco::Track > > &tCH, edm::Handle< reco::GenParticleCollection > &tPCH, const edm::Event event=0, const edm::EventSetup setup=0) const
	compare reco to sim the handle of reco::Track and GenParticle collections More...

	TrackGenAssociatorByChi2 (const edm::ESHandle< MagneticField > mF, const edm::ParameterSet &conf)
	Constructor with PSet. More...

	TrackGenAssociatorByChi2 (const edm::ESHandle< MagneticField > mF, double chi2Cut, bool onlyDiag, const edm::InputTag &beamspotSrc)
	Constructor with magnetic field, double, bool and InputTag. More...

Public Member Functions inherited from TrackGenAssociatorBase
	TrackGenAssociatorBase ()
	Constructor. More...

virtual	~TrackGenAssociatorBase ()

Private Member Functions
double	associateRecoToSim (reco::TrackCollection::const_iterator, TrackingParticleCollection::const_iterator, const reco::BeamSpot &) const
	compare reco::TrackCollection and TrackingParticleCollection iterators: returns the chi2 More...

double	getChi2 (const reco::TrackBase::ParameterVector &rParameters, const reco::TrackBase::CovarianceMatrix &recoTrackCovMatrix, const Basic3DVector< double > &momAtVtx, const Basic3DVector< double > &vert, int charge, const reco::BeamSpot &) const
	basic method where chi2 is computed More...

Private Attributes
edm::InputTag	bsSrc

double	chi2cut

bool	onlyDiagonal

edm::ESHandle< MagneticField >	theMF

Detailed Description

Class that performs the association of reco::Tracks and TrackingParticles evaluating the chi2 of reco tracks parameters and sim tracks parameters. The cut can be tuned from the config file: see data/TrackGenAssociatorByChi2.cfi. Note that the Association Map is filled with -ch2 and not chi2 because it is ordered using std::greater: the track with the lowest association chi2 will be the first in the output map.It is possible to use only diagonal terms (associator by pulls) seeting onlyDiagonal = true in the PSet

Author: cerati, magni

Definition at line 43 of file TrackGenAssociatorByChi2.h.

Member Typedef Documentation

typedef std::map<double, SimTrack> TrackGenAssociatorByChi2::Chi2SimMap

Definition at line 46 of file TrackGenAssociatorByChi2.h.

typedef std::pair< reco::Track, Chi2SimMap> TrackGenAssociatorByChi2::RecoToSimPair

Definition at line 47 of file TrackGenAssociatorByChi2.h.

typedef std::vector< RecoToSimPair > TrackGenAssociatorByChi2::RecoToSimPairAssociation

Definition at line 48 of file TrackGenAssociatorByChi2.h.

Constructor & Destructor Documentation

TrackGenAssociatorByChi2::TrackGenAssociatorByChi2	(	const edm::ESHandle< MagneticField >	mF,
		const edm::ParameterSet &	conf
	)

inline

Constructor with PSet.

Definition at line 51 of file TrackGenAssociatorByChi2.h.

References onlyDiagonal, and theMF.

                                                                                             :
     chi2cut(conf.getParameter<double>("chi2cut")),
     onlyDiagonal(conf.getParameter<bool>("onlyDiagonal")),
     bsSrc(conf.getParameter<edm::InputTag>("beamSpot")) {
     theMF=mF;  
     if (onlyDiagonal)
       edm::LogInfo("TrackAssociator") << " ---- Using Off Diagonal Covariance Terms = 0 ---- " <<  "\n";
     else 
       edm::LogInfo("TrackAssociator") << " ---- Using Off Diagonal Covariance Terms != 0 ---- " <<  "\n";
   }

TrackGenAssociatorByChi2::TrackGenAssociatorByChi2	(	const edm::ESHandle< MagneticField >	mF,
		double	chi2Cut,
		bool	onlyDiag,
		const edm::InputTag &	beamspotSrc
	)

inline

Constructor with magnetic field, double, bool and InputTag.

Definition at line 63 of file TrackGenAssociatorByChi2.h.

References bsSrc, chi2cut, onlyDiagonal, and theMF.

                                                                                                                               {
     chi2cut=chi2Cut;
     onlyDiagonal=onlyDiag;
     theMF=mF;  
     bsSrc = beamspotSrc;
   }

Member Function Documentation

GenToRecoCollection TrackGenAssociatorByChi2::associateGenToReco	(	const edm::RefToBaseVector< reco::Track > &	tC,
		const edm::RefVector< reco::GenParticleCollection > &	tPCH,
		const edm::Event *	event = `0`,
		const edm::EventSetup *	setup = `0`
	)		const

virtual

Association Sim To Reco with Collections (Gen Particle version)

Implements TrackGenAssociatorBase.

Definition at line 83 of file TrackGenAssociatorByChi2.cc.

References edm::RefToBaseVector< T >::begin(), RecoTauCleanerPlugins::charge, edm::RefToBaseVector< T >::end(), edm::Event::getByLabel(), track_associator::getChi2(), i, edm::AssociationMap< Tag >::insert(), j, LogDebug, edm::AssociationMap< Tag >::post_insert(), edm::RefVector< C, T, F >::size(), and mathSSE::sqrt().

Referenced by associateGenToReco().

                                                                      {
 
   edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
   e->getByLabel(bsSrc,recoBeamSpotHandle);
   reco::BeamSpot bs = *recoBeamSpotHandle;      
 
   GenToRecoCollection  outputCollection;
 
   //dereference the edm::Refs only once
   std::vector<GenParticle const*> tPC;
   tPC.reserve(tPCH.size());
   for(auto const& ref: tPCH) {
     tPC.push_back(&(*ref));
   }
 
   int tpindex =0;
   for (auto tp=tPC.begin(); tp!=tPC.end(); tp++, ++tpindex){
     
     //skip tps with a very small pt
     //if (sqrt((*tp)->momentum().perp2())<0.5) continue;
     int charge = (*tp)->charge();
     if (charge==0) continue;
     
     LogDebug("TrackAssociator") << "=========LOOKING FOR ASSOCIATION===========" << "\n"
                 << "TrackingParticle #"<<tpindex<<" with pt=" << sqrt((*tp)->momentum().perp2()) << "\n"
                 << "===========================================" << "\n";
     
     Basic3DVector<double> momAtVtx((*tp)->momentum().x(),(*tp)->momentum().y(),(*tp)->momentum().z());
     Basic3DVector<double> vert((*tp)->vertex().x(),(*tp)->vertex().y(),(*tp)->vertex().z());
       
     int tindex=0;
     for (RefToBaseVector<reco::Track>::const_iterator rt=tC.begin(); rt!=tC.end(); rt++, tindex++){
       
       TrackBase::ParameterVector rParameters = (*rt)->parameters();      
       TrackBase::CovarianceMatrix recoTrackCovMatrix = (*rt)->covariance();
       if (onlyDiagonal) {
     for (unsigned int i=0;i<5;i++){
       for (unsigned int j=0;j<5;j++){
         if (i!=j) recoTrackCovMatrix(i,j)=0;
       }
     }
       }
       recoTrackCovMatrix.Invert();
       
       double chi2 = getChi2(rParameters,recoTrackCovMatrix,momAtVtx,vert,charge,bs);
       
       if (chi2<chi2cut) {
         //NOTE: tPCH and tPC have the same index for the same object
     outputCollection.insert(tPCH[tpindex],
                 std::make_pair(tC[tindex],
                            -chi2));//-chi2 because the Association Map is ordered using std::greater
       }
     }
   }
   outputCollection.post_insert();
   return outputCollection;
 }

virtual reco::GenToRecoCollection TrackGenAssociatorByChi2::associateGenToReco	(	edm::Handle< edm::View< reco::Track > > &	tCH,
		edm::Handle< reco::GenParticleCollection > &	tPCH,
		const edm::Event *	event = `0`,
		const edm::EventSetup *	setup = `0`
	)		const

inlinevirtual

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

Implements TrackGenAssociatorBase.

Definition at line 98 of file TrackGenAssociatorByChi2.h.

References associateGenToReco(), event(), edm::HandleBase::id(), j, edm::RefToBaseVector< T >::push_back(), edm::RefVector< C, T, F >::push_back(), and HcalObjRepresent::setup().

                                                                                               {
     edm::RefToBaseVector<reco::Track> tc(tCH);
     for (unsigned int j=0; j<tCH->size();j++)
       tc.push_back(edm::RefToBase<reco::Track>(tCH,j));
 
     edm::RefVector<reco::GenParticleCollection> tpc(tPCH.id());
     for (unsigned int j=0; j<tPCH->size();j++)
       tpc.push_back(edm::Ref<reco::GenParticleCollection>(tPCH,j));
 
     return associateGenToReco(tc,tpc,event,setup);
   }  

RecoToGenCollection TrackGenAssociatorByChi2::associateRecoToGen	(	const edm::RefToBaseVector< reco::Track > &	tC,
		const edm::RefVector< reco::GenParticleCollection > &	tPCH,
		const edm::Event *	event = `0`,
		const edm::EventSetup *	setup = `0`
	)		const

virtual

Association Sim To Reco with Collections (Gen Particle version)

Implements TrackGenAssociatorBase.

Definition at line 21 of file TrackGenAssociatorByChi2.cc.

References edm::RefToBaseVector< T >::begin(), RecoTauCleanerPlugins::charge, edm::RefToBaseVector< T >::end(), edm::Event::getByLabel(), track_associator::getChi2(), i, edm::AssociationMap< Tag >::insert(), j, LogDebug, edm::AssociationMap< Tag >::post_insert(), and edm::RefVector< C, T, F >::size().

Referenced by associateRecoToGen().

                                                                                                 {
   edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
   e->getByLabel(bsSrc,recoBeamSpotHandle);
   reco::BeamSpot bs = *recoBeamSpotHandle;      
 
   RecoToGenCollection  outputCollection;
 
   //dereference the edm::Ref's only once
   std::vector<const GenParticle*> tPC;
   tPC.reserve(tPCH.size());
   for(auto const& ref: tPCH) {
     tPC.push_back(&(*ref));
   }
 
   int tindex=0;
   for (RefToBaseVector<reco::Track>::const_iterator rt=tC.begin(); rt!=tC.end(); rt++, tindex++){
 
     LogDebug("TrackAssociator") << "=========LOOKING FOR ASSOCIATION===========" << "\n"
                 << "rec::Track #"<<tindex<<" with pt=" << (*rt)->pt() <<  "\n"
                 << "===========================================" << "\n";
  
     TrackBase::ParameterVector rParameters = (*rt)->parameters();
 
     TrackBase::CovarianceMatrix recoTrackCovMatrix = (*rt)->covariance();
     if (onlyDiagonal){
       for (unsigned int i=0;i<5;i++){
     for (unsigned int j=0;j<5;j++){
       if (i!=j) recoTrackCovMatrix(i,j)=0;
     }
       }
     } 
 
     recoTrackCovMatrix.Invert();
 
     int tpindex =0;
     for (auto tp=tPC.begin(); tp!=tPC.end(); tp++, ++tpindex){
     
       //skip tps with a very small pt
       //if (sqrt((*tp)->momentum().perp2())<0.5) continue;
       int charge = (*tp)->charge();
       if (charge==0) continue;
       Basic3DVector<double> momAtVtx((*tp)->momentum().x(),(*tp)->momentum().y(),(*tp)->momentum().z());
       Basic3DVector<double> vert=(Basic3DVector<double>) (*tp)->vertex();
 
       double chi2 = getChi2(rParameters,recoTrackCovMatrix,momAtVtx,vert,charge,bs);
       
       if (chi2<chi2cut) {
         //NOTE: tPCH and tPC have the same index for the same object
     outputCollection.insert(tC[tindex], 
                 std::make_pair(tPCH[tpindex],
                            -chi2));//-chi2 because the Association Map is ordered using std::greater
       }
     }
   }
   outputCollection.post_insert();
   return outputCollection;
 }

virtual reco::RecoToGenCollection TrackGenAssociatorByChi2::associateRecoToGen	(	edm::Handle< edm::View< reco::Track > > &	tCH,
		edm::Handle< reco::GenParticleCollection > &	tPCH,
		const edm::Event *	event = `0`,
		const edm::EventSetup *	setup = `0`
	)		const

inlinevirtual

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

Implements TrackGenAssociatorBase.

Definition at line 82 of file TrackGenAssociatorByChi2.h.

References associateRecoToGen(), event(), edm::HandleBase::id(), j, edm::RefToBaseVector< T >::push_back(), edm::RefVector< C, T, F >::push_back(), and HcalObjRepresent::setup().

                                                                                               {
     edm::RefToBaseVector<reco::Track> tc(tCH);
     for (unsigned int j=0; j<tCH->size();j++)
       tc.push_back(edm::RefToBase<reco::Track>(tCH,j));
 
     edm::RefVector<reco::GenParticleCollection> tpc(tPCH.id());
     for (unsigned int j=0; j<tPCH->size();j++)
       tpc.push_back(edm::Ref<reco::GenParticleCollection>(tPCH,j));
 
     return associateRecoToGen(tc,tpc,event,setup);
   }

double TrackGenAssociatorByChi2::associateRecoToSim	(	reco::TrackCollection::const_iterator	,
		TrackingParticleCollection::const_iterator	,
		const reco::BeamSpot &
	)		const

private

compare reco::TrackCollection and TrackingParticleCollection iterators: returns the chi2

double TrackGenAssociatorByChi2::getChi2	(	const reco::TrackBase::ParameterVector &	rParameters,
		const reco::TrackBase::CovarianceMatrix &	recoTrackCovMatrix,
		const Basic3DVector< double > &	momAtVtx,
		const Basic3DVector< double > &	vert,
		int	charge,
		const reco::BeamSpot &	bs
	)		const

private

basic method where chi2 is computed

Definition at line 12 of file TrackGenAssociatorByChi2.cc.

References track_associator::getChi2().

                                                                       {
   return track_associator::getChi2(rParameters, recoTrackCovMatrix,momAtVtx, vert, charge, *theMF, bs);
 }