TruthSegToTruthPartAssocAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRUTHSEGMENTMAKER_TruthSegToTruthPartAssocAlg_H
#define MUONTRUTHSEGMENTMAKER_TruthSegToTruthPartAssocAlg_H
 
#include <AthenaBaseComps/AthReentrantAlgorithm.h>
 
#include <StoreGate/ReadHandleKey.h>
#include <StoreGate/ReadDecorHandleKeyArray.h>
#include <StoreGate/WriteDecorHandleKey.h>
#include <StoreGate/WriteDecorHandle.h>
 
#include "xAODTruth/TruthParticleContainer.h"
#include "xAODMuon/MuonSegmentContainer.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonReadoutGeometryR4/MuonDetectorManager.h"
#include "ActsGeometryInterfaces/IExtrapolationTool.h"
#include "ActsEvent/ContextUtility.h"
 
namespace MuonR4{
    class TruthSegToTruthPartAssocAlg: public AthReentrantAlgorithm {
        public:
            using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
            virtual StatusCode initialize() override final;
            virtual StatusCode execute(const EventContext& ctx) const override final;
        private:
            using TruthPartLink_t = ElementLink<xAOD::TruthParticleContainer>;
            using TruthPartDecor_t = SG::WriteDecorHandle<xAOD::MuonSegmentContainer, TruthPartLink_t>;
            using TruthSegLinkVec_t = std::vector<ElementLink<xAOD::MuonSegmentContainer>>;
            using TruthSegLinkDecor_t = SG::WriteDecorHandle<xAOD::TruthParticleContainer, TruthSegLinkVec_t>;

            void matchPileupSegments(const EventContext& ctx,
                                     const std::vector<const xAOD::TruthParticle*>& pileUpMuons,
                                     const std::vector<const xAOD::MuonSegment*>& pileUpSegments,
                                     TruthPartDecor_t& truthPartDecor,
                                     TruthSegLinkDecor_t& truthSegDecor) const;
            ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "IdHelperSvc",  "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
            SG::ReadHandleKey<xAOD::TruthParticleContainer> m_truthKey{this, "TruthKey", "MuonTruthParticles"};
            Gaudi::Property<std::vector<std::string>> m_simHitIds{this, "SimHitIds", {}};
            SG::ReadDecorHandleKeyArray<xAOD::TruthParticleContainer> m_simHitKeys{this, "TruthSimHitIdKeys", {}};
            SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> m_segLinkKey{this, "SegmentToPartKey", m_truthKey, "truthSegmentLinks"};
            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_segmentKey{this, "SegmentKey", "MuonTruthSegments"};
            SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_truthLinkKey{this, "TruthToPartKey", m_segmentKey, "truthParticleLink"};
            Gaudi::Property<bool> m_includePileUpObjs{this, "includePileUpObjs", false};
            Gaudi::Property<double> m_pileUpObjDPhiCut{this,"deltaPhiSegBkgMuon", 10.*Gaudi::Units::deg};
            Gaudi::Property<double> m_pileUpObjExtpDxCut{this,"BkgMatchingExtpX0", 10.*Gaudi::Units::cm};
            Gaudi::Property<double> m_pileUpObjExtpDyCut{this,"BkgMatchingExtpY0", 10.*Gaudi::Units::mm};
            Gaudi::Property<double> m_pileUpObjExtpDthetaCut{this,"BkgMatchingExtpTheta", 0.5*Gaudi::Units::deg};
            Gaudi::Property<double> m_pileUpObjExtpDphiCut{this,"BkgMatchingExtpPhi", 5.*Gaudi::Units::deg};
             
            ActsTrk::ContextUtility m_ctxProvider{this};
            ToolHandle<ActsTrk::IExtrapolationTool> m_extrapolationTool{this, "ExtrapolationTool" ,"" };
            const MuonGMR4::MuonDetectorManager* m_detMgr{};
 
    };
}
#endif