MuonPhaseII/MuonDetDescr/MuonReadoutGeometryR4/MuonReadoutGeometryR4/MuonDetectorManager.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONREADOUTGEOMETRY_MUONDETECTORMANAGER_H
#define MUONREADOUTGEOMETRY_MUONDETECTORMANAGER_H
 
#include "MuonReadoutGeometryR4/MuonDetectorDefs.h"
#include "MuonReadoutGeometryR4/MuonReadoutElement.h"
#include "AthenaKernel/CLASS_DEF.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GeoModelKernel/GeoVDetectorManager.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
 
#include <map>
#include <memory>


#define DECLARE_GETTERSETTER(ELE_TYPE, GETTER, SETTER)        \
    ELE_TYPE* GETTER(const IdentifierHash& hash);             \
    ELE_TYPE* GETTER(const Identifier& hash);                 \
                                                              \
    const ELE_TYPE* GETTER(const IdentifierHash& hash) const; \
    const ELE_TYPE* GETTER(const Identifier& hash) const;     \
                                                              \
    StatusCode SETTER(ElementPtr_t<ELE_TYPE> element);
 
#define DECLARE_ELEMENT(ELE_TYPE) \
    DECLARE_GETTERSETTER(ELE_TYPE, get##ELE_TYPE, add##ELE_TYPE)   \
                                                                   \
    std::vector<const ELE_TYPE*> getAll##ELE_TYPE##s() const;      \
    std::vector<ELE_TYPE*>       getAll##ELE_TYPE##s();     
namespace MuonGMR4 {
 
class MdtReadoutElement;
class TgcReadoutElement;
class RpcReadoutElement;
class sTgcReadoutElement;
class MmReadoutElement;
class SpectrometerSector;
class Chamber;
 
class MuonDetectorManager : public GeoVDetectorManager, public AthMessaging {
 
   public:
    MuonDetectorManager();
    ~MuonDetectorManager();
 
    using ChIndex = Muon::MuonStationIndex::ChIndex;
    template <class MuonDetectorType> using ElementPtr_t = std::unique_ptr<MuonDetectorType>;
    template <class MuonDetectorType> using ElementStorage_t = std::vector<ElementPtr_t<MuonDetectorType>>;
    
    DECLARE_ELEMENT(MdtReadoutElement)
    DECLARE_ELEMENT(TgcReadoutElement)    
    DECLARE_ELEMENT(RpcReadoutElement)
    DECLARE_ELEMENT(sTgcReadoutElement)
    DECLARE_ELEMENT(MmReadoutElement)
    
    
    unsigned int getNumTreeTops() const override final; 
    PVConstLink getTreeTop(unsigned int i) const override final;
    void addTreeTop(PVConstLink pv);
    void clearTreeTops();
    const Muon::IMuonIdHelperSvc* idHelperSvc() const;
    
    std::vector<const MuonReadoutElement*> getAllReadoutElements() const;
    std::vector<MuonReadoutElement*> getAllReadoutElements();
    const MuonReadoutElement* getReadoutElement(const Identifier& id) const;
    MuonReadoutElement* getReadoutElement(const Identifier& id);
 
#ifndef SIMULATIONBASE
    void addSpectrometerSector(ElementPtr_t<SpectrometerSector>&& chSector);
    const SpectrometerSector* getSectorEnvelope(const Identifier& channelId) const;
    const SpectrometerSector* getSectorEnvelope(const Muon::MuonStationIndex::ChIndex chIdx,
                                                const unsigned sector,
                                                const int side) const;
    const Chamber* getChamber(const Identifier& channelId) const;
    
    struct MSEnvelopeSorter{
        bool operator()(const SpectrometerSector* a, const SpectrometerSector* b) const;
        bool operator()(const Chamber* a, const Chamber* b) const;
    };
    using MuonSectorSet = std::vector<const SpectrometerSector*>;
    using MuonChamberSet = std::vector<const Chamber*>;
    MuonSectorSet getAllSectors() const;
    MuonChamberSet getAllChambers() const;
#endif

    std::vector<ActsTrk::DetectorType> getDetectorTypes() const;
   private:
    template <class MuonDetectorType> void linkElements(ElementStorage_t<MuonDetectorType>& allStore,
                                                        MuonDetectorType* readOutEle);
 
    ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{"Muon::MuonIdHelperSvc/MuonIdHelperSvc", 
                                                        "MuonDetectorManager"};
#ifndef SIMULATIONBASE
    ElementStorage_t<SpectrometerSector> m_secEnvelopes{};
    using EnvelopeMap_t = std::unordered_map<unsigned, const SpectrometerSector*>;
    EnvelopeMap_t m_envelopesById{};
#endif
    ElementStorage_t<MdtReadoutElement> m_mdtEles{};
    ElementStorage_t<TgcReadoutElement> m_tgcEles{};    
    ElementStorage_t<RpcReadoutElement> m_rpcEles{};
    ElementStorage_t<sTgcReadoutElement> m_sTgcEles{};
    ElementStorage_t<MmReadoutElement> m_mmEles{};
 
    std::vector<PVConstLink> m_treeTopVector{};
 
 
};
 
template <> void MuonDetectorManager::linkElements(ElementStorage_t<MdtReadoutElement>& detStore, MdtReadoutElement* refEle);
 
}  // namespace MuonGMR4
 
CLASS_DEF(MuonGMR4::MuonDetectorManager, 248531088, 1)
#undef DECLARE_GETTERSETTER
#undef DECLARE_ELEMENT
#include <MuonReadoutGeometryR4/MuonDetectorManager.icc>
#endif