TrackProbability/interface/HistogramProbabilityEstimator.h

0001 #ifndef HistogramProbabilityEstimator_H
0002 #define HistogramProbabilityEstimator_H
0003
0004 #include "DataFormats/TrackReco/interface/Track.h"
0005 #include "DataFormats/JetReco/interface/Jet.h"
0006 #include "DataFormats/VertexReco/interface/Vertex.h"
0007
0008 #include "CondFormats/BTauObjects/interface/CalibratedHistogram.h"
0009 #include "CondFormats/BTauObjects/interface/TrackProbabilityCalibration.h"
0010
0011 // #include "RecoBTag/XMLCalibration/interface/AlgorithmCalibration.h"
0012
0013 /** provides the track probability to come from the primary vertex
0014    *  for a given track
0015    */
0016 #include <utility>
0017
0018 class HistogramProbabilityEstimator {
0019 public:
0020   HistogramProbabilityEstimator(const TrackProbabilityCalibration *calib3D, const TrackProbabilityCalibration *calib2D)
0021       : m_calibration3D(calib3D), m_calibration2D(calib2D) {}
0022
0023   /*   HistogramProbabilityEstimator( AlgorithmCalibration<TrackClassFilterCategory,CalibratedHistogramXML>  * calib3D,
0024                                  AlgorithmCalibration<TrackClassFilterCategory,CalibratedHistogramXML> * calib2D)
0025     :   m_calibration3D(calib3D),m_calibrationTransverse(calib2D)
0026
0027   {
0028   }
0029 */
0030
0031   ~HistogramProbabilityEstimator() {
0032     //  if(m_calibration3D!=0) delete m_calibration3D;
0033     //  if(m_calibration2D!=0) delete m_calibration2D;
0034   }
0035   std::pair<bool, double> probability(
0036       bool quality, int ipType, float significance, const reco::Track &, const reco::Jet &, const reco::Vertex &) const;
0037
0038 private:
0039   const TrackProbabilityCalibration *m_calibration3D;
0040   const TrackProbabilityCalibration *m_calibration2D;
0041 };
0042
0043 #endif