#include <ElectronMVAEstimatorRun2Spring15Trig.h>

Inheritance diagram for ElectronMVAEstimatorRun2Spring15Trig:

Classes
struct	AllVariables

Public Types
enum	mvaCategories { UNDEFINED = -1, CAT_EB1 = 0, CAT_EB2 = 1, CAT_EE = 2 }

Public Member Functions
void	constrainMVAVariables (AllVariables &) const

	ElectronMVAEstimatorRun2Spring15Trig (const edm::ParameterSet &conf)

std::vector< float >	fillMVAVariables (const edm::Ptr< reco::Candidate > &particle, const edm::Event &) const override

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

const std::string &	getName () const final

int	getNCategories () const override

const std::string &	getTag () const final

bool	isEndcapCategory (int category) const

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

void	setConsumes (edm::ConsumesCollector &&) const final

	~ElectronMVAEstimatorRun2Spring15Trig () override

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

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

template<typename... Args>
std::vector< float >	packMVAVariables (const Args...args) const

virtual	~AnyMVAEstimatorRun2Base ()

Public Attributes
const int	nCategories = 3

Public Attributes inherited from AnyMVAEstimatorRun2Base
const edm::ParameterSet	_conf

Private Attributes
AllVariables	_allMVAVars

const edm::InputTag	_beamSpotLabel

const edm::InputTag	_conversionsLabelAOD

const edm::InputTag	_conversionsLabelMiniAOD

std::vector< std::unique_ptr< const GBRForest > >	_gbrForests

const std::string	_MethodName

const std::string	_name = "ElectronMVAEstimatorRun2Spring15Trig"

const std::string	_tag

Detailed Description

Definition at line 19 of file ElectronMVAEstimatorRun2Spring15Trig.h.

Member Enumeration Documentation

enum ElectronMVAEstimatorRun2Spring15Trig::mvaCategories

Enumerator
UNDEFINED
CAT_EB1
CAT_EB2
CAT_EE

Definition at line 26 of file ElectronMVAEstimatorRun2Spring15Trig.h.

                      {
     UNDEFINED = -1,
     CAT_EB1  = 0,
     CAT_EB2  = 1,
     CAT_EE   = 2
   };

Constructor & Destructor Documentation

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

Definition at line 13 of file ElectronMVAEstimatorRun2Spring15Trig.cc.

References _gbrForests, GBRForestTools::createGBRForest(), Exception, edm::ParameterSet::getParameter(), mps_fire::i, nCategories, electrons_cff::weightFile, mvaElectronID_Fall17_iso_V1_cff::weightFileNames, and ~ElectronMVAEstimatorRun2Spring15Trig().

                                                                                                      :
   AnyMVAEstimatorRun2Base(conf),
   _tag(conf.getParameter<std::string>("mvaTag")),
   _MethodName("BDTG method"),
   _beamSpotLabel(conf.getParameter<edm::InputTag>("beamSpot")),
   _conversionsLabelAOD(conf.getParameter<edm::InputTag>("conversionsAOD")),
   _conversionsLabelMiniAOD(conf.getParameter<edm::InputTag>("conversionsMiniAOD")) {
   
   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;
 
   _gbrForests.clear();
   // 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] );
     _gbrForests.push_back( GBRForestTools::createGBRForest( weightFile ) );
 
   }
 
 }

ElectronMVAEstimatorRun2Spring15Trig::~ElectronMVAEstimatorRun2Spring15Trig ( )

override

Definition at line 43 of file ElectronMVAEstimatorRun2Spring15Trig.cc.

Referenced by ElectronMVAEstimatorRun2Spring15Trig().

43 {

44 }

Member Function Documentation

void ElectronMVAEstimatorRun2Spring15Trig::constrainMVAVariables ( AllVariables & allMVAVars ) const

Definition at line 307 of file ElectronMVAEstimatorRun2Spring15Trig.cc.

Referenced by fillMVAVariables(), and getTag().

                                                                                                {
 
   // 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.;
   
 }

std::vector< float > ElectronMVAEstimatorRun2Spring15Trig::fillMVAVariables	(	const edm::Ptr< reco::Candidate > &	particle,
		const edm::Event &	iEvent
	)		const

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 140 of file ElectronMVAEstimatorRun2Spring15Trig.cc.

Referenced by getTag(), isEndcapCategory(), and mvaValue().

                                                 {
 
   // 
   // Declare all value maps corresponding to the products we defined earlier
   //
   edm::Handle<reco::BeamSpot> theBeamSpot;
   edm::Handle<reco::ConversionCollection> conversions;
 
   // Get data needed for conversion rejection
   iEvent.getByLabel(_beamSpotLabel, theBeamSpot);
 
   // Conversions in miniAOD and AOD have different names,
   // but the same type, so we use the same handle with different tokens.
   iEvent.getByLabel(_conversionsLabelAOD, conversions);
   if( !conversions.isValid() )
     iEvent.getByLabel(_conversionsLabelMiniAOD, conversions);
 
   // Make sure everything is retrieved successfully
   if(! (theBeamSpot.isValid() 
     && conversions.isValid() ) 
      )
     throw cms::Exception("MVA failure: ")
       << "Failed to retrieve event content needed for this MVA" 
       << std::endl
       << "Check python MVA configuration file."
       << std::endl;
 
   // 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() == nullptr )
     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();
   
   AllVariables allMVAVars;
 
   // 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();
 
   // 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() != nullptr )
     myTrackRef = elePatPtr->closestCtfTrackRef();
   validKF = (myTrackRef.isAvailable() && (myTrackRef.isNonnull()) );  
 
   //Pure tracking variables
   allMVAVars.kfhits         = (validKF) ? myTrackRef->hitPattern().trackerLayersWithMeasurement() : -1. ;
   allMVAVars.kfchi2          = (validKF) ? myTrackRef->normalizedChi2() : 0;
   allMVAVars.gsfchi2         = eleRecoPtr->gsfTrack()->normalizedChi2();
 
   // Energy matching
   allMVAVars.fbrem           = eleRecoPtr->fbrem();
 
   allMVAVars.gsfhits         = eleRecoPtr->gsfTrack()->hitPattern().trackerLayersWithMeasurement();
   allMVAVars.expectedMissingInnerHits = eleRecoPtr->gsfTrack()
     ->hitPattern().numberOfLostHits(reco::HitPattern::MISSING_INNER_HITS);
 
   reco::ConversionRef conv_ref = ConversionTools::matchedConversion(*eleRecoPtr,
                                     conversions, 
                                     theBeamSpot->position());
   double vertexFitProbability = -1.; 
   if(!conv_ref.isNull()) {
     const reco::Vertex &vtx = conv_ref.get()->conversionVertex(); if (vtx.isValid()) {
       vertexFitProbability = TMath::Prob( vtx.chi2(), vtx.ndof());
     } 
   }
   allMVAVars.convVtxFitProbability    = vertexFitProbability;
 
   allMVAVars.EoP             = eleRecoPtr->eSuperClusterOverP();
   allMVAVars.eleEoPout       = eleRecoPtr->eEleClusterOverPout();
   float pAtVertex            = eleRecoPtr->trackMomentumAtVtx().R();
   allMVAVars.IoEmIoP         = (1.0/eleRecoPtr->ecalEnergy()) - (1.0 / pAtVertex );
 
   // 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.SCeta           = scEta;
 
   constrainMVAVariables(allMVAVars);
 
   std::vector<float> vars;
 
   if( isEndcapCategory( findCategory( particle ) ) ) {
     vars = packMVAVariables(allMVAVars.see,
                                        allMVAVars.spp,
                                        allMVAVars.OneMinusE1x5E5x5,
                                        allMVAVars.R9,
                                        allMVAVars.etawidth,
                                        allMVAVars.phiwidth,
                                        allMVAVars.HoE,
                                        // Endcap only variables
                                        allMVAVars.PreShowerOverRaw,                                       
                                        //Pure tracking variables
                                        allMVAVars.kfhits,
                                        allMVAVars.kfchi2,
                                        allMVAVars.gsfchi2,                                       
                                        // Energy matching
                                        allMVAVars.fbrem,                                       
                                        allMVAVars.gsfhits,
                                        allMVAVars.expectedMissingInnerHits,
                                        allMVAVars.convVtxFitProbability,
                                        allMVAVars.EoP,
                                        allMVAVars.eleEoPout,
                                        allMVAVars.IoEmIoP,                                       
                                        // Geometrical matchings
                                        allMVAVars.deta,
                                        allMVAVars.dphi,
                                        allMVAVars.detacalo,                                       
                                        // Spectator variables  
                                        allMVAVars.pt,
                                        allMVAVars.SCeta);
   } else {
     vars = packMVAVariables(allMVAVars.see,
                                        allMVAVars.spp,
                                        allMVAVars.OneMinusE1x5E5x5,
                                        allMVAVars.R9,
                                        allMVAVars.etawidth,
                                        allMVAVars.phiwidth,
                                        allMVAVars.HoE,
                                        //Pure tracking variables
                                        allMVAVars.kfhits,
                                        allMVAVars.kfchi2,
                                        allMVAVars.gsfchi2,                                       
                                        // Energy matching
                                        allMVAVars.fbrem,                                       
                                        allMVAVars.gsfhits,
                                        allMVAVars.expectedMissingInnerHits,
                                        allMVAVars.convVtxFitProbability,
                                        allMVAVars.EoP,
                                        allMVAVars.eleEoPout,
                                        allMVAVars.IoEmIoP,                                       
                                        // Geometrical matchings
                                        allMVAVars.deta,
                                        allMVAVars.dphi,
                                        allMVAVars.detacalo,                                       
                                        // Spectator variables  
                                        allMVAVars.pt,
                                        allMVAVars.SCeta);
   }
   return vars;
 }

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

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 97 of file ElectronMVAEstimatorRun2Spring15Trig.cc.

References funct::abs(), CAT_EB1, CAT_EB2, CAT_EE, mvaElectronID_Fall17_iso_V1_cff::ebeeSplit, mvaElectronID_Fall17_iso_V1_cff::ebSplit, PVValHelper::eta, edm::Ptr< T >::get(), isEndcapCategory(), nullptr, reco::GsfElectron::superCluster(), and UNDEFINED.

Referenced by fillMVAVariables(), getTag(), and 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() == nullptr )
     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 eta = eleRecoPtr->superCluster()->eta();
 
   //
   // Determine the category
   //
   int  iCategory = UNDEFINED;
   const float ebSplit = 0.800;// barrel is split into two regions
   const float ebeeSplit = 1.479; // division between barrel and endcap
 
   if (std::abs(eta) < ebSplit)  
     iCategory = CAT_EB1;
 
   if (std::abs(eta) >= ebSplit && std::abs(eta) < ebeeSplit)
     iCategory = CAT_EB2;
 
   if (std::abs(eta) >= ebeeSplit) 
     iCategory = CAT_EE;
   
   return iCategory;
 }

const std::string& ElectronMVAEstimatorRun2Spring15Trig::getName ( void ) const

inlinefinalvirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 89 of file ElectronMVAEstimatorRun2Spring15Trig.h.

References _name.

Referenced by plotting.Plot::draw().

89 { return _name; }

ElectronMVAEstimatorRun2Spring15Trig::_name

const std::string _name

Definition: ElectronMVAEstimatorRun2Spring15Trig.h:111

int ElectronMVAEstimatorRun2Spring15Trig::getNCategories ( ) const

inlineoverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 87 of file ElectronMVAEstimatorRun2Spring15Trig.h.

References python.rootplot.argparse::category, isEndcapCategory(), and nCategories.

87 { return nCategories; }

ElectronMVAEstimatorRun2Spring15Trig::nCategories

const int nCategories

Definition: ElectronMVAEstimatorRun2Spring15Trig.h:25

const std::string& ElectronMVAEstimatorRun2Spring15Trig::getTag ( ) const

inlinefinalvirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 90 of file ElectronMVAEstimatorRun2Spring15Trig.h.

References _tag, constrainMVAVariables(), fillMVAVariables(), findCategory(), and setConsumes().

90 { return _tag; }

ElectronMVAEstimatorRun2Spring15Trig::_tag

const std::string _tag

Definition: ElectronMVAEstimatorRun2Spring15Trig.h:115

bool ElectronMVAEstimatorRun2Spring15Trig::isEndcapCategory ( int category ) const

Definition at line 129 of file ElectronMVAEstimatorRun2Spring15Trig.cc.

References CAT_EE, fillMVAVariables(), and GeomDetEnumerators::isEndcap().

Referenced by fillMVAVariables(), findCategory(), and getNCategories().

                                       {
 
   bool isEndcap = false;
   if( category == CAT_EE )
     isEndcap = true;
 
   return isEndcap;
 }

float ElectronMVAEstimatorRun2Spring15Trig::mvaValue	(	const edm::Ptr< reco::Candidate > &	particle,
		const edm::Event &	iEvent
	)		const

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 62 of file ElectronMVAEstimatorRun2Spring15Trig.cc.

References _gbrForests, _MethodName, gather_cfg::cout, debug, fillMVAVariables(), findCategory(), and mps_fire::result.

Referenced by setConsumes().

                                                                                {
   
   const int iCategory = findCategory( particle );
   const std::vector<float> vars = fillMVAVariables( particle, iEvent );
   const float result = _gbrForests.at(iCategory)->GetClassifier(vars.data());
 
   const bool debug = false;
   if(debug) {
     std::cout << " *** Inside the class _MethodName " << _MethodName << std::endl;
     std::cout << " bin " << iCategory
           << " fbrem " <<  vars[11]
           << " kfchi2 " << vars[9]
           << " mykfhits " << vars[8]
           << " gsfchi2 " << vars[10]
           << " deta " <<  vars[18]
           << " dphi " << vars[19]
           << " detacalo " << vars[20]
           << " see " << vars[0]
           << " spp " << vars[1]
           << " etawidth " << vars[4]
           << " phiwidth " << vars[5]
           << " OneMinusE1x5E5x5 " << vars[2]
           << " R9 " << vars[3]
           << " HoE " << vars[6]
           << " EoP " << vars[15]
           << " IoEmIoP " << vars[17]
           << " eleEoPout " << vars[16]
           << " eta " << vars[22]
           << " pt " << vars[21] << std::endl;
     std::cout << " ### MVA " << result << std::endl;
   }
 
   return result;
 }

void ElectronMVAEstimatorRun2Spring15Trig::setConsumes ( edm::ConsumesCollector && cc ) const

finalvirtual

Reimplemented from AnyMVAEstimatorRun2Base.

Definition at line 47 of file ElectronMVAEstimatorRun2Spring15Trig.cc.

References _beamSpotLabel, _conversionsLabelAOD, _conversionsLabelMiniAOD, and mvaValue().

Referenced by getTag().

                                                                                       {
 
   // All tokens for event content needed by this MVA
 
   // Beam spot (same for AOD and miniAOD)
    cc.consumes<reco::BeamSpot>(_beamSpotLabel);
 
   // Conversions collection (different names in AOD and miniAOD)
   cc.mayConsume<reco::ConversionCollection>(_conversionsLabelAOD);
   cc.mayConsume<reco::ConversionCollection>(_conversionsLabelMiniAOD);
   
 
 }