CMS 3D CMS Logo

List of all members | Classes | Public Member Functions | Protected Member Functions | Protected Attributes
DualBzeroTrajectoryFactory Class Reference

A factory that produces instances of class ReferenceTrajectory from a given TrajTrackPairCollection. More...

Inheritance diagram for DualBzeroTrajectoryFactory:
TrajectoryFactoryBase

Classes

struct  DualBzeroTrajectoryInput
 

Public Member Functions

DualBzeroTrajectoryFactoryclone () const override
 
 DualBzeroTrajectoryFactory (const edm::ParameterSet &config)
 
const ReferenceTrajectoryCollection trajectories (const edm::EventSetup &setup, const ConstTrajTrackPairCollection &tracks, const ExternalPredictionCollection &external, const reco::BeamSpot &beamSpot) const override
 
const ReferenceTrajectoryCollection trajectories (const edm::EventSetup &setup, const ConstTrajTrackPairCollection &tracks, const reco::BeamSpot &beamSpot) const override
 Produce the reference trajectories. More...
 
 ~DualBzeroTrajectoryFactory () override
 
- Public Member Functions inherited from TrajectoryFactoryBase
const edm::ParameterSetconfiguration () const
 
MaterialEffects materialEffects (void) const
 
PropagationDirection propagationDirection (void) const
 
unsigned int tracksPerTrajectory () const
 
 TrajectoryFactoryBase (const edm::ParameterSet &config)
 
 TrajectoryFactoryBase (const edm::ParameterSet &config, unsigned int tracksPerTrajectory)
 
virtual ~TrajectoryFactoryBase (void)
 

Protected Member Functions

const TrajectoryStateOnSurface propagateExternal (const TrajectoryStateOnSurface &external, const Surface &surface, const MagneticField *magField) const
 
const DualBzeroTrajectoryInput referenceStateAndRecHits (const ConstTrajTrackPair &track) const
 
- Protected Member Functions inherited from TrajectoryFactoryBase
virtual const TrajectoryInput innermostStateAndRecHits (const ConstTrajTrackPair &track) const
 
virtual const Trajectory::DataContainer orderedTrajectoryMeasurements (const Trajectory &trajectory) const
 
bool sameSurface (const Surface &s1, const Surface &s2) const
 
bool useRecHit (const TransientTrackingRecHit::ConstRecHitPointer &hitPtr) const
 

Protected Attributes

double theMass
 
double theMomentumEstimate
 
- Protected Attributes inherited from TrajectoryFactoryBase
const bool allowZeroMaterial_
 
const bool includeAPEs_
 
const bool useBeamSpot_
 

Additional Inherited Members

- Public Types inherited from TrajectoryFactoryBase
typedef AlignmentAlgorithmBase::ConstTrajTrackPair ConstTrajTrackPair
 
typedef AlignmentAlgorithmBase::ConstTrajTrackPairCollection ConstTrajTrackPairCollection
 
typedef std::vector< TrajectoryStateOnSurfaceExternalPredictionCollection
 
typedef ReferenceTrajectoryBase::MaterialEffects MaterialEffects
 
typedef std::vector< ReferenceTrajectoryPtrReferenceTrajectoryCollection
 
typedef ReferenceTrajectoryBase::ReferenceTrajectoryPtr ReferenceTrajectoryPtr
 
typedef std::pair< TrajectoryStateOnSurface, TransientTrackingRecHit::ConstRecHitContainerTrajectoryInput
 

Detailed Description

A factory that produces instances of class ReferenceTrajectory from a given TrajTrackPairCollection.

Definition at line 17 of file DualBzeroTrajectoryFactory.cc.

Constructor & Destructor Documentation

◆ DualBzeroTrajectoryFactory()

DualBzeroTrajectoryFactory::DualBzeroTrajectoryFactory ( const edm::ParameterSet config)

Definition at line 55 of file DualBzeroTrajectoryFactory.cc.

57  theMass = config.getParameter<double>("ParticleMass");
58  theMomentumEstimate = config.getParameter<double>("MomentumEstimate");
59 }

References theMass, and theMomentumEstimate.

Referenced by clone().

◆ ~DualBzeroTrajectoryFactory()

DualBzeroTrajectoryFactory::~DualBzeroTrajectoryFactory ( void  )
override

Definition at line 61 of file DualBzeroTrajectoryFactory.cc.

61 {}

Member Function Documentation

◆ clone()

DualBzeroTrajectoryFactory* DualBzeroTrajectoryFactory::clone ( void  ) const
inlineoverridevirtual

Implements TrajectoryFactoryBase.

Definition at line 32 of file DualBzeroTrajectoryFactory.cc.

32 { return new DualBzeroTrajectoryFactory(*this); }

References DualBzeroTrajectoryFactory().

◆ propagateExternal()

const TrajectoryStateOnSurface DualBzeroTrajectoryFactory::propagateExternal ( const TrajectoryStateOnSurface external,
const Surface surface,
const MagneticField magField 
) const
protected

Definition at line 196 of file DualBzeroTrajectoryFactory.cc.

198  {
200  const std::pair<TrajectoryStateOnSurface, double> tsosWithPath = propagator.propagateWithPath(external, surface);
201  return tsosWithPath.first;
202 }

References anyDirection, and TrackCandidateProducer_cfi::propagator.

Referenced by trajectories().

◆ referenceStateAndRecHits()

const DualBzeroTrajectoryFactory::DualBzeroTrajectoryInput DualBzeroTrajectoryFactory::referenceStateAndRecHits ( const ConstTrajTrackPair track) const
protected

Definition at line 152 of file DualBzeroTrajectoryFactory.cc.

153  {
154  DualBzeroTrajectoryInput input;
155 
156  // get the trajectory measurements in the correct order, i.e. reverse if needed
158  Trajectory::DataContainer usedTrajMeas;
159  Trajectory::DataContainer::iterator itM;
160  // get all relevant trajectory measurements
161  for (itM = allTrajMeas.begin(); itM != allTrajMeas.end(); itM++) {
162  if (useRecHit((*itM).recHit()))
163  usedTrajMeas.push_back(*itM);
164  }
165 
166  unsigned int iMeas = 0;
167  unsigned int nMeas = usedTrajMeas.size();
168  unsigned int nRefStateMeas = nMeas / 2;
169  // get the valid RecHits
170  for (itM = usedTrajMeas.begin(); itM != usedTrajMeas.end(); itM++, iMeas++) {
171  TransientTrackingRecHit::ConstRecHitPointer aRecHit = (*itM).recHit();
172 
173  if (iMeas < nRefStateMeas) {
174  input.bwdRecHits.push_back(aRecHit);
175  } else if (iMeas > nRefStateMeas) {
176  input.fwdRecHits.push_back(aRecHit);
177  } else { // iMeas == nRefStateMeas
178  if ((*itM).updatedState().isValid()) {
179  input.refTsos = (*itM).updatedState();
180  input.bwdRecHits.push_back(aRecHit);
181  input.fwdRecHits.push_back(aRecHit);
182  } else {
183  // if the tsos of the middle hit is not valid, try the next one ...
184  nRefStateMeas++;
185  input.bwdRecHits.push_back(aRecHit);
186  }
187  }
188  }
189 
190  // bring input.fwdRecHits into correct order
191  std::reverse(input.bwdRecHits.begin(), input.bwdRecHits.end());
192 
193  return input;
194 }

References input, TrajectoryFactoryBase::orderedTrajectoryMeasurements(), groupFilesInBlocks::reverse, and TrajectoryFactoryBase::useRecHit().

Referenced by trajectories().

◆ trajectories() [1/2]

const DualBzeroTrajectoryFactory::ReferenceTrajectoryCollection DualBzeroTrajectoryFactory::trajectories ( const edm::EventSetup setup,
const ConstTrajTrackPairCollection tracks,
const ExternalPredictionCollection external,
const reco::BeamSpot beamSpot 
) const
overridevirtual

Implements TrajectoryFactoryBase.

Definition at line 91 of file DualBzeroTrajectoryFactory.cc.

95  {
97 
98  if (tracks.size() != external.size()) {
99  edm::LogInfo("ReferenceTrajectories")
100  << "@SUB=DualBzeroTrajectoryFactory::trajectories"
101  << "Inconsistent input:\n"
102  << "\tnumber of tracks = " << tracks.size() << "\tnumber of external predictions = " << external.size();
103  return trajectories;
104  }
105 
108 
109  ConstTrajTrackPairCollection::const_iterator itTracks = tracks.begin();
110  ExternalPredictionCollection::const_iterator itExternal = external.begin();
111 
112  while (itTracks != tracks.end()) {
113  const DualBzeroTrajectoryInput input = referenceStateAndRecHits(*itTracks);
114  // Check input: If all hits were rejected, the TSOS is initialized as invalid.
115  if (input.refTsos.isValid()) {
116  if ((*itExternal).isValid()) {
117  TrajectoryStateOnSurface propExternal =
118  propagateExternal(*itExternal, input.refTsos.surface(), magneticField.product());
119 
120  if (!propExternal.isValid())
121  continue;
122 
124  config.useBeamSpot = useBeamSpot_;
125  config.includeAPEs = includeAPEs_;
126  config.allowZeroMaterial = allowZeroMaterial_;
128  propExternal, input.fwdRecHits, input.bwdRecHits, magneticField.product(), beamSpot, config));
129 
130  AlgebraicSymMatrix externalParamErrors(asHepMatrix<5>(propExternal.localError().matrix()));
131  ptr->setParameterErrors(externalParamErrors.sub(2, 5));
132  trajectories.push_back(ptr);
133  } else {
135  config.useBeamSpot = useBeamSpot_;
136  config.includeAPEs = includeAPEs_;
137  config.allowZeroMaterial = allowZeroMaterial_;
139  input.refTsos, input.fwdRecHits, input.bwdRecHits, magneticField.product(), beamSpot, config));
140 
141  trajectories.push_back(ptr);
142  }
143  }
144 
145  ++itTracks;
146  ++itExternal;
147  }
148 
149  return trajectories;
150 }

References TrajectoryFactoryBase::allowZeroMaterial_, pwdgSkimBPark_cfi::beamSpot, looper::config, get, TrajectoryFactoryBase::includeAPEs_, input, TrajectoryStateOnSurface::isValid(), TrajectoryStateOnSurface::localError(), HLT_2018_cff::magneticField, TrajectoryFactoryBase::materialEffects(), LocalTrajectoryError::matrix(), propagateExternal(), TrajectoryFactoryBase::propagationDirection(), referenceStateAndRecHits(), singleTopDQM_cfi::setup, theMass, theMomentumEstimate, PDWG_EXOHSCP_cff::tracks, trajectories(), and TrajectoryFactoryBase::useBeamSpot_.

◆ trajectories() [2/2]

const DualBzeroTrajectoryFactory::ReferenceTrajectoryCollection DualBzeroTrajectoryFactory::trajectories ( const edm::EventSetup setup,
const ConstTrajTrackPairCollection tracks,
const reco::BeamSpot beamSpot 
) const
overridevirtual

Produce the reference trajectories.

Implements TrajectoryFactoryBase.

Definition at line 63 of file DualBzeroTrajectoryFactory.cc.

64  {
66 
69 
70  ConstTrajTrackPairCollection::const_iterator itTracks = tracks.begin();
71 
72  while (itTracks != tracks.end()) {
73  const DualBzeroTrajectoryInput input = this->referenceStateAndRecHits(*itTracks);
74  // Check input: If all hits were rejected, the TSOS is initialized as invalid.
75  if (input.refTsos.isValid()) {
77  config.useBeamSpot = useBeamSpot_;
78  config.includeAPEs = includeAPEs_;
79  config.allowZeroMaterial = allowZeroMaterial_;
81  input.refTsos, input.fwdRecHits, input.bwdRecHits, magneticField.product(), beamSpot, config));
82  trajectories.push_back(ptr);
83  }
84 
85  ++itTracks;
86  }
87 
88  return trajectories;
89 }

References TrajectoryFactoryBase::allowZeroMaterial_, pwdgSkimBPark_cfi::beamSpot, looper::config, get, TrajectoryFactoryBase::includeAPEs_, input, HLT_2018_cff::magneticField, TrajectoryFactoryBase::materialEffects(), TrajectoryFactoryBase::propagationDirection(), referenceStateAndRecHits(), singleTopDQM_cfi::setup, theMass, theMomentumEstimate, PDWG_EXOHSCP_cff::tracks, and TrajectoryFactoryBase::useBeamSpot_.

Referenced by trajectories().

Member Data Documentation

◆ theMass

double DualBzeroTrajectoryFactory::theMass
protected

Definition at line 47 of file DualBzeroTrajectoryFactory.cc.

Referenced by DualBzeroTrajectoryFactory(), and trajectories().

◆ theMomentumEstimate

double DualBzeroTrajectoryFactory::theMomentumEstimate
protected

Definition at line 48 of file DualBzeroTrajectoryFactory.cc.

Referenced by DualBzeroTrajectoryFactory(), and trajectories().

TrajectoryFactoryBase::useRecHit
bool useRecHit(const TransientTrackingRecHit::ConstRecHitPointer &hitPtr) const
Definition: TrajectoryFactoryBase.cc:94
PDWG_EXOHSCP_cff.tracks
tracks
Definition: PDWG_EXOHSCP_cff.py:28
anyDirection
Definition: PropagationDirection.h:4
pwdgSkimBPark_cfi.beamSpot
beamSpot
Definition: pwdgSkimBPark_cfi.py:5
input
static const std::string input
Definition: EdmProvDump.cc:48
HLT_2018_cff.magneticField
magneticField
Definition: HLT_2018_cff.py:348
DualBzeroTrajectoryFactory::trajectories
const ReferenceTrajectoryCollection trajectories(const edm::EventSetup &setup, const ConstTrajTrackPairCollection &tracks, const reco::BeamSpot &beamSpot) const override
Produce the reference trajectories.
Definition: DualBzeroTrajectoryFactory.cc:63
edm::LogInfo
Definition: MessageLogger.h:254
TrajectoryFactoryBase::includeAPEs_
const bool includeAPEs_
Definition: TrajectoryFactoryBase.h:71
DualBzeroTrajectoryFactory::theMomentumEstimate
double theMomentumEstimate
Definition: DualBzeroTrajectoryFactory.cc:48
ReferenceTrajectoryBase::Config
Definition: ReferenceTrajectoryBase.h:121
LocalTrajectoryError::matrix
const AlgebraicSymMatrix55 & matrix() const
Definition: LocalTrajectoryError.h:60
DualBzeroTrajectoryFactory::propagateExternal
const TrajectoryStateOnSurface propagateExternal(const TrajectoryStateOnSurface &external, const Surface &surface, const MagneticField *magField) const
Definition: DualBzeroTrajectoryFactory.cc:196
singleTopDQM_cfi.setup
setup
Definition: singleTopDQM_cfi.py:37
groupFilesInBlocks.reverse
reverse
Definition: groupFilesInBlocks.py:131
IdealMagneticFieldRecord
Definition: IdealMagneticFieldRecord.h:11
config
Definition: config.py:1
external
Definition: HEPTopTaggerV2.h:20
TrajectoryStateOnSurface
Definition: TrajectoryStateOnSurface.h:16
TrackCandidateProducer_cfi.propagator
propagator
Definition: TrackCandidateProducer_cfi.py:17
DualBzeroTrajectoryFactory::theMass
double theMass
Definition: DualBzeroTrajectoryFactory.cc:47
Trajectory::DataContainer
std::vector< TrajectoryMeasurement > DataContainer
Definition: Trajectory.h:40
TrajectoryFactoryBase::ReferenceTrajectoryPtr
ReferenceTrajectoryBase::ReferenceTrajectoryPtr ReferenceTrajectoryPtr
Definition: TrajectoryFactoryBase.h:22
TrajectoryFactoryBase::useBeamSpot_
const bool useBeamSpot_
Definition: TrajectoryFactoryBase.h:70
edm::ESHandle< MagneticField >
TrajectoryFactoryBase::propagationDirection
PropagationDirection propagationDirection(void) const
Definition: TrajectoryFactoryBase.h:46
looper.config
config
Definition: looper.py:291
TrajectoryFactoryBase::ReferenceTrajectoryCollection
std::vector< ReferenceTrajectoryPtr > ReferenceTrajectoryCollection
Definition: TrajectoryFactoryBase.h:26
DualBzeroTrajectoryFactory::referenceStateAndRecHits
const DualBzeroTrajectoryInput referenceStateAndRecHits(const ConstTrajTrackPair &track) const
Definition: DualBzeroTrajectoryFactory.cc:152
TrackingRecHit::ConstRecHitPointer
std::shared_ptr< TrackingRecHit const > ConstRecHitPointer
Definition: TrackingRecHit.h:25
AlgebraicSymMatrix
CLHEP::HepSymMatrix AlgebraicSymMatrix
Definition: AlgebraicObjects.h:15
TrajectoryFactoryBase::materialEffects
MaterialEffects materialEffects(void) const
Definition: TrajectoryFactoryBase.h:45
get
#define get
TrajectoryFactoryBase::allowZeroMaterial_
const bool allowZeroMaterial_
Definition: TrajectoryFactoryBase.h:72
DualBzeroTrajectoryFactory::DualBzeroTrajectoryFactory
DualBzeroTrajectoryFactory(const edm::ParameterSet &config)
Definition: DualBzeroTrajectoryFactory.cc:55
AnalyticalPropagator
Definition: AnalyticalPropagator.h:22
DualBzeroReferenceTrajectory
Definition: DualBzeroReferenceTrajectory.h:37
HLT_2018_cff.track
track
Definition: HLT_2018_cff.py:10352
TrajectoryStateOnSurface::localError
const LocalTrajectoryError & localError() const
Definition: TrajectoryStateOnSurface.h:77
TrajectoryFactoryBase::TrajectoryFactoryBase
TrajectoryFactoryBase(const edm::ParameterSet &config)
Definition: TrajectoryFactoryBase.cc:9
TrajectoryStateOnSurface::isValid
bool isValid() const
Definition: TrajectoryStateOnSurface.h:54
TrajectoryFactoryBase::orderedTrajectoryMeasurements
virtual const Trajectory::DataContainer orderedTrajectoryMeasurements(const Trajectory &trajectory) const
Definition: TrajectoryFactoryBase.cc:65