TrackProducer Class Reference

#include <TrackProducer.h>

Inheritance diagram for TrackProducer:

Public Member Functions
std::vector< reco::TransientTrack >	getTransient (edm::Event &, const edm::EventSetup &)
	Get Transient Tracks. More...
void	produce (edm::Event &, const edm::EventSetup &) override
	Implementation of produce method. More...
	TrackProducer (const edm::ParameterSet &iConfig)
	Constructor. 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 Attributes
TrackProducerAlgorithm < reco::Track >	theAlgo
edm::ESGetToken < TrackerTopology, TrackerTopologyRcd >	theTTopoToken

Additional Inherited Members
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
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

Produce Tracks from TrackCandidates

Produce Tracks from TrackCandidates

Author

cerati

Definition at line 16 of file TrackProducer.h.

Constructor & Destructor Documentation

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

explicit

Constructor.

Definition at line 18 of file TrackProducer.cc.

References TrackProducerBase< reco::Track >::alias_, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), TrackProducerBase< reco::Track >::initTrackProducerBase(), TrackProducerBase< reco::Track >::setAlias(), KfTrackProducerBase::setClusterRemovalInfo(), AlCaHLTBitMon_QueryRunRegistry::string, and GlobalPosition_Frontier_DevDB_cff::tag.

    : KfTrackProducerBase(iConfig.getParameter<bool>("TrajectoryInEvent"),
                          iConfig.getParameter<bool>("useHitsSplitting")),
      theAlgo(iConfig),
      theTTopoToken(esConsumes()) {
  initTrackProducerBase(
      iConfig, consumesCollector(), consumes<TrackCandidateCollection>(iConfig.getParameter<edm::InputTag>("src")));
  setAlias(iConfig.getParameter<std::string>("@module_label"));

  if (iConfig.exists("clusterRemovalInfo")) {
    edm::InputTag tag = iConfig.getParameter<edm::InputTag>("clusterRemovalInfo");
    if (!(tag == edm::InputTag())) {
      setClusterRemovalInfo(tag);
    }
  }

  //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<std::vector<int> >();
  produces<TrajTrackAssociationCollection>();
}

Member Function Documentation

std::vector< reco::TransientTrack > TrackProducer::getTransient	(	edm::Event &	theEvent,
		const edm::EventSetup &	setup
	)

Get Transient Tracks.

Definition at line 113 of file TrackProducer.cc.

References cms::cuda::bs, alignCSCRings::e, edm::HandleBase::failedToGet(), TrackProducerBase< reco::Track >::getFromES(), TrackProducerBase< reco::Track >::getFromEvt(), edm::EventBase::id(), LogDebug, edm::ESHandle< class >::product(), TrackProducerAlgorithm< T >::runWithCandidate(), and theAlgo.

                                                                                                        {
  LogDebug("TrackProducer") << "Analyzing event number: " << theEvent.id() << "\n";
  //
  // create empty output collections
  //
  std::vector<reco::TransientTrack> ttks;

  //
  //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);

  //
  //declare and get TrackColection to be retrieved from the event
  //
  AlgoProductCollection algoResults;
  edm::Handle<TrackCandidateCollection> theTCCollection;
  reco::BeamSpot bs;
  getFromEvt(theEvent, theTCCollection, bs);
  //protect against missing product
  if (theTCCollection.failedToGet()) {
    edm::LogError("TrackProducer") << "could not get the TrackCandidateCollection.";
  } else {
    LogDebug("TrackProducer") << "run the algorithm"
                              << "\n";
    try {
      theAlgo.runWithCandidate(theG.product(),
                               theMF.product(),
                               *theTCCollection,
                               theFitter.product(),
                               thePropagator.product(),
                               theBuilder.product(),
                               bs,
                               algoResults);
    } catch (cms::Exception& e) {
      edm::LogError("TrackProducer") << "cms::Exception caught during theAlgo.runWithCandidate."
                                     << "\n"
                                     << e << "\n";
      throw;
    }
  }
  ttks.reserve(algoResults.size());
  for (auto& prod : algoResults) {
    ttks.push_back(reco::TransientTrack(*(prod.track), thePropagator.product()->magneticField()));
  }

  LogDebug("TrackProducer") << "end"
                            << "\n";

  return ttks;
}

void TrackProducer::produce ( edm::Event &  theEvent, const edm::EventSetup &  setup  )

overridevirtual

Implementation of produce method.

Implements TrackProducerBase< reco::Track >.

Definition at line 43 of file TrackProducer.cc.

References cms::cuda::bs, alignCSCRings::e, edm::HandleBase::failedToGet(), edm::EventSetup::getData(), TrackProducerBase< reco::Track >::getFromES(), TrackProducerBase< reco::Track >::getFromEvt(), edm::EventBase::id(), LogDebug, edm::ESHandle< class >::product(), KfTrackProducerBase::putInEvt(), TrackProducerAlgorithm< T >::runWithCandidate(), theAlgo, and theTTopoToken.

                                                                          {
  LogDebug("TrackProducer") << "Analyzing event number: " << theEvent.id() << "\n";
  //
  // create 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<std::vector<int> > outputIndecesInputColl(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);

  TrackerTopology const& ttopo = setup.getData(theTTopoToken);

  //
  //declare and get TrackColection to be retrieved from the event
  //
  AlgoProductCollection algoResults;
  edm::Handle<TrackCandidateCollection> theTCCollection;
  reco::BeamSpot bs;
  getFromEvt(theEvent, theTCCollection, bs);
  //protect against missing product
  if (theTCCollection.failedToGet()) {
    edm::LogError("TrackProducer") << "could not get the TrackCandidateCollection.";
  } else {
    LogDebug("TrackProducer") << "run the algorithm"
                              << "\n";
    try {
      theAlgo.runWithCandidate(theG.product(),
                               theMF.product(),
                               *theTCCollection,
                               theFitter.product(),
                               thePropagator.product(),
                               theBuilder.product(),
                               bs,
                               algoResults);
    } catch (cms::Exception& e) {
      edm::LogError("TrackProducer") << "cms::Exception caught during theAlgo.runWithCandidate."
                                     << "\n"
                                     << e << "\n";
      throw;
    }
  }

  //put everything in the event
  putInEvt(theEvent,
           thePropagator.product(),
           theMeasTk.product(),
           outputRHColl,
           outputTColl,
           outputTEColl,
           outputTrajectoryColl,
           outputIndecesInputColl,
           algoResults,
           theBuilder.product(),
           &ttopo);
  LogDebug("TrackProducer") << "end"
                            << "\n";
}

Member Data Documentation

TrackProducerAlgorithm<reco::Track> TrackProducer::theAlgo

private

Definition at line 36 of file TrackProducer.h.

Referenced by getTransient(), and produce().

edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> TrackProducer::theTTopoToken

private

Definition at line 37 of file TrackProducer.h.

Referenced by produce().