MVAVariableManager< ParticleType > Class Template Reference

#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	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 16 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 79 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("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 47 of file MVAVariableManager.h.

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

{ return names_[index]; }

template<class ParticleType>

int MVAVariableManager< ParticleType >::getNVars ( ) const

inline

Definition at line 49 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 51 of file MVAVariableManager.h.

Referenced by PhotonMVANtuplizer::analyze(), ElectronMVANtuplizer::analyze(), ElectronMVAEstimatorRun2::init(), and PhotonMVAEstimator::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 38 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;
    }
  }

Member Data Documentation

template<class ParticleType>

std::vector<std::string> MVAVariableManager< ParticleType >::formulas_

private

Definition at line 118 of file MVAVariableManager.h.

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

template<class ParticleType>

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

private

Definition at line 117 of file MVAVariableManager.h.

Referenced by MVAVariableManager< reco::GsfElectron >::addVariable(), and MVAVariableManager< reco::GsfElectron >::getValue().

template<class ParticleType>

const MVAVariableIndexMap MVAVariableManager< ParticleType >::indexMap

private

Definition at line 122 of file MVAVariableManager.h.

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

template<class ParticleType>

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

private

Definition at line 120 of file MVAVariableManager.h.

Referenced by MVAVariableManager< reco::GsfElectron >::addVariable(), and MVAVariableManager< reco::GsfElectron >::getVarIndex().

template<class ParticleType>

std::vector<std::string> MVAVariableManager< ParticleType >::names_

private

Definition at line 119 of file MVAVariableManager.h.

Referenced by MVAVariableManager< reco::GsfElectron >::addVariable(), and MVAVariableManager< reco::GsfElectron >::getName().

template<class ParticleType>

int MVAVariableManager< ParticleType >::nVars_

private

Definition at line 112 of file MVAVariableManager.h.

Referenced by MVAVariableManager< reco::GsfElectron >::addVariable(), and MVAVariableManager< reco::GsfElectron >::getNVars().

template<class ParticleType>

std::vector<MVAVariableInfo> MVAVariableManager< ParticleType >::variableInfos_

private

Definition at line 116 of file MVAVariableManager.h.

Referenced by MVAVariableManager< reco::GsfElectron >::addVariable(), and MVAVariableManager< reco::GsfElectron >::getValue().