GsfVertexTrackCompatibilityEstimator.h

Go to the documentation of this file.

#ifndef GsfVertexTrackCompatibilityEstimator_H
#define GsfVertexTrackCompatibilityEstimator_H

#include "RecoVertex/VertexPrimitives/interface/VertexTrackCompatibilityEstimator.h"
#include "RecoVertex/VertexPrimitives/interface/VertexTrack.h"
#include "RecoVertex/VertexPrimitives/interface/CachingVertex.h"
#include "RecoVertex/VertexTools/interface/VertexTrackFactory.h"
#include "RecoVertex/GaussianSumVertexFit/interface/MultiPerigeeLTSFactory.h"
#include "RecoVertex/GaussianSumVertexFit/interface/GsfVertexUpdator.h"

#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexTrackUpdator.h"
#include "RecoVertex/KalmanVertexFit/interface/KVFHelper.h"

class GsfVertexTrackCompatibilityEstimator : public VertexTrackCompatibilityEstimator<5> {
public:
  typedef CachingVertex<5>::RefCountedVertexTrack RefCountedVertexTrack;

  GsfVertexTrackCompatibilityEstimator() {}

  ~GsfVertexTrackCompatibilityEstimator() override {}

  BDpair estimate(const CachingVertex<5>& vrt,
                  const RefCountedVertexTrack track,
                  unsigned int hint = UINT_MAX) const override;

  BDpair estimate(const CachingVertex<5>& v,
                  const RefCountedLinearizedTrackState track,
                  unsigned int hint = UINT_MAX) const override;

  BDpair estimate(const reco::Vertex& vertex, const reco::TransientTrack& track) const override;

  GsfVertexTrackCompatibilityEstimator* clone() const override {
    return new GsfVertexTrackCompatibilityEstimator(*this);
  }

private:
  BDpair estimateFittedTrack(const CachingVertex<5>& v, const RefCountedVertexTrack track) const;
  BDpair estimateNFittedTrack(const CachingVertex<5>& v, const RefCountedVertexTrack track) const;

  GsfVertexUpdator updator;
  //   KalmanVertexTrackUpdator trackUpdator;
  MultiPerigeeLTSFactory lTrackFactory;
  VertexTrackFactory<5> vTrackFactory;
  //   KVFHelper helper;
};

#endif