GlobalPattern.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONR4_MUONFASTRECOEVENT_GLOBALPATTERN__H
#define MUONR4_MUONFASTRECOEVENT_GLOBALPATTERN__H
 
#include "MuonSpacePoint/SpacePoint.h"
#include "MuonStationIndex/MuonStationIndex.h"
#include "MuonTrackEvent/ExpandedSector.h"
#include "AthContainers/DataVector.h"
 
namespace MuonR4 {
class GlobalPattern {
   public:
    using HitType = const SpacePoint*;
    using StIndex = Muon::MuonStationIndex::StIndex;
    using HitCollection = std::unordered_map<StIndex, std::vector<HitType>>;
    
    GlobalPattern(HitCollection&& hitPerStation);
    GlobalPattern() = delete;
    GlobalPattern(const GlobalPattern& other) = default;

    void setTheta(double theta) { m_theta = theta; }
    void setPhi(double phi) { m_phi = phi; }
    void setSector(std::int8_t sector) { m_sector = ExpandedSector{sector}; }
    void setNPrecisionLayers(unsigned n) { m_nPrecisionLayers = n; }
    void setNTriggerLayers(unsigned n) { m_nTriggerLayers = n; }
    void setNPhiLayers(unsigned n) { m_nPhiLayers = n; }
    void setMeanNormResidual2(double res) { m_meanNormResidual2 = res; }
    

    double theta() const { return m_theta; }
    double phi() const { return m_phi; }
    unsigned sector() const { return m_sector.msSector(); }
    unsigned secondarySector() const { return m_sector.adjacentMsSector(); }
    bool isSectorOverlap() const { return sector() != secondarySector(); }
    double sectorPhi() const;
    std::vector<StIndex> getStations() const;
    const std::vector<HitType>& hitsInStation(StIndex station) const;
    unsigned nPrecisionLayers() const { return m_nPrecisionLayers; }
    unsigned nTriggerLayers() const { return m_nTriggerLayers; }
    unsigned nPhiLayers() const { return m_nPhiLayers; }
    double meanNormResidual2() const { return m_meanNormResidual2; }
    const HitCollection& hitsPerStation() const { return m_hitsInStation; }

    friend std::ostream& operator<<(std::ostream& ostr, const GlobalPattern& gp) {
        gp.print(ostr);
        return ostr;
    }

    bool operator==(const GlobalPattern& other) const {
        return m_hitsInStation == other.m_hitsInStation;
    }
 
   private:
    void print(std::ostream& ostr) const;

    double m_theta{0.};
    double m_phi{0.};
    unsigned m_nPrecisionLayers{0};
    unsigned m_nTriggerLayers{0};
    unsigned m_nPhiLayers{0};
    double m_meanNormResidual2{0.};
 
    // The pattern can extend over two sectors in the overlap region
    ExpandedSector m_sector{static_cast<int8_t>(0)};

    const HitCollection m_hitsInStation{};
};

using GlobalPatternContainer = DataVector<GlobalPattern>;
}  // namespace MuonR4
 
#include "AthenaKernel/CLASS_DEF.h"
CLASS_DEF( MuonR4::GlobalPatternContainer , 1197530715 , 1 )
 
#endif