DAFTrackProducer Class Reference

#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::EDGetToken	srcTT_
DAFTrackProducerAlgorithm	theAlgo
bool	TrajAnnSaving_

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 17 of file DAFTrackProducer.h.

Member Typedef Documentation

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

Definition at line 19 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_, edm::ParameterSet::getParameter(), TrackProducerBase< reco::Track >::initTrackProducerBase(), TrackProducerBase< reco::Track >::setAlias(), srcTT_, AlCaHLTBitMon_QueryRunRegistry::string, and TrajAnnSaving_.

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

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

References cms::cuda::bs, alignCSCRings::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< class >::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 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));
}

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 27 of file DAFTrackProducer.h.

Referenced by produce().

bool DAFTrackProducer::TrajAnnSaving_

private

Definition at line 34 of file DAFTrackProducer.h.

Referenced by DAFTrackProducer(), and produce().