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

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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () 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

◆ Constraint

enum GsfTrackRefitter::Constraint

private

Enumerator
none
vertex

Definition at line 24 of file GsfTrackRefitter.h.

                   {
     none,
     //          momentum,
     vertex
   };

Constructor & Destructor Documentation

◆ GsfTrackRefitter()

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

explicit

Constructor.

Definition at line 18 of file GsfTrackRefitter.cc.

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

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.

Member Function Documentation

◆ produce()

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.

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