MultiRefittedTS.h

Go to the documentation of this file.

#ifndef MultiRefittedTS_H
#define MultiRefittedTS_H

#include "RecoVertex/VertexPrimitives/interface/RefittedTrackState.h"
#include "DataFormats/GeometrySurface/interface/ReferenceCounted.h"
#include "RecoVertex/VertexPrimitives/interface/LinearizedTrackState.h"
#include "DataFormats/GeometrySurface/interface/Surface.h"

class Surface;

class MultiRefittedTS : public RefittedTrackState<5> {

public:

  typedef ReferenceCountingPointer<RefittedTrackState<5> > RefCountedRefittedTrackState;
  typedef ReferenceCountingPointer<LinearizedTrackState<5> > RefCountedLinearizedTrackState;

  MultiRefittedTS(const std::vector<RefCountedRefittedTrackState> & prtsComp,
    const Surface & referenceSurface);

  MultiRefittedTS(const std::vector<RefCountedRefittedTrackState> & prtsComp,
    const GlobalPoint & referencePosition);

  ~MultiRefittedTS() override{}

  FreeTrajectoryState freeTrajectoryState() const override;

  TrajectoryStateOnSurface trajectoryStateOnSurface(
        const Surface & surface) const override;

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

  reco::TransientTrack transientTrack() const override;

  AlgebraicVectorN  parameters() const override;

  AlgebraicSymMatrixNN  covariance() const override;

  GlobalPoint position() const override;

  AlgebraicVectorM momentumVector() const override;

  double weight() const override;

  std::vector<ReferenceCountingPointer<RefittedTrackState<5> > > components() const override
  {
    return theComponents;
  }

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


private:

  void computeFreeTrajectoryState() const;


  typedef std::vector<RefCountedRefittedTrackState > RTSvector;

  mutable RTSvector theComponents;
  mutable bool totalWeightAvailable, ftsAvailable;
  mutable double totalWeight;
  mutable FreeTrajectoryState fts;
  const GlobalPoint refPosition;
  ConstReferenceCountingPointer<Surface> refSurface;
  const bool surf;

};
#endif