CMS 3D CMS Logo
List of all members | Classes | Public Types | Public Member Functions | Public Attributes | Private Attributes
PhotonMVAEstimatorRun2Spring16NonTrig Class Reference

#include <PhotonMVAEstimatorRun2Spring16NonTrig.h>

Inheritance diagram for PhotonMVAEstimatorRun2Spring16NonTrig:
AnyMVAEstimatorRun2Base

Classes

struct  AllVariables
 

Public Types

enum  mvaCategories { UNDEFINED = -1, CAT_EB = 0, CAT_EE = 1 }
 

Public Member Functions

void constrainMVAVariables (AllVariables &) const
 
std::unique_ptr< const GBRForest > createSingleReader (const int iCategory, const edm::FileInPath &weightFile)
 
std::vector< float > fillMVAVariables (const edm::Ptr< reco::Candidate > &particle, const edm::Event &iEvent) const override
 
int findCategory (const edm::Ptr< reco::Candidate > &particle) const override
 
virtual const std::string & getName () const override final
 
virtual int getNCategories () const
 
virtual const std::string & getTag () const override final
 
bool isEndcapCategory (int category) const
 
float mvaValue (const edm::Ptr< reco::Candidate > &particle, const edm::Event &) const
 
 PhotonMVAEstimatorRun2Spring16NonTrig (const edm::ParameterSet &conf)
 
void setConsumes (edm::ConsumesCollector &&) const override
 
 ~PhotonMVAEstimatorRun2Spring16NonTrig ()
 
- 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 uint nCategories = 2
 
- Public Attributes inherited from AnyMVAEstimatorRun2Base
const edm::ParameterSet _conf
 

Private Attributes

AllVariables allMVAVars_
 
EffectiveAreas effectiveAreas_
 
std::vector< std::unique_ptr< const GBRForest > > gbrForests_
 
const std::string MethodName_
 
const std::string name_ = "PhotonMVAEstimatorRun2Spring16NonTrig"
 
const edm::InputTag phoChargedIsolationLabel_
 
double phoIsoCutoff_
 
std::vector< double > phoIsoPtScalingCoeff_
 
const edm::InputTag phoPhotonIsolationLabel_
 
const edm::InputTag phoWorstChargedIsolationLabel_
 
const edm::InputTag rhoLabel_
 
std::string tag_
 

Detailed Description

Definition at line 23 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Member Enumeration Documentation

enum PhotonMVAEstimatorRun2Spring16NonTrig::mvaCategories
Enumerator
UNDEFINED 
CAT_EB 
CAT_EE 

Definition at line 30 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

30  {
31  UNDEFINED = -1,
32  CAT_EB = 0,
33  CAT_EE = 1
34  };

Constructor & Destructor Documentation

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

Definition at line 6 of file PhotonMVAEstimatorRun2Spring16NonTrig.cc.

References createSingleReader(), gbrForests_, edm::ParameterSet::getParameter(), mps_fire::i, nCategories, AlCaHLTBitMon_QueryRunRegistry::string, tag_, candidateChargeBTagComputer_cfi::weightFile, mvaElectronID_PHYS14_PU20bx25_nonTrig_V1_cff::weightFileNames, and ~PhotonMVAEstimatorRun2Spring16NonTrig().

6  :
7  AnyMVAEstimatorRun2Base(conf),
8  MethodName_("BDTG method"),
9  phoChargedIsolationLabel_(conf.getParameter<edm::InputTag>("phoChargedIsolation")),
10  phoPhotonIsolationLabel_(conf.getParameter<edm::InputTag>("phoPhotonIsolation")),
11  phoWorstChargedIsolationLabel_(conf.getParameter<edm::InputTag>("phoWorstChargedIsolation")),
12  rhoLabel_(conf.getParameter<edm::InputTag>("rho")),
13  effectiveAreas_( (conf.getParameter<edm::FileInPath>("effAreasConfigFile")).fullPath()),
14  phoIsoPtScalingCoeff_(conf.getParameter<std::vector<double >>("phoIsoPtScalingCoeff")),
15  phoIsoCutoff_(conf.getParameter<double>("phoIsoCutoff"))
16 {
17 
18  //
19  // Construct the MVA estimators
20  //
21  tag_ = conf.getParameter<std::string>("mvaTag");
22 
23  const std::vector <std::string> weightFileNames
24  = conf.getParameter<std::vector<std::string> >("weightFileNames");
25 
26  if( weightFileNames.size() != nCategories )
27  throw cms::Exception("MVA config failure: ")
28  << "wrong number of weightfiles" << std::endl;
29 
30  gbrForests_.clear();
31  // The method name is just a key to retrieve this method later, it is not
32  // a control parameter for a reader (the full definition of the MVA type and
33  // everything else comes from the xml weight files).
34 
35  // Create a TMVA reader object for each category
36  for(uint i=0; i<nCategories; i++){
37 
38  // Use unique_ptr so that all readers are properly cleaned up
39  // when the vector clear() is called in the destructor
40 
41  edm::FileInPath weightFile( weightFileNames[i] );
42  gbrForests_.push_back( createSingleReader(i, weightFile ) );
43 
44  }
45 
46 }
edm::ParameterSet::getParameter
T getParameter(std::string const &) const
AnyMVAEstimatorRun2Base::AnyMVAEstimatorRun2Base
AnyMVAEstimatorRun2Base(const edm::ParameterSet &conf)
Definition: AnyMVAEstimatorRun2Base.h:15
mps_fire.i
i
Definition: mps_fire.py:156
AlCaHLTBitMon_QueryRunRegistry.string
string
Definition: AlCaHLTBitMon_QueryRunRegistry.py:255
PhotonMVAEstimatorRun2Spring16NonTrig::createSingleReader
std::unique_ptr< const GBRForest > createSingleReader(const int iCategory, const edm::FileInPath &weightFile)
Definition: PhotonMVAEstimatorRun2Spring16NonTrig.cc:132
mvaElectronID_PHYS14_PU20bx25_nonTrig_V1_cff.weightFileNames
weightFileNames
Definition: mvaElectronID_PHYS14_PU20bx25_nonTrig_V1_cff.py:90
PhotonMVAEstimatorRun2Spring16NonTrig::gbrForests_
std::vector< std::unique_ptr< const GBRForest > > gbrForests_
Definition: PhotonMVAEstimatorRun2Spring16NonTrig.h:104
candidateChargeBTagComputer_cfi.weightFile
weightFile
Definition: candidateChargeBTagComputer_cfi.py:6
PhotonMVAEstimatorRun2Spring16NonTrig::~PhotonMVAEstimatorRun2Spring16NonTrig ( )

Definition at line 49 of file PhotonMVAEstimatorRun2Spring16NonTrig.cc.

References mvaValue().

Referenced by PhotonMVAEstimatorRun2Spring16NonTrig().

49  {
50 }

Member Function Documentation

void PhotonMVAEstimatorRun2Spring16NonTrig::constrainMVAVariables ( AllVariables &  ) const

Definition at line 321 of file PhotonMVAEstimatorRun2Spring16NonTrig.cc.

Referenced by fillMVAVariables(), and getTag().

321  {
322 
323  // Check that variables do not have crazy values
324 
325  // This function is currently empty as this specific MVA was not
326  // developed with restricting variables to specific physical ranges.
327  return;
328 
329 }
std::unique_ptr< const GBRForest > PhotonMVAEstimatorRun2Spring16NonTrig::createSingleReader ( const int  iCategory,
const edm::FileInPath &  weightFile 
)

Definition at line 132 of file PhotonMVAEstimatorRun2Spring16NonTrig.cc.

References allMVAVars_, edm::FileInPath::fullPath(), isEndcapCategory(), MethodName_, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::scPhi, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varChIsoRaw, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varE2x2overE5x5, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varESEffSigmaRR, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varESEnOverRawE, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varPhoIsoCorr, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varPhoIsoRaw, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varR9, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varRawE, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varRho, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSCEta, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSCEtaWidth, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSCPhiWidth, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSieie, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSieip, and PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varWorstChRaw.

Referenced by isEndcapCategory(), and PhotonMVAEstimatorRun2Spring16NonTrig().

132  {
133 
134  //
135  // Create the reader
136  //
137  TMVA::Reader tmpTMVAReader( "!Color:Silent:Error" );
138 
139  //
140  // Configure all variables and spectators. Note: the order and names
141  // must match what is found in the xml weights file!
142  //
143 
144  tmpTMVAReader.AddVariable("scPhi", &allMVAVars_.scPhi);
145  tmpTMVAReader.AddVariable("r9" , &allMVAVars_.varR9);
146  tmpTMVAReader.AddVariable("sigmaIetaIeta", &allMVAVars_.varSieie);
147  tmpTMVAReader.AddVariable("covIEtaIPhi", &allMVAVars_.varSieip);
148  tmpTMVAReader.AddVariable("s4", &allMVAVars_.varE2x2overE5x5);
149  tmpTMVAReader.AddVariable("scEta" , &allMVAVars_.varSCEta);
150  tmpTMVAReader.AddVariable("SCRawE" , &allMVAVars_.varRawE);
151  tmpTMVAReader.AddVariable("etaWidth", &allMVAVars_.varSCEtaWidth);
152  tmpTMVAReader.AddVariable("phiWidth", &allMVAVars_.varSCPhiWidth);
153  //Pileup
154  tmpTMVAReader.AddVariable("rho" , &allMVAVars_.varRho);
155  //EB only, because of loss of transparency in EE
156  if(!isEndcapCategory(iCategory)){
157  tmpTMVAReader.AddVariable("CITK_isoPhotons" , &allMVAVars_.varPhoIsoRaw);
158  }
159  else{
160  tmpTMVAReader.AddVariable("CITK_isoPhoCorrMax2p5", &allMVAVars_.varPhoIsoCorr);
161  }
162  //isolations in both EB and EE
163  tmpTMVAReader.AddVariable("CITK_isoChargedHad" , &allMVAVars_.varChIsoRaw);
164  tmpTMVAReader.AddVariable("chgIsoWrtWorstVtx", &allMVAVars_.varWorstChRaw);
165 
166  // Endcap only variables
167  if( isEndcapCategory(iCategory) ){
168  tmpTMVAReader.AddVariable("esEffSigmaRR" , &allMVAVars_.varESEffSigmaRR);
169  tmpTMVAReader.AddVariable("esEnergy/SCRawE" , &allMVAVars_.varESEnOverRawE);
170  }
171 
172  //
173  // Book the method and set up the weights file
174  //
175  tmpTMVAReader.BookMVA(MethodName_ , weightFile.fullPath() );
176 
177  return std::unique_ptr<const GBRForest>( new GBRForest( dynamic_cast<TMVA::MethodBDT*>( tmpTMVAReader.FindMVA(MethodName_) ) ) );
178 
179 }
edm::FileInPath::fullPath
std::string fullPath() const
Definition: FileInPath.cc:184
std::vector< float > PhotonMVAEstimatorRun2Spring16NonTrig::fillMVAVariables ( const edm::Ptr< reco::Candidate > &  particle,
const edm::Event &  iEvent 
) const
overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 182 of file PhotonMVAEstimatorRun2Spring16NonTrig.cc.

References funct::abs(), constrainMVAVariables(), reco::Photon::ShowerShape::e2x2, heepElectronID_HEEPV50_CSA14_25ns_cff::e5x5, effectiveAreas_, mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff::esEffSigmaRRMap, Exception, findCategory(), reco::Photon::full5x5_showerShapeVariables(), mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff::full5x5E2x2Map, mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff::full5x5E5x5Map, cutBasedElectronID_CSA14_50ns_V1_cff::full5x5SigmaIEtaIEtaMap, mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff::full5x5SigmaIEtaIPhiMap, edm::Event::getByLabel(), EffectiveAreas::getEffectiveArea(), isEndcapCategory(), edm::Ptr< T >::isNull(), edm::HandleBase::isValid(), hpstanc_transforms::max, Max(), eostools::move(), AnyMVAEstimatorRun2Base::packMVAVariables(), phoChargedIsolationLabel_, phoIsoCutoff_, phoIsoPtScalingCoeff_, phoPhotonIsolationLabel_, phoWorstChargedIsolationLabel_, reco::LeafCandidate::pt(), reco::Photon::r9(), rho, rhoLabel_, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::scPhi, reco::Photon::superCluster(), PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varChIsoRaw, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varE2x2overE5x5, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varESEffSigmaRR, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varESEnOverRawE, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varPhoIsoCorr, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varPhoIsoRaw, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varR9, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varRawE, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varRho, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSCEta, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSCEtaWidth, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSCPhiWidth, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSieie, PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varSieip, and PhotonMVAEstimatorRun2Spring16NonTrig::AllVariables::varWorstChRaw.

Referenced by getTag(), and mvaValue().

182  {
183 
184  //
185  // Declare all value maps corresponding to the above tokens
186  //
187  edm::Handle<edm::ValueMap<float> > full5x5SigmaIEtaIEtaMap;
188  edm::Handle<edm::ValueMap<float> > full5x5SigmaIEtaIPhiMap;
189  edm::Handle<edm::ValueMap<float> > full5x5E2x2Map;
190  edm::Handle<edm::ValueMap<float> > full5x5E5x5Map;
191  edm::Handle<edm::ValueMap<float> > esEffSigmaRRMap;
192  //
193  edm::Handle<edm::ValueMap<float> > phoChargedIsolationMap;
194  edm::Handle<edm::ValueMap<float> > phoPhotonIsolationMap;
195  edm::Handle<edm::ValueMap<float> > phoWorstChargedIsolationMap;
196 
197  // Rho will be pulled from the event content
198  edm::Handle<double> rho;
199 
200  // Get the isolation maps
201  iEvent.getByLabel(phoChargedIsolationLabel_, phoChargedIsolationMap);
202  iEvent.getByLabel(phoPhotonIsolationLabel_, phoPhotonIsolationMap);
203  iEvent.getByLabel(phoWorstChargedIsolationLabel_, phoWorstChargedIsolationMap);
204 
205  // Get rho
206  iEvent.getByLabel(rhoLabel_,rho);
207 
208  // Make sure everything is retrieved successfully
209  if(! ( phoChargedIsolationMap.isValid()
210  && phoPhotonIsolationMap.isValid()
211  && phoWorstChargedIsolationMap.isValid()
212  && rho.isValid() ) )
213  throw cms::Exception("MVA failure: ")
214  << "Failed to retrieve event content needed for this MVA"
215  << std::endl
216  << "Check python MVA configuration file and make sure all needed"
217  << std::endl
218  << "producers are running upstream" << std::endl;
219 
220  // Try to cast the particle into a reco particle.
221  // This should work for both reco and pat.
222  const edm::Ptr<reco::Photon> phoRecoPtr(particle);
223  if( phoRecoPtr.isNull() )
224  throw cms::Exception("MVA failure: ")
225  << " given particle is expected to be reco::Photon or pat::Photon," << std::endl
226  << " but appears to be neither" << std::endl;
227 
228  // Both pat and reco particles have exactly the same accessors.
229  auto superCluster = phoRecoPtr->superCluster();
230  // Full 5x5 cluster shapes. We could take some of this directly from
231  // the photon object, but some of these are not available.
232  float e2x2 = std::numeric_limits<float>::max();
233  float e5x5 = std::numeric_limits<float>::max();
234  float full5x5_sigmaIetaIeta = std::numeric_limits<float>::max();
235  float full5x5_sigmaIetaIphi = std::numeric_limits<float>::max();
236  float effSigmaRR = std::numeric_limits<float>::max();
237 
238  AllVariables allMVAVars;
239 
240  const auto& full5x5_pss = phoRecoPtr->full5x5_showerShapeVariables();
241  e2x2 = full5x5_pss.e2x2;
242  e5x5 = full5x5_pss.e5x5;
243  full5x5_sigmaIetaIeta = full5x5_pss.sigmaIetaIeta;
244  full5x5_sigmaIetaIphi = full5x5_pss.sigmaIetaIphi;
245  effSigmaRR = full5x5_pss.effSigmaRR;
246 
247  allMVAVars.scPhi = superCluster->phi();
248  allMVAVars.varR9 = phoRecoPtr->r9() ;
249  allMVAVars.varSieie = full5x5_sigmaIetaIeta;
250  allMVAVars.varSieip = full5x5_sigmaIetaIphi;
251  allMVAVars.varE2x2overE5x5 = e2x2/e5x5;
252  allMVAVars.varSCEta = superCluster->eta();
253  allMVAVars.varRawE = superCluster->rawEnergy();
254  allMVAVars.varSCEtaWidth = superCluster->etaWidth();
255  allMVAVars.varSCPhiWidth = superCluster->phiWidth();
256  allMVAVars.varESEnOverRawE = superCluster->preshowerEnergy() / superCluster->rawEnergy();
257  allMVAVars.varESEffSigmaRR = effSigmaRR;
258  allMVAVars.varRho = *rho;
259  allMVAVars.varPhoIsoRaw = (*phoPhotonIsolationMap)[phoRecoPtr];
260  allMVAVars.varChIsoRaw = (*phoChargedIsolationMap)[phoRecoPtr];
261  allMVAVars.varWorstChRaw = (*phoWorstChargedIsolationMap)[phoRecoPtr];
262 
263  //photon iso corrected:
264 
265  double eA = effectiveAreas_.getEffectiveArea( std::abs(superCluster->eta()) );
266  double phoIsoPtScalingCoeffVal = 0;
267  if( !isEndcapCategory( findCategory ( particle ) ) )
268  phoIsoPtScalingCoeffVal = phoIsoPtScalingCoeff_.at(0); // barrel case
269  else
270  phoIsoPtScalingCoeffVal = phoIsoPtScalingCoeff_.at(1); //endcap case
271  double phoIsoCorr = (*phoPhotonIsolationMap)[phoRecoPtr] - eA*(*rho) - phoIsoPtScalingCoeffVal*phoRecoPtr->pt();
272 
273  allMVAVars.varPhoIsoCorr = TMath::Max(phoIsoCorr, phoIsoCutoff_);
274 
275  constrainMVAVariables(allMVAVars);
276  //
277  // Important: the order of variables in the "vars" vector has to be EXACTLY
278  // the same as in the .xml file defining the MVA.
279  //
280  std::vector<float> vars;
281  if( isEndcapCategory( findCategory( particle ) ) ) {
282  vars = std::move( packMVAVariables(
283  allMVAVars.scPhi,
284  allMVAVars.varR9,
285  allMVAVars.varSieie,
286  allMVAVars.varSieip,
287  allMVAVars.varE2x2overE5x5,
288  allMVAVars.varSCEta,
289  allMVAVars.varRawE,
290  allMVAVars.varSCEtaWidth,
291  allMVAVars.varSCPhiWidth,
292  allMVAVars.varRho,
293  allMVAVars.varPhoIsoCorr,
294  allMVAVars.varChIsoRaw,
295  allMVAVars.varWorstChRaw,
296  allMVAVars.varESEffSigmaRR,
297  allMVAVars.varESEnOverRawE
298  )
299  );
300  } else {
301  vars = std::move( packMVAVariables(
302  allMVAVars.scPhi,
303  allMVAVars.varR9,
304  allMVAVars.varSieie,
305  allMVAVars.varSieip,
306  allMVAVars.varE2x2overE5x5,
307  allMVAVars.varSCEta,
308  allMVAVars.varRawE,
309  allMVAVars.varSCEtaWidth,
310  allMVAVars.varSCPhiWidth,
311  allMVAVars.varRho,
312  allMVAVars.varPhoIsoRaw,
313  allMVAVars.varChIsoRaw,
314  allMVAVars.varWorstChRaw
315  )
316  );
317  }
318  return vars;
319 }
hpstanc_transforms.max
max
Definition: hpstanc_transforms.py:5
EffectiveAreas::getEffectiveArea
const float getEffectiveArea(float eta) const
Definition: EffectiveAreas.cc:50
PhotonMVAEstimatorRun2Spring16NonTrig::findCategory
int findCategory(const edm::Ptr< reco::Candidate > &particle) const override
Definition: PhotonMVAEstimatorRun2Spring16NonTrig.cc:92
mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff.full5x5E2x2Map
full5x5E2x2Map
Definition: mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff.py:75
AnyMVAEstimatorRun2Base::packMVAVariables
std::vector< float > packMVAVariables(const Args...args) const
Definition: AnyMVAEstimatorRun2Base.h:43
mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff.full5x5SigmaIEtaIPhiMap
full5x5SigmaIEtaIPhiMap
Definition: mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff.py:73
mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff.full5x5E5x5Map
full5x5E5x5Map
Definition: mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff.py:77
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
edm::HandleBase::isValid
bool isValid() const
Definition: HandleBase.h:74
edm::Event::getByLabel
bool getByLabel(InputTag const &tag, Handle< PROD > &result) const
Definition: Event.h:416
Max
T Max(T a, T b)
Definition: MathUtil.h:44
cutBasedElectronID_CSA14_50ns_V1_cff.full5x5SigmaIEtaIEtaMap
full5x5SigmaIEtaIEtaMap
Definition: cutBasedElectronID_CSA14_50ns_V1_cff.py:48
heepElectronID_HEEPV50_CSA14_25ns_cff.e5x5
e5x5
Definition: heepElectronID_HEEPV50_CSA14_25ns_cff.py:55
mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff.esEffSigmaRRMap
esEffSigmaRRMap
Definition: mvaPhotonID_PHYS14_PU20bx25_nonTrig_V1_cff.py:78
eostools.move
def move(src, dest)
Definition: eostools.py:510
int PhotonMVAEstimatorRun2Spring16NonTrig::findCategory ( const edm::Ptr< reco::Candidate > &  particle) const
overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 92 of file PhotonMVAEstimatorRun2Spring16NonTrig.cc.

References funct::abs(), CAT_EB, CAT_EE, PVValHelper::eta, isEndcapCategory(), edm::Ptr< T >::isNull(), reco::Photon::superCluster(), and UNDEFINED.

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

92  {
93 
94  // Try to cast the particle into a reco particle.
95  // This should work for both reco and pat.
96  const edm::Ptr<reco::Photon> phoRecoPtr = ( edm::Ptr<reco::Photon> )particle;
97  if( phoRecoPtr.isNull() )
98  throw cms::Exception("MVA failure: ")
99  << " given particle is expected to be reco::Photon or pat::Photon," << std::endl
100  << " but appears to be neither" << std::endl;
101 
102  float eta = phoRecoPtr->superCluster()->eta();
103 
104  //
105  // Determine the category
106  //
107  int iCategory = UNDEFINED;
108  const float ebeeSplit = 1.479; // division between barrel and endcap
109 
110  if ( std::abs(eta) < ebeeSplit)
111  iCategory = CAT_EB;
112  else
113  iCategory = CAT_EE;
114 
115  return iCategory;
116 }
reco::Photon::superCluster
reco::SuperClusterRef superCluster() const
Ref to SuperCluster.
edm::Ptr::isNull
bool isNull() const
Checks for null.
Definition: Ptr.h:164
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
virtual const std::string& PhotonMVAEstimatorRun2Spring16NonTrig::getName ( void  ) const
inlinefinaloverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 72 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

References name_.

Referenced by plotting.Plot::draw().

72 { return name_; }
virtual int PhotonMVAEstimatorRun2Spring16NonTrig::getNCategories ( ) const
inlinevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 70 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

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

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

Implements AnyMVAEstimatorRun2Base.

Definition at line 73 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

References constrainMVAVariables(), fillMVAVariables(), findCategory(), iEvent, setConsumes(), and tag_.

73 { return tag_; }
bool PhotonMVAEstimatorRun2Spring16NonTrig::isEndcapCategory ( int  category) const

Definition at line 119 of file PhotonMVAEstimatorRun2Spring16NonTrig.cc.

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

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

119  {
120 
121  // For this specific MVA the function is trivial, but kept for possible
122  // future evolution to an MVA with more categories in eta
123  bool isEndcap = false;
124  if( category == CAT_EE )
125  isEndcap = true;
126 
127  return isEndcap;
128 }
GeomDetEnumerators::isEndcap
bool isEndcap(GeomDetEnumerators::SubDetector m)
Definition: GeomDetEnumerators.cc:40
float PhotonMVAEstimatorRun2Spring16NonTrig::mvaValue ( const edm::Ptr< reco::Candidate > &  particle,
const edm::Event &  iEvent 
) const
virtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 53 of file PhotonMVAEstimatorRun2Spring16NonTrig.cc.

References debug, fillMVAVariables(), findCategory(), gbrForests_, isEndcapCategory(), eostools::move(), and mps_fire::result.

Referenced by ~PhotonMVAEstimatorRun2Spring16NonTrig().

53  {
54 
55  const int iCategory = findCategory( particle );
56  const std::vector<float> vars = std::move( fillMVAVariables( particle, iEvent ) );
57 
58  const float result = gbrForests_.at(iCategory)->GetClassifier(vars.data());
59 
60  // DEBUG
61  const bool debug = false;
62  // The list of variables here must match EXACTLY the list and order in the
63  // packMVAVariables() call for barrel and endcap in this file.
64  if( debug ){
65  printf("Printout of the photon variable inputs for MVA:\n");
66  printf(" varSCPhi_ %f\n", vars[0] );
67  printf(" varR9_ %f\n", vars[1] );
68  printf(" varSieie_ %f\n", vars[2] );
69  printf(" varSieip_ %f\n", vars[3] );
70  printf(" varE2x2overE5x5_ %f\n", vars[4] );
71  printf(" varSCEta_ %f\n", vars[5] );
72  printf(" varRawE_ %f\n", vars[6] );
73  printf(" varSCEtaWidth_ %f\n", vars[7] );
74  printf(" varSCPhiWidth_ %f\n", vars[8] );
75  printf(" varRho_ %f\n", vars[9] );
76  if( !isEndcapCategory( iCategory ) ) {
77  printf(" varPhoIsoRaw_ %f\n", vars[10] );
78  } else {
79  printf(" varPhoIsoRawCorr_ %f\n", vars[10] ); // for endcap MVA only in 2016
80  }
81  printf(" varChIsoRaw_ %f\n", vars[11] );
82  printf(" varWorstChRaw_ %f\n", vars[12] );
83  if( isEndcapCategory( iCategory ) ) {
84  printf(" varESEnOverRawE_ %f\n", vars[13] ); // for endcap MVA only
85  printf(" varESEffSigmaRR_ %f\n", vars[14] ); // for endcap MVA only
86  }
87  }
88 
89  return result;
90 }
mps_fire.result
result
Definition: mps_fire.py:113
PhotonMVAEstimatorRun2Spring16NonTrig::findCategory
int findCategory(const edm::Ptr< reco::Candidate > &particle) const override
Definition: PhotonMVAEstimatorRun2Spring16NonTrig.cc:92
PhotonMVAEstimatorRun2Spring16NonTrig::fillMVAVariables
std::vector< float > fillMVAVariables(const edm::Ptr< reco::Candidate > &particle, const edm::Event &iEvent) const override
Definition: PhotonMVAEstimatorRun2Spring16NonTrig.cc:182
PhotonMVAEstimatorRun2Spring16NonTrig::gbrForests_
std::vector< std::unique_ptr< const GBRForest > > gbrForests_
Definition: PhotonMVAEstimatorRun2Spring16NonTrig.h:104
debug
#define debug
Definition: HDRShower.cc:19
eostools.move
def move(src, dest)
Definition: eostools.py:510
void PhotonMVAEstimatorRun2Spring16NonTrig::setConsumes ( edm::ConsumesCollector &&  cc) const
overridevirtual

Reimplemented from AnyMVAEstimatorRun2Base.

Definition at line 331 of file PhotonMVAEstimatorRun2Spring16NonTrig.cc.

References DEFINE_EDM_PLUGIN, phoChargedIsolationLabel_, phoPhotonIsolationLabel_, phoWorstChargedIsolationLabel_, and rhoLabel_.

Referenced by getTag().

331  {
332  cc.consumes<edm::ValueMap<float> >(phoChargedIsolationLabel_);
333  cc.consumes<edm::ValueMap<float> >(phoPhotonIsolationLabel_);
334  cc.consumes<edm::ValueMap<float> >( phoWorstChargedIsolationLabel_);
335  cc.consumes<double>(rhoLabel_);
336 }
edm::ConsumesCollector::consumes
EDGetTokenT< ProductType > consumes(edm::InputTag const &tag)
Definition: ConsumesCollector.h:52

Member Data Documentation

AllVariables PhotonMVAEstimatorRun2Spring16NonTrig::allMVAVars_
private

Definition at line 108 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by createSingleReader().

EffectiveAreas PhotonMVAEstimatorRun2Spring16NonTrig::effectiveAreas_
private

Definition at line 123 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by fillMVAVariables().

std::vector< std::unique_ptr<const GBRForest> > PhotonMVAEstimatorRun2Spring16NonTrig::gbrForests_
private

Definition at line 104 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by mvaValue(), and PhotonMVAEstimatorRun2Spring16NonTrig().

const std::string PhotonMVAEstimatorRun2Spring16NonTrig::MethodName_
private

Definition at line 107 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by createSingleReader().

const std::string PhotonMVAEstimatorRun2Spring16NonTrig::name_ = "PhotonMVAEstimatorRun2Spring16NonTrig"
private

Definition at line 96 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by getName().

const uint PhotonMVAEstimatorRun2Spring16NonTrig::nCategories = 2

Definition at line 29 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by getNCategories(), and PhotonMVAEstimatorRun2Spring16NonTrig().

const edm::InputTag PhotonMVAEstimatorRun2Spring16NonTrig::phoChargedIsolationLabel_
private

Definition at line 117 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

double PhotonMVAEstimatorRun2Spring16NonTrig::phoIsoCutoff_
private

Definition at line 125 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by fillMVAVariables().

std::vector<double> PhotonMVAEstimatorRun2Spring16NonTrig::phoIsoPtScalingCoeff_
private

Definition at line 124 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by fillMVAVariables().

const edm::InputTag PhotonMVAEstimatorRun2Spring16NonTrig::phoPhotonIsolationLabel_
private

Definition at line 118 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring16NonTrig::phoWorstChargedIsolationLabel_
private

Definition at line 119 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring16NonTrig::rhoLabel_
private

Definition at line 120 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

std::string PhotonMVAEstimatorRun2Spring16NonTrig::tag_
private

Definition at line 101 of file PhotonMVAEstimatorRun2Spring16NonTrig.h.

Referenced by getTag(), and PhotonMVAEstimatorRun2Spring16NonTrig().

Generated for CMSSW Reference Manual by doxygen 1.8.11