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