#include <GsfTrajectoryFitter.h>

Inheritance diagram for GsfTrajectoryFitter:

Public Member Functions
std::unique_ptr< TrajectoryFitter >	clone () const override

const MeasurementEstimator *	estimator () const

Trajectory	fitOne (const Trajectory &t, fitType type) const override

Trajectory	fitOne (const TrajectorySeed &aSeed, const RecHitContainer &hits, const TrajectoryStateOnSurface &firstPredTsos, fitType type) const override

Trajectory	fitOne (const TrajectorySeed &aSeed, const RecHitContainer &hits, fitType type) const override

	GsfTrajectoryFitter (const Propagator &aPropagator, const TrajectoryStateUpdator &aUpdator, const MeasurementEstimator &aEstimator, const MultiTrajectoryStateMerger &aMerger, const DetLayerGeometry *detLayerGeometry=nullptr)

const MultiTrajectoryStateMerger *	merger () const

const Propagator *	propagator () const

void	setHitCloner (TkCloner const *hc) override

const TrajectoryStateUpdator *	updator () const

	~GsfTrajectoryFitter () override

Public Member Functions inherited from TrajectoryFitter
std::vector< Trajectory >	fit (const Trajectory &traj, fitType type=standard) const

std::vector< Trajectory >	fit (const TrajectorySeed &seed, const RecHitContainer &hits, fitType type=standard) const

std::vector< Trajectory >	fit (const TrajectorySeed &seed, const RecHitContainer &hits, const TrajectoryStateOnSurface &tsos, fitType type=standard) const

virtual	~TrajectoryFitter ()

Private Types
typedef TrajectoryMeasurement	TM

typedef TrajectoryStateOnSurface	TSOS

Private Attributes
const DetLayerGeometry	dummyGeometry

const MeasurementEstimator *	theEstimator

const DetLayerGeometry *	theGeometry

TkCloner const *	theHitCloner = nullptr

const MultiTrajectoryStateMerger *	theMerger

const Propagator *	thePropagator

bool	theTiming

const TrajectoryStateUpdator *	theUpdator

Additional Inherited Members
Public Types inherited from TrajectoryFitter
enum	fitType { standard = 0, looper = 1 }

typedef Trajectory::RecHitContainer	RecHitContainer

typedef TrajectoryFitterRecord	Record

Detailed Description

A GSF fitter, similar to KFTrajectoryFitter.

Definition at line 20 of file GsfTrajectoryFitter.h.

Member Typedef Documentation

◆ TM

typedef TrajectoryMeasurement GsfTrajectoryFitter::TM

private

Definition at line 23 of file GsfTrajectoryFitter.h.

◆ TSOS

typedef TrajectoryStateOnSurface GsfTrajectoryFitter::TSOS

private

Definition at line 22 of file GsfTrajectoryFitter.h.

Constructor & Destructor Documentation

◆ GsfTrajectoryFitter()

GsfTrajectoryFitter::GsfTrajectoryFitter	(	const Propagator &	aPropagator,
		const TrajectoryStateUpdator &	aUpdator,
		const MeasurementEstimator &	aEstimator,
		const MultiTrajectoryStateMerger &	aMerger,
		const DetLayerGeometry *	detLayerGeometry = `nullptr`
	)

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

Definition at line 14 of file GsfTrajectoryFitter.cc.

References dummyGeometry, and theGeometry.

Referenced by clone().

     : thePropagator(aPropagator.clone()),
       theUpdator(aUpdator.clone()),
       theEstimator(aEstimator.clone()),
       theMerger(aMerger.clone()),
       theGeometry(detLayerGeometry) {
   if (!theGeometry)
     theGeometry = &dummyGeometry;
 }

◆ ~GsfTrajectoryFitter()

GsfTrajectoryFitter::~GsfTrajectoryFitter ( )

override

Definition at line 28 of file GsfTrajectoryFitter.cc.

References theEstimator, theMerger, thePropagator, and theUpdator.

                                           {
   delete thePropagator;
   delete theUpdator;
   delete theEstimator;
   delete theMerger;
 }

Member Function Documentation

◆ clone()

std::unique_ptr<TrajectoryFitter> GsfTrajectoryFitter::clone ( void ) const

inlineoverridevirtual

Implements TrajectoryFitter.

Definition at line 49 of file GsfTrajectoryFitter.h.

References GsfTrajectoryFitter(), theEstimator, theGeometry, theMerger, thePropagator, and theUpdator.

                                                          {
     return std::unique_ptr<TrajectoryFitter>(
         new GsfTrajectoryFitter(*thePropagator, *theUpdator, *theEstimator, *theMerger, theGeometry));
   }

◆ estimator()

const MeasurementEstimator* GsfTrajectoryFitter::estimator ( void ) const

inline

Definition at line 46 of file GsfTrajectoryFitter.h.

References theEstimator.

Referenced by fitOne().

46 { return theEstimator; }

GsfTrajectoryFitter::theEstimator

const MeasurementEstimator * theEstimator

Definition: GsfTrajectoryFitter.h:59

◆ fitOne() [1/3]

Trajectory GsfTrajectoryFitter::fitOne	(	const Trajectory &	t,
		fitType	type
	)		const

overridevirtual

Implements TrajectoryFitter.

Definition at line 35 of file GsfTrajectoryFitter.cc.

References Trajectory::empty(), Trajectory::firstMeasurement(), Trajectory::recHits(), Trajectory::seed(), and TrajectoryMeasurement::updatedState().

                                                                                   {
   if (aTraj.empty())
     return Trajectory();
 
   TM const& firstTM = aTraj.firstMeasurement();
   TSOS firstTsos = TrajectoryStateWithArbitraryError()(firstTM.updatedState());
 
   return fitOne(aTraj.seed(), aTraj.recHits(), firstTsos, type);
 }

◆ fitOne() [2/3]

Trajectory GsfTrajectoryFitter::fitOne	(	const TrajectorySeed &	aSeed,
		const RecHitContainer &	hits,
		const TrajectoryStateOnSurface &	firstPredTsos,
		fitType	type
	)		const

overridevirtual

Implements TrajectoryFitter.

Definition at line 51 of file GsfTrajectoryFitter.cc.

References cms::cuda::assert(), nano_mu_local_reco_cff::chi2, GCP_Ntuples_cfg::dump, MeasurementEstimator::estimate(), estimator(), Trajectory::foundHits(), hfClusterShapes_cfi::hits, DetLayerGeometry::idToLayer(), TrajectoryStateOnSurface::isValid(), LogDebug, TkCloner::makeShared(), MultiTrajectoryStateMerger::merge(), merger(), Propagator::propagate(), propagator(), Trajectory::push(), TrajectoryStateOnSurface::surface(), theGeometry, theHitCloner, TrajectoryStateUpdator::update(), and updator().

                                                       {
   if (hits.empty())
     return Trajectory();
 
   Trajectory myTraj(aSeed, propagator()->propagationDirection());
 
   TSOS predTsos(firstPredTsos);
   if (!predTsos.isValid()) {
     edm::LogInfo("GsfTrackFitters") << "GsfTrajectoryFitter: predicted tsos of first measurement not valid!";
     return Trajectory();
   }
 
   TSOS currTsos;
   if (hits.front()->isValid()) {
     auto const& ihit = hits.front();
     //update
     assert((!(ihit)->canImproveWithTrack()) || (nullptr != theHitCloner));
     assert((!(ihit)->canImproveWithTrack()) || (nullptr != dynamic_cast<BaseTrackerRecHit const*>(ihit.get())));
     auto preciseHit = theHitCloner->makeShared(ihit, predTsos);
     dump(*preciseHit, 1, "GsfTrackFitters");
     { currTsos = updator()->update(predTsos, *preciseHit); }
     if (!predTsos.isValid() || !currTsos.isValid()) {
       edm::LogError("InvalidState") << "first hit";
       return Trajectory();
     }
     myTraj.push(TM(predTsos, currTsos, preciseHit, 0., theGeometry->idToLayer(preciseHit->geographicalId())),
                 estimator()->estimate(predTsos, *preciseHit).second);
   } else {
     currTsos = predTsos;
     if (!predTsos.isValid()) {
       edm::LogError("InvalidState") << "first invalid hit";
       return Trajectory();
     }
     myTraj.push(TM(predTsos, hits.front(), 0., theGeometry->idToLayer((hits.front())->geographicalId())));
   }
 
   int hitcounter = 1;
   for (RecHitContainer::const_iterator ihit = hits.begin() + 1; ihit != hits.end(); ihit++) {
     ++hitcounter;
 
     //
     // temporary protection copied from KFTrajectoryFitter.
     //
     if ((**ihit).isValid() == false && (**ihit).det() == nullptr) {
       LogDebug("GsfTrackFitters") << " Error: invalid hit with no GeomDet attached .... skipping";
       continue;
     }
 
     {
       //       TimeMe t(*propTimer,false);
       predTsos = propagator()->propagate(currTsos, (**ihit).det()->surface());
     }
     if (!predTsos.isValid()) {
       if (myTraj.foundHits() >= 3) {
         edm::LogInfo("GsfTrackFitters") << "GsfTrajectoryFitter: predicted tsos not valid! \n"
                                         << "Returning trajectory with " << myTraj.foundHits() << " found hits.";
         return myTraj;
       } else {
         edm::LogInfo("GsfTrackFitters") << "GsfTrajectoryFitter: predicted tsos not valid after " << myTraj.foundHits()
                                         << " hits, discarding candidate!";
         return Trajectory();
       }
     }
     if (merger())
       predTsos = merger()->merge(predTsos);
 
     if ((**ihit).isValid()) {
       //update
       assert((!(*ihit)->canImproveWithTrack()) || (nullptr != theHitCloner));
       assert((!(*ihit)->canImproveWithTrack()) || (nullptr != dynamic_cast<BaseTrackerRecHit const*>((*ihit).get())));
       if (!predTsos.isValid()) {
         return Trajectory();
       }
       auto preciseHit = theHitCloner->makeShared(*ihit, predTsos);
       dump(*preciseHit, hitcounter, "GsfTrackFitters");
       currTsos = updator()->update(predTsos, *preciseHit);
       if (!predTsos.isValid() || !currTsos.isValid()) {
         edm::LogError("InvalidState") << "inside hit";
         return Trajectory();
       }
       auto chi2 = estimator()->estimate(predTsos, *preciseHit).second;
       myTraj.push(TM(predTsos, currTsos, preciseHit, chi2, theGeometry->idToLayer(preciseHit->geographicalId())));
       LogDebug("GsfTrackFitters") << "added measurement with chi2 " << chi2;
     } else {
       currTsos = predTsos;
       if (!predTsos.isValid()) {
         edm::LogError("InvalidState") << "inside invalid hit";
         return Trajectory();
       }
       myTraj.push(TM(predTsos, *ihit, 0., theGeometry->idToLayer((*ihit)->geographicalId())));
     }
     dump(predTsos, "predTsos", "GsfTrackFitters");
     dump(currTsos, "currTsos", "GsfTrackFitters");
   }
   return myTraj;
 }

◆ fitOne() [3/3]

Trajectory GsfTrajectoryFitter::fitOne	(	const TrajectorySeed &	aSeed,
		const RecHitContainer &	hits,
		fitType	type
	)		const

overridevirtual

Implements TrajectoryFitter.

Definition at line 45 of file GsfTrajectoryFitter.cc.

                                                                                                                    {
   edm::LogError("GsfTrajectoryFitter") << "GsfTrajectoryFitter::fit(TrajectorySeed, vector<RecHit>) not implemented";
 
   return Trajectory();
 }