File indexing completed on 2024-04-06 12:24:35
0001
0002
0003
0004
0005 #ifndef RecoBTag_SoftLepton_MuonTagger_h
0006 #define RecoBTag_SoftLepton_MuonTagger_h
0007
0008 #include "FWCore/Framework/interface/ESConsumesCollector.h"
0009 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0010 #include "CommonTools/MVAUtils/interface/TMVAEvaluator.h"
0011 #include "CondFormats/DataRecord/interface/GBRWrapperRcd.h"
0012 #include "RecoBTau/JetTagComputer/interface/JetTagComputer.h"
0013 #include "RecoBTag/SoftLepton/interface/LeptonSelector.h"
0014 #include <memory>
0015
0016 class MuonTagger : public JetTagComputer {
0017 public:
0018 struct Tokens {
0019 Tokens(const edm::ParameterSet& cfg, edm::ESConsumesCollector&& cc);
0020 edm::ESGetToken<GBRForest, GBRWrapperRcd> gbrForest_;
0021 };
0022
0023 MuonTagger(const edm::ParameterSet&, Tokens);
0024 void initialize(const JetTagComputerRecord&) override;
0025 float discriminator(const TagInfoHelper& tagInfo) const override;
0026
0027 private:
0028 btag::LeptonSelector m_selector;
0029 const edm::FileInPath m_weightFile;
0030 const bool m_useGBRForest;
0031 const bool m_useAdaBoost;
0032 const Tokens m_tokens;
0033
0034 std::unique_ptr<TMVAEvaluator> mvaID;
0035 };
0036
0037 #endif