PhotonMVAEstimatorRun2Spring15NonTrig Class Reference

#include <PhotonMVAEstimatorRun2Spring15NonTrig.h>

Inheritance diagram for PhotonMVAEstimatorRun2Spring15NonTrig:

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 overridefinal
virtual int	getNCategories () const
virtual const std::string &	getTag () const overridefinal
bool	isEndcapCategory (int category) const
float	mvaValue (const edm::Ptr< reco::Candidate > &particle, const edm::Event &) const
	PhotonMVAEstimatorRun2Spring15NonTrig (const edm::ParameterSet &conf)
void	setConsumes (edm::ConsumesCollector &&) const override
	~PhotonMVAEstimatorRun2Spring15NonTrig ()
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 = 2
Public Attributes inherited from AnyMVAEstimatorRun2Base
const edm::ParameterSet	_conf

Private Attributes
AllVariables	_allMVAVars
const edm::InputTag	_esEffSigmaRRMapLabel
const edm::InputTag	_full5x5E1x3MapLabel
const edm::InputTag	_full5x5E2x2MapLabel
const edm::InputTag	_full5x5E2x5MaxMapLabel
const edm::InputTag	_full5x5E5x5MapLabel
const edm::InputTag	_full5x5SigmaIEtaIEtaMapLabel
const edm::InputTag	_full5x5SigmaIEtaIPhiMapLabel
std::vector< std::unique_ptr < const GBRForest > >	_gbrForests
const std::string	_MethodName
const std::string	_name = "PhotonMVAEstimatorRun2Spring15NonTrig"
const edm::InputTag	_phoChargedIsolationLabel
const edm::InputTag	_phoPhotonIsolationLabel
const edm::InputTag	_phoWorstChargedIsolationLabel
const edm::InputTag	_rhoLabel
std::string	_tag

Detailed Description

Definition at line 21 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Member Enumeration Documentation

enum PhotonMVAEstimatorRun2Spring15NonTrig::mvaCategories

Enumerator
UNDEFINED
CAT_EB
CAT_EE

Definition at line 28 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

                     {
    UNDEFINED = -1,
    CAT_EB  = 0,
    CAT_EE  = 1
  };

Constructor & Destructor Documentation

PhotonMVAEstimatorRun2Spring15NonTrig::PhotonMVAEstimatorRun2Spring15NonTrig

(

const edm::ParameterSet &

conf

)

Definition at line 7 of file PhotonMVAEstimatorRun2Spring15NonTrig.cc.

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

                                                                                                       :
  AnyMVAEstimatorRun2Base(conf),
  _MethodName("BDTG method"),
  _full5x5SigmaIEtaIEtaMapLabel(conf.getParameter<edm::InputTag>("full5x5SigmaIEtaIEtaMap")), 
  _full5x5SigmaIEtaIPhiMapLabel(conf.getParameter<edm::InputTag>("full5x5SigmaIEtaIPhiMap")), 
  _full5x5E1x3MapLabel(conf.getParameter<edm::InputTag>("full5x5E1x3Map")), 
  _full5x5E2x2MapLabel(conf.getParameter<edm::InputTag>("full5x5E2x2Map")), 
  _full5x5E2x5MaxMapLabel(conf.getParameter<edm::InputTag>("full5x5E2x5MaxMap")), 
  _full5x5E5x5MapLabel(conf.getParameter<edm::InputTag>("full5x5E5x5Map")), 
  _esEffSigmaRRMapLabel(conf.getParameter<edm::InputTag>("esEffSigmaRRMap")), 
  _phoChargedIsolationLabel(conf.getParameter<edm::InputTag>("phoChargedIsolation")), 
  _phoPhotonIsolationLabel(conf.getParameter<edm::InputTag>("phoPhotonIsolation")), 
  _phoWorstChargedIsolationLabel(conf.getParameter<edm::InputTag>("phoWorstChargedIsolation")), 
  _rhoLabel(conf.getParameter<edm::InputTag>("rho"))
{

  //
  // Construct the MVA estimators
  //
  _tag = conf.getParameter<std::string>("mvaTag");

  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();
  // The method name is just a key to retrieve this method later, it is not
  // a control parameter for a reader (the full definition of the MVA type and
  // everything else comes from the xml weight files).
   
  // 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( createSingleReader(i, weightFile ) );

  }

}

PhotonMVAEstimatorRun2Spring15NonTrig::~PhotonMVAEstimatorRun2Spring15NonTrig

(

)

Definition at line 54 of file PhotonMVAEstimatorRun2Spring15NonTrig.cc.

                                        {
}

Member Function Documentation

void PhotonMVAEstimatorRun2Spring15NonTrig::constrainMVAVariables

(

AllVariables &

)

const

Definition at line 338 of file PhotonMVAEstimatorRun2Spring15NonTrig.cc.

Referenced by fillMVAVariables().

                                                                                    {

  // Check that variables do not have crazy values
  
  // This function is currently empty as this specific MVA was not
  // developed with restricting variables to specific physical ranges.
  return;

}

std::unique_ptr< const GBRForest > PhotonMVAEstimatorRun2Spring15NonTrig::createSingleReader	(	const int	iCategory,
		const edm::FileInPath &	weightFile
	)

Definition at line 141 of file PhotonMVAEstimatorRun2Spring15NonTrig.cc.

References _allMVAVars, _MethodName, edm::FileInPath::fullPath(), isEndcapCategory(), groupFilesInBlocks::temp, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varChIsoRaw, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varE1x3overE5x5, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varE2x2overE5x5, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varE2x5overE5x5, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varESEffSigmaRR, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varESEnOverRawE, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varEta, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varPhi, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varPhoIsoRaw, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varPt, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varR9, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varRawE, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varRho, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSCEta, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSCEtaWidth, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSCPhiWidth, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSieie, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSieip, and PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varWorstChRaw.

Referenced by PhotonMVAEstimatorRun2Spring15NonTrig().

                                                                        {

  //
  // Create the reader  
  //
  TMVA::Reader tmpTMVAReader( "!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("recoPhi"   , &_allMVAVars.varPhi);
  tmpTMVAReader.AddVariable("r9"        , &_allMVAVars.varR9);
  tmpTMVAReader.AddVariable("sieieFull5x5", &_allMVAVars.varSieie);
  tmpTMVAReader.AddVariable("sieipFull5x5", &_allMVAVars.varSieip);
  tmpTMVAReader.AddVariable("e1x3Full5x5/e5x5Full5x5", &_allMVAVars.varE1x3overE5x5);
  tmpTMVAReader.AddVariable("e2x2Full5x5/e5x5Full5x5", &_allMVAVars.varE2x2overE5x5);
  tmpTMVAReader.AddVariable("e2x5Full5x5/e5x5Full5x5", &_allMVAVars.varE2x5overE5x5);
  tmpTMVAReader.AddVariable("recoSCEta" , &_allMVAVars.varSCEta);
  tmpTMVAReader.AddVariable("rawE"      , &_allMVAVars.varRawE);
  tmpTMVAReader.AddVariable("scEtaWidth", &_allMVAVars.varSCEtaWidth);
  tmpTMVAReader.AddVariable("scPhiWidth", &_allMVAVars.varSCPhiWidth);

  // Endcap only variables
  if( isEndcapCategory(iCategory) ){
    tmpTMVAReader.AddVariable("esEn/rawE" , &_allMVAVars.varESEnOverRawE);
    tmpTMVAReader.AddVariable("esRR"      , &_allMVAVars.varESEffSigmaRR);
  }

  // Pileup
  tmpTMVAReader.AddVariable("rho"       , &_allMVAVars.varRho);

  // Isolations
  tmpTMVAReader.AddVariable("phoIsoRaw" , &_allMVAVars.varPhoIsoRaw);
  tmpTMVAReader.AddVariable("chIsoRaw"  , &_allMVAVars.varChIsoRaw);
  tmpTMVAReader.AddVariable("chWorstRaw", &_allMVAVars.varWorstChRaw);

  // Spectators
  tmpTMVAReader.AddSpectator("recoPt" , &_allMVAVars.varPt);
  tmpTMVAReader.AddSpectator("recoEta", &_allMVAVars.varEta);

  //
  // Book the method and set up the weights file
  //
  std::unique_ptr<TMVA::IMethod> temp( tmpTMVAReader.BookMVA(_MethodName , weightFile.fullPath() ) );

  return std::unique_ptr<const GBRForest>( new GBRForest( dynamic_cast<TMVA::MethodBDT*>( tmpTMVAReader.FindMVA(_MethodName) ) ) );

}

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

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 193 of file PhotonMVAEstimatorRun2Spring15NonTrig.cc.

References _esEffSigmaRRMapLabel, _full5x5E1x3MapLabel, _full5x5E2x2MapLabel, _full5x5E2x5MaxMapLabel, _full5x5E5x5MapLabel, _full5x5SigmaIEtaIEtaMapLabel, _full5x5SigmaIEtaIPhiMapLabel, _phoChargedIsolationLabel, _phoPhotonIsolationLabel, _phoWorstChargedIsolationLabel, _rhoLabel, constrainMVAVariables(), Exception, findCategory(), edm::Event::getByLabel(), isEndcapCategory(), edm::Ptr< T >::isNull(), edm::HandleBase::isValid(), AnyMVAEstimatorRun2Base::packMVAVariables(), rho, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varChIsoRaw, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varE1x3overE5x5, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varE2x2overE5x5, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varE2x5overE5x5, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varESEffSigmaRR, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varESEnOverRawE, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varEta, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varPhi, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varPhoIsoRaw, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varPt, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varR9, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varRawE, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varRho, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSCEta, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSCEtaWidth, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSCPhiWidth, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSieie, PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varSieip, and PhotonMVAEstimatorRun2Spring15NonTrig::AllVariables::varWorstChRaw.

Referenced by mvaValue().

                                                                                                                                            {
  
  // 
  // Declare all value maps corresponding to the above tokens
  //
  edm::Handle<edm::ValueMap<float> > full5x5SigmaIEtaIEtaMap;
  edm::Handle<edm::ValueMap<float> > full5x5SigmaIEtaIPhiMap;
  edm::Handle<edm::ValueMap<float> > full5x5E1x3Map;
  edm::Handle<edm::ValueMap<float> > full5x5E2x2Map;
  edm::Handle<edm::ValueMap<float> > full5x5E2x5MaxMap;
  edm::Handle<edm::ValueMap<float> > full5x5E5x5Map;
  edm::Handle<edm::ValueMap<float> > esEffSigmaRRMap;
  //
  edm::Handle<edm::ValueMap<float> > phoChargedIsolationMap;
  edm::Handle<edm::ValueMap<float> > phoPhotonIsolationMap;
  edm::Handle<edm::ValueMap<float> > phoWorstChargedIsolationMap;

  // Rho will be pulled from the event content
  edm::Handle<double> rho;

  // Get the full5x5 and ES maps
  iEvent.getByLabel(_full5x5SigmaIEtaIEtaMapLabel, full5x5SigmaIEtaIEtaMap);
  iEvent.getByLabel(_full5x5SigmaIEtaIPhiMapLabel, full5x5SigmaIEtaIPhiMap);
  iEvent.getByLabel(_full5x5E1x3MapLabel, full5x5E1x3Map);
  iEvent.getByLabel(_full5x5E2x2MapLabel, full5x5E2x2Map);
  iEvent.getByLabel(_full5x5E2x5MaxMapLabel, full5x5E2x5MaxMap);
  iEvent.getByLabel(_full5x5E5x5MapLabel, full5x5E5x5Map);
  iEvent.getByLabel(_esEffSigmaRRMapLabel, esEffSigmaRRMap);

  // Get the isolation maps
  iEvent.getByLabel(_phoChargedIsolationLabel, phoChargedIsolationMap);
  iEvent.getByLabel(_phoPhotonIsolationLabel, phoPhotonIsolationMap);
  iEvent.getByLabel(_phoWorstChargedIsolationLabel, phoWorstChargedIsolationMap);

  // Get rho
  iEvent.getByLabel(_rhoLabel,rho);

  // Make sure everything is retrieved successfully
  if(! (full5x5SigmaIEtaIEtaMap.isValid()
    && full5x5SigmaIEtaIPhiMap.isValid()
    && full5x5E1x3Map.isValid()
    && full5x5E2x2Map.isValid()
    && full5x5E2x5MaxMap.isValid()
    && full5x5E5x5Map.isValid()
    && esEffSigmaRRMap.isValid()
    && phoChargedIsolationMap.isValid()
    && phoPhotonIsolationMap.isValid()
    && phoWorstChargedIsolationMap.isValid()
    && rho.isValid() ) )
    throw cms::Exception("MVA failure: ")
      << "Failed to retrieve event content needed for this MVA" 
      << std::endl
      << "Check python MVA configuration file and make sure all needed"
      << std::endl
      << "producers are running upstream" << std::endl;

  // Try to cast the particle into a reco particle.
  // This should work for both reco and pat.
  const edm::Ptr<reco::Photon> phoRecoPtr(particle);
  if( phoRecoPtr.isNull() )
    throw cms::Exception("MVA failure: ")
      << " given particle is expected to be reco::Photon or pat::Photon," << std::endl
      << " but appears to be neither" << std::endl;

  // Both pat and reco particles have exactly the same accessors.
  auto superCluster = phoRecoPtr->superCluster();
  // Full 5x5 cluster shapes. We could take some of this directly from
  // the photon object, but some of these are not available.
  const float e1x3 = (*full5x5E1x3Map   )[ phoRecoPtr ];
  const float e2x2 = (*full5x5E2x2Map   )[ phoRecoPtr ];
  const float e2x5 = (*full5x5E2x5MaxMap)[ phoRecoPtr ];
  const float e5x5 = (*full5x5E5x5Map   )[ phoRecoPtr ];

  AllVariables allMVAVars;

  allMVAVars.varPhi          = phoRecoPtr->phi();
  allMVAVars.varR9           = phoRecoPtr->r9() ;
  allMVAVars.varSieie        = (*full5x5SigmaIEtaIEtaMap)[ phoRecoPtr ]; // in principle, in the photon object as well
  allMVAVars.varSieip        = (*full5x5SigmaIEtaIPhiMap)[ phoRecoPtr ]; // not in the photon object
  allMVAVars.varE1x3overE5x5 = e1x3/e5x5;
  allMVAVars.varE2x2overE5x5 = e2x2/e5x5;
  allMVAVars.varE2x5overE5x5 = e2x5/e5x5;
  allMVAVars.varSCEta        = superCluster->eta(); 
  allMVAVars.varRawE         = superCluster->rawEnergy(); 
  allMVAVars.varSCEtaWidth   = superCluster->etaWidth(); 
  allMVAVars.varSCPhiWidth   = superCluster->phiWidth(); 
  allMVAVars.varESEnOverRawE = superCluster->preshowerEnergy() / superCluster->rawEnergy();
  allMVAVars.varESEffSigmaRR = (*esEffSigmaRRMap)[ phoRecoPtr ];
  allMVAVars.varRho          = *rho; 
  allMVAVars.varPhoIsoRaw    = (*phoPhotonIsolationMap)[phoRecoPtr];  
  allMVAVars.varChIsoRaw     = (*phoChargedIsolationMap)[phoRecoPtr];
  allMVAVars.varWorstChRaw   = (*phoWorstChargedIsolationMap)[phoRecoPtr];
  // Declare spectator vars
  allMVAVars.varPt = phoRecoPtr->pt(); 
  allMVAVars.varEta = phoRecoPtr->eta();

  constrainMVAVariables(allMVAVars);

  std::vector<float> vars;
  if( isEndcapCategory( findCategory( particle ) ) ) {
    vars = std::move( packMVAVariables(allMVAVars.varPhi,
                                       allMVAVars.varR9,
                                       allMVAVars.varSieie,
                                       allMVAVars.varSieip,
                                       allMVAVars.varE1x3overE5x5,
                                       allMVAVars.varE2x2overE5x5,
                                       allMVAVars.varE2x5overE5x5,
                                       allMVAVars.varSCEta,
                                       allMVAVars.varRawE,
                                       allMVAVars.varSCEtaWidth,
                                       allMVAVars.varSCPhiWidth,
                                       allMVAVars.varESEnOverRawE,
                                       allMVAVars.varESEffSigmaRR,
                                       allMVAVars.varRho,
                                       allMVAVars.varPhoIsoRaw,
                                       allMVAVars.varChIsoRaw,
                                       allMVAVars.varWorstChRaw,
                                       // Declare spectator vars
                                       allMVAVars.varPt,
                                       allMVAVars.varEta) 
                      ); 
  } else {
    vars = std::move( packMVAVariables(allMVAVars.varPhi,
                                       allMVAVars.varR9,
                                       allMVAVars.varSieie,
                                       allMVAVars.varSieip,
                                       allMVAVars.varE1x3overE5x5,
                                       allMVAVars.varE2x2overE5x5,
                                       allMVAVars.varE2x5overE5x5,
                                       allMVAVars.varSCEta,
                                       allMVAVars.varRawE,
                                       allMVAVars.varSCEtaWidth,
                                       allMVAVars.varSCPhiWidth,
                                       allMVAVars.varRho,
                                       allMVAVars.varPhoIsoRaw,
                                       allMVAVars.varChIsoRaw,
                                       allMVAVars.varWorstChRaw,
                                       // Declare spectator vars
                                       allMVAVars.varPt,
                                       allMVAVars.varEta) 
                      );
  }
  return vars;
}

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

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 100 of file PhotonMVAEstimatorRun2Spring15NonTrig.cc.

References funct::abs(), CAT_EB, CAT_EE, eta, edm::Ptr< T >::isNull(), and UNDEFINED.

Referenced by fillMVAVariables(), and mvaValue().

                                                                                                    {
  
  // Try to cast the particle into a reco particle.
  // This should work for both reco and pat.
  const edm::Ptr<reco::Photon> phoRecoPtr = ( edm::Ptr<reco::Photon> )particle;
  if( phoRecoPtr.isNull() )
    throw cms::Exception("MVA failure: ")
      << " given particle is expected to be reco::Photon or pat::Photon," << std::endl
      << " but appears to be neither" << std::endl;

  float eta = phoRecoPtr->superCluster()->eta();

  //
  // Determine the category
  //
  int  iCategory = UNDEFINED;
  const float ebeeSplit = 1.479; // division between barrel and endcap

  if ( std::abs(eta) < ebeeSplit)  
    iCategory = CAT_EB;

  if (std::abs(eta) >= ebeeSplit) 
    iCategory = CAT_EE;

  return iCategory;
}

virtual const std::string& PhotonMVAEstimatorRun2Spring15NonTrig::getName ( void  ) const

inlinefinaloverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 73 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

References _name.

{ return _name; }

virtual int PhotonMVAEstimatorRun2Spring15NonTrig::getNCategories ( ) const

inlinevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 71 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

References nCategories.

{ return nCategories; }

virtual const std::string& PhotonMVAEstimatorRun2Spring15NonTrig::getTag ( void  ) const

inlinefinaloverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 74 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

References _tag.

{ return _tag; }

bool PhotonMVAEstimatorRun2Spring15NonTrig::isEndcapCategory

(

int

category

)

const

Definition at line 128 of file PhotonMVAEstimatorRun2Spring15NonTrig.cc.

References CAT_EE, and GeomDetEnumerators::isEndcap().

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

                                     {

  // For this specific MVA the function is trivial, but kept for possible
  // future evolution to an MVA with more categories in eta
  bool isEndcap = false;
  if( category == CAT_EE )
    isEndcap = true;

  return isEndcap;
}

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

virtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 58 of file PhotonMVAEstimatorRun2Spring15NonTrig.cc.

References _gbrForests, debug, fillMVAVariables(), findCategory(), isEndcapCategory(), and query::result.

                                                                              {  

  const int iCategory = findCategory( particle );
  const std::vector<float> vars = std::move( fillMVAVariables( particle, iEvent ) );  
  
  const float result = _gbrForests.at(iCategory)->GetClassifier(vars.data());

  // DEBUG
  const bool debug = false;
  if( debug ){
    printf("Printout of the photon variable inputs for MVA:\n");
    printf("  varPhi_           %f\n", vars[0]   );
    printf("  varR9_            %f\n", vars[1]   ); 
    printf("  varSieie_         %f\n", vars[2]   );
    printf("  varSieip_         %f\n", vars[3]   ); 
    printf("  varE1x3overE5x5_  %f\n", vars[4]   ); 
    printf("  varE2x2overE5x5_  %f\n", vars[5]   ); 
    printf("  varE2x5overE5x5_  %f\n", vars[6]   ); 
    printf("  varSCEta_         %f\n", vars[7]   ); 
    printf("  varRawE_          %f\n", vars[8]   ); 
    printf("  varSCEtaWidth_    %f\n", vars[9]   ); 
    printf("  varSCPhiWidth_    %f\n", vars[10]  ); 
    printf("  varRho_           %f\n", vars[11]  );
    printf("  varPhoIsoRaw_     %f\n", vars[12]  );
    printf("  varChIsoRaw_      %f\n", vars[13]  ); 
    printf("  varWorstChRaw_    %f\n", vars[14]  );
    if( isEndcapCategory( iCategory ) ) {
      printf("  varESEnOverRawE_  %f\n", vars[15]  ); // for endcap MVA only
      printf("  varESEffSigmaRR_  %f\n", vars[16]  ); // for endcap MVA only
      // The spectators
      printf("  varPt_    %f\n", vars[17]          ); 
      printf("  varEta_  %f\n",  vars[18]          );
    } else {
      // The spectators
      printf("  varPt_    %f\n", vars[15]          ); 
      printf("  varEta_  %f\n",  vars[16]          );
    }    
  }
  
  return result;
}

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

overridevirtual

Reimplemented from AnyMVAEstimatorRun2Base.

Definition at line 348 of file PhotonMVAEstimatorRun2Spring15NonTrig.cc.

References _esEffSigmaRRMapLabel, _full5x5E1x3MapLabel, _full5x5E2x2MapLabel, _full5x5E2x5MaxMapLabel, _full5x5E5x5MapLabel, _full5x5SigmaIEtaIEtaMapLabel, _full5x5SigmaIEtaIPhiMapLabel, _phoChargedIsolationLabel, _phoPhotonIsolationLabel, _phoWorstChargedIsolationLabel, and _rhoLabel.

                                                                                       {
  cc.consumes<edm::ValueMap<float> >(_full5x5SigmaIEtaIEtaMapLabel);
  cc.consumes<edm::ValueMap<float> >(_full5x5SigmaIEtaIPhiMapLabel); 
  cc.consumes<edm::ValueMap<float> >(_full5x5E1x3MapLabel); 
  cc.consumes<edm::ValueMap<float> >(_full5x5E2x2MapLabel);
  cc.consumes<edm::ValueMap<float> >(_full5x5E2x5MaxMapLabel);
  cc.consumes<edm::ValueMap<float> >(_full5x5E5x5MapLabel);
  cc.consumes<edm::ValueMap<float> >(_esEffSigmaRRMapLabel);
  cc.consumes<edm::ValueMap<float> >(_phoChargedIsolationLabel);
  cc.consumes<edm::ValueMap<float> >(_phoPhotonIsolationLabel);
  cc.consumes<edm::ValueMap<float> >( _phoWorstChargedIsolationLabel);
  cc.consumes<double>(_rhoLabel);
}

Member Data Documentation

AllVariables PhotonMVAEstimatorRun2Spring15NonTrig::_allMVAVars

private

Definition at line 109 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by createSingleReader().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_esEffSigmaRRMapLabel

private

Definition at line 124 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_full5x5E1x3MapLabel

private

Definition at line 120 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_full5x5E2x2MapLabel

private

Definition at line 121 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_full5x5E2x5MaxMapLabel

private

Definition at line 122 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_full5x5E5x5MapLabel

private

Definition at line 123 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_full5x5SigmaIEtaIEtaMapLabel

private

Definition at line 118 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_full5x5SigmaIEtaIPhiMapLabel

private

Definition at line 119 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

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

private

Definition at line 105 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by mvaValue(), and PhotonMVAEstimatorRun2Spring15NonTrig().

const std::string PhotonMVAEstimatorRun2Spring15NonTrig::_MethodName

private

Definition at line 108 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by createSingleReader().

const std::string PhotonMVAEstimatorRun2Spring15NonTrig::_name = "PhotonMVAEstimatorRun2Spring15NonTrig"

private

Definition at line 97 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by plotting.FakeDuplicate::__str__(), plotting.AggregateBins::__str__(), plotting.AggregateHistos::__str__(), plotting.Plot::_createOne(), plotting.PlotGroup::_save(), plotting.FakeDuplicate::create(), plotting.AggregateBins::create(), plotting.AggregateHistos::create(), plotting.Plot::draw(), getName(), and plotting.Plot::getName().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_phoChargedIsolationLabel

private

Definition at line 126 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_phoPhotonIsolationLabel

private

Definition at line 127 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_phoWorstChargedIsolationLabel

private

Definition at line 128 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

const edm::InputTag PhotonMVAEstimatorRun2Spring15NonTrig::_rhoLabel

private

Definition at line 129 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by fillMVAVariables(), and setConsumes().

std::string PhotonMVAEstimatorRun2Spring15NonTrig::_tag

private

Definition at line 102 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by getTag(), and PhotonMVAEstimatorRun2Spring15NonTrig().

const int PhotonMVAEstimatorRun2Spring15NonTrig::nCategories = 2

Definition at line 27 of file PhotonMVAEstimatorRun2Spring15NonTrig.h.

Referenced by getNCategories(), and PhotonMVAEstimatorRun2Spring15NonTrig().