MVAVariableHelper.h

Go to the documentation of this file.

#ifndef RecoEgamma_EgammaTools_MVAVariableHelper_H
#define RecoEgamma_EgammaTools_MVAVariableHelper_H
 
#include "DataFormats/Common/interface/ValueMap.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Framework/interface/Event.h"
 
#include <unordered_map>
#include <vector>
#include <string>
 
class MVAVariableIndexMap {
public:
  MVAVariableIndexMap();
 
  int getIndex(std::string const& name) const { return indexMap_.at(name); }
 
private:
  const std::unordered_map<std::string, int> indexMap_;
};
 
class MVAVariableHelper {
public:
  MVAVariableHelper(edm::ConsumesCollector&& cc);
 
  const std::vector<float> getAuxVariables(const edm::Event& iEvent) const;
 
private:
  static float getVariableFromDoubleToken(edm::EDGetToken const& token, const edm::Event& iEvent) {
    edm::Handle<double> handle;
    iEvent.getByToken(token, handle);
    return *handle;
  }
 
  const std::vector<edm::EDGetToken> tokens_;
};
 
#endif