MuonLRTOverlapRemovalTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef LRTMUONANALYSISTOOLS_MuonLRTOverlapRemovalTool_H
#define LRTMUONANALYSISTOOLS_MuonLRTOverlapRemovalTool_H
 
#include "MuonAnalysisInterfaces/IMuonLRTOverlapRemovalTool.h"
#include "MuonAnalysisInterfaces/IMuonSelectionTool.h"
#include <string>
#include <map>
 
#include "xAODTracking/TrackParticle.h"
#include "xAODMuon/MuonContainer.h"
#include "xAODMuon/Muon.h"
 
#include <AsgTools/ToolHandle.h>
#include <AsgTools/AsgTool.h>
#include <AsgTools/PropertyWrapper.h>
 
namespace CP {

  class MuonLRTOverlapRemovalTool : virtual public CP::IMuonLRTOverlapRemovalTool, public asg::AsgTool {
 
    public:


 
      MuonLRTOverlapRemovalTool(const std::string& name);
      virtual ~MuonLRTOverlapRemovalTool() = default;
      ASG_TOOL_CLASS( MuonLRTOverlapRemovalTool, CP::IMuonLRTOverlapRemovalTool )
      virtual StatusCode initialize();

      virtual std::pair<bool, bool> resolveOverlap( const xAOD::Muon* promptMuon,
                                                  const xAOD::Muon* lrtMuon) const;

      virtual std::tuple<int, int> checkOverlapForDecor(const xAOD::Muon* promptMuon,
                                                         const xAOD::Muon* lrtMuon) const;

      virtual void checkOverlap(const xAOD::MuonContainer & promptCollection,
                                  const xAOD::MuonContainer &  lrtCollection,
                                  std::vector<bool>& promptMuonsSelectedToKeep,
                                  std::vector<bool>& lrtMuonsSelectedToKeep ) const;

      virtual bool hasOverlap(const xAOD::Muon* promptMuon,
                              const xAOD::Muon* lrtMuon) const;

      virtual float getIDMEdEta(const xAOD::Muon* muon) const;
 
    private:
      Gaudi::Property<int>  m_strategy{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."}; 
      ToolHandle<CP::IMuonSelectionTool> m_muonSelectionTool{this, "MuonSelectionTool", "", "tool to determine the working point of the muons"};
 
  };
 
} // end namespace CP
#endif // LRTMUONANALYSISTOOLS_MuonLRTOverlapRemovalTool_H