#include <MVAVariableManager.h>

Classes
struct	MVAVariableInfo

Public Member Functions
std::vector< edm::InputTag >	getGlobalInputTags () const

std::vector< edm::InputTag >	getHelperInputTags () const

const std::string	getName (int index) const

const int	getNVars () const

float	getValue (int index, const edm::Ptr< ParticleType > &ptclPtr, const edm::EventBase &iEvent) const

int	getVarIndex (std::string &name)

int	init (const std::string &variableDefinitionFileName)

	MVAVariableManager ()

	MVAVariableManager (const std::string &variableDefinitionFileName)

Private Member Functions
void	addVariable (std::string &name, std::string &formula, std::string &lowerClip, std::string &upperClip)

Private Attributes
std::vector< std::string >	formulas_

std::vector< StringObjectFunction< ParticleType > >	functions_

std::vector< edm::InputTag >	globalInputTags_

std::vector< edm::InputTag >	helperInputTags_

std::map< std::string, int >	indexMap_

std::vector< std::string >	names_

int	nVars_

std::vector< MVAVariableInfo >	variableInfos_

Detailed Description

template<class ParticleType>
class MVAVariableManager< ParticleType >

Definition at line 19 of file MVAVariableManager.h.

Constructor & Destructor Documentation

template<class ParticleType>

MVAVariableManager< ParticleType >::MVAVariableManager ( )

inline

Definition at line 22 of file MVAVariableManager.h.

                          {
         nVars_ = 0;
     };

template<class ParticleType>

MVAVariableManager< ParticleType >::MVAVariableManager ( const std::string & variableDefinitionFileName )

inline

Definition at line 26 of file MVAVariableManager.h.

                                                                     {
         init(variableDefinitionFileName);
     };

Member Function Documentation

template<class ParticleType>

void MVAVariableManager< ParticleType >::addVariable	(	std::string &	name,
		std::string &	formula,
		std::string &	lowerClip,
		std::string &	upperClip
	)

inlineprivate

Definition at line 118 of file MVAVariableManager.h.

Referenced by MVAVariableManager< reco::GsfElectron >::init().

                                                                                                       {
         bool hasLowerClip = lowerClip.find("None") == std::string::npos;
         bool hasUpperClip = upperClip.find("None") == std::string::npos;
         bool fromVariableHelper = formula.find("MVAVariableHelper") != std::string::npos ||
                                   formula.find("IDValueMapProducer") != std::string::npos ||
                                   formula.find("egmPhotonIsolation") != std::string::npos;
         float lowerClipValue = hasLowerClip ? (float)::atof(lowerClip.c_str()) : 0.;
         float upperClipValue = hasUpperClip ? (float)::atof(upperClip.c_str()) : 0.;
 
         // *Rho* is the only global variable used ever, so its hardcoded...
         bool isGlobalVariable = formula.find("Rho") != std::string::npos;
 
         if ( !(fromVariableHelper || isGlobalVariable) ) {
             functions_.push_back(StringObjectFunction<ParticleType>(formula));
         } else {
             // Push back a dummy function since we won't use the
             // StringObjectFunction to evaluate a variable form the helper or a
             // global variable
             functions_.push_back(StringObjectFunction<ParticleType>("pt"));
         }
 
         formulas_.push_back(formula);
         if (fromVariableHelper) {
             helperInputTags_.push_back(edm::InputTag(formula));
         }
         if (isGlobalVariable) {
             globalInputTags_.push_back(edm::InputTag(formula));
         }
         MVAVariableInfo varInfo = {
             .hasLowerClip       = hasLowerClip,
             .hasUpperClip       = hasUpperClip,
             .lowerClipValue     = lowerClipValue,
             .upperClipValue     = upperClipValue,
             .fromVariableHelper = fromVariableHelper,
             .isGlobalVariable   = isGlobalVariable
         };
         variableInfos_.push_back(varInfo);
         names_.push_back(name);
         indexMap_[name] = nVars_;
         nVars_++;
     };

template<class ParticleType>

std::vector<edm::InputTag> MVAVariableManager< ParticleType >::getGlobalInputTags ( ) const

inline

Definition at line 80 of file MVAVariableManager.h.

Referenced by ElectronMVANtuplizer::ElectronMVANtuplizer(), ElectronMVAEstimatorRun2::setConsumes(), and PhotonMVAEstimator::setConsumes().

                                                       {
         return globalInputTags_;
     }

template<class ParticleType>

std::vector<edm::InputTag> MVAVariableManager< ParticleType >::getHelperInputTags ( ) const

inline

Definition at line 76 of file MVAVariableManager.h.

Referenced by ElectronMVANtuplizer::ElectronMVANtuplizer(), ElectronMVAEstimatorRun2::setConsumes(), and PhotonMVAEstimator::setConsumes().

                                                       {
         return helperInputTags_;
     }

template<class ParticleType>

const std::string MVAVariableManager< ParticleType >::getName ( int index ) const

inline

Definition at line 68 of file MVAVariableManager.h.

Referenced by plotting.Plot::draw(), ElectronMVANtuplizer::ElectronMVANtuplizer(), ElectronMVAEstimatorRun2::mvaValue(), and PhotonMVAEstimator::mvaValue().

                                              {
         return names_[index];
     }

template<class ParticleType>

const int MVAVariableManager< ParticleType >::getNVars ( ) const

inline

Definition at line 72 of file MVAVariableManager.h.

Referenced by ElectronMVANtuplizer::ElectronMVANtuplizer().

                                {
         return nVars_;
     }

template<class ParticleType>

float MVAVariableManager< ParticleType >::getValue	(	int	index,
		const edm::Ptr< ParticleType > &	ptclPtr,
		const edm::EventBase &	iEvent
	)		const

inline

Definition at line 84 of file MVAVariableManager.h.

Referenced by ElectronMVANtuplizer::analyze(), ElectronMVAEstimatorRun2::mvaValue(), and PhotonMVAEstimator::mvaValue().

                                                                                                      {
         float value;
         MVAVariableInfo varInfo = variableInfos_[index];
         if (varInfo.fromVariableHelper) {
             edm::Handle<edm::ValueMap<float>> vMap;
             iEvent.getByLabel(edm::InputTag(formulas_[index]), vMap);
             value = (*vMap)[ptclPtr];
         } else if (varInfo.isGlobalVariable) {
             edm::Handle<double> valueHandle;
             iEvent.getByLabel(edm::InputTag(formulas_[index]), valueHandle);
             value = *valueHandle;
         } else {
             value = functions_[index](*ptclPtr);
         }
         if (varInfo.hasLowerClip && value < varInfo.lowerClipValue) {
             value = varInfo.lowerClipValue;
         }
         if (varInfo.hasUpperClip && value > varInfo.upperClipValue) {
             value = varInfo.upperClipValue;
         }
         return value;
     }

template<class ParticleType>

int MVAVariableManager< ParticleType >::getVarIndex ( std::string & name )

inline

Definition at line 59 of file MVAVariableManager.h.

Referenced by ElectronMVAEstimatorRun2::init(), and PhotonMVAEstimator::PhotonMVAEstimator().

                                      {
         std::map<std::string,int>::iterator it = indexMap_.find(name);
         if (it == indexMap_.end()) {
             return -1;
         } else {
             return it->second;
         }
     }

template<class ParticleType>

int MVAVariableManager< ParticleType >::init ( const std::string & variableDefinitionFileName )

inline

Definition at line 30 of file MVAVariableManager.h.

Referenced by MVAVariableManager< reco::GsfElectron >::MVAVariableManager().

                                                           {
         nVars_ = 0;
 
         variableInfos_.clear();
         functions_.clear();
         formulas_.clear();
         names_.clear();
         helperInputTags_.clear();
         globalInputTags_.clear();
 
         edm::FileInPath variableDefinitionFileEdm(variableDefinitionFileName);
         std::ifstream file(variableDefinitionFileEdm.fullPath());
 
         std::string name, formula, upper, lower;
         while( true ) {
             file >> name;
             if (name.find("#") != std::string::npos) {
                 file.ignore(std::numeric_limits<std::streamsize>::max(), '\n');
                 continue;
             }
             file >> formula >> lower >> upper;
             if (file.eof()) {
                 break;
             }
             addVariable(name, formula, lower, upper);
         }
         return nVars_;
     };