GsfTrajectorySmoother Class Reference

#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=nullptr)
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

TM

typedef TrajectoryMeasurement GsfTrajectorySmoother::TM

private

Definition at line 24 of file GsfTrajectorySmoother.h.

TSOS

typedef TrajectoryStateOnSurface GsfTrajectorySmoother::TSOS

private

Definition at line 23 of file GsfTrajectorySmoother.h.

Constructor & Destructor Documentation

GsfTrajectorySmoother()

GsfTrajectorySmoother::GsfTrajectorySmoother	(	const GsfPropagatorWithMaterial &	aPropagator,
		const TrajectoryStateUpdator &	aUpdator,
		const MeasurementEstimator &	aEstimator,
		const MultiTrajectoryStateMerger &	merger,
		float	errorRescaling,
		const bool	materialBeforeUpdate = true,
		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). It clones the algorithms.

Definition at line 11 of file GsfTrajectorySmoother.cc.

    : 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;
}

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().

~GsfTrajectorySmoother()

GsfTrajectorySmoother::~GsfTrajectorySmoother ( )

override

Definition at line 43 of file GsfTrajectorySmoother.cc.

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

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

Member Function Documentation

clone()

GsfTrajectorySmoother* GsfTrajectorySmoother::clone ( void  ) const

inlineoverridevirtual

Implements TrajectorySmoother.

Definition at line 45 of file GsfTrajectorySmoother.h.

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

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

estimator()

const MeasurementEstimator* GsfTrajectorySmoother::estimator ( void  ) const

inline

Definition at line 43 of file GsfTrajectorySmoother.h.

{ return theEstimator; }

References theEstimator.

Referenced by trajectory().

setHitCloner()

void GsfTrajectorySmoother::setHitCloner ( TkCloner const *  hc )

inlineoverridevirtual

Implements TrajectorySmoother.

Definition at line 50 of file GsfTrajectorySmoother.h.

{}

trajectory()

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

overridevirtual

Implements TrajectorySmoother.

Definition at line 53 of file GsfTrajectorySmoother.cc.

                                                                          {
  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;
}

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().

updator()

const TrajectoryStateUpdator* GsfTrajectorySmoother::updator ( ) const

inline

Definition at line 42 of file GsfTrajectorySmoother.h.

{ return theUpdator; }

References theUpdator.

Referenced by trajectory().

Member Data Documentation

dummyGeometry

const DetLayerGeometry GsfTrajectorySmoother::dummyGeometry

private

Definition at line 65 of file GsfTrajectorySmoother.h.

Referenced by GsfTrajectorySmoother().

theAlongPropagator

const GsfPropagatorWithMaterial* GsfTrajectorySmoother::theAlongPropagator

private

Definition at line 53 of file GsfTrajectorySmoother.h.

Referenced by clone(), GsfTrajectorySmoother(), trajectory(), and ~GsfTrajectorySmoother().

theConvolutor

const FullConvolutionWithMaterial* GsfTrajectorySmoother::theConvolutor

private

Definition at line 56 of file GsfTrajectorySmoother.h.

Referenced by GsfTrajectorySmoother(), trajectory(), and ~GsfTrajectorySmoother().

theErrorRescaling

float GsfTrajectorySmoother::theErrorRescaling

private

Definition at line 63 of file GsfTrajectorySmoother.h.

Referenced by clone(), and trajectory().

theEstimator

const MeasurementEstimator* GsfTrajectorySmoother::theEstimator

private

Definition at line 58 of file GsfTrajectorySmoother.h.

Referenced by clone(), estimator(), and ~GsfTrajectorySmoother().

theGeometry

const DetLayerGeometry* GsfTrajectorySmoother::theGeometry

private

Definition at line 66 of file GsfTrajectorySmoother.h.

Referenced by clone(), GsfTrajectorySmoother(), and trajectory().

theGeomPropagator

const GsfPropagatorAdapter* GsfTrajectorySmoother::theGeomPropagator

private

Definition at line 55 of file GsfTrajectorySmoother.h.

Referenced by GsfTrajectorySmoother(), trajectory(), and ~GsfTrajectorySmoother().

theMatBeforeUpdate

bool GsfTrajectorySmoother::theMatBeforeUpdate

private

Definition at line 62 of file GsfTrajectorySmoother.h.

Referenced by clone(), GsfTrajectorySmoother(), and trajectory().

theMerger

const MultiTrajectoryStateMerger* GsfTrajectorySmoother::theMerger

private

Definition at line 59 of file GsfTrajectorySmoother.h.

Referenced by clone(), trajectory(), and ~GsfTrajectorySmoother().

theOppositePropagator

const GsfPropagatorWithMaterial* GsfTrajectorySmoother::theOppositePropagator

private

Definition at line 54 of file GsfTrajectorySmoother.h.

Referenced by GsfTrajectorySmoother(), trajectory(), and ~GsfTrajectorySmoother().

theTiming

bool GsfTrajectorySmoother::theTiming

private

Definition at line 61 of file GsfTrajectorySmoother.h.

theUpdator

const TrajectoryStateUpdator* GsfTrajectorySmoother::theUpdator

private

Definition at line 57 of file GsfTrajectorySmoother.h.

Referenced by clone(), updator(), and ~GsfTrajectorySmoother().