PerigeeRefittedTrackState.h

Go to the documentation of this file.

#ifndef PerigeeRefittedTrackState_H
#define PerigeeRefittedTrackState_H

#include "DataFormats/GeometrySurface/interface/ReferenceCounted.h"
#include "RecoVertex/VertexPrimitives/interface/RefittedTrackState.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateClosestToPoint.h"

class FreeTrajectoryState;
class TrajectoryStateOnSurface;
class Surface;
class Propagator;

class PerigeeRefittedTrackState : public RefittedTrackState<5> {

public:

  typedef ReferenceCountingPointer<RefittedTrackState<5> > RefCountedRefittedTrackState;

  PerigeeRefittedTrackState(const TrajectoryStateClosestToPoint & tscp,
                const AlgebraicVector3 & aMomentumAtVertex,
                const double aWeight = 1.) :
    theState(tscp), momentumAtVertex(aMomentumAtVertex), theWeight(aWeight) {}

 ~PerigeeRefittedTrackState() override{}

  FreeTrajectoryState freeTrajectoryState() const override
    {return theState.theState();}

  TrajectoryStateOnSurface trajectoryStateOnSurface(
        const Surface & surface) const override;

  TrajectoryStateOnSurface trajectoryStateOnSurface(
        const Surface & surface, const Propagator & propagator) const override;

  AlgebraicVector5 parameters() const override
    {return theState.perigeeParameters().vector();}

  AlgebraicSymMatrix55  covariance() const override
    {return theState.perigeeError().covarianceMatrix();}

  GlobalPoint position() const override
    {return theState.referencePoint();}

  AlgebraicVector3 momentumVector() const override;

  double weight() const override {return theWeight;}

  ReferenceCountingPointer<RefittedTrackState<5> > stateWithNewWeight
    (const double newWeight) const override;

  std::vector<ReferenceCountingPointer<RefittedTrackState<5> > > components() const override;

  reco::TransientTrack transientTrack() const override;

private:

  TrajectoryStateClosestToPoint theState;
  AlgebraicVector3 momentumAtVertex;
  double theWeight;
};
#endif