File indexing completed on 2024-04-06 12:25:01
0001 #ifndef ConversionLikelihoodCalculator_h
0002 #define ConversionLikelihoodCalculator_h
0003
0004 #include "DataFormats/EgammaCandidates/interface/ConversionFwd.h"
0005
0006 #include "TMVA/Reader.h"
0007 #include <memory>
0008
0009 class ConversionLikelihoodCalculator {
0010 public:
0011 ConversionLikelihoodCalculator();
0012 void setWeightsFile(const char* weightsFile);
0013
0014 double calculateLikelihood(reco::ConversionRef conversion);
0015 double calculateLikelihood(reco::Conversion& conversion);
0016
0017 private:
0018 std::unique_ptr<TMVA::Reader> reader_;
0019 float log_e_over_p_;
0020 float log_abs_cot_theta_;
0021 float log_abs_delta_phi_;
0022 float log_chi2_max_pt_;
0023 float log_chi2_min_pt_;
0024 };
0025
0026 #endif
