SingleTrackVertexConstraint.h

Go to the documentation of this file.

#ifndef SingleTrackVertexConstraint_H
#define SingleTrackVertexConstraint_H

#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexUpdator.h"
#include "RecoVertex/VertexPrimitives/interface/VertexSmoother.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexTrackUpdator.h"

#include "RecoVertex/VertexTools/interface/LinearizedTrackStateFactory.h"
#include "RecoVertex/VertexTools/interface/VertexTrackFactory.h"
#include "TrackingTools/TransientTrack/interface/TransientTrackFromFTSFactory.h"

#include <tuple>
class SingleTrackVertexConstraint {
public:
  typedef std::pair<reco::TransientTrack, float> TrackFloatPair;
  typedef std::tuple<bool, reco::TransientTrack, float> BTFtuple;

  SingleTrackVertexConstraint(bool doTrackerBoundCheck = true) : doTrackerBoundCheck_(doTrackerBoundCheck) {}

  BTFtuple constrain(const reco::TransientTrack& track,
                     const GlobalPoint& priorPos,
                     const GlobalError& priorError) const;

  BTFtuple constrain(const FreeTrajectoryState& fts, const GlobalPoint& priorPos, const GlobalError& priorError) const;

  BTFtuple constrain(const reco::TransientTrack& track, const VertexState priorVertex) const;

  BTFtuple constrain(const reco::TransientTrack& track, const reco::BeamSpot& spot) const;

  BTFtuple constrain(const FreeTrajectoryState& fts, const reco::BeamSpot& spot) const;

private:
  KalmanVertexUpdator<5> vertexUpdator;
  KalmanVertexTrackUpdator<5> theVertexTrackUpdator;

  LinearizedTrackStateFactory theLTrackFactory;
  VertexTrackFactory<5> theVTrackFactory;
  TransientTrackFromFTSFactory ttFactory;
  bool doTrackerBoundCheck_;
};

#endif