SpacePointWriter.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
 */
#ifndef MUONACTSDUMP_SPACEPOINTWRITER_H
#define MUONACTSDUMP_SPACEPOINTWRITER_H
 
 
#include "AthenaBaseComps/AthHistogramAlgorithm.h"
#include "MuonTesterTree/MuonTesterTree.h"
#include "MuonTesterTree/ThreeVectorBranch.h"
#include "MuonTesterTree/CoordTransformBranch.h"
#include "MuonSpacePoint/SpacePointContainer.h"
 
 
#include "StoreGate/ReadHandleKeyArray.h"
#include "ActsGeometryInterfaces/ITrackingGeometryTool.h"
 
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
 
#include "Acts/Geometry/GeometryIdentifier.hpp"
 
namespace MuonValR4{
    class SpacePointWriter : 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:
            std::uint32_t encodeId(const MuonR4::SpacePoint& spacePoint,
                                   std::uint32_t gasGap) const;
            ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};

            PublicToolHandle<ActsTrk::ITrackingGeometryTool> m_trackingGeometryTool{this, "TrackingGeometryTool", ""};
            
            SG::ReadHandleKeyArray<MuonR4::SpacePointContainer> m_spacePointKeys{this, "SpacePointKeys", {"MuonSpacePoints", "NswSpacePoints"} };

            MuonVal::MuonTesterTree m_tree{"MuonSpacePoints", "ActsMuonSpacePointDump"};
            MuonVal::ScalarBranch<std::uint32_t>& m_eventId{m_tree.newScalar<std::uint32_t>("event_id")};

            using GeoId_t = Acts::GeometryIdentifier::Value;
            MuonVal::VectorBranch<GeoId_t>& m_geometryId{m_tree.newVector<GeoId_t>("spacePoint_geometryId")};
            MuonVal::VectorBranch<std::uint16_t>& m_bucketId{m_tree.newVector<std::uint16_t>("spacePoint_bucketId")};
            MuonVal::VectorBranch<std::uint32_t>& m_muonId{m_tree.newVector<std::uint32_t>("spacePoint_muonId")};
            MuonVal::ThreeVectorBranch m_localPosition{m_tree, "spacePoint_localPos"};
            MuonVal::UnitThreeVectorBranch m_sensorDirection{m_tree, "spacePoint_sensorDir"};
            MuonVal::UnitThreeVectorBranch m_toNextSensor{m_tree, "spacePoint_toNextDir"};
            MuonVal::VectorBranch<float>& m_covLoc0{m_tree.newVector<float>( "spacePoint_covLoc0")};
            MuonVal::VectorBranch<float>& m_covLoc1{m_tree.newVector<float>( "spacePoint_covLoc1")};
            MuonVal::VectorBranch<float>& m_covT{m_tree.newVector<float>( "spacePoint_covT")};
            MuonVal::VectorBranch<float>& m_driftR{m_tree.newVector<float>( "spacePoint_driftRadius")};
            MuonVal::VectorBranch<float>& m_time{m_tree.newVector<float>( "spacePoint_time")};
            MuonVal::CoordSystemsBranch m_toMeasFrame{m_tree, "spacePoint_toSectorFrame"};
    };
}
 
#endif