JetTagComputer/interface/GenericMVAJetTagComputer.h

0001 #ifndef RecoBTau_JetTagComputer_GenericMVAJetTagComputer_h
0002 #define RecoBTau_JetTagComputer_GenericMVAJetTagComputer_h
0003
0004 #include <memory>
0005
0006 #include "FWCore/Framework/interface/ESConsumesCollector.h"
0007 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0008 #include "CondFormats/DataRecord/interface/BTauGenericMVAJetTagComputerRcd.h"
0009 #include "DataFormats/BTauReco/interface/BaseTagInfo.h"
0010 #include "DataFormats/BTauReco/interface/TaggingVariable.h"
0011 #include "RecoBTau/JetTagComputer/interface/JetTagComputer.h"
0012 #include "RecoBTau/JetTagComputer/interface/GenericMVAComputerCache.h"
0013 #include "RecoBTau/JetTagComputer/interface/TagInfoMVACategorySelector.h"
0014
0015 class JetTagComputerRecord;
0016
0017 class GenericMVAJetTagComputer : public JetTagComputer {
0018 public:
0019   struct Tokens {
0020     Tokens(const edm::ParameterSet &parameters, edm::ESConsumesCollector &&cc);
0021     edm::ESGetToken<PhysicsTools::Calibration::MVAComputerContainer, BTauGenericMVAJetTagComputerRcd> calib_;
0022   };
0023
0024   GenericMVAJetTagComputer(const edm::ParameterSet &parameters, Tokens tokens);
0025   ~GenericMVAJetTagComputer() override;
0026
0027   void initialize(const JetTagComputerRecord &) override;
0028
0029   float discriminator(const TagInfoHelper &info) const override;
0030
0031   virtual reco::TaggingVariableList taggingVariables(const reco::BaseTagInfo &tagInfo) const;
0032   virtual reco::TaggingVariableList taggingVariables(const TagInfoHelper &info) const;
0033
0034 private:
0035   std::unique_ptr<TagInfoMVACategorySelector> categorySelector_;
0036   GenericMVAComputerCache computerCache_;
0037   Tokens tokens_;
0038 };
0039
0040 #endif  // RecoBTau_JetTagComputer_GenericMVAJetTagComputer_h