#include <DAFTrackProducer.h>

Inheritance diagram for DAFTrackProducer:

Public Types
typedef std::vector< Trajectory >	TrajectoryCollection

Public Types inherited from TrackProducerBase< reco::Track >
using	AlgoProductCollection = typename Base::AlgoProductCollection

using	Base = AlgoProductTraits< reco::Track >

using	TrackCollection = typename Base::TrackCollection

using	TrackView = typename Base::TrackView

Public Types inherited from AlgoProductTraits< reco::Track >
using	AlgoProductCollection = std::vector< AlgoProduct >

using	TrackCollection = std::vector< reco::Track >

using	TrackView = edm::View< reco::Track >

Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Public Member Functions
	DAFTrackProducer (const edm::ParameterSet &iConfig)

void	produce (edm::Event &, const edm::EventSetup &) override
	Method where the procduction take place. To be implemented in concrete classes. More...

Public Member Functions inherited from KfTrackProducerBase
	KfTrackProducerBase (bool trajectoryInEvent, bool split)
	Constructor. More...

virtual void	putInEvt (edm::Event &, const Propagator prop, const MeasurementTracker measTk, std::unique_ptr< TrackingRecHitCollection > &, std::unique_ptr< reco::TrackCollection > &, std::unique_ptr< reco::TrackExtraCollection > &, std::unique_ptr< std::vector< Trajectory > > &, std::unique_ptr< std::vector< int > > &, AlgoProductCollection &, TransientTrackingRecHitBuilder const , const TrackerTopology ttopo, int BeforeOrAfter=0)
	Put produced collections in the event. More...

Public Member Functions inherited from TrackProducerBase< reco::Track >
const edm::ParameterSet &	getConf () const

virtual void	getFromES (const edm::EventSetup &, edm::ESHandle< TrackerGeometry > &, edm::ESHandle< MagneticField > &, edm::ESHandle< TrajectoryFitter > &, edm::ESHandle< Propagator > &, edm::ESHandle< MeasurementTracker > &, edm::ESHandle< TransientTrackingRecHitBuilder > &)
	Get needed services from the Event Setup. More...

virtual void	getFromEvt (edm::Event &, edm::Handle< TrackCandidateCollection > &, reco::BeamSpot &)
	Get TrackCandidateCollection from the Event (needed by TrackProducer) More...

virtual void	getFromEvt (edm::Event &, edm::Handle< TrackView > &, reco::BeamSpot &)
	Get TrackCollection from the Event (needed by TrackRefitter) More...

void	setAlias (std::string alias)
	set the aliases of produced collections More...

void	setClusterRemovalInfo (const edm::InputTag &clusterRemovalInfo)
	Sets the information on cluster removal, and turns it on. More...

void	setConf (const edm::ParameterSet &conf)
	Set parameter set. More...

void	setSecondHitPattern (Trajectory traj, reco::Track &track, const Propagator prop, const MeasurementTrackerEvent measTk, const TrackerTopology ttopo)

void	setSrc (const edm::EDGetToken &src, const edm::EDGetTokenT< reco::BeamSpot > &bsSrc, const edm::EDGetTokenT< MeasurementTrackerEvent > &mteSrc)
	set label of source collection More...

	TrackProducerBase (bool trajectoryInEvent=false)
	Constructor. More...

virtual	~TrackProducerBase ()(false)
	Destructor. More...

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Private Member Functions
void	getFromEvt (edm::Event &, edm::Handle< TrajTrackAssociationCollection > &, reco::BeamSpot &)

void	putInEvtTrajAnn (edm::Event &theEvent, TrajAnnealingCollection &trajannResults, std::unique_ptr< TrajAnnealingCollection > &selTrajAnn)

Private Attributes
edm::EDGetToken	srcTT_

DAFTrackProducerAlgorithm	theAlgo

bool	TrajAnnSaving_

Additional Inherited Members
Protected Attributes inherited from TrackProducerBase< reco::Track >
std::string	alias_

edm::EDGetTokenT< reco::BeamSpot >	bsSrc_

edm::InputTag	clusterRemovalInfo_

edm::ParameterSet	conf_

edm::EDGetTokenT< MeasurementTrackerEvent >	mteSrc_

bool	rekeyClusterRefs_

edm::OrphanHandle< TrackCollection >	rTracks_

edm::EDGetToken	src_

edm::ESHandle< NavigationSchool >	theSchool

bool	trajectoryInEvent_

Detailed Description

EDProducer for DAFTrackProducerAlgorithm.

Author: tropiano, genta in May 2014 by brondolin

Definition at line 17 of file DAFTrackProducer.h.

Member Typedef Documentation

typedef std::vector<Trajectory> DAFTrackProducer::TrajectoryCollection

Definition at line 20 of file DAFTrackProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 21 of file DAFTrackProducer.cc.

References TrackProducerBase< reco::Track >::alias_, edm::ParameterSet::getParameter(), TrackProducerBase< reco::Track >::setAlias(), TrackProducerBase< reco::Track >::setConf(), TrackProducerBase< reco::Track >::setSrc(), srcTT_, AlCaHLTBitMon_QueryRunRegistry::string, and TrajAnnSaving_.

                                                                 :
   KfTrackProducerBase(iConfig.getParameter<bool>("TrajectoryInEvent"),false),
   theAlgo(iConfig)
 {
   setConf(iConfig);
   setSrc( consumes<TrackCandidateCollection>(iConfig.getParameter<edm::InputTag>( "src" )),
           consumes<reco::BeamSpot>(iConfig.getParameter<edm::InputTag>( "beamSpot" )),
           consumes<MeasurementTrackerEvent>(iConfig.getParameter<edm::InputTag>( "MeasurementTrackerEvent") ));
   srcTT_ = consumes<TrajTrackAssociationCollection>(iConfig.getParameter<edm::InputTag>( "src" ));
   setAlias( iConfig.getParameter<std::string>( "@module_label" ) );
 
   //register your products
   produces<reco::TrackCollection>().setBranchAlias( alias_ + "Tracks" );
   produces<reco::TrackExtraCollection>().setBranchAlias( alias_ + "TrackExtras" );
   produces<TrackingRecHitCollection>().setBranchAlias( alias_ + "RecHits" );
   produces<std::vector<Trajectory> >();
   produces<TrajTrackAssociationCollection>();
   produces<TrajAnnealingCollection>().setBranchAlias( alias_ + "TrajectoryAnnealing" );
   produces<reco::TrackCollection>("beforeDAF").setBranchAlias( alias_ + "TracksBeforeDAF" );
   produces<reco::TrackExtraCollection>("beforeDAF").setBranchAlias( alias_ + "TrackExtrasBeforeDAF" );
   produces<reco::TrackCollection>("afterDAF").setBranchAlias( alias_ + "TracksAfterDAF" );
   produces<reco::TrackExtraCollection>("afterDAF").setBranchAlias( alias_ + "TrackExtrasAfterDAF" );
 
   TrajAnnSaving_ = iConfig.getParameter<bool>("TrajAnnealingSaving");
 }

Member Function Documentation

void DAFTrackProducer::getFromEvt	(	edm::Event &	theEvent,
		edm::Handle< TrajTrackAssociationCollection > &	trajTrackAssociationHandle,
		reco::BeamSpot &	bs
	)

private

Definition at line 145 of file DAFTrackProducer.cc.

References TrackProducerBase< reco::Track >::bsSrc_, edm::Event::getByToken(), and srcTT_.

Referenced by produce().

 {
 
   //get the TrajTrackMap from the event
   //WARNING: src has always to be redefined in cfg file
   theEvent.getByToken(srcTT_,trajTrackAssociationHandle);
 
   //get the BeamSpot
   edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
   theEvent.getByToken(bsSrc_,recoBeamSpotHandle);
   bs = *recoBeamSpotHandle;
 
 }

void DAFTrackProducer::produce	(	edm::Event &	,
		const edm::EventSetup &
	)

overridevirtual

Method where the procduction take place. To be implemented in concrete classes.

Implements TrackProducerBase< reco::Track >.

Definition at line 48 of file DAFTrackProducer.cc.

References MillePedeFileConverter_cfg::e, edm::EventSetup::get(), edm::Event::getByToken(), TrackProducerBase< reco::Track >::getConf(), TrackProducerBase< reco::Track >::getFromES(), getFromEvt(), edm::ParameterSet::getParameter(), edm::EventBase::id(), LogDebug, TrackProducerBase< reco::Track >::mteSrc_, edm::ESHandle< T >::product(), KfTrackProducerBase::putInEvt(), putInEvtTrajAnn(), DAFTrackProducerAlgorithm::runWithCandidate(), AlCaHLTBitMon_QueryRunRegistry::string, theAlgo, and TrajAnnSaving_.

 {
   edm::LogInfo("DAFTrackProducer") << "Analyzing event number: " << theEvent.id() << "\n";
   
   //empty output collections
   std::unique_ptr<TrackingRecHitCollection>    outputRHColl (new TrackingRecHitCollection);
   std::unique_ptr<reco::TrackCollection>       outputTColl(new reco::TrackCollection);
   std::unique_ptr<reco::TrackExtraCollection>  outputTEColl(new reco::TrackExtraCollection);
   std::unique_ptr<std::vector<Trajectory> >    outputTrajectoryColl(new std::vector<Trajectory>);
   std::unique_ptr<TrajAnnealingCollection>     outputTrajAnnColl(new TrajAnnealingCollection);
   std::unique_ptr<std::vector<int> >           outputIndecesInputColl(new std::vector<int>);
 
  
   //new tracks collections (changes before and after DAF)
   std::unique_ptr<TrackingRecHitCollection>    outputRHCollBeforeDAF (new TrackingRecHitCollection);
   std::unique_ptr<reco::TrackCollection>       outputTCollBeforeDAF(new reco::TrackCollection);
   std::unique_ptr<reco::TrackExtraCollection>  outputTECollBeforeDAF(new reco::TrackExtraCollection);
   std::unique_ptr<std::vector<Trajectory> >    outputTrajectoryCollBeforeDAF(new std::vector<Trajectory>);
   std::unique_ptr<std::vector<int> >           outputIndecesInputCollBeforeDAF(new std::vector<int>);
   //----
   std::unique_ptr<TrackingRecHitCollection>    outputRHCollAfterDAF (new TrackingRecHitCollection);
   std::unique_ptr<reco::TrackCollection>       outputTCollAfterDAF(new reco::TrackCollection);
   std::unique_ptr<reco::TrackExtraCollection>  outputTECollAfterDAF(new reco::TrackExtraCollection);
   std::unique_ptr<std::vector<Trajectory> >    outputTrajectoryCollAfterDAF(new std::vector<Trajectory>);
   std::unique_ptr<std::vector<int> >           outputIndecesInputCollAfterDAF(new std::vector<int>);
 
   //declare and get stuff to be retrieved from ES
   edm::ESHandle<TrackerGeometry> theG;
   edm::ESHandle<MagneticField> theMF;
   edm::ESHandle<TrajectoryFitter> theFitter;
   edm::ESHandle<Propagator> thePropagator;
   edm::ESHandle<MeasurementTracker>  theMeasTk;
   edm::ESHandle<TransientTrackingRecHitBuilder> theBuilder;
   getFromES(setup,theG,theMF,theFitter,thePropagator,theMeasTk,theBuilder);
 
   edm::ESHandle<TrackerTopology> httopo;
   setup.get<TrackerTopologyRcd>().get(httopo);
 
   //get additional es_modules needed by the DAF 
   edm::ESHandle<MultiRecHitCollector> measurementCollectorHandle;
   std::string measurementCollectorName = getConf().getParameter<std::string>("MeasurementCollector");
   setup.get<MultiRecHitRecord>().get(measurementCollectorName, measurementCollectorHandle);
   edm::ESHandle<SiTrackerMultiRecHitUpdator> updatorHandle;
   std::string updatorName = getConf().getParameter<std::string>("UpdatorName"); 
   setup.get<MultiRecHitRecord>().get(updatorName, updatorHandle);    
 
   //get MeasurementTrackerEvent
   edm::Handle<MeasurementTrackerEvent> mte;
   theEvent.getByToken(mteSrc_, mte);
 
 
   //declare and get TrackCollection 
   AlgoProductCollection algoResults;
   reco::BeamSpot bs;
   TrajAnnealingCollection trajannResults;
 
   //declare and get  new tracks collections
   AlgoProductCollection algoResultsBeforeDAF;
   AlgoProductCollection algoResultsAfterDAF;
   try{
 
     edm::Handle<TrajTrackAssociationCollection> trajTrackAssociationHandle;
     getFromEvt(theEvent,trajTrackAssociationHandle,bs);
 
 
     //run the algorithm  
     LogDebug("DAFTrackProducer") << "run the DAF algorithm" << "\n";
     theAlgo.runWithCandidate(theG.product(), theMF.product(),  
                  *trajTrackAssociationHandle, 
                  &*mte,
                              theFitter.product(), theBuilder.product(), 
                  measurementCollectorHandle.product(), updatorHandle.product(), bs, 
                  algoResults, trajannResults, TrajAnnSaving_,
                  algoResultsBeforeDAF, algoResultsAfterDAF);
     
   } catch (cms::Exception &e){ 
     edm::LogInfo("DAFTrackProducer") << "cms::Exception caught!!!" << "\n" << e << "\n"; 
     throw; 
   }
 
   //put everything in the event
   putInEvt(theEvent, thePropagator.product(),theMeasTk.product(), 
            outputRHColl, outputTColl, outputTEColl, 
            outputTrajectoryColl,  outputIndecesInputColl, algoResults, theBuilder.product(), httopo.product());
   putInEvtTrajAnn(theEvent, trajannResults, outputTrajAnnColl);
 
   //put in theEvent before and after DAF tracks collections
   putInEvt(theEvent, thePropagator.product(),theMeasTk.product(), 
            outputRHCollBeforeDAF, outputTCollBeforeDAF, outputTECollBeforeDAF, 
            outputTrajectoryCollBeforeDAF,  outputIndecesInputCollBeforeDAF, algoResultsBeforeDAF, theBuilder.product(), httopo.product(), 1);
   putInEvt(theEvent, thePropagator.product(),theMeasTk.product(), 
            outputRHCollAfterDAF, outputTCollAfterDAF, outputTECollAfterDAF, 
            outputTrajectoryCollAfterDAF,  outputIndecesInputCollAfterDAF, algoResultsAfterDAF, theBuilder.product(), httopo.product(), 2);
 
   LogDebug("DAFTrackProducer") << "end the DAF algorithm." << "\n";
 }

void DAFTrackProducer::putInEvtTrajAnn	(	edm::Event &	theEvent,
		TrajAnnealingCollection &	trajannResults,
		std::unique_ptr< TrajAnnealingCollection > &	selTrajAnn
	)

private

Definition at line 159 of file DAFTrackProducer.cc.

References mps_fire::i, eostools::move(), edm::Event::put(), and findQualityFiles::size.

Referenced by produce().

                                                                           {
   const int size = trajannResults.size();
   outputTrajAnnColl->reserve(size);
 
   for(unsigned int i = 0; i < trajannResults.size() ; i++){
     outputTrajAnnColl->push_back(trajannResults[i]);
   }
 
   theEvent.put( std::move(outputTrajAnnColl) );
 }

Member Data Documentation

edm::EDGetToken DAFTrackProducer::srcTT_

private

Definition at line 35 of file DAFTrackProducer.h.

Referenced by DAFTrackProducer(), and getFromEvt().

DAFTrackProducerAlgorithm DAFTrackProducer::theAlgo

private

Definition at line 28 of file DAFTrackProducer.h.

Referenced by produce().

bool DAFTrackProducer::TrajAnnSaving_

private

Definition at line 34 of file DAFTrackProducer.h.

Referenced by DAFTrackProducer(), and produce().

Public Types

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation