ConversionLikelihoodCalculator.h

Go to the documentation of this file.

#ifndef ConversionLikelihoodCalculator_h
#define ConversionLikelihoodCalculator_h
 
#include "DataFormats/EgammaCandidates/interface/ConversionFwd.h"
#include "DataFormats/EgammaCandidates/interface/Conversion.h"
 
#include "TMVA/Reader.h"
#include <memory>
 
class ConversionLikelihoodCalculator {
public:
  ConversionLikelihoodCalculator();
  void setWeightsFile(const char* weightsFile);
 
  double calculateLikelihood(reco::ConversionRef conversion);
  double calculateLikelihood(reco::Conversion& conversion);
 
private:
  std::unique_ptr<TMVA::Reader> reader_;
  float log_e_over_p_;
  float log_abs_cot_theta_;
  float log_abs_delta_phi_;
  float log_chi2_max_pt_;
  float log_chi2_min_pt_;
};
 
#endif