#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<>
using	CacheTypes = CacheContexts< T...>

using	GlobalCache = typename CacheTypes::GlobalCache

using	HasAbility = AbilityChecker< T...>

using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache

using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache

using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >

using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache

using	RunCache = typename CacheTypes::RunCache

using	RunContext = RunContextT< RunCache, GlobalCache >

using	RunSummaryCache = typename CacheTypes::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	initTrackProducerBase (const edm::ParameterSet &conf, edm::ConsumesCollector cc, const edm::EDGetToken &src)
	Call this method in inheriting class' constructor. More...

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

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

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

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

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

	EDProducer (const EDProducer &)=delete

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

const EDProducer &	operator= (const EDProducer &)=delete

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::ESGetToken < MultiRecHitCollector, MultiRecHitRecord >	measurementCollectorToken_

edm::EDGetTokenT < TrajTrackAssociationCollection >	srcTT_

DAFTrackProducerAlgorithm	theAlgo

bool	TrajAnnSaving_

edm::ESGetToken < TrackerTopology, TrackerTopologyRcd >	ttopoToken_

edm::ESGetToken < SiTrackerMultiRecHitUpdator, MultiRecHitRecord >	updatorToken_

Additional Inherited Members
Protected Member Functions inherited from KfTrackProducerBase
void	setClusterRemovalInfo (const edm::InputTag &clusterRemovalInfo)
	Sets the information on cluster removal, and turns it on. More...

Protected Attributes inherited from KfTrackProducerBase
edm::InputTag	clusterRemovalInfo_

bool	rekeyClusterRefs_

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

edm::EDGetTokenT< reco::BeamSpot >	bsSrc_

edm::ESGetToken < TransientTrackingRecHitBuilder, TransientRecHitRecord >	builderSrc_

edm::ParameterSet	conf_

edm::ESGetToken < TrajectoryFitter, TrajectoryFitter::Record >	fitterSrc_

edm::ESGetToken < MeasurementTracker, CkfComponentsRecord >	measTkSrc_

edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	mfSrc_

edm::EDGetTokenT < MeasurementTrackerEvent >	mteSrc_

edm::ESGetToken< Propagator, TrackingComponentsRecord >	propagatorSrc_

edm::OrphanHandle < TrackCollection >	rTracks_

edm::ESGetToken < NavigationSchool, NavigationSchoolRecord >	schoolSrc_

edm::EDGetToken	src_

edm::ESHandle< NavigationSchool >	theSchool

edm::ESGetToken < TrackerGeometry, TrackerDigiGeometryRecord >	trackGeomSrc_

bool	trajectoryInEvent_

bool	useSchool_

Detailed Description

EDProducer for DAFTrackProducerAlgorithm.

Author: tropiano, genta in May 2014 by brondolin

Definition at line 19 of file DAFTrackProducer.h.

Member Typedef Documentation

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

Definition at line 21 of file DAFTrackProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 20 of file DAFTrackProducer.cc.

References TrackProducerBase< reco::Track >::alias_, DeDxTools::esConsumes(), TrackProducerBase< reco::Track >::getConf(), edm::ParameterSet::getParameter(), TrackProducerBase< reco::Track >::initTrackProducerBase(), measurementCollectorToken_, TrackProducerBase< reco::Track >::setAlias(), srcTT_, AlCaHLTBitMon_QueryRunRegistry::string, TrajAnnSaving_, ttopoToken_, and updatorToken_.

     : KfTrackProducerBase(iConfig.getParameter<bool>("TrajectoryInEvent"), false), theAlgo(iConfig) {
   initTrackProducerBase(
       iConfig, consumesCollector(), consumes<TrackCandidateCollection>(iConfig.getParameter<edm::InputTag>("src")));
   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");
   ttopoToken_ = esConsumes();
   std::string measurementCollectorName = getConf().getParameter<std::string>("MeasurementCollector");
   measurementCollectorToken_ = esConsumes(edm::ESInputTag("", measurementCollectorName));
   std::string updatorName = getConf().getParameter<std::string>("UpdatorName");
   updatorToken_ = esConsumes(edm::ESInputTag("", updatorName));
 }

Member Function Documentation

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

private

Definition at line 171 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 47 of file DAFTrackProducer.cc.

References cms::cuda::bs, alignCSCRings::e, edm::Event::getByToken(), TrackProducerBase< reco::Track >::getFromES(), getFromEvt(), edm::EventSetup::getHandle(), edm::EventBase::id(), LogDebug, measurementCollectorToken_, TrackProducerBase< reco::Track >::mteSrc_, edm::ESHandle< class >::product(), KfTrackProducerBase::putInEvt(), putInEvtTrajAnn(), DAFTrackProducerAlgorithm::runWithCandidate(), theAlgo, TrajAnnSaving_, ttopoToken_, and updatorToken_.

                                                                              {
   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.getHandle(ttopoToken_);
 
   //get additional es_modules needed by the DAF
   edm::ESHandle<MultiRecHitCollector> measurementCollectorHandle = setup.getHandle(measurementCollectorToken_);
   edm::ESHandle<SiTrackerMultiRecHitUpdator> updatorHandle = setup.getHandle(updatorToken_);
 
   //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 184 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));
 }