CMS 3D CMS Logo
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
List of all members | Classes | Public Types | Public Member Functions | Static Public Attributes | Private Attributes
ElectronMVAEstimatorRun2Phys14NonTrig Class Reference

#include <ElectronMVAEstimatorRun2Phys14NonTrig.h>

Inheritance diagram for ElectronMVAEstimatorRun2Phys14NonTrig:
AnyMVAEstimatorRun2Base

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 (AllVariables &vars) const
 
std::unique_ptr< const GBRForest > createSingleReader (const int iCategory, const edm::FileInPath &weightFile)
 
 ElectronMVAEstimatorRun2Phys14NonTrig (const edm::ParameterSet &conf)
 
std::vector< float > fillMVAVariables (const edm::Ptr< reco::Candidate > &particle, const edm::Event &) const
 
int findCategory (const edm::Ptr< reco::Candidate > &particle) const
 
virtual const std::string & getName () const overridefinal
 
virtual int getNCategories () const overridefinal
 
virtual const std::string & getTag () const overridefinal
 
bool isEndcapCategory (int category) const
 
float mvaValue (const edm::Ptr< reco::Candidate > &particle, const edm::Event &evt) const
 
 ~ElectronMVAEstimatorRun2Phys14NonTrig ()
 
- 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 void setConsumes (edm::ConsumesCollector &&cc) const
 
virtual ~AnyMVAEstimatorRun2Base ()
 

Static Public Attributes

static int nCategories = 6
 

Private Attributes

AllVariables _allMVAVars
 
std::vector< std::unique_ptr
< const GBRForest > > 		_gbrForests
 
std::string _MethodName
 
const std::string _name = "ElectronMVAEstimatorRun2Phys14NonTrig"
 
std::string _tag
 

Additional Inherited Members

- Public Attributes inherited from AnyMVAEstimatorRun2Base
const edm::ParameterSet _conf
 

Detailed Description

Definition at line 19 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 26 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

26  {
27  UNDEFINED = -1,
28  CAT_EB1_PT5to10 = 0,
29  CAT_EB2_PT5to10 = 1,
30  CAT_EE_PT5to10 = 2,
31  CAT_EB1_PT10plus = 3,
32  CAT_EB2_PT10plus = 4,
33  CAT_EE_PT10plus = 5
34  };

Constructor & Destructor Documentation

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

Definition at line 10 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

References _gbrForests, _MethodName, _tag, createSingleReader(), Exception, edm::ParameterSet::getParameter(), i, nCategories, and AlCaHLTBitMon_QueryRunRegistry::string.

10  :
11  AnyMVAEstimatorRun2Base(conf) {
12 
13  _tag = conf.getParameter<std::string>("mvaTag");
14 
15  const std::vector <std::string> weightFileNames
16  = conf.getParameter<std::vector<std::string> >("weightFileNames");
17 
18  if( (int)(weightFileNames.size()) != nCategories )
19  throw cms::Exception("MVA config failure: ")
20  << "wrong number of weightfiles" << std::endl;
21 
22  _gbrForests.clear();
23  _MethodName = "BDTG method";
24  // Create a TMVA reader object for each category
25  for(int i=0; i<nCategories; i++){
26 
27  // Use unique_ptr so that all readers are properly cleaned up
28  // when the vector clear() is called in the destructor
29 
30  edm::FileInPath weightFile( weightFileNames[i] );
31  _gbrForests.push_back( createSingleReader(i, weightFile ) );
32 
33  }
34 
35 }
edm::ParameterSet::getParameter
T getParameter(std::string const &) const
i
int i
Definition: DBlmapReader.cc:9
AnyMVAEstimatorRun2Base::AnyMVAEstimatorRun2Base
AnyMVAEstimatorRun2Base(const edm::ParameterSet &conf)
Definition: AnyMVAEstimatorRun2Base.h:16
AlCaHLTBitMon_QueryRunRegistry.string
string string
Definition: AlCaHLTBitMon_QueryRunRegistry.py:255
ElectronMVAEstimatorRun2Phys14NonTrig::_gbrForests
std::vector< std::unique_ptr< const GBRForest > > _gbrForests
Definition: ElectronMVAEstimatorRun2Phys14NonTrig.h:103
ElectronMVAEstimatorRun2Phys14NonTrig::createSingleReader
std::unique_ptr< const GBRForest > createSingleReader(const int iCategory, const edm::FileInPath &weightFile)
Definition: ElectronMVAEstimatorRun2Phys14NonTrig.cc:137
ElectronMVAEstimatorRun2Phys14NonTrig::~ElectronMVAEstimatorRun2Phys14NonTrig ( )

Definition at line 38 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

38  {
39 }

Member Function Documentation

void ElectronMVAEstimatorRun2Phys14NonTrig::constrainMVAVariables ( AllVariables &  vars) const

Definition at line 311 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

References funct::abs(), ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::deta, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::detacalo, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::dphi, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::eleEoPout, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::EoP, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::fbrem, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::gsfchi2, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::kfchi2, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::OneMinusE1x5E5x5, and ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::R9.

Referenced by fillMVAVariables().

311  {
312 
313  // Check that variables do not have crazy values
314 
315  if(vars.fbrem < -1.)
316  vars.fbrem = -1.;
317 
318  vars.deta = std::abs(vars.deta);
319  if(vars.deta > 0.06)
320  vars.deta = 0.06;
321 
322  vars.dphi = std::abs(vars.dphi);
323  if(vars.dphi > 0.6)
324  vars.dphi = 0.6;
325 
326  if(vars.EoP > 20.)
327  vars.EoP = 20.;
328 
329  if(vars.eleEoPout > 20.)
330  vars.eleEoPout = 20.;
331 
332  vars.detacalo = std::abs(vars.detacalo);
333  if(vars.detacalo > 0.2)
334  vars.detacalo = 0.2;
335 
336  if(vars.OneMinusE1x5E5x5 < -1.)
337  vars.OneMinusE1x5E5x5 = -1;
338 
339  if(vars.OneMinusE1x5E5x5 > 2.)
340  vars.OneMinusE1x5E5x5 = 2.;
341 
342  if(vars.R9 > 5)
343  vars.R9 = 5;
344 
345  if(vars.gsfchi2 > 200.)
346  vars.gsfchi2 = 200;
347 
348  if(vars.kfchi2 > 10.)
349  vars.kfchi2 = 10.;
350 }
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
std::unique_ptr< const GBRForest > ElectronMVAEstimatorRun2Phys14NonTrig::createSingleReader ( const int  iCategory,
const edm::FileInPath &  weightFile 
)

Definition at line 137 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

References _allMVAVars, _MethodName, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::deta, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::detacalo, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::dphi, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::eleEoPout, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::EoP, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::etawidth, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::fbrem, edm::FileInPath::fullPath(), ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::gsfchi2, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::HoE, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::IoEmIoP, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::isBarrel, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::isEndcap, isEndcapCategory(), ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::kfchi2, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::kfhits, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::OneMinusE1x5E5x5, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::phiwidth, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::PreShowerOverRaw, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::pt, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::R9, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::SCeta, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::see, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::spp, and groupFilesInBlocks::temp.

Referenced by ElectronMVAEstimatorRun2Phys14NonTrig().

137  {
138 
139  //
140  // Create the reader
141  //
142  TMVA::Reader tmpTMVAReader( "!Color:Silent:Error" );
143 
144  //
145  // Configure all variables and spectators. Note: the order and names
146  // must match what is found in the xml weights file!
147  //
148 
149  tmpTMVAReader.AddVariable("ele_kfhits", &_allMVAVars.kfhits);
150 
151  // Pure ECAL -> shower shapes
152  tmpTMVAReader.AddVariable("ele_oldsigmaietaieta", &_allMVAVars.see);
153  tmpTMVAReader.AddVariable("ele_oldsigmaiphiiphi", &_allMVAVars.spp);
154  tmpTMVAReader.AddVariable("ele_oldcircularity", &_allMVAVars.OneMinusE1x5E5x5);
155  tmpTMVAReader.AddVariable("ele_oldr9", &_allMVAVars.R9);
156  tmpTMVAReader.AddVariable("ele_scletawidth", &_allMVAVars.etawidth);
157  tmpTMVAReader.AddVariable("ele_sclphiwidth", &_allMVAVars.phiwidth);
158  tmpTMVAReader.AddVariable("ele_he", &_allMVAVars.HoE);
159  // Endcap only variables
160  if( isEndcapCategory(iCategory) )
161  tmpTMVAReader.AddVariable("ele_psEoverEraw", &_allMVAVars.PreShowerOverRaw);
162 
163  //Pure tracking variables
164  tmpTMVAReader.AddVariable("ele_kfchi2", &_allMVAVars.kfchi2);
165  tmpTMVAReader.AddVariable("ele_chi2_hits", &_allMVAVars.gsfchi2);
166 
167  // Energy matching
168  tmpTMVAReader.AddVariable("ele_fbrem", &_allMVAVars.fbrem);
169  tmpTMVAReader.AddVariable("ele_ep", &_allMVAVars.EoP);
170  tmpTMVAReader.AddVariable("ele_eelepout", &_allMVAVars.eleEoPout);
171  tmpTMVAReader.AddVariable("ele_IoEmIop", &_allMVAVars.IoEmIoP);
172 
173  // Geometrical matchings
174  tmpTMVAReader.AddVariable("ele_deltaetain", &_allMVAVars.deta);
175  tmpTMVAReader.AddVariable("ele_deltaphiin", &_allMVAVars.dphi);
176  tmpTMVAReader.AddVariable("ele_deltaetaseed", &_allMVAVars.detacalo);
177 
178  // Spectator variables
179  tmpTMVAReader.AddSpectator("ele_pT", &_allMVAVars.pt);
180  tmpTMVAReader.AddSpectator("ele_isbarrel", &_allMVAVars.isBarrel);
181  tmpTMVAReader.AddSpectator("ele_isendcap", &_allMVAVars.isEndcap);
182  tmpTMVAReader.AddSpectator("scl_eta", &_allMVAVars.SCeta);
183 
184  //
185  // Book the method and set up the weights file
186  //
187  std::unique_ptr<TMVA::IMethod> temp( tmpTMVAReader.BookMVA(_MethodName , weightFile.fullPath() ) );
188 
189  return std::unique_ptr<const GBRForest> ( new GBRForest( dynamic_cast<TMVA::MethodBDT*>( tmpTMVAReader.FindMVA(_MethodName) ) ) );
190 }
edm::FileInPath::fullPath
std::string fullPath() const
Definition: FileInPath.cc:184
std::vector< float > ElectronMVAEstimatorRun2Phys14NonTrig::fillMVAVariables ( const edm::Ptr< reco::Candidate > &  particle,
const edm::Event &   
) const
virtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 194 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

References funct::abs(), constexpr, constrainMVAVariables(), ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::deta, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::detacalo, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::dphi, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::eleEoPout, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::EoP, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::etawidth, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::fbrem, findCategory(), edm::Ptr< T >::get(), ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::gsfchi2, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::HoE, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::IoEmIoP, edm::Ref< C, T, F >::isAvailable(), ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::isBarrel, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::isEndcap, isEndcapCategory(), edm::Ref< C, T, F >::isNonnull(), ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::kfchi2, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::kfhits, eostools::move(), NULL, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::OneMinusE1x5E5x5, AnyMVAEstimatorRun2Base::packMVAVariables(), ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::phiwidth, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::PreShowerOverRaw, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::pt, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::R9, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::SCeta, ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::see, and ElectronMVAEstimatorRun2Phys14NonTrig::AllVariables::spp.

Referenced by mvaValue().

195  {
196 
197  // Try to cast the particle into a reco particle.
198  // This should work for both reco and pat.
199  const edm::Ptr<reco::GsfElectron> eleRecoPtr = ( edm::Ptr<reco::GsfElectron> )particle;
200  if( eleRecoPtr.get() == NULL )
201  throw cms::Exception("MVA failure: ")
202  << " given particle is expected to be reco::GsfElectron or pat::Electron," << std::endl
203  << " but appears to be neither" << std::endl;
204 
205  // the instance of the variables that we will manipulate
206  AllVariables allMVAVars;
207 
208  // Both pat and reco particles have exactly the same accessors, so we use a reco ptr
209  // throughout the code, with a single exception as of this writing, handled separately below.
210  auto superCluster = eleRecoPtr->superCluster();
211 
212  // To get to CTF track information in pat::Electron, we have to have the pointer
213  // to pat::Electron, it is not accessible from the pointer to reco::GsfElectron.
214  // This behavior is reported and is expected to change in the future (post-7.4.5 some time).
215  bool validKF= false;
216  reco::TrackRef myTrackRef = eleRecoPtr->closestCtfTrackRef();
217  const edm::Ptr<pat::Electron> elePatPtr(eleRecoPtr);
218  // Check if this is really a pat::Electron, and if yes, get the track ref from this new
219  // pointer instead
220  if( elePatPtr.get() != NULL )
221  myTrackRef = elePatPtr->closestCtfTrackRef();
222  validKF = (myTrackRef.isAvailable() && (myTrackRef.isNonnull()) );
223 
224  allMVAVars.kfhits = (validKF) ? myTrackRef->hitPattern().trackerLayersWithMeasurement() : -1. ;
225  // Pure ECAL -> shower shapes
226  allMVAVars.see = eleRecoPtr->full5x5_sigmaIetaIeta();
227  allMVAVars.spp = eleRecoPtr->full5x5_sigmaIphiIphi();
228  allMVAVars.OneMinusE1x5E5x5 = 1. - eleRecoPtr->full5x5_e1x5() / eleRecoPtr->full5x5_e5x5();
229  allMVAVars.R9 = eleRecoPtr->full5x5_r9();
230  allMVAVars.etawidth = superCluster->etaWidth();
231  allMVAVars.phiwidth = superCluster->phiWidth();
232  allMVAVars.HoE = eleRecoPtr->hadronicOverEm();
233  // Endcap only variables
234  allMVAVars.PreShowerOverRaw = superCluster->preshowerEnergy() / superCluster->rawEnergy();
235  //Pure tracking variables
236  allMVAVars.kfchi2 = (validKF) ? myTrackRef->normalizedChi2() : 0;
237  allMVAVars.gsfchi2 = eleRecoPtr->gsfTrack()->normalizedChi2();
238  // Energy matching
239  allMVAVars.fbrem = eleRecoPtr->fbrem();
240  allMVAVars.EoP = eleRecoPtr->eSuperClusterOverP();
241  allMVAVars.eleEoPout = eleRecoPtr->eEleClusterOverPout();
242  allMVAVars.IoEmIoP = (1.0/eleRecoPtr->ecalEnergy()) - (1.0 / eleRecoPtr->p());
243  // Geometrical matchings
244  allMVAVars.deta = eleRecoPtr->deltaEtaSuperClusterTrackAtVtx();
245  allMVAVars.dphi = eleRecoPtr->deltaPhiSuperClusterTrackAtVtx();
246  allMVAVars.detacalo = eleRecoPtr->deltaEtaSeedClusterTrackAtCalo();
247  // Spectator variables
248  allMVAVars.pt = eleRecoPtr->pt();
249  const float scEta = superCluster->eta();
250  constexpr float ebeeSplit = 1.479;
251  allMVAVars.isBarrel = ( std::abs(scEta) < ebeeSplit );
252  allMVAVars.isEndcap = ( std::abs(scEta) >= ebeeSplit );
253  allMVAVars.SCeta = scEta;
254 
255  constrainMVAVariables(allMVAVars);
256 
257  std::vector<float> vars;
258 
259  if( isEndcapCategory( findCategory(particle) ) ) {
260  vars = std::move( packMVAVariables( allMVAVars.kfhits,
261  allMVAVars.see,
262  allMVAVars.spp,
263  allMVAVars.OneMinusE1x5E5x5,
264  allMVAVars.R9,
265  allMVAVars.etawidth,
266  allMVAVars.phiwidth,
267  allMVAVars.HoE,
268  allMVAVars.PreShowerOverRaw,
269  allMVAVars.kfchi2,
270  allMVAVars.gsfchi2,
271  allMVAVars.fbrem,
272  allMVAVars.EoP,
273  allMVAVars.eleEoPout,
274  allMVAVars.IoEmIoP,
275  allMVAVars.deta,
276  allMVAVars.dphi,
277  allMVAVars.detacalo,
278  allMVAVars.pt,
279  allMVAVars.isBarrel,
280  allMVAVars.isEndcap,
281  allMVAVars.SCeta )
282  );
283  } else {
284  vars = std::move( packMVAVariables( allMVAVars.kfhits,
285  allMVAVars.see,
286  allMVAVars.spp,
287  allMVAVars.OneMinusE1x5E5x5,
288  allMVAVars.R9,
289  allMVAVars.etawidth,
290  allMVAVars.phiwidth,
291  allMVAVars.HoE,
292  allMVAVars.kfchi2,
293  allMVAVars.gsfchi2,
294  allMVAVars.fbrem,
295  allMVAVars.EoP,
296  allMVAVars.eleEoPout,
297  allMVAVars.IoEmIoP,
298  allMVAVars.deta,
299  allMVAVars.dphi,
300  allMVAVars.detacalo,
301  allMVAVars.pt,
302  allMVAVars.isBarrel,
303  allMVAVars.isEndcap,
304  allMVAVars.SCeta )
305  );
306  }
307 
308  return vars;
309 }
edm::Ref::isAvailable
bool isAvailable() const
Definition: Ref.h:576
edm::Ref::isNonnull
bool isNonnull() const
Checks for non-null.
Definition: Ref.h:252
edm::Ptr::get
T const * get() const
Returns C++ pointer to the item.
Definition: Ptr.h:160
NULL
#define NULL
Definition: scimark2.h:8
constexpr
#define constexpr
AnyMVAEstimatorRun2Base::packMVAVariables
std::vector< float > packMVAVariables(const Args...args) const
Definition: AnyMVAEstimatorRun2Base.h:44
ElectronMVAEstimatorRun2Phys14NonTrig::findCategory
int findCategory(const edm::Ptr< reco::Candidate > &particle) const
Definition: ElectronMVAEstimatorRun2Phys14NonTrig.cc:82
eostools.move
def move
Definition: eostools.py:510
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
int ElectronMVAEstimatorRun2Phys14NonTrig::findCategory ( const edm::Ptr< reco::Candidate > &  particle) const
virtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 82 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, constexpr, eta, edm::Ptr< T >::get(), NULL, EnergyCorrector::pt, and UNDEFINED.

Referenced by fillMVAVariables(), and mvaValue().

82  {
83 
84  // Try to cast the particle into a reco particle.
85  // This should work for both reco and pat.
86  const edm::Ptr<reco::GsfElectron> eleRecoPtr = ( edm::Ptr<reco::GsfElectron> )particle;
87  if( eleRecoPtr.get() == NULL )
88  throw cms::Exception("MVA failure: ")
89  << " given particle is expected to be reco::GsfElectron or pat::Electron," << std::endl
90  << " but appears to be neither" << std::endl;
91 
92  const float pt = eleRecoPtr->pt();
93  const float eta = eleRecoPtr->superCluster()->eta();
94 
95  //
96  // Determine the category
97  //
98  int iCategory = UNDEFINED;
99  constexpr float ptSplit = 10; // we have above and below 10 GeV categories
100  constexpr float ebSplit = 0.800;// barrel is split into two regions
101  constexpr float ebeeSplit = 1.479; // division between barrel and endcap
102 
103  if (pt < ptSplit && std::abs(eta) < ebSplit)
104  iCategory = CAT_EB1_PT5to10;
105 
106  if (pt < ptSplit && std::abs(eta) >= ebSplit && std::abs(eta) < ebeeSplit)
107  iCategory = CAT_EB2_PT5to10;
108 
109  if (pt < ptSplit && std::abs(eta) >= ebeeSplit)
110  iCategory = CAT_EE_PT5to10;
111 
112  if (pt >= ptSplit && std::abs(eta) < ebSplit)
113  iCategory = CAT_EB1_PT10plus;
114 
115  if (pt >= ptSplit && std::abs(eta) >= ebSplit && std::abs(eta) < ebeeSplit)
116  iCategory = CAT_EB2_PT10plus;
117 
118  if (pt >= ptSplit && std::abs(eta) >= ebeeSplit)
119  iCategory = CAT_EE_PT10plus;
120 
121  return iCategory;
122 }
edm::Ptr::get
T const * get() const
Returns C++ pointer to the item.
Definition: Ptr.h:160
NULL
#define NULL
Definition: scimark2.h:8
constexpr
#define constexpr
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
virtual const std::string& ElectronMVAEstimatorRun2Phys14NonTrig::getName ( void  ) const
inlinefinaloverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 80 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

References _name.

Referenced by plotting.Plot::draw().

80 { return _name; }
virtual int ElectronMVAEstimatorRun2Phys14NonTrig::getNCategories ( ) const
inlinefinaloverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 78 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

References nCategories.

78 { return nCategories; }
virtual const std::string& ElectronMVAEstimatorRun2Phys14NonTrig::getTag ( void  ) const
inlinefinaloverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 81 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

References _tag.

81 { return _tag; }
bool ElectronMVAEstimatorRun2Phys14NonTrig::isEndcapCategory ( int  category) const

Definition at line 125 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

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

Referenced by createSingleReader(), and fillMVAVariables().

125  {
126 
127  bool isEndcap = false;
128  if( category == CAT_EE_PT5to10 || category == CAT_EE_PT10plus )
129  isEndcap = true;
130 
131  return isEndcap;
132 }
GeomDetEnumerators::isEndcap
bool isEndcap(GeomDetEnumerators::SubDetector m)
Definition: GeomDetEnumerators.cc:37
float ElectronMVAEstimatorRun2Phys14NonTrig::mvaValue ( const edm::Ptr< reco::Candidate > &  particle,
const edm::Event &  evt 
) const
virtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 42 of file ElectronMVAEstimatorRun2Phys14NonTrig.cc.

References _gbrForests, _MethodName, constexpr, gather_cfg::cout, debug, fillMVAVariables(), findCategory(), eostools::move(), and query::result.

42  {
43 
44  const int iCategory = findCategory( particle );
45  const std::vector<float> vars = std::move( fillMVAVariables( particle, evt ) );
46  const float result = _gbrForests.at(iCategory)->GetClassifier(vars.data());
47 
48  constexpr bool debug = false;
49  if(debug) {
50  std::cout << " *** Inside the class _MethodName " << _MethodName << std::endl;
51  std::cout << " bin " << iCategory
52  << " fbrem " << vars[11] //_allMVAVars.fbrem
53  << " kfchi2 " << vars[9] //_allMVAVars.kfchi2
54  << " mykfhits " << vars[0] //_allMVAVars.kfhits
55  << " gsfchi2 " << vars[10] //_allMVAVars.gsfchi2
56  << " deta " << vars[15] //_allMVAVars.deta
57  << " dphi " << vars[16] //_allMVAVars.dphi
58  << " detacalo " << vars[17] //_allMVAVars.detacalo
59  << " see " << vars[1] //_allMVAVars.see
60  << " spp " << vars[2] //_allMVAVars.spp
61  << " etawidth " << vars[5] //_allMVAVars.etawidth
62  << " phiwidth " << vars[6] // _allMVAVars.phiwidth
63  << " OneMinusE1x5E5x5 " << vars[3] //_allMVAVars.OneMinusE1x5E5x5
64  << " R9 " << vars[4] //_allMVAVars.R9
65  << " HoE " << vars[7] //_allMVAVars.HoE
66  << " EoP " << vars[12] //_allMVAVars.EoP
67  << " IoEmIoP " << vars[14] //_allMVAVars.IoEmIoP
68  << " eleEoPout " << vars[13] // _allMVAVars.eleEoPout
69  //<< " d0 " << _allMVAVars.d0
70  // << " ip3d " << _allMVAVars.ip3d
71  << " eta " << vars[21] //_allMVAVars.SCeta
72  << " isBarrel " << vars[19] //_allMVAVars.isBarrel
73  << " isEndcap " << vars[20] //_allMVAVars.isEndcap
74  << " pt " << vars[18] //_allMVAVars.pt
75  << std::endl;
76  std::cout << " ### MVA " << result << std::endl;
77  }
78 
79  return result;
80 }
ElectronMVAEstimatorRun2Phys14NonTrig::fillMVAVariables
std::vector< float > fillMVAVariables(const edm::Ptr< reco::Candidate > &particle, const edm::Event &) const
Definition: ElectronMVAEstimatorRun2Phys14NonTrig.cc:194
constexpr
#define constexpr
ElectronMVAEstimatorRun2Phys14NonTrig::findCategory
int findCategory(const edm::Ptr< reco::Candidate > &particle) const
Definition: ElectronMVAEstimatorRun2Phys14NonTrig.cc:82
ElectronMVAEstimatorRun2Phys14NonTrig::_gbrForests
std::vector< std::unique_ptr< const GBRForest > > _gbrForests
Definition: ElectronMVAEstimatorRun2Phys14NonTrig.h:103
query.result
tuple result
Definition: query.py:137
eostools.move
def move
Definition: eostools.py:510
debug
#define debug
Definition: HDRShower.cc:19
gather_cfg.cout
tuple cout
Definition: gather_cfg.py:145

Member Data Documentation

AllVariables ElectronMVAEstimatorRun2Phys14NonTrig::_allMVAVars
private

Definition at line 107 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

Referenced by createSingleReader().

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

Definition at line 103 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

Referenced by ElectronMVAEstimatorRun2Phys14NonTrig(), and mvaValue().

std::string ElectronMVAEstimatorRun2Phys14NonTrig::_MethodName
private

Definition at line 106 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

Referenced by createSingleReader(), ElectronMVAEstimatorRun2Phys14NonTrig(), and mvaValue().

const std::string ElectronMVAEstimatorRun2Phys14NonTrig::_name = "ElectronMVAEstimatorRun2Phys14NonTrig"
private

Definition at line 96 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

Referenced by plotting.Subtract::__str__(), plotting.FakeDuplicate::__str__(), plotting.AggregateBins::__str__(), plotting.AggregateHistos::__str__(), plotting.ROC::__str__(), plotting.PlotGroup::_drawSeparate(), plotting.PlotGroup::_save(), plotting.Subtract::create(), plotting.FakeDuplicate::create(), plotting.AggregateBins::create(), plotting.AggregateHistos::create(), plotting.Plot::create(), plotting.PlotGroup::draw(), getName(), plotting.Plot::getName(), plotting.PlotGroup::getName(), plotting.PlotterFolder::getName(), plotting.PlotterItem::getName(), plotting.PlotterFolder::getSelectionNameIterator(), validation.SimpleSample::name(), trackingPlots.Iteration::name(), plotting.PlotterItem::readDirs(), and plotting.PlotGroup::remove().

std::string ElectronMVAEstimatorRun2Phys14NonTrig::_tag
private

Definition at line 100 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

Referenced by ElectronMVAEstimatorRun2Phys14NonTrig(), and getTag().

int ElectronMVAEstimatorRun2Phys14NonTrig::nCategories = 6
static

Definition at line 25 of file ElectronMVAEstimatorRun2Phys14NonTrig.h.

Referenced by ElectronMVAEstimatorRun2Phys14NonTrig(), and getNCategories().

Generated for CMSSW Reference Manual by doxygen 1.8.5