NswSegmentFinderAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONR4_MUONPATTERNRECOGNITIONALGS_NSWSEGMENTFINDERALG_H
#define MUONR4_MUONPATTERNRECOGNITIONALGS_NSWSEGMENTFINDERALG_H
 
#include <AthenaBaseComps/AthReentrantAlgorithm.h>
#include <StoreGate/WriteHandleKey.h>
#include <StoreGate/ReadCondHandleKey.h>
 
#include <MuonSpacePoint/SpacePointContainer.h>
#include <MuonPatternEvent/MuonPatternContainer.h>
#include <MuonPatternHelpers/SegmentLineFitter.h>
 
#include <MuonIdHelpers/MmIdHelper.h>
#include <MuonReadoutGeometryR4/MuonDetectorManager.h>
#include <MuonPatternEvent/MuonHoughDefs.h>
#include <MuonRecToolInterfacesR4/IPatternVisualizationTool.h>
 
#include "MuonRecToolInterfacesR4/ISpacePointCalibrator.h"
 
 
#include <MuonSpacePoint/SpacePointPerLayerSplitter.h>
 
#include <span>
#include <vector>
 
 
namespace MuonR4{
 
 
class NswSegmentFinderAlg : public AthReentrantAlgorithm {
  
    public:
        using AthReentrantAlgorithm::AthReentrantAlgorithm;
        virtual ~NswSegmentFinderAlg() = default;
        virtual StatusCode initialize() override;
        virtual StatusCode execute(const EventContext& ctx) const override;    
        virtual StatusCode finalize() override;
 
    private:
        
        enum class StripOrient{
          U, 
          V, 
          X, 
          P, 
          C, 
          Unknown
        };
 
        class SeedStatistics{
 
        public:
        
        using chIdx_t = Muon::MuonStationIndex::ChIndex;
        
        SeedStatistics() = default;
 
         //dump seed statistics to the map
        void addToStat(const MuonGMR4::SpectrometerSector* msSector,
                       unsigned int nSeeds, 
                       unsigned int nExtSeeds,
                       unsigned int nSegments);
 
        // print the seed counting stats in the end of the algorithm */
        void printTableSeedStats(MsgStream& msg) const;
        
        private:
 
        struct SeedField{
         unsigned int nSeeds{0};
          unsigned int nExtSeeds{0};
          unsigned int nSegments{0};
        };
 
        struct SectorField{  
          chIdx_t chIdx{0};      
          int phi{0};
          int eta{0};
          int8_t side{1};
 
          bool operator<(SectorField const& o) const noexcept {
             if(chIdx != o.chIdx) return chIdx < o.chIdx;
             if(eta != o.eta) return eta < o.eta;              
             if(side != o.side) return side < o.side;
              return phi < o.phi;
          }
 
        };
 
        using SeedStatistic_T = std::map<SectorField, SeedField>;
        SeedStatistic_T m_seedStat{};
 
        std::mutex m_mutex{};
 
        };
 
        StripOrient classifyStrip(const SpacePoint& spacePoint) const;
 
        using HitVec = SpacePointPerLayerSplitter::HitVec;
 
        using HitLayVec = SpacePointPerLayerSplitter::HitLayVec;
        using HitLaySpan_t = std::vector<std::reference_wrapper<const HitVec>>;
        using UsedHitMarker_t = std::vector<std::vector<unsigned int>>;
        using UsedHitSpan_t = std::vector<std::reference_wrapper<std::vector<unsigned int>>>;
        using InitialSeed_t = std::array<const SpacePoint*, 4>;
        using InitialSeedVec_t = std::vector<InitialSeed_t>;
        UsedHitMarker_t emptyBookKeeper(const HitLayVec& sortedSp) const;
        
        enum class HitWindow{tooLow, 
                             inside, 
                             tooHigh}; 
 
        HitWindow hitFromIPCorridor(const SpacePoint& testHit, 
                                    const Amg::Vector3D& beamSpotPos, 
                                    const Amg::Vector3D& dirEstUp,
                                    const Amg::Vector3D& dirEstDn) const;
        void constructPrelimnarySeeds(const Amg::Vector3D& beamSpot,
                                      const HitLaySpan_t& combinatoricLayers,
                                      const UsedHitSpan_t& usedHits,
                                      InitialSeedVec_t& outVec) const;
        std::unique_ptr<SegmentSeed> buildSegmentSeed(const InitialSeed_t& initialSeed,
                                                      const AmgSymMatrix(2)& bMatrix, 
                                                      const HoughMaximum& max, 
                                                      const HitLaySpan_t& extensionLayers,
                                                      const UsedHitSpan_t& usedHits) const;
        std::unique_ptr<Segment> fitSegmentSeed(const EventContext& ctx,
                                                const ActsTrk::GeometryContext& gctx, 
                                                const SegmentSeed *patternSeed) const;
 
        void markHitsAsUsed(const HitVec& spacePoints,
                            const HitLayVec& allSortHits,
                            UsedHitMarker_t& usedHitMarker,
                            unsigned int increase,
                            bool markNeighborHits) const;
 
       
        HitVec extendHits(const Amg::Vector3D& startPos, 
                          const Amg::Vector3D& direction, 
                          const HitLaySpan_t& extensionLayers,
                          const UsedHitSpan_t& usedHits) const;
 
        std::pair<std::vector<std::unique_ptr<SegmentSeed>>, std::vector<std::unique_ptr<Segment>>>
              findSegmentsFromMaximum(const HoughMaximum& max, 
                                   const ActsTrk::GeometryContext& gctx,
                                   const EventContext& ctx) const;
 
        // read handle key for the input maxima (from a previous eta-transform)
        SG::ReadHandleKey<EtaHoughMaxContainer> m_etaKey{this, "CombinatorialReadKey", "MuonHoughNswMaxima"};
 
        //write handle key for the segment seeds container
        SG::WriteHandleKey<SegmentSeedContainer> m_writeSegmentSeedKey{this, "MuonNswSegmentSeedWriteKey", "MuonNswSegmentSeeds"};
 
        // write handle key for the segments container
        SG::WriteHandleKey<SegmentContainer> m_writeSegmentKey{this, "MuonNswSegmentWriteKey", "MuonNswSegments"};
 
        // access to the ACTS geometry context 
        SG::ReadHandleKey<ActsTrk::GeometryContext> m_geoCtxKey{this, "AlignmentKey", "ActsAlignment", "cond handle key"};
 
        // access to the Muon Id Helper
        ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc {this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
 
        ToolHandle<MuonValR4::IPatternVisualizationTool> m_visionTool{this, "VisualizationTool", ""};
 
        //Space point calibration tool 
        ToolHandle<ISpacePointCalibrator> m_calibTool{this, "Calibrator", "" };
 
        // Pointer to the line segment fitter 
        std::unique_ptr<SegmentFit::SegmentLineFitter> m_lineFitter{};
  
        //the window in theta to search for hits in the seed extension
        DoubleProperty m_windowTheta {this, "thetaWindow", 0.5 * Gaudi::Units::deg};
        
        //apply a cut threshold in the pulls during the hit extension
        DoubleProperty m_minPullThreshold{this, "maxPull", 5.};
        
        //minimum number of hits required to form a seed after extension
        DoubleProperty m_minSeedHits{this, "minSeedHits", 6.};
 
        //maximum number of MM Clusters that are invalid in the seed
        DoubleProperty m_maxInvalidClusters{this, "maxInvalidClusters", 4.};
 
        //reject also hits from the seed even if it does not lead to succesful segment
        BooleanProperty m_markHitsFromSeed{this, "markHitsFromSeed", true};
 
        //maximum number that hit is allowed to be used
        UnsignedIntegerProperty m_maxUsed{this, "maxHitIsUsed", 6};
 
        //minimum number of strips required for MMClusers not to be invalid
        DoubleProperty m_minClusSize{this, "minClusterSize", 1.};
 
        //maximum number of chi2 cut for the segment
        DoubleProperty m_maxChi2{this, "maxChi2", 6.};
 
        // maximum number of clusters in the layer for the seed finding
        DoubleProperty m_maxClustersInLayer{this, "maxClustersInLayer", 8};
 
        //maximum number of dY window size for killing hits on the layer from the segments 
        DoubleProperty m_maxdYWindow{this, "maxdYWindow", 4.*Gaudi::Units::cm};  
 
        //dump statistics for the seeds per sector
        BooleanProperty m_dumpSeedStatistics{this, "dumpStatistics", true};
 
        std::unique_ptr<SeedStatistics> m_seedCounter ATLAS_THREAD_SAFE{};
        
       
};
 
}
 
#endif