#include <GsfTrackRefitter.h>

Inheritance diagram for GsfTrackRefitter:

Public Member Functions
	GsfTrackRefitter (const edm::ParameterSet &iConfig)
	Constructor. More...

void	produce (edm::Event &, const edm::EventSetup &) override
	Implementation of produce method. More...

Public Member Functions inherited from GsfTrackProducerBase
	GsfTrackProducerBase (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::GsfTrackCollection > &, std::unique_ptr< reco::TrackExtraCollection > &, std::unique_ptr< reco::GsfTrackExtraCollection > &, std::unique_ptr< std::vector< Trajectory > > &, AlgoProductCollection &, TransientTrackingRecHitBuilder const , const reco::BeamSpot &, const TrackerTopology ttopo)
	Put produced collections in the event. More...

Public Member Functions inherited from TrackProducerBase< reco::GsfTrack >
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::GsfTrack &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	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () const final

Private Types
enum	Constraint { none, vertex }

Private Attributes
Constraint	constraint_

edm::EDGetTokenT< GsfTrackVtxConstraintAssociationCollection >	gsfTrackVtxConstraintTag_

TrackProducerAlgorithm< reco::GsfTrack >	theAlgo

Additional Inherited Members
Public Types inherited from TrackProducerBase< reco::GsfTrack >
using	AlgoProductCollection = typename Base::AlgoProductCollection

using	Base = AlgoProductTraits< reco::GsfTrack >

using	TrackCollection = typename Base::TrackCollection

using	TrackView = typename Base::TrackView

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

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

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

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

Protected Member Functions inherited from GsfTrackProducerBase
void	fillMode (reco::GsfTrack &track, const TrajectoryStateOnSurface innertsos, const Propagator &gsfProp, const TransverseImpactPointExtrapolator &tipExtrapolator, TrajectoryStateClosestToBeamLineBuilder &tscblBuilder, const reco::BeamSpot &bs) const

void	fillStates (TrajectoryStateOnSurface tsos, std::vector< reco::GsfComponent5D > &states) const

Protected Attributes inherited from TrackProducerBase< reco::GsfTrack >
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

Refit GSF Tracks. Based on the TrackRefitter.

Definition at line 14 of file GsfTrackRefitter.h.

Member Enumeration Documentation

enum GsfTrackRefitter::Constraint

private

Enumerator
none
vertex

Definition at line 24 of file GsfTrackRefitter.h.

                   {
     none,
     //          momentum,
     vertex
   };

Constructor & Destructor Documentation

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

explicit

Constructor.

Definition at line 18 of file GsfTrackRefitter.cc.

References TrackProducerBase< reco::GsfTrack >::alias_, constraint_, Exception, edm::ParameterSet::getParameter(), gsfTrackVtxConstraintTag_, none, TrackProducerBase< reco::GsfTrack >::setAlias(), TrackProducerBase< reco::GsfTrack >::setConf(), TrackProducerBase< reco::GsfTrack >::setSrc(), AlCaHLTBitMon_QueryRunRegistry::string, and vertex.

     : GsfTrackProducerBase(iConfig.getParameter<bool>("TrajectoryInEvent"),
                            iConfig.getParameter<bool>("useHitsSplitting")),
       theAlgo(iConfig) {
   setConf(iConfig);
   setSrc(consumes<edm::View<reco::GsfTrack>>(iConfig.getParameter<edm::InputTag>("src")),
          consumes<reco::BeamSpot>(iConfig.getParameter<edm::InputTag>("beamSpot")),
          consumes<MeasurementTrackerEvent>(iConfig.getParameter<edm::InputTag>("MeasurementTrackerEvent")));
   setAlias(iConfig.getParameter<std::string>("@module_label"));
   std::string constraint_str = iConfig.getParameter<std::string>("constraint");
 
   if (constraint_str.empty())
     constraint_ = none;
   //   else if (constraint_str == "momentum") constraint_ = momentum;
   else if (constraint_str == "vertex") {
     constraint_ = vertex;
     gsfTrackVtxConstraintTag_ = consumes<GsfTrackVtxConstraintAssociationCollection>(
         iConfig.getParameter<edm::InputTag>("gsfTrackVtxConstraintTag"));
   } else {
     edm::LogError("GsfTrackRefitter") << "constraint: " << constraint_str
                                       << " not understood. Set it to 'momentum', 'vertex' or leave it empty";
     throw cms::Exception("GsfTrackRefitter")
         << "unknown type of contraint! Set it to 'momentum', 'vertex' or leave it empty";
   }
 
   //register your products
   produces<reco::GsfTrackCollection>().setBranchAlias(alias_ + "GsfTracks");
   produces<reco::TrackExtraCollection>().setBranchAlias(alias_ + "TrackExtras");
   produces<reco::GsfTrackExtraCollection>().setBranchAlias(alias_ + "GsfTrackExtras");
   produces<TrackingRecHitCollection>().setBranchAlias(alias_ + "RecHits");
   produces<std::vector<Trajectory>>();
   produces<TrajGsfTrackAssociationCollection>();
 }

Member Function Documentation

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

overridevirtual

Implementation of produce method.

Implements TrackProducerBase< reco::GsfTrack >.

Definition at line 52 of file GsfTrackRefitter.cc.

References TrackProducerBase< reco::GsfTrack >::bsSrc_, constraint_, MillePedeFileConverter_cfg::e, edm::HandleBase::failedToGet(), edm::EventSetup::get(), edm::Event::getByToken(), TrackProducerBase< reco::GsfTrack >::getFromES(), TrackProducerBase< reco::GsfTrack >::getFromEvt(), gsfTrackVtxConstraintTag_, edm::EventBase::id(), LogDebug, none, edm::ESHandle< T >::product(), GsfTrackProducerBase::putInEvt(), TrackProducerAlgorithm< T >::runWithTrack(), TrackProducerAlgorithm< T >::runWithVertex(), theAlgo, and vertex.

                                                                              {
   edm::LogInfo("GsfTrackRefitter") << "Analyzing event number: " << theEvent.id() << "\n";
   //
   // create empty output collections
   //
   std::unique_ptr<TrackingRecHitCollection> outputRHColl(new TrackingRecHitCollection);
   std::unique_ptr<reco::GsfTrackCollection> outputTColl(new reco::GsfTrackCollection);
   std::unique_ptr<reco::TrackExtraCollection> outputTEColl(new reco::TrackExtraCollection);
   std::unique_ptr<reco::GsfTrackExtraCollection> outputGsfTEColl(new reco::GsfTrackExtraCollection);
   std::unique_ptr<std::vector<Trajectory>> outputTrajectoryColl(new std::vector<Trajectory>);
 
   //
   //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;
   //  getFromES(setup,theG,theMF,theFitter,thePropagator);
   edm::ESHandle<TransientTrackingRecHitBuilder> theBuilder;
   getFromES(setup, theG, theMF, theFitter, thePropagator, theMeasTk, theBuilder);
 
   edm::ESHandle<TrackerTopology> httopo;
   setup.get<TrackerTopologyRcd>().get(httopo);
 
   //
   //declare and get TrackCollection to be retrieved from the event
   //
   AlgoProductCollection algoResults;
   reco::BeamSpot bs;
   switch (constraint_) {
     case none: {
       edm::Handle<edm::View<reco::GsfTrack>> theTCollection;
       getFromEvt(theEvent, theTCollection, bs);
       if (theTCollection.failedToGet()) {
         edm::LogError("GsfTrackRefitter") << "could not get the reco::GsfTrackCollection.";
         return;
       }
       LogDebug("GsfTrackRefitter") << "run the algorithm"
                                    << "\n";
       try {
         theAlgo.runWithTrack(theG.product(),
                              theMF.product(),
                              *theTCollection,
                              theFitter.product(),
                              thePropagator.product(),
                              theBuilder.product(),
                              bs,
                              algoResults);
       } catch (cms::Exception& e) {
         edm::LogError("TrackProducer") << "cms::Exception caught during theAlgo.runWithTrack."
                                        << "\n"
                                        << e << "\n";
         throw;
       }
       break;
     }
     case vertex: {
       edm::Handle<GsfTrackVtxConstraintAssociationCollection> theTCollectionWithConstraint;
       theEvent.getByToken(gsfTrackVtxConstraintTag_, theTCollectionWithConstraint);
       edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
       theEvent.getByToken(bsSrc_, recoBeamSpotHandle);
       bs = *recoBeamSpotHandle;
       if (theTCollectionWithConstraint.failedToGet()) {
         edm::LogError("TrackRefitter") << "could not get TrackVtxConstraintAssociationCollection product.";
         break;
       }
       LogDebug("TrackRefitter") << "run the algorithm"
                                 << "\n";
       try {
         theAlgo.runWithVertex(theG.product(),
                               theMF.product(),
                               *theTCollectionWithConstraint,
                               theFitter.product(),
                               thePropagator.product(),
                               theBuilder.product(),
                               bs,
                               algoResults);
       } catch (cms::Exception& e) {
         edm::LogError("TrackProducer") << "cms::Exception caught during theAlgo.runWithTrack."
                                        << "\n"
                                        << e << "\n";
         throw;
       }
     }
       //default... there cannot be any other possibility due to the check in the ctor
   }
 
   //put everything in th event
   putInEvt(theEvent,
            thePropagator.product(),
            theMeasTk.product(),
            outputRHColl,
            outputTColl,
            outputTEColl,
            outputGsfTEColl,
            outputTrajectoryColl,
            algoResults,
            theBuilder.product(),
            bs,
            httopo.product());
   LogDebug("GsfTrackRefitter") << "end"
                                << "\n";
 }