RecoSegToTruthAssocAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRUTHALGSR4_RECOSEGTOTRUTHASSOCALG_H
#define MUONTRUTHALGSR4_RECOSEGTOTRUTHASSOCALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
 
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "xAODMuon/MuonSegmentContainer.h"
#include "xAODMuonSimHit/MuonSimHit.h"
#include "xAODMeasurementBase/UncalibratedMeasurementContainer.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/ReadDecorHandleKey.h"
#include "StoreGate/WriteDecorHandleKey.h"
#include "StoreGate/ReadDecorHandleKeyArray.h"
 
namespace MuonR4{
    class RecoSegToTruthAssocAlg : public AthReentrantAlgorithm{
        public:
            using AthReentrantAlgorithm::AthReentrantAlgorithm;
            virtual StatusCode initialize() override final;
            virtual StatusCode execute(const EventContext& ctx) const override final;
        private:
            struct SegmentWithTruth{
                const xAOD::MuonSegment* segment{nullptr};
                std::unordered_set<const xAOD::MuonSimHit*> hits{};
            };
            using SegWithTruthVec_t = std::vector<SegmentWithTruth>;
            std::vector<SegmentWithTruth> matchSimHits(const xAOD::MuonSegmentContainer& segments) const;
            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_truthSegKey{this, "TruthSegKey", "MuonTruthSegments"};
            SG::ReadDecorHandleKey<xAOD::MuonSegmentContainer> m_truthSegLinkKey{this, "TruthSegLinkKey", m_truthSegKey, "truthParticleLink"};
            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_segmentKey{this, "SegmentKey", "RecoSegs"};
            SG::ReadDecorHandleKey<xAOD::MuonSegmentContainer> m_segPrdLinkKey{this, "SegPrdLinkKey", m_segmentKey, "prdLinks"};
            SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_segTruthSegLinkKey{this, "SegToTruthSegLinkKey", m_segmentKey, "truthSegmentLink"};
            SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_segTruthLinkKey{this, "SegTruthLinkKey", m_segmentKey, "truthParticleLink"};
            Gaudi::Property<std::string> m_simLink{this, "SimHitLink", "simHitLink"};
            SG::ReadHandleKeyArray<xAOD::UncalibratedMeasurementContainer> m_measKeys{this, "PrdContainer",{}};
            SG::ReadDecorHandleKeyArray<xAOD::UncalibratedMeasurementContainer> m_prdLinkKeys{this, "PrdLinkKeys", {}};
    };
}
#endif