JetTagComputer/interface/TagInfoMVACategorySelector.h

0001 #ifndef RecoBTau_JetTagComputer_TagInfoMVACategorySelector_h
0002 #define RecoBTau_JetTagComputer_TagInfoMVACategorySelector_h
0003
0004 #include <string>
0005 #include <vector>
0006
0007 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0008 #include "DataFormats/BTauReco/interface/TaggingVariable.h"
0009 #include "PhysicsTools/MVAComputer/interface/Calibration.h"
0010
0011 class TagInfoMVACategorySelector {
0012 public:
0013   TagInfoMVACategorySelector(const edm::ParameterSet &params);
0014   ~TagInfoMVACategorySelector();
0015
0016   inline const std::vector<std::string> &getCategoryLabels() const { return categoryLabels; }
0017
0018   int findCategory(const reco::TaggingVariableList &taggingVariables) const;
0019
0020 private:
0021   reco::TaggingVariableName categoryVariable;
0022   std::vector<std::string> categoryLabels;
0023 };
0024
0025 #endif  // RecoBTau_JetTagComputer_TagInfoMVACategorySelector_h