MVAComputer/interface/MVAComputerESSourceBase.h

0001 #ifndef PhysicsTools_MVAComputer_MVAComputerESSourceBase_h
0002 #define PhysicsTools_MVAComputer_MVAComputerESSourceBase_h
0003
0004 #include <string>
0005 #include <map>
0006
0007 #include <memory>
0008
0009 #include "FWCore/Framework/interface/ESProducer.h"
0010 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0011
0012 #include "PhysicsTools/MVAComputer/interface/Calibration.h"
0013 #include "PhysicsTools/MVAComputer/interface/MVAComputer.h"
0014
0015 namespace PhysicsTools {
0016
0017   class MVAComputerESSourceBase : public edm::ESProducer {
0018   public:
0019     using ReturnType = std::unique_ptr<Calibration::MVAComputerContainer>;
0020
0021     MVAComputerESSourceBase(const edm::ParameterSet &params);
0022     ~MVAComputerESSourceBase() override;
0023
0024   protected:
0025     ReturnType produce() const;
0026
0027     typedef std::map<std::string, std::string> LabelFileMap;
0028
0029     LabelFileMap mvaCalibrations;
0030   };
0031
0032 }  // namespace PhysicsTools
0033
0034 #endif  // PhysicsTools_MVAComputer_MVAComputerESSourceBase_h