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

Inheritance diagram for DualBzeroTrajectoryFactory:

Classes
struct	DualBzeroTrajectoryInput

Public Member Functions
DualBzeroTrajectoryFactory *	clone () const override

	DualBzeroTrajectoryFactory (const edm::ParameterSet &config, edm::ConsumesCollector &iC)

const ReferenceTrajectoryCollection	trajectories (const edm::EventSetup &setup, const ConstTrajTrackPairCollection &tracks, const reco::BeamSpot &beamSpot) const override
	Produce the reference trajectories. More...

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

	~DualBzeroTrajectoryFactory () override

Public Member Functions inherited from TrajectoryFactoryBase
const edm::ParameterSet &	configuration () const

MaterialEffects	materialEffects (void) const

PropagationDirection	propagationDirection (void) const

unsigned int	tracksPerTrajectory () const

	TrajectoryFactoryBase (const edm::ParameterSet &config, const edm::ConsumesCollector &iC)

	TrajectoryFactoryBase (const edm::ParameterSet &config, unsigned int tracksPerTrajectory, const edm::ConsumesCollector &iC)

virtual	~TrajectoryFactoryBase (void)

Public Attributes
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	m_MagFieldToken

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< TrajectoryStateOnSurface >	ExternalPredictionCollection

typedef ReferenceTrajectoryBase::MaterialEffects	MaterialEffects

typedef std::vector< ReferenceTrajectoryPtr >	ReferenceTrajectoryCollection

typedef ReferenceTrajectoryBase::ReferenceTrajectoryPtr	ReferenceTrajectoryPtr

typedef std::pair< TrajectoryStateOnSurface, TransientTrackingRecHit::ConstRecHitContainer >	TrajectoryInput

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,
		edm::ConsumesCollector &	iC
	)

Definition at line 56 of file DualBzeroTrajectoryFactory.cc.

References theMass, and theMomentumEstimate.

Referenced by clone().

     : TrajectoryFactoryBase(config, iC), m_MagFieldToken(iC.esConsumes()) {
   theMass = config.getParameter<double>("ParticleMass");
   theMomentumEstimate = config.getParameter<double>("MomentumEstimate");
 }

◆ ~DualBzeroTrajectoryFactory()

DualBzeroTrajectoryFactory::~DualBzeroTrajectoryFactory ( void )

override

Definition at line 62 of file DualBzeroTrajectoryFactory.cc.

62 {}

Member Function Documentation

◆ clone()

DualBzeroTrajectoryFactory* DualBzeroTrajectoryFactory::clone ( void ) const

inlineoverridevirtual

Implements TrajectoryFactoryBase.

Definition at line 33 of file DualBzeroTrajectoryFactory.cc.

References DualBzeroTrajectoryFactory().

33 { return new DualBzeroTrajectoryFactory(*this); }

DualBzeroTrajectoryFactory::DualBzeroTrajectoryFactory

DualBzeroTrajectoryFactory(const edm::ParameterSet &config, edm::ConsumesCollector &iC)

Definition: DualBzeroTrajectoryFactory.cc:56

◆ propagateExternal()

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

protected

Definition at line 194 of file DualBzeroTrajectoryFactory.cc.

References anyDirection, and TrackCandidateProducer_cfi::propagator.

Referenced by trajectories().

                                                                                                                   {
   AnalyticalPropagator propagator(magField, anyDirection);
   const std::pair<TrajectoryStateOnSurface, double> tsosWithPath = propagator.propagateWithPath(external, surface);
   return tsosWithPath.first;
 }

◆ referenceStateAndRecHits()

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

protected

Definition at line 150 of file DualBzeroTrajectoryFactory.cc.

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

Referenced by trajectories().

                                            {
   DualBzeroTrajectoryInput input;
 
   // get the trajectory measurements in the correct order, i.e. reverse if needed
   Trajectory::DataContainer allTrajMeas = this->orderedTrajectoryMeasurements(*track.first);
   Trajectory::DataContainer usedTrajMeas;
   Trajectory::DataContainer::iterator itM;
   // get all relevant trajectory measurements
   for (itM = allTrajMeas.begin(); itM != allTrajMeas.end(); itM++) {
     if (useRecHit((*itM).recHit()))
       usedTrajMeas.push_back(*itM);
   }
 
   unsigned int iMeas = 0;
   unsigned int nMeas = usedTrajMeas.size();
   unsigned int nRefStateMeas = nMeas / 2;
   // get the valid RecHits
   for (itM = usedTrajMeas.begin(); itM != usedTrajMeas.end(); itM++, iMeas++) {
     TransientTrackingRecHit::ConstRecHitPointer aRecHit = (*itM).recHit();
 
     if (iMeas < nRefStateMeas) {
       input.bwdRecHits.push_back(aRecHit);
     } else if (iMeas > nRefStateMeas) {
       input.fwdRecHits.push_back(aRecHit);
     } else {  // iMeas == nRefStateMeas
       if ((*itM).updatedState().isValid()) {
         input.refTsos = (*itM).updatedState();
         input.bwdRecHits.push_back(aRecHit);
         input.fwdRecHits.push_back(aRecHit);
       } else {
         // if the tsos of the middle hit is not valid, try the next one ...
         nRefStateMeas++;
         input.bwdRecHits.push_back(aRecHit);
       }
     }
   }
 
   // bring input.fwdRecHits into correct order
   std::reverse(input.bwdRecHits.begin(), input.bwdRecHits.end());
 
   return input;
 }

◆ trajectories() [1/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 64 of file DualBzeroTrajectoryFactory.cc.

References TrajectoryFactoryBase::allowZeroMaterial_, pwdgSkimBPark_cfi::beamSpot, DiMuonV_cfg::config, TrajectoryFactoryBase::includeAPEs_, input, m_MagFieldToken, HLT_2024v14_cff::magneticField, TrajectoryFactoryBase::materialEffects(), TrajectoryFactoryBase::propagationDirection(), referenceStateAndRecHits(), singleTopDQM_cfi::setup, theMass, theMomentumEstimate, DiMuonV_cfg::tracks, and TrajectoryFactoryBase::useBeamSpot_.

Referenced by trajectories().

                                                                                                                 {
   ReferenceTrajectoryCollection trajectories;
 
   const MagneticField *magneticField = &setup.getData(m_MagFieldToken);
 
   ConstTrajTrackPairCollection::const_iterator itTracks = tracks.begin();
 
   while (itTracks != tracks.end()) {
     const DualBzeroTrajectoryInput input = this->referenceStateAndRecHits(*itTracks);
     // Check input: If all hits were rejected, the TSOS is initialized as invalid.
     if (input.refTsos.isValid()) {
       ReferenceTrajectoryBase::Config config(materialEffects(), propagationDirection(), theMass, theMomentumEstimate);
       config.useBeamSpot = useBeamSpot_;
       config.includeAPEs = includeAPEs_;
       config.allowZeroMaterial = allowZeroMaterial_;
       ReferenceTrajectoryPtr ptr(new DualBzeroReferenceTrajectory(
           input.refTsos, input.fwdRecHits, input.bwdRecHits, magneticField, beamSpot, config));
       trajectories.push_back(ptr);
     }
 
     ++itTracks;
   }
 
   return trajectories;
 }

◆ trajectories() [2/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.

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

                                         {
   ReferenceTrajectoryCollection trajectories;
 
   if (tracks.size() != external.size()) {
     edm::LogInfo("ReferenceTrajectories")
         << "@SUB=DualBzeroTrajectoryFactory::trajectories"
         << "Inconsistent input:\n"
         << "\tnumber of tracks = " << tracks.size() << "\tnumber of external predictions = " << external.size();
     return trajectories;
   }
 
   const MagneticField *magneticField = &setup.getData(m_MagFieldToken);
 
   ConstTrajTrackPairCollection::const_iterator itTracks = tracks.begin();
   ExternalPredictionCollection::const_iterator itExternal = external.begin();
 
   while (itTracks != tracks.end()) {
     const DualBzeroTrajectoryInput input = referenceStateAndRecHits(*itTracks);
     // Check input: If all hits were rejected, the TSOS is initialized as invalid.
     if (input.refTsos.isValid()) {
       if ((*itExternal).isValid()) {
         TrajectoryStateOnSurface propExternal = propagateExternal(*itExternal, input.refTsos.surface(), magneticField);
 
         if (!propExternal.isValid())
           continue;
 
         ReferenceTrajectoryBase::Config config(materialEffects(), propagationDirection(), theMass, theMomentumEstimate);
         config.useBeamSpot = useBeamSpot_;
         config.includeAPEs = includeAPEs_;
         config.allowZeroMaterial = allowZeroMaterial_;
         ReferenceTrajectoryPtr ptr(new DualBzeroReferenceTrajectory(
             propExternal, input.fwdRecHits, input.bwdRecHits, magneticField, beamSpot, config));
 
         AlgebraicSymMatrix externalParamErrors(asHepMatrix<5>(propExternal.localError().matrix()));
         ptr->setParameterErrors(externalParamErrors.sub(2, 5));
         trajectories.push_back(ptr);
       } else {
         ReferenceTrajectoryBase::Config config(materialEffects(), propagationDirection(), theMass, theMomentumEstimate);
         config.useBeamSpot = useBeamSpot_;
         config.includeAPEs = includeAPEs_;
         config.allowZeroMaterial = allowZeroMaterial_;
         ReferenceTrajectoryPtr ptr(new DualBzeroReferenceTrajectory(
             input.refTsos, input.fwdRecHits, input.bwdRecHits, magneticField, beamSpot, config));
 
         trajectories.push_back(ptr);
       }
     }
 
     ++itTracks;
     ++itExternal;
   }
 
   return trajectories;
 }