JetQGTaggerBDT.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETQGTAGGERBDT_H_
#define JETQGTAGGERBDT_H_
 
#include "AsgTools/SlotSpecificObj.h"
#include "BoostedJetTaggers/JSSTaggerBase.h"
#include "CxxUtils/checker_macros.h"
 
#include "TMVA/Tools.h"
#include "TMVA/Reader.h"
#include "TMVA/MethodCuts.h"
 
namespace InDet {
  class IInDetTrackSelectionTool;
}
 
namespace CP {
 
  class JetQGTaggerBDT :
    public JSSTaggerBase {
      ASG_TOOL_CLASS0(JetQGTaggerBDT)
 
      public:
 
        JetQGTaggerBDT(const std::string& name);
 
        virtual StatusCode initialize() override;
 
        virtual StatusCode tag(const xAOD::Jet& jet) const override;
 
      private:
 
        float getScore( const xAOD::Jet& jet, asg::AcceptData &acceptData ) const;
 
        bool getJetProperties( const xAOD::Jet& jet, asg::AcceptData &acceptData ) const;
 
        bool getPrecomputedVariables( const xAOD::Jet& jet, asg::AcceptData &acceptData ) const;
 
        bool calculateVariables( const xAOD::Jet& jet, asg::AcceptData &acceptData ) const;
 
        bool isCorrectNumberOfTracks( int expectedNTracks, int nTracksFromGhostTracks ) const;
 
        struct Tagger {
          std::unique_ptr<TMVA::Reader> tmva;
          float pt;
          float eta;
          float ntracks;
          float trackwidth;
          float trackC1;
        };
 
        mutable SG::SlotSpecificObj<Tagger> m_bdtTagger ATLAS_THREAD_SAFE;
        std::string m_BDTmethod;
 
        asg::AnaToolHandle<InDet::IInDetTrackSelectionTool> m_trkSelectionTool;
 
        int m_mode;
 
        SG::WriteDecorHandleKey<xAOD::JetContainer> m_decScoreKey{this, "QGTaggerBDTScore", "QGTaggerBDTScore", "SG key for QGTaggerBDTScore"};
 
    };
 
} /* namespace CP */
 
#endif