EGammaMvaEleEstimatorFWLite.h

Go to the documentation of this file.

#ifndef PhysicsTools_Heppy_EGammaMvaEleEstimatorFWLite_h
#define PhysicsTools_Heppy_EGammaMvaEleEstimatorFWLite_h
 
class EGammaMvaEleEstimator;
class EGammaMvaEleEstimatorCSA14;
namespace reco {
  class Vertex;
}
namespace pat {
  class Electron;
}
#include <vector>
#include <string>
 
namespace heppy {
 
  class EGammaMvaEleEstimatorFWLite {
  public:
    EGammaMvaEleEstimatorFWLite();
    ~EGammaMvaEleEstimatorFWLite();
 
    enum MVAType {
      kTrig = 0,           // MVA for triggering electrons
      kTrigNoIP = 1,       // MVA for triggering electrons without IP info
      kNonTrig = 2,        // MVA for non-triggering electrons
      kTrigCSA14 = 3,      // MVA for non-triggering electrons
      kNonTrigCSA14 = 4,   // MVA for non-triggering electrons
      kNonTrigPhys14 = 5,  // MVA for non-triggering electrons
    };
 
    void initialize(std::string methodName, MVAType type, bool useBinnedVersion, std::vector<std::string> weightsfiles);
 
    float mvaValue(
        const pat::Electron& ele, const reco::Vertex& vertex, double rho, bool full5x5, bool printDebug = false);
 
  private:
    EGammaMvaEleEstimator* estimator_;
    EGammaMvaEleEstimatorCSA14* estimatorCSA14_;
  };
}  // namespace heppy
#endif