JetManagerTool.h

Go to the documentation of this file.

/*
Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef Jet_Manager_H
#define Jet_Manager_H
 
#include "src/TrigBtagEmulationJet.h"
 
#include "AsgMessaging/AsgMessaging.h"
#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"
#include "AsgTools/PropertyWrapper.h"
#include "AsgTools/CurrentContext.h"
#include "AsgDataHandles/ReadHandleKey.h"
 
#include "xAODJet/JetAttributes.h"
#include "xAODJet/JetContainer.h"
#include "xAODJet/JetAuxContainer.h"
#include "xAODBTagging/BTaggingContainer.h"
#include "xAODBTagging/BTaggingAuxContainer.h"
#include "xAODBTagging/BTaggingUtilities.h"
 
#include "TrigBtagEmulationTool/EmulContext.h"
 
#include "TLorentzVector.h"
#include <string>
#include <vector>
 
namespace Trig {
 
  class JetManagerTool : 
    public asg::AsgTool {
      ASG_TOOL_CLASS0(JetManagerTool)
  public:
    JetManagerTool(const std::string& name);
    virtual ~JetManagerTool() = default;
    
    virtual StatusCode initialize() override;
        
    StatusCode retrieveByContainer(const EventContext& ctx,
                   EmulContext& emulCtx) const;
    
    const std::string& jetContainerName() const;    
    const std::string& bjetContainerName() const;
        
    const std::vector<TrigBtagEmulationJet>& getJets(const EmulContext& emulCtx) const;
    const std::vector<TrigBtagEmulationJet>& getSortedPreselJets(const EmulContext& emulCtx) const;
    
  private:
    // These are set in the initialize method given JetContainerName
    SG::ReadHandleKey< xAOD::JetContainer > m_jetInputKey {this, "InputJets", "", "Input Jet Collection Key, retrieved from reconstructed jets"};
    SG::ReadHandleKey< xAOD::JetContainer > m_bjetInputKey {this, "InputBJets", "", "Input b-Jet Collection Key, retrieved from reconstructed jets"};
    SG::ReadHandleKey< xAOD::BTaggingContainer > m_btagInputKey {this, "InputBTag", "HLT_xAOD__BTaggingContainer_HLTBjetFex", "Run2 input b-tag Collection Key"};
 
    Gaudi::Property<std::string> m_btagging_link {this, "BTaggingLink", "btaggingLink"};    
    Gaudi::Property<std::string> m_jetcontainer {this, "JetContainerName", "", "Jet Container"};
    Gaudi::Property<int> m_LHCPeriod {this, "LHCPeriod", 3, "LHC Period Run2 or Run3"};
 
    bool matchedSPLITjet(const xAOD::Jet*, const xAOD::Jet*) const;
  };
 
  inline const std::string& JetManagerTool::jetContainerName() const { return m_jetInputKey.key(); }
  inline const std::string& JetManagerTool::bjetContainerName() const { return m_bjetInputKey.key(); }
} // namespace
 
#endif