MuonIdHelperSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUON_MUONIDHELPERSVC_H
#define MUON_MUONIDHELPERSVC_H
 
#include "AthenaBaseComps/AthService.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
 
 
#include <unordered_set>
#include <array>
namespace Muon {
    class MuonIdHelperSvc : public extends<AthService, IMuonIdHelperSvc> {
    public:
        MuonIdHelperSvc(const std::string& name, ISvcLocator* svc);

        virtual ~MuonIdHelperSvc() override = default;

        virtual StatusCode initialize() override;

        virtual std::string toString(const Identifier& id) const override;

        virtual std::string toStringTech(const Identifier& id) const override;

        virtual std::string toStringStation(const Identifier& id) const override;

        virtual std::string toStringChamber(const Identifier& id) const override;

        virtual std::string toStringDetEl(const Identifier& id) const override;

        virtual std::string toStringGasGap(const Identifier& id) const override;

        virtual std::string chamberNameString(const Identifier& id) const override;

        virtual bool isMuon(const Identifier& id) const override;

        virtual bool isMdt(const Identifier& id) const override;

        virtual bool isMM(const Identifier& id) const override;

        virtual bool isCsc(const Identifier& id) const override;

        virtual bool isRpc(const Identifier& id) const override;

        virtual bool isTgc(const Identifier& id) const override;

        virtual bool issTgc(const Identifier& id) const override;

        virtual bool issMdt(const Identifier& id) const override;

        virtual bool hasHPTDC(const Identifier& id) const override;

        virtual bool measuresPhi(const Identifier& id) const override;

        virtual bool isEndcap(const Identifier& id) const override;

        virtual bool isTrigger(const Identifier& id) const override;

        virtual bool isSmallChamber(const Identifier& id) const override;

        virtual int gasGap(const Identifier& id) const override;

        virtual Identifier chamberId(const Identifier& id) const override;

        virtual Identifier detElId(const Identifier& id) const override;

        virtual Identifier gasGapId(const Identifier& id) const override;

        virtual Identifier layerId(const Identifier& id) const override;

        virtual const MdtIdHelper& mdtIdHelper() const override { return *m_mdtIdHelper; }

        virtual const RpcIdHelper& rpcIdHelper() const override { return *m_rpcIdHelper; }

        virtual const TgcIdHelper& tgcIdHelper() const override { return *m_tgcIdHelper; }

        virtual const CscIdHelper& cscIdHelper() const override { return *m_cscIdHelper; }

        virtual const sTgcIdHelper& stgcIdHelper() const override { return *m_stgcIdHelper; }

        virtual const MmIdHelper& mmIdHelper() const override { return *m_mmIdHelper; }

        virtual MuonStationIndex::ChIndex chamberIndex(const Identifier& id) const override;

        virtual MuonStationIndex::StIndex stationIndex(const Identifier& id) const override;

        virtual MuonStationIndex::PhiIndex phiIndex(const Identifier& id) const override;

        virtual MuonStationIndex::DetectorRegionIndex regionIndex(const Identifier& id) const override;

        virtual MuonStationIndex::LayerIndex layerIndex(const Identifier& id) const override;

        virtual MuonStationIndex::TechnologyIndex technologyIndex(const Identifier& id) const override;

        virtual int stationPhi(const Identifier& id) const override;

        virtual int stationEta(const Identifier& id) const override;

        virtual std::string stationNameString(const Identifier& id) const override;

        virtual int stationName(const Identifier& id) const override;

        virtual int stationRegion(const Identifier& id) const override;

        virtual int sector(const Identifier& id) const override;
        virtual const std::set<MuonStationIndex::TechnologyIndex>& 
                    technologiesInStation(MuonStationIndex::StIndex stIndex) const override;
 
        bool hasRPC() const override;
        bool hasTGC() const override;
        bool hasMDT() const override;
 
        bool hasCSC() const override;
        bool hasSTGC() const override;
        bool hasMM() const override;

        virtual IdentifierHash moduleHash(const Identifier& id) const override;
        virtual IdentifierHash detElementHash(const Identifier& id) const override;
 
        
    private:
        IdentifierHash moduleHash(const MuonIdHelper& idHelper, const Identifier& id) const;
        IdentifierHash detElementHash(const MuonIdHelper& idHelper, const Identifier& id) const;
        
        const RpcIdHelper* m_rpcIdHelper{nullptr};
        const TgcIdHelper* m_tgcIdHelper{nullptr};
        const CscIdHelper* m_cscIdHelper{nullptr};
        const MdtIdHelper* m_mdtIdHelper{nullptr};
        const MmIdHelper* m_mmIdHelper{nullptr};
        const sTgcIdHelper* m_stgcIdHelper{nullptr};
 
        const MuonIdHelper* m_primaryHelper{nullptr};
 
        Gaudi::Property<bool> m_hasMDT{this, "HasMDT", true, "Specify whether the current detector layour contains MDT chambers"};
        Gaudi::Property<bool> m_hasRPC{this, "HasRPC", true, "Specify whether the current detector layour contains RPC chambers"};
        Gaudi::Property<bool> m_hasTGC{this, "HasTGC", true, "Specify whether the current detector layour contains TGC chambers"};        
        Gaudi::Property<bool> m_hasCSC{this, "HasCSC", true, "Specify whether current detector layout contains CSC chambers"};
        Gaudi::Property<bool> m_hasSTGC{this, "HasSTGC", true, "Specify whether current detector layout contains STGC chambers"};
        Gaudi::Property<bool> m_hasMM{this, "HasMM", true, "Specify whether current detector layout contains MicroMegas chambers"};
 
        ServiceHandle<StoreGateSvc> m_detStore{this, "DetectorStore", "DetectorStore"};
 
        using TechIdx = MuonStationIndex::TechnologyIndex;
        using ChIdx = MuonStationIndex::ChIndex;
        using StIdx = MuonStationIndex::StIndex;
        using PhiIdx = MuonStationIndex::PhiIndex;
        struct StationNameData {
            std::string stationName{};
            bool isEndcap{false};
            bool isSmall{false};
            ChIdx chIndex{ChIdx::ChUnknown};
            StIdx stIndex{StIdx::StUnknown};           
        };
        std::vector<StationNameData> m_stationNameData{};
        std::vector<TechIdx> m_technologies{};
 
        int m_BIS_stat{-1};
        std::unordered_set<Identifier> m_smdt_stat{};
        std::array<std::set<TechIdx>, static_cast<int>(StIdx::StIndexMax)> m_techPerStation{};
    };
 
}  // namespace Muon
 
#endif  // MUON_MUONIDHELPERSVC_H