#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 () noexcept(false)
	Destructor. More...

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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () 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 \review in May 2014 by brondolin

Definition at line 17 of file DAFTrackProducer.h.

Member Typedef Documentation

◆ TrajectoryCollection

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

Definition at line 19 of file DAFTrackProducer.h.

Constructor & Destructor Documentation

◆ DAFTrackProducer()

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

explicit

Definition at line 20 of file DAFTrackProducer.cc.

     : 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");
 }

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_.

Member Function Documentation

◆ getFromEvt()

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

private

Definition at line 173 of file DAFTrackProducer.cc.

                                                     {
   //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;
 }

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

Referenced by produce().

◆ produce()

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 44 of file DAFTrackProducer.cc.

                                                                              {
   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";
 }

References cms::cuda::bs, MillePedeFileConverter_cfg::e, 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(), singleTopDQM_cfi::setup, AlCaHLTBitMon_QueryRunRegistry::string, theAlgo, and TrajAnnSaving_.

◆ putInEvtTrajAnn()

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

private

Definition at line 186 of file DAFTrackProducer.cc.

                                                                                                   {
   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));
 }

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

Referenced by produce().

Member Data Documentation

◆ srcTT_

edm::EDGetToken DAFTrackProducer::srcTT_

private

Definition at line 35 of file DAFTrackProducer.h.

Referenced by DAFTrackProducer(), and getFromEvt().

◆ theAlgo

DAFTrackProducerAlgorithm DAFTrackProducer::theAlgo

private

Definition at line 27 of file DAFTrackProducer.h.

Referenced by produce().

◆ TrajAnnSaving_

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

◆ TrajectoryCollection

Constructor & Destructor Documentation

◆ DAFTrackProducer()

Member Function Documentation

◆ getFromEvt()

◆ produce()

◆ putInEvtTrajAnn()

Member Data Documentation

◆ srcTT_

◆ theAlgo

◆ TrajAnnSaving_