#include <TrackAlgoCompareUtil.h>

Inheritance diagram for TrackAlgoCompareUtil:

Public Member Functions
	TrackAlgoCompareUtil (const edm::ParameterSet &)

	~TrackAlgoCompareUtil ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
virtual void	beginJob ()

virtual void	endJob ()

virtual void	produce (edm::Event &, const edm::EventSetup &) override

void	SetTrackingParticleD0Dz (TrackingParticleRef tp, const reco::BeamSpot &bs, const MagneticField *bf, TPtoRecoTrack &TPRT)

void	SetTrackingParticleD0Dz (TrackingParticleRef tp, const reco::BeamSpot &bs, const MagneticField *bf, RecoTracktoTP &RTTP)

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 57 of file TrackAlgoCompareUtil.h.

Constructor & Destructor Documentation

TrackAlgoCompareUtil::TrackAlgoCompareUtil ( const edm::ParameterSet & iConfig )

explicit

Definition at line 8 of file TrackAlgoCompareUtil.cc.

References edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), and AlCaHLTBitMon_QueryRunRegistry::string.

 {
     //now do what ever other initialization is needed
     trackLabel_algoA = iConfig.getParameter<edm::InputTag>("trackLabel_algoA");
     trackLabel_algoB = iConfig.getParameter<edm::InputTag>("trackLabel_algoB");
     trackingParticleLabel_fakes = iConfig.getParameter<edm::InputTag>("trackingParticleLabel_fakes");
     trackingParticleLabel_effic = iConfig.getParameter<edm::InputTag>("trackingParticleLabel_effic");
     vertexLabel_algoA = iConfig.getParameter<edm::InputTag>("vertexLabel_algoA");
     vertexLabel_algoB = iConfig.getParameter<edm::InputTag>("vertexLabel_algoB");
     beamSpotLabel = iConfig.getParameter<edm::InputTag>("beamSpotLabel");
     assocLabel_algoA = iConfig.getUntrackedParameter<std::string>("assocLabel_algoA", "TrackAssociatorByHits");
     assocLabel_algoB = iConfig.getUntrackedParameter<std::string>("assocLabel_algoB", "TrackAssociatorByHits");
     associatormap_algoA = iConfig.getParameter< edm::InputTag >("associatormap_algoA");
     associatormap_algoB = iConfig.getParameter< edm::InputTag >("associatormap_algoB");
     UseAssociators = iConfig.getParameter< bool >("UseAssociators");
     UseVertex = iConfig.getParameter< bool >("UseVertex");
   
     produces<RecoTracktoTPCollection>("AlgoA");
     produces<RecoTracktoTPCollection>("AlgoB");
     produces<TPtoRecoTrackCollection>("TP");
 }

TrackAlgoCompareUtil::~TrackAlgoCompareUtil ( )

Definition at line 31 of file TrackAlgoCompareUtil.cc.

32 {

33 }

Member Function Documentation

void TrackAlgoCompareUtil::beginJob ( void )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 37 of file TrackAlgoCompareUtil.cc.

38 {

39 }

void TrackAlgoCompareUtil::endJob ( void )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 266 of file TrackAlgoCompareUtil.cc.

267 {

268 }

void TrackAlgoCompareUtil::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 44 of file TrackAlgoCompareUtil.cc.

References SiPixelRawToDigiRegional_cfi::beamSpot, edm::AssociationMap< Tag >::end(), edm::AssociationMap< Tag >::find(), edm::EventSetup::get(), edm::Event::getByLabel(), i, reco::BeamSpot::position(), edm::Handle< T >::product(), edm::ESHandle< class >::product(), edm::Event::put(), RecoTracktoTP::SetBeamSpot(), TPtoRecoTrack::SetBeamSpot(), RecoTracktoTP::SetRecoTrack(), TPtoRecoTrack::SetRecoTrack_AlgoA(), TPtoRecoTrack::SetRecoTrack_AlgoB(), RecoTracktoTP::SetRecoVertex(), TPtoRecoTrack::SetRecoVertex_AlgoA(), TPtoRecoTrack::SetRecoVertex_AlgoB(), RecoTracktoTP::SetShared(), TPtoRecoTrack::SetShared_AlgoA(), TPtoRecoTrack::SetShared_AlgoB(), RecoTracktoTP::SetTrackingParticle(), and TPtoRecoTrack::SetTrackingParticle().

 {
      // create output collection instance
     std::auto_ptr<RecoTracktoTPCollection> outputAlgoA(new RecoTracktoTPCollection());
     std::auto_ptr<RecoTracktoTPCollection> outputAlgoB(new RecoTracktoTPCollection());
     std::auto_ptr<TPtoRecoTrackCollection> outputTP(new TPtoRecoTrackCollection());
   
     // Get Inputs
     edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
     iEvent.getByLabel(beamSpotLabel, recoBeamSpotHandle);
     reco::BeamSpot beamSpot = *recoBeamSpotHandle; 
   
     edm::Handle<View<reco::Track> > trackCollAlgoA;
     iEvent.getByLabel(trackLabel_algoA, trackCollAlgoA);
   
     edm::Handle< View<reco::Track> > trackCollAlgoB;
     iEvent.getByLabel(trackLabel_algoB, trackCollAlgoB);
   
     edm::Handle<TrackingParticleCollection> trackingParticleCollFakes;
     iEvent.getByLabel(trackingParticleLabel_fakes, trackingParticleCollFakes);
   
     edm::Handle<TrackingParticleCollection> trackingParticleCollEffic;
     iEvent.getByLabel(trackingParticleLabel_effic, trackingParticleCollEffic);
   
     edm::Handle<reco::VertexCollection> vertexCollAlgoA;
     edm::Handle<reco::VertexCollection> vertexCollAlgoB;
     if(UseVertex) 
     {
         iEvent.getByLabel(vertexLabel_algoA, vertexCollAlgoA);
         iEvent.getByLabel(vertexLabel_algoB, vertexCollAlgoB);
     }
   
     // call the associator functions:
     reco::RecoToSimCollection recSimColl_AlgoA; 
     reco::RecoToSimCollection recSimColl_AlgoB;
   
     reco::SimToRecoCollection simRecColl_AlgoA;
     reco::SimToRecoCollection simRecColl_AlgoB;
 
     if(UseAssociators)
     {
         edm::ESHandle<TrackAssociatorBase> theAssociator_algoA;
         iSetup.get<TrackAssociatorRecord>().get(assocLabel_algoA, theAssociator_algoA);
   
         edm::ESHandle<TrackAssociatorBase> theAssociator_algoB;
         iSetup.get<TrackAssociatorRecord>().get(assocLabel_algoB, theAssociator_algoB);
   
         recSimColl_AlgoA = theAssociator_algoA->associateRecoToSim(trackCollAlgoA, trackingParticleCollFakes, &iEvent, &iSetup);
         recSimColl_AlgoB = theAssociator_algoB->associateRecoToSim(trackCollAlgoB, trackingParticleCollFakes, &iEvent, &iSetup);
 
         simRecColl_AlgoA = theAssociator_algoA->associateSimToReco(trackCollAlgoA, trackingParticleCollEffic, &iEvent, &iSetup);
         simRecColl_AlgoB = theAssociator_algoB->associateSimToReco(trackCollAlgoB, trackingParticleCollEffic, &iEvent, &iSetup);
     }
     else
     {
         Handle<reco::RecoToSimCollection > recotosimCollectionH_AlgoA;
         iEvent.getByLabel(associatormap_algoA,recotosimCollectionH_AlgoA);
         recSimColl_AlgoA  = *(recotosimCollectionH_AlgoA.product());
         
         Handle<reco::RecoToSimCollection > recotosimCollectionH_AlgoB;
         iEvent.getByLabel(associatormap_algoB,recotosimCollectionH_AlgoB);
         recSimColl_AlgoB  = *(recotosimCollectionH_AlgoB.product());
         
         Handle<reco::SimToRecoCollection > simtorecoCollectionH_AlgoA;
         iEvent.getByLabel(associatormap_algoA, simtorecoCollectionH_AlgoA);
         simRecColl_AlgoA = *(simtorecoCollectionH_AlgoA.product());
 
         Handle<reco::SimToRecoCollection > simtorecoCollectionH_AlgoB;
         iEvent.getByLabel(associatormap_algoB, simtorecoCollectionH_AlgoB);
         simRecColl_AlgoB = *(simtorecoCollectionH_AlgoB.product());
     }
     
     // define the vector of references to trackingParticleColl associated with a given reco::Track
     std::vector<std::pair<TrackingParticleRef, double> > associatedTrackingParticles;
   
     // define the vector of references to trackColl associated with a given TrackingParticle
     std::vector<std::pair<reco::TrackBaseRef, double> > associatedRecoTracks;
   
     // Get the magnetic field data from the event (used to calculate the point of closest TrackingParticle)
     edm::ESHandle<MagneticField> theMagneticField;
     iSetup.get<IdealMagneticFieldRecord>().get(theMagneticField);
     const MagneticField *magneticField = theMagneticField.product();
     
     // fill collection algoA
     for(View<reco::Track>::size_type i = 0; i < trackCollAlgoA->size(); ++i)
     {
         // get recoTrack algo A
         reco::TrackBaseRef recoTrack(trackCollAlgoA, i);
         RecoTracktoTP  recoTracktoTP;
         recoTracktoTP.SetRecoTrack(recoTrack);
         recoTracktoTP.SetBeamSpot(beamSpot.position());
         
         // get the associated trackingParticle
         if(recSimColl_AlgoA.find(recoTrack) != recSimColl_AlgoA.end())
         {
             associatedTrackingParticles = recSimColl_AlgoA[recoTrack];
             recoTracktoTP.SetTrackingParticle( associatedTrackingParticles.begin()->first );
             recoTracktoTP.SetShared( associatedTrackingParticles.begin()->second );
             SetTrackingParticleD0Dz(associatedTrackingParticles.begin()->first, beamSpot, magneticField, recoTracktoTP);
         }
         else
         {           
             recoTracktoTP.SetTrackingParticle(TrackingParticleRef());
             recoTracktoTP.SetShared(-1.0);
         }       
         
         // get the reco primary vertex info
         if(UseVertex && vertexCollAlgoA->size())  
         {
             recoTracktoTP.SetRecoVertex( reco::VertexRef(vertexCollAlgoA, 0) );
         }
         else
         {
             recoTracktoTP.SetRecoVertex( reco::VertexRef() );
         }   
        
         outputAlgoA->push_back(recoTracktoTP);
     }
   
   
     // fill collection algoB
     for(reco::TrackCollection::size_type i = 0; i < trackCollAlgoB->size(); ++i)
     {
         // get recoTrack algo B
         reco::TrackBaseRef recoTrack(trackCollAlgoB, i);
         RecoTracktoTP  recoTracktoTP;
         recoTracktoTP.SetRecoTrack(recoTrack);
         recoTracktoTP.SetBeamSpot(beamSpot.position());
       
         // get the associated trackingParticle
         if(recSimColl_AlgoB.find(recoTrack) != recSimColl_AlgoB.end())
         {
             associatedTrackingParticles = recSimColl_AlgoB[recoTrack];
             recoTracktoTP.SetTrackingParticle( associatedTrackingParticles.begin()->first );
             recoTracktoTP.SetShared( associatedTrackingParticles.begin()->second );
             SetTrackingParticleD0Dz(associatedTrackingParticles.begin()->first, beamSpot, magneticField, recoTracktoTP);
         }
         else
         {           
             recoTracktoTP.SetTrackingParticle(TrackingParticleRef());
             recoTracktoTP.SetShared(-1.0);
         }       
         
         // get the reco primary vertex info
         if(UseVertex && vertexCollAlgoB->size())  
         {
             recoTracktoTP.SetRecoVertex( reco::VertexRef(vertexCollAlgoB, 0) );
         }
         else
         {
             recoTracktoTP.SetRecoVertex( reco::VertexRef() );
         }
         
         outputAlgoB->push_back(recoTracktoTP);
     }
   
   
     for(TrackingParticleCollection::size_type i = 0; i < trackingParticleCollEffic->size(); ++i)
     {
         // initialize the trackingParticle (sim) info
         TrackingParticleRef tparticle(trackingParticleCollEffic, i);
         TPtoRecoTrack  tptoRecoTrack;
         tptoRecoTrack.SetBeamSpot(beamSpot.position());
         tptoRecoTrack.SetTrackingParticle(tparticle);
         SetTrackingParticleD0Dz(tparticle, beamSpot,  magneticField, tptoRecoTrack);
       
         // get the assocated recoTrack algoA
         if(simRecColl_AlgoA.find(tparticle) != simRecColl_AlgoA.end())
         {
             associatedRecoTracks = simRecColl_AlgoA[tparticle];
             tptoRecoTrack.SetRecoTrack_AlgoA(associatedRecoTracks.begin()->first );
             tptoRecoTrack.SetShared_AlgoA(associatedRecoTracks.begin()->second );
         }
         else
         {
             tptoRecoTrack.SetRecoTrack_AlgoA(reco::TrackBaseRef());
             tptoRecoTrack.SetShared_AlgoA(-1.0);
         }
         
         // get the recoVertex algo A
         if(UseVertex && vertexCollAlgoA->size())
         {
             tptoRecoTrack.SetRecoVertex_AlgoA( reco::VertexRef(vertexCollAlgoA, 0) );
         }
         else
         {
             tptoRecoTrack.SetRecoVertex_AlgoA( reco::VertexRef() );
         }
       
         // get the assocated recoTrack algoB
         if(simRecColl_AlgoB.find(tparticle) != simRecColl_AlgoB.end())
         {
             associatedRecoTracks = simRecColl_AlgoB[tparticle];
             tptoRecoTrack.SetRecoTrack_AlgoB(associatedRecoTracks.begin()->first );
             tptoRecoTrack.SetShared_AlgoB(associatedRecoTracks.begin()->second );
         }
         else
         {
             tptoRecoTrack.SetRecoTrack_AlgoB(reco::TrackBaseRef());
             tptoRecoTrack.SetShared_AlgoB(-1.0);
         }
         // get the recoVertex algo B
         if(UseVertex && vertexCollAlgoB->size())
         {
             tptoRecoTrack.SetRecoVertex_AlgoB( reco::VertexRef(vertexCollAlgoB, 0) );
         }
         else
         {
             tptoRecoTrack.SetRecoVertex_AlgoB( reco::VertexRef() );
         }
       
         outputTP->push_back(tptoRecoTrack);
     }
 
 
     // put the collection in the event record
     iEvent.put(outputAlgoA, "AlgoA");
     iEvent.put(outputAlgoB, "AlgoB");
     iEvent.put(outputTP, "TP");
 }

void TrackAlgoCompareUtil::SetTrackingParticleD0Dz	(	TrackingParticleRef	tp,
		const reco::BeamSpot &	bs,
		const MagneticField *	bf,
		TPtoRecoTrack &	TPRT
	)

private

Definition at line 272 of file TrackAlgoCompareUtil.cc.

References TrajectoryStateClosestToBeamLine::isValid(), FreeTrajectoryState::momentum(), AlCaHLTBitMon_ParallelJobs::p, FreeTrajectoryState::position(), TPtoRecoTrack::SetTrackingParticleMomentumPCA(), TPtoRecoTrack::SetTrackingParticlePCA(), and TrajectoryStateClosestToBeamLine::trackStateAtPCA().

 {
     GlobalPoint trackingParticleVertex( tp->vertex().x(), tp->vertex().y(), tp->vertex().z() );
     GlobalVector trackingParticleP3(tp->g4Track_begin()->momentum().x(),
                                     tp->g4Track_begin()->momentum().y(),
                                     tp->g4Track_begin()->momentum().z() );
     TrackCharge trackingParticleCharge(tp->charge());
   
     FreeTrajectoryState ftsAtProduction( trackingParticleVertex, trackingParticleP3, trackingParticleCharge, bf );
     TSCBLBuilderNoMaterial tscblBuilder;
     TrajectoryStateClosestToBeamLine tsAtClosestApproach = tscblBuilder(ftsAtProduction, bs);  //as in TrackProducerAlgorithm
   
     if(tsAtClosestApproach.isValid())
     {
         GlobalPoint v1 = tsAtClosestApproach.trackStateAtPCA().position();
         GlobalVector p = tsAtClosestApproach.trackStateAtPCA().momentum();
 
         TPRT.SetTrackingParticleMomentumPCA(p);
         TPRT.SetTrackingParticlePCA(v1);
     }
     else
     {
         TPRT.SetTrackingParticleMomentumPCA(GlobalVector(-9999.0, -9999.0, -9999.0));
         TPRT.SetTrackingParticlePCA(GlobalPoint(-9999.0, -9999.0, -9999.0));
     }
 }

void TrackAlgoCompareUtil::SetTrackingParticleD0Dz	(	TrackingParticleRef	tp,
		const reco::BeamSpot &	bs,
		const MagneticField *	bf,
		RecoTracktoTP &	RTTP
	)

private

Definition at line 300 of file TrackAlgoCompareUtil.cc.

References TrajectoryStateClosestToBeamLine::isValid(), FreeTrajectoryState::momentum(), AlCaHLTBitMon_ParallelJobs::p, FreeTrajectoryState::position(), RecoTracktoTP::SetTrackingParticleMomentumPCA(), RecoTracktoTP::SetTrackingParticlePCA(), and TrajectoryStateClosestToBeamLine::trackStateAtPCA().

 {
     GlobalPoint trackingParticleVertex( tp->vertex().x(), tp->vertex().y(), tp->vertex().z() );
     GlobalVector trackingParticleP3(tp->g4Track_begin()->momentum().x(),
                                     tp->g4Track_begin()->momentum().y(),
                                     tp->g4Track_begin()->momentum().z() );
     TrackCharge trackingParticleCharge(tp->charge());
 
     FreeTrajectoryState ftsAtProduction( trackingParticleVertex, trackingParticleP3, trackingParticleCharge, bf );
     TSCBLBuilderNoMaterial tscblBuilder;
     TrajectoryStateClosestToBeamLine tsAtClosestApproach = tscblBuilder(ftsAtProduction, bs);  //as in TrackProducerAlgorithm
 
     if(tsAtClosestApproach.isValid())
     {
         GlobalPoint v1 = tsAtClosestApproach.trackStateAtPCA().position();
         GlobalVector p = tsAtClosestApproach.trackStateAtPCA().momentum();
 
         RTTP.SetTrackingParticleMomentumPCA(p);
         RTTP.SetTrackingParticlePCA(v1);
     }
     else
     {
         RTTP.SetTrackingParticleMomentumPCA(GlobalVector(-9999.0, -9999.0, -9999.0));
         RTTP.SetTrackingParticlePCA(GlobalPoint(-9999.0, -9999.0, -9999.0));
     }
 }

Member Data Documentation

edm::InputTag TrackAlgoCompareUtil::associatormap_algoA

private

Definition at line 82 of file TrackAlgoCompareUtil.h.

edm::InputTag TrackAlgoCompareUtil::associatormap_algoB

private

Definition at line 83 of file TrackAlgoCompareUtil.h.

std::string TrackAlgoCompareUtil::assocLabel_algoA

private

Definition at line 86 of file TrackAlgoCompareUtil.h.

std::string TrackAlgoCompareUtil::assocLabel_algoB

private

Definition at line 87 of file TrackAlgoCompareUtil.h.

edm::InputTag TrackAlgoCompareUtil::beamSpotLabel

private

Definition at line 81 of file TrackAlgoCompareUtil.h.

edm::InputTag TrackAlgoCompareUtil::trackingParticleLabel_effic

private

Definition at line 77 of file TrackAlgoCompareUtil.h.

edm::InputTag TrackAlgoCompareUtil::trackingParticleLabel_fakes

private

Definition at line 76 of file TrackAlgoCompareUtil.h.

edm::InputTag TrackAlgoCompareUtil::trackingVertexLabel

private

Definition at line 80 of file TrackAlgoCompareUtil.h.

edm::InputTag TrackAlgoCompareUtil::trackLabel_algoA

private

Definition at line 74 of file TrackAlgoCompareUtil.h.

edm::InputTag TrackAlgoCompareUtil::trackLabel_algoB

private

Definition at line 75 of file TrackAlgoCompareUtil.h.

bool TrackAlgoCompareUtil::UseAssociators

private

Definition at line 84 of file TrackAlgoCompareUtil.h.

bool TrackAlgoCompareUtil::UseVertex

private

Definition at line 85 of file TrackAlgoCompareUtil.h.

edm::InputTag TrackAlgoCompareUtil::vertexLabel_algoA

private

Definition at line 78 of file TrackAlgoCompareUtil.h.

edm::InputTag TrackAlgoCompareUtil::vertexLabel_algoB

private

Definition at line 79 of file TrackAlgoCompareUtil.h.

Private Attributes
edm::InputTag	associatormap_algoA

edm::InputTag	associatormap_algoB

std::string	assocLabel_algoA

std::string	assocLabel_algoB

edm::InputTag	beamSpotLabel

edm::InputTag	trackingParticleLabel_effic

edm::InputTag	trackingParticleLabel_fakes

edm::InputTag	trackingVertexLabel

edm::InputTag	trackLabel_algoA

edm::InputTag	trackLabel_algoB

bool	UseAssociators

bool	UseVertex

edm::InputTag	vertexLabel_algoA

edm::InputTag	vertexLabel_algoB

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation