#include <PhotonMVAEstimator.h>

Inheritance diagram for PhotonMVAEstimator:

Public Member Functions
int	findCategory (const reco::Candidate *candPtr) const override

float	mvaValue (const reco::Candidate *candPtr, std::vector< float > const &auxVars, int &iCategory) const override

	PhotonMVAEstimator (const edm::ParameterSet &conf)

	~PhotonMVAEstimator () override

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

	AnyMVAEstimatorRun2Base (const ::std::string &mvaName, const ::std::string &mvaTag, int nCategories, bool debug)

const std::string &	getName () const

int	getNCategories () const

const std::string &	getTag () const

bool	isDebug () const

float	mvaValue (const reco::Candidate *candidate, std::vector< float > const &auxVariables) const

virtual	~AnyMVAEstimatorRun2Base ()

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Member Functions
int	findCategory (reco::Photon const &photon) const

Private Attributes
std::vector< ThreadSafeStringCut< StringCutObjectSelector< reco::Photon >, reco::Photon > >	categoryFunctions_

std::unique_ptr< EffectiveAreas >	effectiveAreas_

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

MVAVariableManager< reco::Photon >	mvaVarMngr_

int	nCategories_

std::vector< int >	nVariables_

double	phoIsoCutoff_

std::vector< double >	phoIsoPtScalingCoeff_

std::vector< std::vector< int > >	variables_

Detailed Description

Definition at line 14 of file PhotonMVAEstimator.h.

Constructor & Destructor Documentation

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

Definition at line 5 of file PhotonMVAEstimator.cc.

References categoryFunctions_, createGBRForest(), TkAlMuonSelectors_cfi::cut, effectiveAreas_, Exception, gbrForests_, AnyMVAEstimatorRun2Base::getNCategories(), edm::ParameterSet::getParameter(), AnyMVAEstimatorRun2Base::getTag(), MVAVariableManager< ParticleType >::getVarIndex(), mps_fire::i, mvaValue(), mvaVarMngr_, nVariables_, phoIsoCutoff_, phoIsoPtScalingCoeff_, variables_, and mvaElectronID_Fall17_iso_V1_cff::weightFileNames.

   : AnyMVAEstimatorRun2Base(conf)
   , mvaVarMngr_ (conf.getParameter<std::string>("variableDefinition"))
 {
 
   //
   // Construct the MVA estimators
   //
   if (getTag() == "Run2Spring16NonTrigV1") {
       effectiveAreas_ = std::make_unique<EffectiveAreas>((conf.getParameter<edm::FileInPath>("effAreasConfigFile")).fullPath());
       phoIsoPtScalingCoeff_ = conf.getParameter<std::vector<double >>("phoIsoPtScalingCoeff");
       phoIsoCutoff_ = conf.getParameter<double>("phoIsoCutoff");
   }
 
   const auto weightFileNames = conf.getParameter<std::vector<std::string> >("weightFileNames");
   const auto categoryCutStrings = conf.getParameter<std::vector<std::string> >("categoryCuts");
 
   if( (int)(categoryCutStrings.size()) != getNCategories() )
     throw cms::Exception("MVA config failure: ")
       << "wrong number of category cuts in PhotonMVAEstimator" << getTag() << std::endl;
 
   for (auto const& cut : categoryCutStrings) categoryFunctions_.emplace_back(cut);
 
   // Initialize GBRForests
   if( static_cast<int>(weightFileNames.size()) != getNCategories() )
     throw cms::Exception("MVA config failure: ")
       << "wrong number of weightfiles in PhotonMVAEstimator" << getTag() << std::endl;
 
   gbrForests_.clear();
   // Create a TMVA reader object for each category
   for(int i=0; i<getNCategories(); i++){
 
     std::vector<int> variablesInCategory;
 
     std::vector<std::string> variableNamesInCategory;
     gbrForests_.push_back(createGBRForest(weightFileNames[i], variableNamesInCategory));
 
     nVariables_.push_back(variableNamesInCategory.size());
 
     variables_.push_back(variablesInCategory);
 
     for (int j=0; j<nVariables_[i];++j) {
         int index = mvaVarMngr_.getVarIndex(variableNamesInCategory[j]);
         if(index == -1) {
             throw cms::Exception("MVA config failure: ")
                << "Concerning PhotonMVAEstimator" << getTag() << std::endl
                << "Variable " << variableNamesInCategory[j]
                << " not found in variable definition file!" << std::endl;
         }
         variables_[i].push_back(index);
 
     }
   }
 }

PhotonMVAEstimator::~PhotonMVAEstimator ( )

inlineoverride

Definition at line 20 of file PhotonMVAEstimator.h.

References fillDescriptions(), findCategory(), mvaValue(), and muons2muons_cfi::photon.

20 {};

Member Function Documentation

static void PhotonMVAEstimator::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Referenced by ~PhotonMVAEstimator().

int PhotonMVAEstimator::findCategory ( const reco::Candidate * candPtr ) const

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 104 of file PhotonMVAEstimator.cc.

References Exception.

Referenced by mvaValue(), and ~PhotonMVAEstimator().

                                                                         {
 
   const reco::Photon* phoPtr = dynamic_cast<const reco::Photon*>(candPtr);
   if( phoPtr == nullptr ) {
     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;
   }
 
   return findCategory(*phoPtr);
 
 }

int PhotonMVAEstimator::findCategory ( reco::Photon const & photon ) const

private

Definition at line 117 of file PhotonMVAEstimator.cc.

References categoryFunctions_, DEFINE_EDM_PLUGIN, AnyMVAEstimatorRun2Base::getNCategories(), AnyMVAEstimatorRun2Base::getTag(), mps_fire::i, reco::LeafCandidate::pt(), and reco::Photon::superCluster().

                                                                    {
 
   for (int i = 0; i < getNCategories(); ++i) {
       if (categoryFunctions_[i](photon)) return i;
   }
 
   edm::LogWarning  ("MVA warning") <<
       "category not defined for particle with pt " << photon.pt() << " GeV, eta " <<
           photon.superCluster()->eta() << " in PhotonMVAEstimator" << getTag();
 
   return -1;
 
 }

float PhotonMVAEstimator::mvaValue	(	const reco::Candidate *	candPtr,
		std::vector< float > const &	auxVars,
		int &	iCategory
	)		const

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 61 of file PhotonMVAEstimator.cc.

References funct::abs(), gather_cfg::cout, effectiveAreas_, Exception, findCategory(), gbrForests_, MVAVariableManager< ParticleType >::getName(), AnyMVAEstimatorRun2Base::getTag(), MVAVariableManager< ParticleType >::getValue(), mps_fire::i, AnyMVAEstimatorRun2Base::isDebug(), Max(), mvaVarMngr_, nVariables_, phoIsoCutoff_, phoIsoPtScalingCoeff_, reco::LeafCandidate::pt(), reco::Photon::superCluster(), and variables_.

Referenced by PhotonMVAEstimator(), and ~PhotonMVAEstimator().

                                                                                               {
 
   const reco::Photon* phoPtr = dynamic_cast<const reco::Photon*>(candPtr);
   if( phoPtr == nullptr) {
     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;
   }
 
   iCategory = findCategory( phoPtr );
 
   std::vector<float> vars;
 
   for (int i = 0; i < nVariables_[iCategory]; ++i) {
       vars.push_back(mvaVarMngr_.getValue(variables_[iCategory][i], *phoPtr, auxVars));
   }
 
   // Special case for Spring16!
   if (getTag() == "Run2Spring16NonTrigV1" and iCategory == 1) { // Endcap category
       // Raw value for EB only, because of loss of transparency in EE
       // for endcap MVA only in 2016
       double eA = effectiveAreas_->getEffectiveArea( std::abs(phoPtr->superCluster()->eta()) );
       double phoIsoCorr = vars[10] - eA*(double)vars[9] - phoIsoPtScalingCoeff_.at(1) * phoPtr->pt();
       vars[10] = TMath::Max( phoIsoCorr, phoIsoCutoff_);
   }
 
   if(isDebug()) {
     std::cout << " *** Inside PhotonMVAEstimator" << getTag() << std::endl;
     std::cout << " category " << iCategory << std::endl;
     for (int i = 0; i < nVariables_[iCategory]; ++i) {
         std::cout << " " << mvaVarMngr_.getName(variables_[iCategory][i]) << " " << vars[i] << std::endl;
     }
   }
 
   const float response = gbrForests_.at(iCategory)->GetResponse(vars.data());
 
   if(isDebug()) {
     std::cout << " ### MVA " << response << std::endl << std::endl;
   }
 
   return response;
 }