#include <MVAVariableManager.h>

Classes
struct	MVAVariableInfo

Public Member Functions
const std::string &	getName (int index) const

int	getNVars () const

float	getValue (int index, const ParticleType &particle, const std::vector< float > &auxVariables) const

int	getVarIndex (const std::string &name)

	MVAVariableManager (const std::string &variableDefinitionFileName)

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

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

std::vector< ThreadSafeStringCut< StringObjectFunction< ParticleType >, ParticleType > >	functions_

const MVAVariableIndexMap< ParticleType >	indexMap

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 14 of file MVAVariableManager.h.

Constructor & Destructor Documentation

template<class ParticleType>

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

inline

Definition at line 18 of file MVAVariableManager.h.

       : nVars_ (0)
     {
         edm::FileInPath variableDefinitionFileEdm(variableDefinitionFileName);
         std::ifstream file(variableDefinitionFileEdm.fullPath());
 
         std::string name, formula, upper, lower;
         while( true ) {
             file >> name;
             if(file.eof()) {
                 break;
             }
             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);
         }
     }

Member Function Documentation

template<class ParticleType>

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

inlineprivate

Definition at line 87 of file MVAVariableManager.h.

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

     {
         bool hasLowerClip = lowerClip.find("None") == std::string::npos;
         bool hasUpperClip = upperClip.find("None") == std::string::npos;
         bool isAuxiliary = formula.find("MVAVariableHelper")  != std::string::npos ||
                            formula.find("IDValueMapProducer") != std::string::npos ||
                            formula.find("egmPhotonIsolation") != std::string::npos ||
                            formula.find("Rho") != std::string::npos;
                            // *Rho* is still hardcoded...
         float lowerClipValue = hasLowerClip ? (float)::atof(lowerClip.c_str()) : 0.;
         float upperClipValue = hasUpperClip ? (float)::atof(upperClip.c_str()) : 0.;
 
         if ( !isAuxiliary ) functions_.emplace_back(formula);
         // Else push back a dummy function since we won't use the
         // StringObjectFunction to evaluate an auxiliary variable
         else functions_.emplace_back("pt");
 
         formulas_.push_back(formula);
 
         int auxIndex = isAuxiliary ? indexMap.getIndex(formula) : -1;
 
         MVAVariableInfo varInfo {
             .hasLowerClip   = hasLowerClip,
             .hasUpperClip   = hasUpperClip,
             .lowerClipValue = lowerClipValue,
             .upperClipValue = upperClipValue,
             .auxIndex       = auxIndex,
         };
 
         variableInfos_.push_back(varInfo);
         names_.push_back(name);
         indexMap_[name] = nVars_;
         nVars_++;
     };

template<class ParticleType>

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

inline

Definition at line 52 of file MVAVariableManager.h.

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

                                               {
         return names_[index];
     }

template<class ParticleType>

int MVAVariableManager< ParticleType >::getNVars ( ) const

inline

Definition at line 56 of file MVAVariableManager.h.

                          {
         return nVars_;
     }

template<class ParticleType>

float MVAVariableManager< ParticleType >::getValue	(	int	index,
		const ParticleType &	particle,
		const std::vector< float > &	auxVariables
	)		const

inline

Definition at line 60 of file MVAVariableManager.h.

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

                                                                                                         {
         float value;
 
         MVAVariableInfo varInfo = variableInfos_[index];
 
         if (varInfo.auxIndex >= 0) value = auxVariables[varInfo.auxIndex];
         else value = functions_[index](particle);
 
         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 ( const std::string & name )

inline

Definition at line 42 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;
         }
     }