TrackToTruthPartAssocAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRUTHSEGMENTMAKER_TrackToTruthPartAssocAlg_H
#define MUONTRUTHSEGMENTMAKER_TrackToTruthPartAssocAlg_H
 
#include <AthenaBaseComps/AthReentrantAlgorithm.h>
 
#include <xAODTruth/TruthParticleContainer.h>
#include <xAODTracking/TrackParticleContainer.h>
 
#include <StoreGate/ReadDecorHandleKeyArray.h>
#include <StoreGate/WriteDecorHandleKey.h>
 
#include <MuonIdHelpers/IMuonIdHelperSvc.h>
#include <MuonRecHelperTools/IMuonEDMHelperSvc.h>
 
namespace MuonR4{
    class TrackToTruthPartAssocAlg : public AthReentrantAlgorithm {
        public:
            using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
            StatusCode initialize() override final;
            StatusCode execute(const EventContext& ctx) const override final;
        private:
            ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "IdHelperSvc",  "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
            ServiceHandle<Muon::IMuonEDMHelperSvc> m_edmHelperSvc{this, "EdmHelperSvc", "Muon::MuonEDMHelperSvc/MuonEDMHelperSvc"};
 
 
            using TrkReadKey_t = SG::ReadHandleKey<xAOD::TrackParticleContainer>;
            using TrkWriteDecorKey_t = SG::WriteDecorHandleKey<xAOD::TrackParticleContainer>;
 
            TrkReadKey_t m_trkKey{this, "TrackCollection", "MSTrks"};
            TrkWriteDecorKey_t m_originWriteKey{this, "TruthOriginWriteKey", m_trkKey, "truthOrigin"};
            TrkWriteDecorKey_t m_typeWriteKey{this, "TruthTypeWriteKey", m_trkKey, "truthType"};
            TrkWriteDecorKey_t m_classificationWriteKey{this, "TruthClassificationWriteKey", m_trkKey, "truthClassification"};
            TrkWriteDecorKey_t m_linkWriteKey{this, "TruthLinkWriteKey", m_trkKey, "truthParticleLink"};
            using TruthReadKey_t = SG::ReadHandleKey<xAOD::TruthParticleContainer>;
            using TruthReadDecorKey_t = SG::ReadDecorHandleKey<xAOD::TruthParticleContainer>;
            using TruthReadDecorKeyArr_t =  SG::ReadDecorHandleKeyArray<xAOD::TruthParticleContainer>;
 
            TruthReadKey_t m_truthMuonKey{this, "TruthMuonKey", "MuonTruthParticles"};

            Gaudi::Property<std::vector<std::string>> m_simHitIds{this, "SimHitIds", {}};
            TruthReadDecorKeyArr_t m_simHitKeys{this, "TruthSimHitIdKeys", {}};
            TruthReadDecorKey_t m_truMuOriginKey{this, "TruthMuonOriginKey", m_truthMuonKey, "truthOrigin"};
            TruthReadDecorKey_t m_truMuTypeKey{this, "TruthMuonTypeKey", m_truthMuonKey, "truthType"};
            TruthReadDecorKey_t m_truMuClassificationKey{this, "TruthMuonClassificationKey", m_truthMuonKey, "truthClassification"};
 
    };
}
 
#endif