MVAComputerESSourceBase.cc

Go to the documentation of this file.

#include <iostream>
#include <string>
#include <memory>
#include <vector>
#include <map>
 
#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/Framework/interface/ESProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/Entry.h"
#include "FWCore/ParameterSet/interface/FileInPath.h"
 
#include "PhysicsTools/MVAComputer/interface/Calibration.h"
#include "PhysicsTools/MVAComputer/interface/MVAComputer.h"
#include "PhysicsTools/MVAComputer/interface/MVAComputerESSourceBase.h"
 
namespace PhysicsTools {
 
  MVAComputerESSourceBase::MVAComputerESSourceBase(const edm::ParameterSet &params) {
    std::vector<std::string> names = params.getParameterNames();
    for (std::vector<std::string>::const_iterator iter = names.begin(); iter != names.end(); iter++) {
      if (iter->c_str()[0] == '@')
        continue;
 
      const edm::Entry &entry = params.retrieve(*iter);
 
      std::string path;
      if (entry.typeCode() == 'F')
        path = entry.getFileInPath().fullPath();
      else
        path = entry.getString();
 
      mvaCalibrations[*iter] = path;
    }
  }
 
  MVAComputerESSourceBase::~MVAComputerESSourceBase() {}
 
  MVAComputerESSourceBase::ReturnType MVAComputerESSourceBase::produce() const {
    auto container = std::make_unique<Calibration::MVAComputerContainer>();
 
    for (LabelFileMap::const_iterator iter = mvaCalibrations.begin(); iter != mvaCalibrations.end(); iter++) {
      std::unique_ptr<Calibration::MVAComputer> calibration(MVAComputer::readCalibration(iter->second.c_str()));
 
      container->add(iter->first) = *calibration;
    }
 
    return container;
  }
 
}  // namespace PhysicsTools