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<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();
 
    template <class MuonDetectorType>
    using ElementPtr = std::unique_ptr<MuonDetectorType>;
    template <class MuonDetectorType>
    using ElementStorage = std::vector<ElementPtr<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);
    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<SpectrometerSector>&& chSector);
    const SpectrometerSector* getSectorEnvelope(const Identifier& channelId) 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::set<const SpectrometerSector*, MSEnvelopeSorter>;
    using MuonChamberSet = std::set<const Chamber*, MSEnvelopeSorter>;
    MuonSectorSet getAllSectors() const;
    MuonChamberSet getAllChambers() const;
#endif
 
    std::vector<ActsTrk::DetectorType> getDetectorTypes() const;
   private:
    template <class MuonDetectorType> void linkElements(ElementStorage<MuonDetectorType>& allStore,
                                                        MuonDetectorType* readOutEle);
 
    ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{"Muon::MuonIdHelperSvc/MuonIdHelperSvc", 
                                                        "MuonDetectorManager"};
#ifndef SIMULATIONBASE
    ElementStorage<SpectrometerSector> m_secEnvelopes{};
#endif
    ElementStorage<MdtReadoutElement> m_mdtEles{};
    ElementStorage<TgcReadoutElement> m_tgcEles{};    
    ElementStorage<RpcReadoutElement> m_rpcEles{};
    ElementStorage<sTgcReadoutElement> m_sTgcEles{};
    ElementStorage<MmReadoutElement> m_mmEles{};
 
    std::vector<PVConstLink> m_treeTopVector{};
 
 
};
 
template <> void MuonDetectorManager::linkElements(ElementStorage<MdtReadoutElement>& detStore, MdtReadoutElement* refEle);
 
}  // namespace MuonGMR4
 
CLASS_DEF(MuonGMR4::MuonDetectorManager, 248531088, 1)
#undef DECLARE_GETTERSETTER
#undef DECLARE_ELEMENT
#include <MuonReadoutGeometryR4/MuonDetectorManager.icc>
#endif