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 "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(int sector) { m_sector1 = sector; }
    void setSecondarySector(int sector) { m_sector2 = sector; }
    void setNPrecisionHits(unsigned n) { m_nPrecisionHits = n; }
    void setNEtaNonPrecisionHits(unsigned n) { m_nEtaNonPrecisionHits = n; }
    void setNPhiHits(unsigned n) { m_nPhiHits = n; }
    void setTotalResidual(double res) { m_totalResidual = res; }
    

    double theta() const { return m_theta; }
    double phi() const { return m_phi; }
    int sector() const { return m_sector1; }
    bool isSectorOverlap() const { return m_sector2 != m_sector1; }
    int secondarySector() const { return m_sector2; }
    double sectorPhi() const;
    std::vector<StIndex> getStations() const;
    const std::vector<HitType>& hitsInStation(StIndex station) const;
    unsigned nPrecisionHits() const { return m_nPrecisionHits; }
    unsigned nEtaNonPrecisionHits() const { return m_nEtaNonPrecisionHits; }
    unsigned nPhiHits() const { return m_nPhiHits; }
    double totalResidual() const { return m_totalResidual; }
    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_nPrecisionHits{0};
    unsigned m_nEtaNonPrecisionHits{0};
    unsigned m_nPhiHits{0};
    double m_totalResidual{0.};
 
    // The pattern can extend over two sectors in the overlap region
    int m_sector1{-1};
    int m_sector2{-1};

    const HitCollection m_hitsInStation{};
};

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