#include <ElectronMVAEstimatorRun2Phys14NonTrig.h>

Inheritance diagram for ElectronMVAEstimatorRun2Phys14NonTrig:

Classes
struct	AllVariables

Public Types
enum	mvaCategories { UNDEFINED = -1, CAT_EB1_PT5to10 = 0, CAT_EB2_PT5to10 = 1, CAT_EE_PT5to10 = 2, CAT_EB1_PT10plus = 3, CAT_EB2_PT10plus = 4, CAT_EE_PT10plus = 5 }

Public Member Functions
void	constrainMVAVariables ()

TMVA::Reader *	createSingleReader (const int iCategory, const edm::FileInPath &weightFile)

	ElectronMVAEstimatorRun2Phys14NonTrig (const edm::ParameterSet &conf)

void	fillMVAVariables (const edm::Ptr< reco::Candidate > &particle)

int	findCategory (const edm::Ptr< reco::Candidate > &particle)

const std::string	getName ()

int	getNCategories ()

bool	isEndcapCategory (int category)

float	mvaValue (const edm::Ptr< reco::Candidate > &particle)

	~ElectronMVAEstimatorRun2Phys14NonTrig ()

Public Member Functions inherited from AnyMVAEstimatorRun2Base
	AnyMVAEstimatorRun2Base (const edm::ParameterSet &conf)

virtual void	getEventContent (const edm::Event &iEvent)

virtual void	setConsumes (edm::ConsumesCollector &&cc)

virtual	~AnyMVAEstimatorRun2Base ()

Public Attributes
const int	nCategories = 6

Public Attributes inherited from AnyMVAEstimatorRun2Base
const edm::ParameterSet	_conf

Private Attributes
AllVariables	_allMVAVars

std::string	_MethodName

std::vector< std::unique_ptr < TMVA::Reader > >	_tmvaReaders

const std::string	name_ = "ElectronMVAEstimatorRun2Phys14NonTrig"

Detailed Description

Definition at line 15 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

Member Enumeration Documentation

enum ElectronMVAEstimatorRun2Phys14NonTrig::mvaCategories

Enumerator
UNDEFINED
CAT_EB1_PT5to10
CAT_EB2_PT5to10
CAT_EE_PT5to10
CAT_EB1_PT10plus
CAT_EB2_PT10plus
CAT_EE_PT10plus

Definition at line 22 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

                      {
     UNDEFINED = -1,
     CAT_EB1_PT5to10  = 0,
     CAT_EB2_PT5to10  = 1,
     CAT_EE_PT5to10   = 2,
     CAT_EB1_PT10plus = 3,
     CAT_EB2_PT10plus = 4,
     CAT_EE_PT10plus  = 5
   };

Constructor & Destructor Documentation

ElectronMVAEstimatorRun2Phys14NonTrig::ElectronMVAEstimatorRun2Phys14NonTrig ( const edm::ParameterSet & conf )

Definition at line 10 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

References _MethodName, _tmvaReaders, createSingleReader(), edm::hlt::Exception, edm::ParameterSet::getParameter(), i, and nCategories.

                                                                                                        :
   AnyMVAEstimatorRun2Base(conf){
 
   const std::vector <std::string> weightFileNames
     = conf.getParameter<std::vector<std::string> >("weightFileNames");
 
   if( (int)(weightFileNames.size()) != nCategories )
     throw cms::Exception("MVA config failure: ")
       << "wrong number of weightfiles" << std::endl;
 
   _tmvaReaders.clear();
   _MethodName = "BDTG method";
   // Create a TMVA reader object for each category
   for(int i=0; i<nCategories; i++){
 
     // Use unique_ptr so that all readers are properly cleaned up
     // when the vector clear() is called in the destructor
 
     edm::FileInPath weightFile( weightFileNames[i] );
     _tmvaReaders.push_back( std::unique_ptr<TMVA::Reader> ( createSingleReader(i, weightFile ) ) );
 
   }
 
 }

ElectronMVAEstimatorRun2Phys14NonTrig::~ElectronMVAEstimatorRun2Phys14NonTrig ( )

Definition at line 36 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

References _tmvaReaders.

                                         {
   
   _tmvaReaders.clear();
 }

Member Function Documentation

void ElectronMVAEstimatorRun2Phys14NonTrig::constrainMVAVariables ( )

Definition at line 249 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

Referenced by mvaValue().

                                                                  {
 
   // Check that variables do not have crazy values
 
   if(_allMVAVars.fbrem < -1.)
     _allMVAVars.fbrem = -1.;
   
   _allMVAVars.deta = fabs(_allMVAVars.deta);
   if(_allMVAVars.deta > 0.06)
     _allMVAVars.deta = 0.06;
   
   _allMVAVars.dphi = fabs(_allMVAVars.dphi);
   if(_allMVAVars.dphi > 0.6)
     _allMVAVars.dphi = 0.6;
   
   if(_allMVAVars.EoP > 20.)
     _allMVAVars.EoP = 20.;
   
   if(_allMVAVars.eleEoPout > 20.)
     _allMVAVars.eleEoPout = 20.;
   
   _allMVAVars.detacalo = fabs(_allMVAVars.detacalo);
   if(_allMVAVars.detacalo > 0.2)
     _allMVAVars.detacalo = 0.2;
   
   if(_allMVAVars.OneMinusE1x5E5x5 < -1.)
     _allMVAVars.OneMinusE1x5E5x5 = -1;
   
   if(_allMVAVars.OneMinusE1x5E5x5 > 2.)
     _allMVAVars.OneMinusE1x5E5x5 = 2.; 
   
   if(_allMVAVars.R9 > 5)
     _allMVAVars.R9 = 5;
   
   if(_allMVAVars.gsfchi2 > 200.)
     _allMVAVars.gsfchi2 = 200;
   
   if(_allMVAVars.kfchi2 > 10.)
     _allMVAVars.kfchi2 = 10.;
   
 }

TMVA::Reader * ElectronMVAEstimatorRun2Phys14NonTrig::createSingleReader	(	const int	iCategory,
		const edm::FileInPath &	weightFile
	)

Definition at line 134 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

Referenced by ElectronMVAEstimatorRun2Phys14NonTrig().

                                                                         {
 
   //
   // Create the reader  
   //
   TMVA::Reader *tmpTMVAReader = new TMVA::Reader( "!Color:Silent:Error" );
 
   //
   // Configure all variables and spectators. Note: the order and names
   // must match what is found in the xml weights file!
   //
 
   tmpTMVAReader->AddVariable("ele_kfhits",           &_allMVAVars.kfhits);
   
   // Pure ECAL -> shower shapes
   tmpTMVAReader->AddVariable("ele_oldsigmaietaieta", &_allMVAVars.see);
   tmpTMVAReader->AddVariable("ele_oldsigmaiphiiphi", &_allMVAVars.spp);
   tmpTMVAReader->AddVariable("ele_oldcircularity",   &_allMVAVars.OneMinusE1x5E5x5);
   tmpTMVAReader->AddVariable("ele_oldr9",            &_allMVAVars.R9);
   tmpTMVAReader->AddVariable("ele_scletawidth",      &_allMVAVars.etawidth);
   tmpTMVAReader->AddVariable("ele_sclphiwidth",      &_allMVAVars.phiwidth);
   tmpTMVAReader->AddVariable("ele_he",               &_allMVAVars.HoE);
   // Endcap only variables
   if( isEndcapCategory(iCategory) )
     tmpTMVAReader->AddVariable("ele_psEoverEraw",    &_allMVAVars.PreShowerOverRaw);
   
   //Pure tracking variables
   tmpTMVAReader->AddVariable("ele_kfchi2",           &_allMVAVars.kfchi2);
   tmpTMVAReader->AddVariable("ele_chi2_hits",        &_allMVAVars.gsfchi2);
 
   // Energy matching
   tmpTMVAReader->AddVariable("ele_fbrem",           &_allMVAVars.fbrem);
   tmpTMVAReader->AddVariable("ele_ep",              &_allMVAVars.EoP);
   tmpTMVAReader->AddVariable("ele_eelepout",        &_allMVAVars.eleEoPout);
   tmpTMVAReader->AddVariable("ele_IoEmIop",         &_allMVAVars.IoEmIoP);
   
   // Geometrical matchings
   tmpTMVAReader->AddVariable("ele_deltaetain",      &_allMVAVars.deta);
   tmpTMVAReader->AddVariable("ele_deltaphiin",      &_allMVAVars.dphi);
   tmpTMVAReader->AddVariable("ele_deltaetaseed",    &_allMVAVars.detacalo);
   
   // Spectator variables  
   tmpTMVAReader->AddSpectator("ele_pT",             &_allMVAVars.pt);
   tmpTMVAReader->AddSpectator("ele_isbarrel",       &_allMVAVars.isBarrel);
   tmpTMVAReader->AddSpectator("ele_isendcap",       &_allMVAVars.isEndcap);
   tmpTMVAReader->AddSpectator("scl_eta",            &_allMVAVars.SCeta);
 
   //
   // Book the method and set up the weights file
   //
   tmpTMVAReader->BookMVA(_MethodName , weightFile.fullPath() );
 
   return tmpTMVAReader;
 }

void ElectronMVAEstimatorRun2Phys14NonTrig::fillMVAVariables ( const edm::Ptr< reco::Candidate > & particle )

virtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 190 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

Referenced by mvaValue().

                                                                                                  {
 
   // Try to cast the particle into a reco particle.
   // This should work for both reco and pat.
   const edm::Ptr<reco::GsfElectron> eleRecoPtr = ( edm::Ptr<reco::GsfElectron> )particle;
   if( eleRecoPtr.get() == NULL )
     throw cms::Exception("MVA failure: ")
       << " given particle is expected to be reco::GsfElectron or pat::Electron," << std::endl
       << " but appears to be neither" << std::endl;
 
   // Both pat and reco particles have exactly the same accessors, so we use a reco ptr 
   // throughout the code, with a single exception as of this writing, handled separately below.
   auto superCluster = eleRecoPtr->superCluster();
 
   // To get to CTF track information in pat::Electron, we have to have the pointer
   // to pat::Electron, it is not accessible from the pointer to reco::GsfElectron.
   // This behavior is reported and is expected to change in the future (post-7.4.5 some time).
   bool validKF= false; 
   reco::TrackRef myTrackRef = eleRecoPtr->closestCtfTrackRef();
   const edm::Ptr<pat::Electron> elePatPtr(eleRecoPtr);
   // Check if this is really a pat::Electron, and if yes, get the track ref from this new
   // pointer instead
   if( elePatPtr.get() != NULL )
     myTrackRef = elePatPtr->closestCtfTrackRef();
   validKF = (myTrackRef.isAvailable() && (myTrackRef.isNonnull()) );  
 
   _allMVAVars.kfhits         = (validKF) ? myTrackRef->hitPattern().trackerLayersWithMeasurement() : -1. ;
   // Pure ECAL -> shower shapes
   _allMVAVars.see            = eleRecoPtr->full5x5_sigmaIetaIeta();
   _allMVAVars.spp            = eleRecoPtr->full5x5_sigmaIphiIphi();
   _allMVAVars.OneMinusE1x5E5x5 = 1. - eleRecoPtr->full5x5_e1x5() / eleRecoPtr->full5x5_e5x5();
   _allMVAVars.R9             = eleRecoPtr->full5x5_r9();
   _allMVAVars.etawidth       = superCluster->etaWidth();
   _allMVAVars.phiwidth       = superCluster->phiWidth();
   _allMVAVars.HoE            = eleRecoPtr->hadronicOverEm();
   // Endcap only variables
   _allMVAVars.PreShowerOverRaw  = superCluster->preshowerEnergy() / superCluster->rawEnergy();
   //Pure tracking variables
   _allMVAVars.kfchi2          = (validKF) ? myTrackRef->normalizedChi2() : 0;
   _allMVAVars.gsfchi2         = eleRecoPtr->gsfTrack()->normalizedChi2();
   // Energy matching
   _allMVAVars.fbrem           = eleRecoPtr->fbrem();
   _allMVAVars.EoP             = eleRecoPtr->eSuperClusterOverP();
   _allMVAVars.eleEoPout       = eleRecoPtr->eEleClusterOverPout();
   _allMVAVars.IoEmIoP         = (1.0/eleRecoPtr->ecalEnergy()) - (1.0 / eleRecoPtr->p());
   // Geometrical matchings
   _allMVAVars.deta            = eleRecoPtr->deltaEtaSuperClusterTrackAtVtx();
   _allMVAVars.dphi            = eleRecoPtr->deltaPhiSuperClusterTrackAtVtx();
   _allMVAVars.detacalo        = eleRecoPtr->deltaEtaSeedClusterTrackAtCalo();
   // Spectator variables  
   _allMVAVars.pt              = eleRecoPtr->pt();
   float scEta = superCluster->eta();
   _allMVAVars.isBarrel        = ( std::abs(scEta) < 1.479 );
   _allMVAVars.isEndcap        = ( std::abs(scEta) >= 1.479);
   _allMVAVars.SCeta           = scEta;
 
 
 }

int ElectronMVAEstimatorRun2Phys14NonTrig::findCategory ( const edm::Ptr< reco::Candidate > & particle )

virtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 80 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

References funct::abs(), CAT_EB1_PT10plus, CAT_EB1_PT5to10, CAT_EB2_PT10plus, CAT_EB2_PT5to10, CAT_EE_PT10plus, CAT_EE_PT5to10, eta(), edm::Ptr< T >::get(), NULL, EnergyCorrector::pt, and UNDEFINED.

Referenced by mvaValue().

                                                                                              {
   
   // Try to cast the particle into a reco particle.
   // This should work for both reco and pat.
   const edm::Ptr<reco::GsfElectron> eleRecoPtr = ( edm::Ptr<reco::GsfElectron> )particle;
   if( eleRecoPtr.get() == NULL )
     throw cms::Exception("MVA failure: ")
       << " given particle is expected to be reco::GsfElectron or pat::Electron," << std::endl
       << " but appears to be neither" << std::endl;
 
   float pt = eleRecoPtr->pt();
   float eta = eleRecoPtr->superCluster()->eta();
 
   //
   // Determine the category
   //
   int  iCategory = UNDEFINED;
   const float ptSplit = 10;   // we have above and below 10 GeV categories
   const float ebSplit = 0.800;// barrel is split into two regions
   const float ebeeSplit = 1.479; // division between barrel and endcap
 
   if (pt < ptSplit && std::abs(eta) < ebSplit)  
     iCategory = CAT_EB1_PT5to10;
 
   if (pt < ptSplit && std::abs(eta) >= ebSplit && std::abs(eta) < ebeeSplit)
     iCategory = CAT_EB2_PT5to10;
 
   if (pt < ptSplit && std::abs(eta) >= ebeeSplit) 
     iCategory = CAT_EE_PT5to10;
 
   if (pt >= ptSplit && std::abs(eta) < ebSplit) 
     iCategory = CAT_EB1_PT10plus;
 
   if (pt >= ptSplit && std::abs(eta) >= ebSplit && std::abs(eta) < ebeeSplit)
     iCategory = CAT_EB2_PT10plus;
 
   if (pt >= ptSplit && std::abs(eta) >= ebeeSplit) 
     iCategory = CAT_EE_PT10plus;
   
   return iCategory;
 }

const std::string ElectronMVAEstimatorRun2Phys14NonTrig::getName ( void )

inlinevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 76 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

References name_.

76 {return name_;};

ElectronMVAEstimatorRun2Phys14NonTrig::name_

const std::string name_

Definition: ElectronMVAEstimatorRun2Phys14NonTrig.h:91

int ElectronMVAEstimatorRun2Phys14NonTrig::getNCategories ( )

inlinevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 74 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

References nCategories.

74 {return nCategories;};

ElectronMVAEstimatorRun2Phys14NonTrig::nCategories

const int nCategories

Definition: ElectronMVAEstimatorRun2Phys14NonTrig.h:21

bool ElectronMVAEstimatorRun2Phys14NonTrig::isEndcapCategory ( int category )

Definition at line 123 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

References CAT_EE_PT10plus, CAT_EE_PT5to10, and GeomDetEnumerators::isEndcap().

Referenced by createSingleReader().

                                {
 
   bool isEndcap = false;
   if( category == CAT_EE_PT5to10 || category == CAT_EE_PT10plus )
     isEndcap = true;
 
   return isEndcap;
 }

float ElectronMVAEstimatorRun2Phys14NonTrig::mvaValue ( const edm::Ptr< reco::Candidate > & particle )

virtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 42 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

                                                 {
   
   int iCategory = findCategory( particle );
   fillMVAVariables( particle );  
   constrainMVAVariables();
   float result = _tmvaReaders.at(iCategory)->EvaluateMVA(_MethodName);
 
   bool debug = false;
   if(debug) {
     std::cout << " *** Inside the class _MethodName " << _MethodName << std::endl;
     std::cout << " bin " << iCategory
           << " fbrem " <<  _allMVAVars.fbrem
          << " kfchi2 " << _allMVAVars.kfchi2
          << " mykfhits " << _allMVAVars.kfhits
          << " gsfchi2 " << _allMVAVars.gsfchi2
          << " deta " <<  _allMVAVars.deta
          << " dphi " << _allMVAVars.dphi
          << " detacalo " << _allMVAVars.detacalo
          << " see " << _allMVAVars.see
          << " spp " << _allMVAVars.spp
          << " etawidth " << _allMVAVars.etawidth
          << " phiwidth " << _allMVAVars.phiwidth
          << " OneMinusE1x5E5x5 " << _allMVAVars.OneMinusE1x5E5x5
          << " R9 " << _allMVAVars.R9
          << " HoE " << _allMVAVars.HoE
          << " EoP " << _allMVAVars.EoP
          << " IoEmIoP " << _allMVAVars.IoEmIoP
          << " eleEoPout " << _allMVAVars.eleEoPout
       //<< " d0 " << _allMVAVars.d0
       //   << " ip3d " << _allMVAVars.ip3d
          << " eta " << _allMVAVars.SCeta
           << " pt " << _allMVAVars.pt << std::endl;
     std::cout << " ### MVA " << result << std::endl;
   }
 
   return result;
 }