#include <GsfTrajectorySmoother.h>

Inheritance diagram for GsfTrajectorySmoother:

Public Member Functions
GsfTrajectorySmoother *	clone () const override

const MeasurementEstimator *	estimator () const

	GsfTrajectorySmoother (const GsfPropagatorWithMaterial &aPropagator, const TrajectoryStateUpdator &aUpdator, const MeasurementEstimator &aEstimator, const MultiTrajectoryStateMerger &merger, float errorRescaling, const bool materialBeforeUpdate=true, const DetLayerGeometry *detLayerGeometry=0)

void	setHitCloner (TkCloner const *hc) override

Trajectory	trajectory (const Trajectory &aTraj) const override

const TrajectoryStateUpdator *	updator () const

	~GsfTrajectorySmoother () override

Public Member Functions inherited from TrajectorySmoother
virtual TrajectoryContainer	trajectories (const Trajectory &traj) const

virtual	~TrajectorySmoother ()

Private Types
typedef TrajectoryMeasurement	TM

typedef TrajectoryStateOnSurface	TSOS

Private Attributes
const DetLayerGeometry	dummyGeometry

const GsfPropagatorWithMaterial *	theAlongPropagator

const FullConvolutionWithMaterial *	theConvolutor

float	theErrorRescaling

const MeasurementEstimator *	theEstimator

const DetLayerGeometry *	theGeometry

const GsfPropagatorAdapter *	theGeomPropagator

bool	theMatBeforeUpdate

const MultiTrajectoryStateMerger *	theMerger

const GsfPropagatorWithMaterial *	theOppositePropagator

bool	theTiming

const TrajectoryStateUpdator *	theUpdator

Additional Inherited Members
Public Types inherited from TrajectorySmoother
typedef std::vector< Trajectory >	TrajectoryContainer

typedef TrajectoryContainer::iterator	TrajectoryIterator

Detailed Description

A GSF smoother, similar to KFTrajectorySmoother, but (for testing purposes) without combination with the forward fit.

Definition at line 21 of file GsfTrajectorySmoother.h.

Member Typedef Documentation

typedef TrajectoryMeasurement GsfTrajectorySmoother::TM

private

Definition at line 24 of file GsfTrajectorySmoother.h.

typedef TrajectoryStateOnSurface GsfTrajectorySmoother::TSOS

private

Definition at line 23 of file GsfTrajectorySmoother.h.

Constructor & Destructor Documentation

GsfTrajectorySmoother::GsfTrajectorySmoother	(	const GsfPropagatorWithMaterial &	aPropagator,
		const TrajectoryStateUpdator &	aUpdator,
		const MeasurementEstimator &	aEstimator,
		const MultiTrajectoryStateMerger &	merger,
		float	errorRescaling,
		const bool	materialBeforeUpdate = `true`,
		const DetLayerGeometry *	detLayerGeometry = `0`
	)

Constructor with explicit components for propagation, update, chi2 calculation, merging and flag for merging before / after the update (i.e. fully configured). It clones the algorithms.

Definition at line 11 of file GsfTrajectorySmoother.cc.

References alongMomentum, FullConvolutionWithMaterial::clone(), GsfPropagatorWithMaterial::clone(), GsfPropagatorWithMaterial::convolutionWithMaterial(), dummyGeometry, GsfPropagatorWithMaterial::geometricalPropagator(), oppositeToMomentum, AlCaHLTBitMon_ParallelJobs::p, theAlongPropagator, theConvolutor, theGeometry, theGeomPropagator, theMatBeforeUpdate, and theOppositePropagator.

Referenced by clone().

     : theAlongPropagator(nullptr),
       theOppositePropagator(nullptr),
       theGeomPropagator(nullptr),
       theConvolutor(nullptr),
       theUpdator(aUpdator.clone()),
       theEstimator(aEstimator.clone()),
       theMerger(aMerger.clone()),
       theMatBeforeUpdate(materialBeforeUpdate),
       theErrorRescaling(errorRescaling),
       theGeometry(detLayerGeometry) {
   auto p = aPropagator.clone();
   p->setPropagationDirection(alongMomentum);
   theAlongPropagator = p;
   p = aPropagator.clone();
   p->setPropagationDirection(oppositeToMomentum);
   theOppositePropagator = p;
   if (!theMatBeforeUpdate) {
     theGeomPropagator = new GsfPropagatorAdapter(aPropagator.geometricalPropagator());
     theConvolutor = aPropagator.convolutionWithMaterial().clone();
   }
 
   if (!theGeometry)
     theGeometry = &dummyGeometry;
 }

GsfTrajectorySmoother::~GsfTrajectorySmoother ( )

override

Definition at line 43 of file GsfTrajectorySmoother.cc.

References theAlongPropagator, theConvolutor, theEstimator, theGeomPropagator, theMerger, theOppositePropagator, and theUpdator.

                                               {
   delete theAlongPropagator;
   delete theOppositePropagator;
   delete theGeomPropagator;
   delete theConvolutor;
   delete theUpdator;
   delete theEstimator;
   delete theMerger;
 }

Member Function Documentation

GsfTrajectorySmoother* GsfTrajectorySmoother::clone ( void ) const

inlineoverridevirtual

Implements TrajectorySmoother.

Definition at line 45 of file GsfTrajectorySmoother.h.

References GsfTrajectorySmoother(), theAlongPropagator, theErrorRescaling, theEstimator, theGeometry, theMatBeforeUpdate, theMerger, and theUpdator.

                                                 {
     return new GsfTrajectorySmoother(
         *theAlongPropagator, *theUpdator, *theEstimator, *theMerger, theErrorRescaling, theMatBeforeUpdate, theGeometry);
   }

const MeasurementEstimator* GsfTrajectorySmoother::estimator ( void ) const

inline

Definition at line 43 of file GsfTrajectorySmoother.h.

References theEstimator.

Referenced by trajectory().

43 { return theEstimator; }

GsfTrajectorySmoother::theEstimator

const MeasurementEstimator * theEstimator

Definition: GsfTrajectorySmoother.h:58

void GsfTrajectorySmoother::setHitCloner ( TkCloner const * hc )

inlineoverridevirtual

Implements TrajectorySmoother.

Definition at line 50 of file GsfTrajectorySmoother.h.

50 {}

Trajectory GsfTrajectorySmoother::trajectory ( const Trajectory & aTraj ) const

overridevirtual

Implements TrajectorySmoother.

Definition at line 53 of file GsfTrajectorySmoother.cc.

References alongMomentum, hltPixelTracks_cff::chi2, heavyFlavorValidationHarvestingSequence_cff::combiner, Trajectory::direction(), FrontierConditions_GlobalTag_cff::dump, Trajectory::empty(), MeasurementEstimator::estimate(), TrajectoryMeasurement::estimate(), estimator(), TrajectoryStateOnSurface::globalMomentum(), TrajectoryStateOnSurface::globalPosition(), DetLayerGeometry::idToLayer(), TrajectoryStateOnSurface::isValid(), LogDebug, Trajectory::measurements(), MultiTrajectoryStateMerger::merge(), TrajectoryStateOnSurface::rescaleError(), edm::second(), Trajectory::seed(), TrajectoryStateOnSurface::surface(), theAlongPropagator, theConvolutor, theErrorRescaling, theGeometry, theGeomPropagator, theMatBeforeUpdate, theMerger, theOppositePropagator, TrajectoryStateUpdator::update(), and updator().

                                                                           {
   if (aTraj.empty())
     return Trajectory();
 
   const Propagator* usePropagator = theAlongPropagator;
   if (aTraj.direction() == alongMomentum) {
     usePropagator = theOppositePropagator;
   }
   if (not usePropagator) {
     usePropagator = theGeomPropagator;
   }
 
   Trajectory myTraj(aTraj.seed(), usePropagator->propagationDirection());
 
   std::vector<TM> const& avtm = aTraj.measurements();
 
   TSOS predTsos = avtm.back().forwardPredictedState();
   predTsos.rescaleError(theErrorRescaling);
 
   if (!predTsos.isValid()) {
     edm::LogInfo("GsfTrackFitters") << "GsfTrajectorySmoother: predicted tsos of last measurement not valid!";
     return Trajectory();
   }
   TSOS currTsos;
 
   //first smoothed tm is last fitted
   if (avtm.back().recHit()->isValid()) {
     {
       //       TimeMe t(*updateTimer,false);
       currTsos = updator()->update(predTsos, *avtm.back().recHit());
     }
 
     if (!currTsos.isValid()) {
       edm::LogInfo("GsfTrajectorySmoother") << "GsfTrajectorySmoother: tsos not valid after update!";
       return Trajectory();
     }
 
     //check validity
     if (!avtm.back().forwardPredictedState().isValid() || !predTsos.isValid() ||
         !avtm.back().updatedState().isValid()) {
       edm::LogError("InvalidState") << "first hit";
       return Trajectory();
     }
 
     myTraj.push(TM(avtm.back().forwardPredictedState(),
                    predTsos,
                    avtm.back().updatedState(),
                    avtm.back().recHit(),
                    avtm.back().estimate(),
                    theGeometry->idToLayer(avtm.back().recHit()->geographicalId())),
                 avtm.back().estimate());
   } else {
     currTsos = predTsos;
     //check validity
     if (!avtm.back().forwardPredictedState().isValid()) {
       edm::LogError("InvalidState") << "first hit on invalid hit";
       return Trajectory();
     }
 
     myTraj.push(TM(avtm.back().forwardPredictedState(),
                    avtm.back().recHit(),
                    0.,
                    theGeometry->idToLayer(avtm.back().recHit()->geographicalId())));
   }
 
   TrajectoryStateCombiner combiner;
 
   int hitcounter = avtm.size() - 1;
   for (std::vector<TM>::const_reverse_iterator itm = avtm.rbegin() + 1; itm < avtm.rend() - 1; ++itm) {
     predTsos = usePropagator->propagate(currTsos, *(*itm).recHit()->surface());
     if (predTsos.isValid() && theConvolutor && theMatBeforeUpdate)
       predTsos = (*theConvolutor)(predTsos, usePropagator->propagationDirection());
     if (!predTsos.isValid()) {
       edm::LogInfo("GsfTrackFitters") << "GsfTrajectorySmoother: predicted tsos not valid!";
       return Trajectory();
     }
     if (theMerger)
       predTsos = theMerger->merge(predTsos);
 
     if ((*itm).recHit()->isValid()) {
       //update
       currTsos = updator()->update(predTsos, *(*itm).recHit());
 
       if (currTsos.isValid() && theConvolutor && !theMatBeforeUpdate)
         currTsos = (*theConvolutor)(currTsos, usePropagator->propagationDirection());
       if (!currTsos.isValid()) {
         edm::LogInfo("GsfTrackFitters") << "GsfTrajectorySmoother: tsos not valid after update / material effects!";
         return Trajectory();
       }
       //3 different possibilities to calculate smoothed state:
       //1: update combined predictions with hit
       //2: combine fwd-prediction with bwd-filter
       //3: combine bwd-prediction with fwd-filter
       TSOS combTsos = combiner(predTsos, (*itm).forwardPredictedState());
       if (!combTsos.isValid()) {
         LogDebug("GsfTrackFitters") << "KFTrajectorySmoother: combined tsos not valid!\n"
                                     << "pred Tsos pos: " << predTsos.globalPosition() << "\n"
                                     << "pred Tsos mom: " << predTsos.globalMomentum() << "\n"
                                     << "TrackingRecHit: "
                                     << (*itm).recHit()->surface()->toGlobal((*itm).recHit()->localPosition()) << "\n";
         return Trajectory();
       }
 
       TSOS smooTsos = combiner((*itm).updatedState(), predTsos);
 
       if (!smooTsos.isValid()) {
         LogDebug("GsfTrackFitters") << "KFTrajectorySmoother: smoothed tsos not valid!";
         return Trajectory();
       }
 
       if (!(*itm).forwardPredictedState().isValid() || !predTsos.isValid() || !smooTsos.isValid()) {
         edm::LogError("InvalidState") << "inside hits with combination.";
         return Trajectory();
       }
 
       auto chi2 = estimator()->estimate(combTsos, *(*itm).recHit()).second;
       myTraj.push(TM((*itm).forwardPredictedState(),
                      predTsos,
                      smooTsos,
                      (*itm).recHit(),
                      chi2,
                      theGeometry->idToLayer((*itm).recHit()->geographicalId())),
                   (*itm).estimate());
       LogDebug("GsfTrackFitters") << "added measurement #" << hitcounter-- << " with chi2 " << chi2;
       dump(predTsos, "predTsos", "GsfTrackFitters");
       dump(smooTsos, "smooTsos", "GsfTrackFitters");
 
     } else {
       currTsos = predTsos;
       TSOS combTsos = combiner(predTsos, (*itm).forwardPredictedState());
 
       if (!combTsos.isValid()) {
         LogDebug("GsfTrackFitters") << "KFTrajectorySmoother: combined tsos not valid!";
         return Trajectory();
       }
 
       if (!(*itm).forwardPredictedState().isValid() || !predTsos.isValid() || !combTsos.isValid()) {
         edm::LogError("InvalidState") << "inside hits with invalid rechit.";
         return Trajectory();
       }
 
       myTraj.push(TM((*itm).forwardPredictedState(),
                      predTsos,
                      combTsos,
                      (*itm).recHit(),
                      0.,
                      theGeometry->idToLayer((*itm).recHit()->geographicalId())));
       LogDebug("GsfTrackFitters") << "added invalid measurement #" << hitcounter--;
       dump(predTsos, "predTsos", "GsfTrackFitters");
       dump(combTsos, "smooTsos", "GsfTrackFitters");
     }
     if (theMerger)
       currTsos = theMerger->merge(currTsos);
 
     dump(currTsos, "currTsos", "GsfTrackFitters");
   }
 
   //last smoothed tm is last filtered
   predTsos = usePropagator->propagate(currTsos, *avtm.front().recHit()->surface());
   if (predTsos.isValid() && theConvolutor && theMatBeforeUpdate)
     predTsos = (*theConvolutor)(predTsos, usePropagator->propagationDirection());
   if (!predTsos.isValid()) {
     edm::LogInfo("GsfTrackFitters") << "GsfTrajectorySmoother: predicted tsos not valid!";
     return Trajectory();
   }
   if (theMerger)
     predTsos = theMerger->merge(predTsos);
 
   if (avtm.front().recHit()->isValid()) {
     //update
     currTsos = updator()->update(predTsos, *avtm.front().recHit());
     if (currTsos.isValid() && theConvolutor && !theMatBeforeUpdate)
       currTsos = (*theConvolutor)(currTsos, usePropagator->propagationDirection());
     if (!currTsos.isValid()) {
       edm::LogInfo("GsfTrackFitters") << "GsfTrajectorySmoother: tsos not valid after update / material effects!";
       return Trajectory();
     }
 
     if (!avtm.front().forwardPredictedState().isValid() || !predTsos.isValid() || !currTsos.isValid()) {
       edm::LogError("InvalidState") << "last hit";
       return Trajectory();
     }
 
     auto chi2 = estimator()->estimate(predTsos, *avtm.front().recHit()).second;
     myTraj.push(TM(avtm.front().forwardPredictedState(),
                    predTsos,
                    currTsos,
                    avtm.front().recHit(),
                    chi2,
                    theGeometry->idToLayer(avtm.front().recHit()->geographicalId())),
                 avtm.front().estimate());
     LogDebug("GsfTrackFitters") << "added measurement #" << hitcounter-- << " with chi2 " << chi2;
     dump(predTsos, "predTsos", "GsfTrackFitters");
     dump(currTsos, "smooTsos", "GsfTrackFitters");
 
   } else {
     if (!avtm.front().forwardPredictedState().isValid()) {
       edm::LogError("InvalidState") << "last invalid hit";
       return Trajectory();
     }
     myTraj.push(TM(avtm.front().forwardPredictedState(),
                    avtm.front().recHit(),
                    0.,
                    theGeometry->idToLayer(avtm.front().recHit()->geographicalId())));
     LogDebug("GsfTrackFitters") << "added invalid measurement #" << hitcounter--;
   }
 
   return myTraj;
 }