MvaSoftElectronEstimator.h

Go to the documentation of this file.

//--------------------------------------------------------------------------------------------------
// $Id $
//
// MvaSoftEleEstimator
//
// Helper Class for applying MVA electron ID selection
//
// Authors: S. de Visscher
//--------------------------------------------------------------------------------------------------



//#define STANDALONE   // <---- this line

#ifndef MvaSoftEleEstimator_H
#define MvaSoftEleEstimator_H

#include "DataFormats/PatCandidates/interface/Electron.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "RecoEcal/EgammaCoreTools/interface/EcalClusterLazyTools.h"
#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
#include "EgammaAnalysis/ElectronTools/interface/ElectronEffectiveArea.h"
#include <vector>
#include <TROOT.h>
#include "TMVA/Factory.h"
#include "TMVA/Tools.h"
#include "TMVA/Reader.h"

class MvaSoftEleEstimator{
  public:
    MvaSoftEleEstimator(std::string);
    ~MvaSoftEleEstimator(); 
    
    Double_t mvaValue(Float_t, Float_t, Float_t,Float_t,Float_t,Float_t);

  private:
    TMVA::Reader* TMVAReader;
    float mva_sip3d, mva_sip2d, mva_ptRel, mva_deltaR, mva_ratio, mva_e_pi;
};

#endif