JetSignalVertexCompatibilityAlgo.h

Go to the documentation of this file.

#ifndef JetSignalVertexCompatibilityAlgo_h
#define JetSignalVertexCompatibilityAlgo_h
 
#include <vector>
#include <map>
 
#include "DataFormats/Common/interface/RefToBase.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
 
#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
 
namespace reco {
 
  class JetSignalVertexCompatibilityAlgo {
  public:
    JetSignalVertexCompatibilityAlgo(double cut, double temperature);
    ~JetSignalVertexCompatibilityAlgo();
 
    std::vector<float> compatibility(const reco::VertexCollection &vertices, const reco::TrackRefVector &tracks) const;
 
    void resetEvent(const TransientTrackBuilder *trackBuilder);
 
  private:
    template <typename T>
    struct RefToBaseLess {
      bool operator()(const edm::RefToBase<T> &r1, const edm::RefToBase<T> &r2) const;
    };
 
    typedef std::map<reco::TrackBaseRef, reco::TransientTrack, RefToBaseLess<reco::Track> > TransientTrackMap;
 
    const TransientTrack &convert(const reco::TrackBaseRef &track) const;
    double activation(double compat) const;
 
    static double trackVertexCompat(const reco::Vertex &vtx, const TransientTrack &track);
 
    mutable TransientTrackMap trackMap;
    const TransientTrackBuilder *trackBuilder;
 
    const double cut;
    const double temperature;
  };
 
}  // namespace reco
 
#endif  // JetSignalVertexCompatibilityAlgo_h