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