MuonLRTMergingAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/

 
 
 
#ifndef MUON_ANALYSIS_ALGORITHMS__MUON_LRT_MERGING_ALGORITHM__H
#define MUON_ANALYSIS_ALGORITHMS__MUON_LRT_MERGING_ALGORITHM__H
 
#include <AnaAlgorithm/AnaReentrantAlgorithm.h>
#include <xAODMuon/MuonContainer.h>
#include <MuonAnalysisInterfaces/IMuonLRTOverlapRemovalTool.h>
#include <AsgTools/ToolHandle.h>
 
#include <AsgTools/PropertyWrapper.h>
#include <AsgDataHandles/WriteHandleKey.h>
#include <AsgDataHandles/ReadHandleKey.h>
#include <AsgDataHandles/WriteDecorHandleKey.h>
#include <AthContainers/ConstDataVector.h>
 
namespace CP
{
 
  class MuonLRTMergingAlg final : public EL::AnaReentrantAlgorithm
  {
  public:
    MuonLRTMergingAlg (const std::string& name,
                             ISvcLocator* pSvcLocator);
    StatusCode initialize () override;
    StatusCode execute (const EventContext &ctx) const  override;
 
  private:




      SG::ReadHandleKey<xAOD::MuonContainer>      m_promptMuonLocation{this, "PromptMuonLocation","Muons", "Prompt muons to merge"};
      SG::ReadHandleKey<xAOD::MuonContainer>      m_lrtMuonLocation{this, "LRTMuonLocation","MuonsLRT", "LRT muons to merge"};
      SG::WriteHandleKey<xAOD::MuonContainer>     m_outMuonLocation{this, "OutputMuonLocation", "StdWithLRTMuons", "name of the muon container to write"};
      SG::WriteHandleKey<ConstDataVector<xAOD::MuonContainer>> m_outMuonViewLocation;

      Gaudi::Property<bool>  m_createViewCollection{this, "CreateViewCollection", true};     
      ToolHandle<CP::IMuonLRTOverlapRemovalTool>  m_overlapRemovalTool{this, "overlapRemovalTool", "", "tool to determine overlaps between regular and LRT muons"};

      Gaudi::Property<int>  m_ORstrategy{this, "overlapStrategy", CP::IMuonLRTOverlapRemovalTool::defaultStrategy, "Overlap removal strategy to use (0 = default, 1 = write all muons and decorate with overlap type)"};
      Gaudi::Property<bool>  m_useRun3WP{this, "UseRun3WP", false, "Switch to toggle the run 2 & run 3 geometry for the muon ID working points used in the prompt-LRT overlap resolution. Set to true for run 3 and false for run 2."};



      StatusCode mergeMuon(const xAOD::MuonContainer & muonCol,
                           const std::vector<bool> & muonIsGood,
                           ConstDataVector<xAOD::MuonContainer>* outputCol) const;

      StatusCode mergeMuon(const xAOD::MuonContainer & muonCol,
                           const std::vector<bool> & muonIsGood,
                           xAOD::MuonContainer* outputCol) const;
 
 
      SG::WriteDecorHandleKey<xAOD::MuonContainer> m_promptIsLRTKey
      { this, "PromptIsLRTKey", m_promptMuonLocation, "isLRT" };
      SG::WriteDecorHandleKey<xAOD::MuonContainer> m_lrtIsLRTKey
      { this, "LRTIsLRTKey", m_lrtMuonLocation, "isLRT" };
  };
}
 
#endif