HistogramProbabilityEstimator.h

Go to the documentation of this file.

#ifndef HistogramProbabilityEstimator_H
#define HistogramProbabilityEstimator_H

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

#include "CondFormats/BTauObjects/interface/CalibratedHistogram.h"
#include "CondFormats/BTauObjects/interface/TrackProbabilityCalibration.h"

// #include "RecoBTag/XMLCalibration/interface/AlgorithmCalibration.h"

#include <utility>

class HistogramProbabilityEstimator {
public:
  HistogramProbabilityEstimator(const TrackProbabilityCalibration *calib3D, const TrackProbabilityCalibration *calib2D)
      : m_calibration3D(calib3D), m_calibration2D(calib2D) {}

  /*   HistogramProbabilityEstimator( AlgorithmCalibration<TrackClassFilterCategory,CalibratedHistogramXML>  * calib3D,
                                 AlgorithmCalibration<TrackClassFilterCategory,CalibratedHistogramXML> * calib2D)
    :   m_calibration3D(calib3D),m_calibrationTransverse(calib2D)

  {
  }
*/

  ~HistogramProbabilityEstimator() {
    //  if(m_calibration3D!=0) delete m_calibration3D;
    //  if(m_calibration2D!=0) delete m_calibration2D;
  }
  std::pair<bool, double> probability(
      bool quality, int ipType, float significance, const reco::Track &, const reco::Jet &, const reco::Vertex &) const;

private:
  const TrackProbabilityCalibration *m_calibration3D;
  const TrackProbabilityCalibration *m_calibration2D;
};

#endif