KalmanVertexTrackCompatibilityEstimator.h

Go to the documentation of this file.

#ifndef KalmanVertexTrackCompatibilityEstimator_H
#define KalmanVertexTrackCompatibilityEstimator_H


#include "RecoVertex/VertexPrimitives/interface/VertexTrackCompatibilityEstimator.h"
//#include "TrackingTools/PatternTools/interface/TSCPBuilderNoMaterial.h"
#include "RecoVertex/VertexPrimitives/interface/VertexTrack.h"
#include "RecoVertex/VertexPrimitives/interface/CachingVertex.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexUpdator.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexTrackUpdator.h"
#include "RecoVertex/VertexTools/interface/VertexTrackFactory.h"
#include "RecoVertex/KalmanVertexFit/interface/KVFHelper.h"

template <unsigned int N>
class KalmanVertexTrackCompatibilityEstimator : public VertexTrackCompatibilityEstimator<N>
{

public:

  typedef typename CachingVertex<N>::RefCountedVertexTrack RefCountedVertexTrack;
  typedef typename VertexTrack<N>::RefCountedLinearizedTrackState RefCountedLinearizedTrackState;
  typedef typename VertexTrack<N>::RefCountedRefittedTrackState RefCountedRefittedTrackState;
  typedef typename std::pair<bool, double> BDpair;

  KalmanVertexTrackCompatibilityEstimator(){}

  ~KalmanVertexTrackCompatibilityEstimator() override{}

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

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

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

  KalmanVertexTrackCompatibilityEstimator<N> * clone() const override
  {
    return new KalmanVertexTrackCompatibilityEstimator(* this);
  }


private:

  BDpair estimateFittedTrack(const CachingVertex<N> & v, const RefCountedVertexTrack track) const;
  BDpair estimateNFittedTrack(const CachingVertex<N> & v, const RefCountedVertexTrack track) const;  
  BDpair estimateDifference(const CachingVertex<N> & more, const CachingVertex<N> & less, 
                                                       const RefCountedVertexTrack track) const;
  KalmanVertexUpdator<N> updator;
  KalmanVertexTrackUpdator<N> trackUpdator;
  VertexTrackFactory<N> vTrackFactory;
  KVFHelper<N> helper;

};

#endif