TruthSegmentWriter.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
 */
#ifndef MUONACTSDUMP_TRUTHSEGMENTWRITER_H
#define MUONACTSDUMP_TRUTHSEGMENTWRITER_H
 
#include "AthenaBaseComps/AthHistogramAlgorithm.h"
 
#include "MuonTesterTree/MuonTesterTree.h"
#include "MuonTesterTree/ThreeVectorBranch.h"
#include "MuonPRDTest/SegmentVariables.h"
 
#include "ActsEvent/ContextUtility.h"
#include "MuonReadoutGeometryR4/MuonDetectorManager.h"
#include "xAODMuon/MuonSegmentContainer.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/ReadDecorHandleKey.h"
 
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
 
 
namespace Acts{
    class Surface;
}
 
namespace MuonValR4{
    class TruthSegmentWriter : public AthHistogramAlgorithm{
        public:
            using AthHistogramAlgorithm::AthHistogramAlgorithm;
            virtual StatusCode initialize() override final;
            virtual StatusCode execute(const EventContext& ctx)override final;
            virtual StatusCode finalize() override final;
        private:
            const Acts::Surface& getSurface(const Identifier& simHitId) const;
            const Acts::Surface& getSurface(const xAOD::MuonSegment& segment) const;
            ActsTrk::ContextUtility m_ctxProvider{this};

            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_segmentKey{this, "segmentKey", "MuonTruthSegments"};
            SG::ReadDecorHandleKey<xAOD::MuonSegmentContainer> m_truthLinkKey{this, "truthLinkKey", m_segmentKey, "truthParticleLink"};
            
            ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
            const MuonGMR4::MuonDetectorManager* m_detMgr{nullptr};

            MuonVal::MuonTesterTree m_tree{"MuonTruth", "ActsMuonTruthDump"};
            MuonVal::ScalarBranch<std::uint32_t>& m_eventId{m_tree.newScalar<std::uint32_t>("event_id")};
            std::shared_ptr<MuonPRDTest::SegmentVariables> m_segmentBranches{};
            std::shared_ptr<MuonVal::CoordSystemsBranch> m_segmentFrames{};
            std::shared_ptr<MuonVal::IParticleFourMomBranch> m_truthTrks{};
 
 
    
    };
}
 
 
#endif