File indexing completed on 2023-10-25 09:58:56
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 ¶meters, edm::ESConsumesCollector &&cc);
0021 edm::ESGetToken<PhysicsTools::Calibration::MVAComputerContainer, BTauGenericMVAJetTagComputerRcd> calib_;
0022 };
0023
0024 GenericMVAJetTagComputer(const edm::ParameterSet ¶meters, 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