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