xAODSegmentCnvAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONSEGMENTCNV_XAODSEGMENTCNVALG_H
#define MUONSEGMENTCNV_XAODSEGMENTCNVALG_H
 
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/WriteDecorHandleKey.h"
 
#include "xAODMuonViews/FillContainer.h"
#include "StoreGate/WriteDecorHandle.h"
 
#include "ActsGeometryInterfaces/ITrackingGeometrySvc.h"
#include "ActsGeometryInterfaces/IExtrapolationTool.h"
 
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonPatternEvent/MuonPatternContainer.h"
 
#include "xAODMuon/MuonSegmentContainer.h"
#include "xAODMuonPrepData/CombinedMuonStripContainer.h"
#include "xAODMuonPrepData/CombinedMuonStripAuxContainer.h"
 
#include "xAODMeasurementBase/UncalibratedMeasurementContainer.h" 
#include "ActsEvent/AuxiliaryMeasurementHandler.h"
#include "ActsEvent/ContextUtility.h"
 
 
namespace MuonR4{
    class xAODSegmentCnvAlg: public AthReentrantAlgorithm {
        public:
            using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
            virtual StatusCode initialize() override final;
            virtual StatusCode execute(const EventContext& ctx) const override final;
        
        private:
            using PrdCont_t = xAOD::UncalibratedMeasurementContainer;
            using PrdLink_t = ElementLink<PrdCont_t>;
            using PrdLinkVec_t = std::vector<PrdLink_t>;

            struct PrepDataCollectorShip {
                PrepDataCollectorShip(const xAODSegmentCnvAlg& parent,
                                      const EventContext& ctx,
                                      const Acts::GeometryContext& tgContext):
                    dec_prdLinks{parent.m_prdLinkKey, ctx},
                    dec_prdStates{parent.m_prdStateKey, ctx},
                    beamSpotMeasCreator{parent.m_auxMeasProv.makeHandle(ctx, tgContext)}{}

                SG::WriteDecorHandle<xAOD::MuonSegmentContainer, PrdLinkVec_t> dec_prdLinks;
                SG::WriteDecorHandle<xAOD::MuonSegmentContainer, std::vector<char>> dec_prdStates;
                Acts::Result<ActsTrk::AuxiliaryMeasurementHandler::MeasurementProvider, 
                            ActsTrk::AuxiliaryMeasurementHandler::HandleStatus> beamSpotMeasCreator;
                xAOD::FillContainer<xAOD::CombinedMuonStripContainer,
                            xAOD::CombinedMuonStripAuxContainer> prdCombContainer{};
                const xAOD::UncalibratedMeasurement* beamSpot{};
            };

            struct Counter{
                std::uint8_t precision{0};
                std::uint8_t triggerPhi{0};
                std::uint8_t triggerEta{0};
            };

            StatusCode linkMeasurements(const ActsTrk::GeometryContext& gctx,                       
                                        const Segment& inSegment,
                                        xAOD::MuonSegment& targetSegment,
                                        PrepDataCollectorShip& ship) const;

            void evaluateSummary(const EventContext& ctx, 
                                 xAOD::MuonSegment& segment) const;
                   
            void findHoles(const EventContext& ctx,
                           const Acts::BoundTrackParameters& startPars,
                           const Acts::Surface* target,
                           const std::unordered_set<Identifier>& geoIdsWithHits,
                           Counter& holeCounter) const;

            ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc",  
                                                                "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};

            SG::ReadHandleKeyArray<SegmentContainer> m_readKeys{this, "InSegmentKeys", {"R4MuonSegments"}};
            SG::WriteHandleKey<xAOD::MuonSegmentContainer> m_writeKey{this, "OutSegmentKey", "MuonSegmentsFromR4"};            
            ActsTrk::GeoContextReadKey_t m_geoCtxKey{this, "AlignmentKey", "ActsAlignment", "cond handle key"};
           ServiceHandle<ActsTrk::ITrackingGeometrySvc> m_trackingGeometrySvc{this, "TrackingGeometrySvc", "ActsTrackingGeometrySvc"};
            ToolHandle<ActsTrk::IExtrapolationTool> m_extrapolationTool{this, "ExtrapolationTool", ""};

            using DecorKey_t = SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer>;
            DecorKey_t m_prdLinkKey{this, "PrdLinkKey",  m_writeKey, "prdLinks" };
            DecorKey_t m_prdStateKey{this, "PrdStateKey", m_writeKey, "prdState"};
            DecorKey_t m_localSegParKey{this, "LocalSegParKey", m_writeKey, "localSegPars"};
            DecorKey_t m_localSegCovKey{this, "LocalCovParKey", m_writeKey, "localSegCov"};
            DecorKey_t m_parentSegKey{this, "ParentSegmentKey", m_writeKey, "parentSegment"};
            SG::WriteHandleKey<xAOD::CombinedMuonStripContainer> m_combMeasKey{this, "combinedPrdKey", "CombinedMuonPrds"};
            ActsTrk::AuxiliaryMeasurementHandler m_auxMeasProv{this};
            ActsTrk::ContextUtility m_ctxProvider{this};
            Gaudi::Property<bool> m_convertBeamSpot{this, "convertBeamSpot", false};
            Gaudi::Property<bool> m_estimateHoles{this, "estimateHoles", true};
            Gaudi::Property<double> m_extraHolePath{this, "extraHolePath", 20.*Gaudi::Units::cm};
    };
}
#endif