CMS 3D CMS Logo

List of all members | Public Member Functions | Static Public Member Functions | Private Attributes
TracksToTrajectories Class Reference
Inheritance diagram for TracksToTrajectories:
edm::stream::EDProducer< edm::GlobalCache< Count > >

Public Member Functions

void produce (edm::Event &, const edm::EventSetup &) override
 Convert a reco::TrackCollection into std::vector<Trajectory> More...
 
 TracksToTrajectories (const edm::ParameterSet &, const Count *)
 Constructor. More...
 
 ~TracksToTrajectories () override
 Destructor. More...
 
- Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< Count > >
 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
 

Static Public Member Functions

static void globalEndJob (Count const *iCount)
 
static std::unique_ptr< CountinitializeGlobalCache (edm::ParameterSet const &)
 

Private Attributes

edm::EDGetTokenT< reco::TrackCollectiontheTracksToken
 
std::unique_ptr< TrackTransformerBasetheTrackTransformer
 

Additional Inherited Members

- Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< Count > >
typedef CacheContexts< T... > CacheTypes
 
typedef CacheTypes::GlobalCache GlobalCache
 
typedef AbilityChecker< T... > HasAbility
 
typedef CacheTypes::LuminosityBlockCache LuminosityBlockCache
 
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCacheLuminosityBlockContext
 
typedef CacheTypes::LuminosityBlockSummaryCache LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache, GlobalCacheRunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 

Detailed Description

This class, which is a EDProducer, takes a reco::TrackCollection from the Event and refits the rechits strored in the reco::Tracks. The final result is a std::vector of Trajectories (objs of the type "Trajectory"), which is loaded into the Event in a transient way

Author
R. Bellan - INFN Torino ricca.nosp@m.rdo..nosp@m.bella.nosp@m.n@ce.nosp@m.rn.ch

Definition at line 43 of file TracksToTrajectories.cc.

Constructor & Destructor Documentation

◆ TracksToTrajectories()

TracksToTrajectories::TracksToTrajectories ( const edm::ParameterSet parameterSet,
const Count  
)

Constructor.

Definition at line 68 of file TracksToTrajectories.cc.

68  {
69  theTracksToken = consumes<reco::TrackCollection>(parameterSet.getParameter<InputTag>("Tracks"));
70 
71  ParameterSet trackTransformerParam = parameterSet.getParameter<ParameterSet>("TrackTransformer");
72 
73  string type = parameterSet.getParameter<string>("Type");
74 
75  if (type == "Default")
76  theTrackTransformer = std::make_unique<TrackTransformer>(trackTransformerParam, consumesCollector());
77  else if (type == "GlobalCosmicMuonsForAlignment")
78  theTrackTransformer = std::make_unique<TrackTransformerForGlobalCosmicMuons>(trackTransformerParam);
79  else if (type == "CosmicMuonsForAlignment")
80  theTrackTransformer = std::make_unique<TrackTransformerForCosmicMuons>(trackTransformerParam);
81  else {
82  throw cms::Exception("TracksToTrajectories")
83  << "The selected algorithm does not exist"
84  << "\n"
85  << "Possible choices are:"
86  << "\n"
87  << "Type = [Default, GlobalCosmicMuonsForAlignment, CosmicMuonsForAlignment]";
88  }
89 
90  produces<vector<Trajectory>>("Refitted");
91  produces<TrajTrackAssociationCollection>("Refitted");
92 }

References Exception, edm::ParameterSet::getParameter(), and edm::parameterSet().

◆ ~TracksToTrajectories()

TracksToTrajectories::~TracksToTrajectories ( )
override

Destructor.

Definition at line 95 of file TracksToTrajectories.cc.

95 {}

Member Function Documentation

◆ globalEndJob()

void TracksToTrajectories::globalEndJob ( Count const *  iCount)
static

Definition at line 97 of file TracksToTrajectories.cc.

97  {
98  constexpr char metname[] = "Reco|TrackingTools|TracksToTrajectories";
99 
100  auto theNFailures = iCount->theNFailures.load();
101  auto theNTracks = iCount->theNTracks.load();
102 
103  if (theNFailures != 0)
104  LogWarning(metname) << "During the refit there were " << theNFailures << " out of " << theNTracks
105  << " tracks, i.e. failure rate is: " << double(theNFailures) / theNTracks;
106  else {
107  LogTrace(metname) << "Refit of the tracks done without any failure";
108  }
109 }

References LogTrace, metname, tracksToTrajectories::Count::theNFailures, and tracksToTrajectories::Count::theNTracks.

◆ initializeGlobalCache()

static std::unique_ptr<Count> TracksToTrajectories::initializeGlobalCache ( edm::ParameterSet const &  )
inlinestatic

Definition at line 51 of file TracksToTrajectories.cc.

51 { return std::make_unique<Count>(); }

◆ produce()

void TracksToTrajectories::produce ( edm::Event event,
const edm::EventSetup setup 
)
override

Convert a reco::TrackCollection into std::vector<Trajectory>

Convert Tracks into Trajectories.

Definition at line 112 of file TracksToTrajectories.cc.

112  {
113 #ifdef EDM_ML_DEBUG
114  constexpr char metname[] = "Reco|TrackingTools|TracksToTrajectories";
115 #endif
116 
117  theTrackTransformer->setServices(setup);
118 
119  // Collection of Trajectory
120  auto trajectoryCollection = std::make_unique<vector<Trajectory>>();
121 
122  // Get the reference
123  RefProd<vector<Trajectory>> trajectoryCollectionRefProd = event.getRefBeforePut<vector<Trajectory>>("Refitted");
124 
125  // Get the RecTrack collection from the event
127  event.getByToken(theTracksToken, tracks);
128 
129  // Association map between Trajectory and Track
130  auto trajTrackMap = std::make_unique<TrajTrackAssociationCollection>(trajectoryCollectionRefProd, tracks);
131 
132  Ref<vector<Trajectory>>::key_type trajectoryIndex = 0;
133  reco::TrackRef::key_type trackIndex = 0;
134 
135  // Loop over the Rec tracks
136  for (auto const& newTrack : *tracks) {
137  ++(globalCache()->theNTracks);
138 
139  auto const& trajectoriesSM = theTrackTransformer->transform(newTrack);
140 
141  if (!trajectoriesSM.empty()) {
142  // Load the trajectory in the Trajectory Container
143  trajectoryCollection->push_back(trajectoriesSM.front());
144 
145  // Make the association between the Trajectory and the original Track
146  trajTrackMap->insert(Ref<vector<Trajectory>>(trajectoryCollectionRefProd, trajectoryIndex++),
147  reco::TrackRef(tracks, trackIndex++));
148  } else {
149  LogTrace(metname) << "Error in the Track refitting. This should not happen";
150  ++(globalCache()->theNFailures);
151  }
152  }
153  LogTrace(metname) << "Load the Trajectory Collection";
154  event.put(move(trajectoryCollection), "Refitted");
155  event.put(move(trajTrackMap), "Refitted");
156 }

References LogTrace, metname, eostools::move(), singleTopDQM_cfi::setup, and tracks.

Member Data Documentation

◆ theTracksToken

edm::EDGetTokenT<reco::TrackCollection> TracksToTrajectories::theTracksToken
private

Definition at line 60 of file TracksToTrajectories.cc.

◆ theTrackTransformer

std::unique_ptr<TrackTransformerBase> TracksToTrajectories::theTrackTransformer
private

Definition at line 61 of file TracksToTrajectories.cc.

edm::RefProd
Definition: EDProductfwd.h:25
TracksToTrajectories::theTrackTransformer
std::unique_ptr< TrackTransformerBase > theTrackTransformer
Definition: TracksToTrajectories.cc:61
edm::Handle< reco::TrackCollection >
edm::LogWarning
Log< level::Warning, false > LogWarning
Definition: MessageLogger.h:122
singleTopDQM_cfi.setup
setup
Definition: singleTopDQM_cfi.py:37
edm::Ref
Definition: AssociativeIterator.h:58
edm::parameterSet
ParameterSet const & parameterSet(StableProvenance const &provenance, ProcessHistory const &history)
Definition: Provenance.cc:11
TracksToTrajectories::theTracksToken
edm::EDGetTokenT< reco::TrackCollection > theTracksToken
Definition: TracksToTrajectories.cc:60
edm::ParameterSet
Definition: ParameterSet.h:47
tracks
const uint32_t *__restrict__ const HitContainer *__restrict__ TkSoA *__restrict__ tracks
Definition: CAHitNtupletGeneratorKernelsImpl.h:159
type
type
Definition: SiPixelVCal_PayloadInspector.cc:37
edm::Ref< TrackCollection >::key_type
std::remove_cv< typename std::remove_reference< argument_type >::type >::type key_type
Definition: Ref.h:164
eostools.move
def move(src, dest)
Definition: eostools.py:511
Exception
Definition: hltDiff.cc:245
edm::ParameterSet::getParameter
T getParameter(std::string const &) const
Definition: ParameterSet.h:303
LogTrace
#define LogTrace(id)
Definition: MessageLogger.h:234
edm::InputTag
Definition: InputTag.h:15
metname
const std::string metname
Definition: MuonSeedOrcaPatternRecognition.cc:40