DualTrajectoryFactory Class Reference

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

Inheritance diagram for DualTrajectoryFactory:

Classes
struct	DualTrajectoryInput

Public Member Functions
DualTrajectoryFactory *	clone () const override
	DualTrajectoryFactory (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
	~DualTrajectoryFactory () 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 DualTrajectoryInput	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
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 DualTrajectoryFactory.cc.

Constructor & Destructor Documentation

DualTrajectoryFactory()

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

Definition at line 55 of file DualTrajectoryFactory.cc.

References theMass.

Referenced by clone().

    : TrajectoryFactoryBase(config, iC),
      m_MagFieldToken(iC.esConsumes()),
      theMass(config.getParameter<double>("ParticleMass")) {
  edm::LogInfo("Alignment") << "@SUB=DualTrajectoryFactory"
                            << "mass: " << theMass;
}

~DualTrajectoryFactory()

DualTrajectoryFactory::~DualTrajectoryFactory ( void  )

override

Definition at line 63 of file DualTrajectoryFactory.cc.

{}

Member Function Documentation

clone()

DualTrajectoryFactory* DualTrajectoryFactory::clone ( void  ) const

inlineoverridevirtual

Implements TrajectoryFactoryBase.

Definition at line 33 of file DualTrajectoryFactory.cc.

References DualTrajectoryFactory().

{ return new DualTrajectoryFactory(*this); }

propagateExternal()

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

protected

Definition at line 207 of file DualTrajectoryFactory.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 DualTrajectoryFactory::DualTrajectoryInput DualTrajectoryFactory::referenceStateAndRecHits ( const ConstTrajTrackPair &  track ) const

protected

Definition at line 163 of file DualTrajectoryFactory.cc.

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

Referenced by trajectories().

                                           {
  DualTrajectoryInput 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 DualTrajectoryFactory::ReferenceTrajectoryCollection DualTrajectoryFactory::trajectories ( const edm::EventSetup &  setup, const ConstTrajTrackPairCollection &  tracks, const reco::BeamSpot &  beamSpot  ) const

overridevirtual

Produce the reference trajectories.

Implements TrajectoryFactoryBase.

Definition at line 65 of file DualTrajectoryFactory.cc.

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

Referenced by trajectories().

                                                                                                                {
  ReferenceTrajectoryCollection trajectories;

  const MagneticField *magneticField = &setup.getData(m_MagFieldToken);

  if (magneticField->inTesla(GlobalPoint(0., 0., 0.)).mag2() < 1.e-6) {
    edm::LogWarning("Alignment") << "@SUB=DualTrajectoryFactory::trajectories"
                                 << "B-field in z is " << magneticField->inTesla(GlobalPoint(0., 0., 0.)).z()
                                 << ": You should probably use the DualBzeroTrajectoryFactory\n"
                                 << "or fix this code to switch automatically as the ReferenceTrajectoryFactory does.";
  }

  ConstTrajTrackPairCollection::const_iterator itTracks = tracks.begin();

  while (itTracks != tracks.end()) {
    const DualTrajectoryInput 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);
      config.useBeamSpot = useBeamSpot_;
      config.includeAPEs = includeAPEs_;
      config.allowZeroMaterial = allowZeroMaterial_;
      ReferenceTrajectoryPtr ptr(new DualReferenceTrajectory(
          input.refTsos, input.fwdRecHits, input.bwdRecHits, magneticField, beamSpot, config));
      trajectories.push_back(ptr);
    }

    ++itTracks;
  }

  return trajectories;
}

trajectories() [2/2]

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

overridevirtual

Implements TrajectoryFactoryBase.

Definition at line 99 of file DualTrajectoryFactory.cc.

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

                                        {
  ReferenceTrajectoryCollection trajectories;

  if (tracks.size() != external.size()) {
    edm::LogInfo("ReferenceTrajectories")
        << "@SUB=DualTrajectoryFactory::trajectories"
        << "Inconsistent input:\n"
        << "\tnumber of tracks = " << tracks.size() << "\tnumber of external predictions = " << external.size();
    return trajectories;
  }
  const MagneticField *magneticField = &setup.getData(m_MagFieldToken);

  if (magneticField->inTesla(GlobalPoint(0., 0., 0.)).mag2() < 1.e-6) {
    edm::LogWarning("Alignment") << "@SUB=DualTrajectoryFactory::trajectories"
                                 << "B-field in z is " << magneticField->inTesla(GlobalPoint(0., 0., 0.)).z()
                                 << ": You should probably use the DualBzeroTrajectoryFactory\n"
                                 << "or fix this code to switch automatically as the ReferenceTrajectoryFactory does.";
  }

  ConstTrajTrackPairCollection::const_iterator itTracks = tracks.begin();
  ExternalPredictionCollection::const_iterator itExternal = external.begin();

  while (itTracks != tracks.end()) {
    const DualTrajectoryInput 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);
        config.useBeamSpot = useBeamSpot_;
        config.includeAPEs = includeAPEs_;
        config.allowZeroMaterial = allowZeroMaterial_;
        ReferenceTrajectoryPtr ptr(new DualReferenceTrajectory(
            propExternal, input.fwdRecHits, input.bwdRecHits, magneticField, beamSpot, config));

        AlgebraicSymMatrix externalParamErrors(asHepMatrix<5>(propExternal.localError().matrix()));
        ptr->setParameterErrors(externalParamErrors);
        trajectories.push_back(ptr);
      } else {
        ReferenceTrajectoryBase::Config config(materialEffects(), propagationDirection(), theMass);
        config.useBeamSpot = useBeamSpot_;
        config.includeAPEs = includeAPEs_;
        config.allowZeroMaterial = allowZeroMaterial_;
        ReferenceTrajectoryPtr ptr(new DualReferenceTrajectory(
            input.refTsos, input.fwdRecHits, input.bwdRecHits, magneticField, beamSpot, config));
        trajectories.push_back(ptr);
      }
    }

    ++itTracks;
    ++itExternal;
  }

  return trajectories;
}

Member Data Documentation

m_MagFieldToken

const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> DualTrajectoryFactory::m_MagFieldToken

Definition at line 21 of file DualTrajectoryFactory.cc.

Referenced by trajectories().

theMass

double DualTrajectoryFactory::theMass

protected

Definition at line 48 of file DualTrajectoryFactory.cc.

Referenced by DualTrajectoryFactory(), and trajectories().