TRT_DetectorManager.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRT_READOUTGEOMETRY_TRT_DETECTORMANAGER_H
#define TRT_READOUTGEOMETRY_TRT_DETECTORMANAGER_H

 
#include "TRT_DetElementContainer.h"
 
#include "InDetReadoutGeometry/InDetDetectorManager.h"
#include "TRT_ReadoutGeometry/TRT_Conditions.h"
#include "TRT_ReadoutGeometry/TRT_BarrelElement.h"
#include "TRT_ReadoutGeometry/TRT_EndcapElement.h"
#include "ReadoutGeometryBase/InDetDD_Defs.h"
#include "TRT_ReadoutGeometry/TRT_DetElementCollection.h"
#include "GeoPrimitives/GeoPrimitives.h"
#include "GeoModelKernel/GeoVPhysVol.h"
#include "InDetIdentifier/TRT_ID.h"
#include "GeoModelKernel/GeoXF.h"
 
#include "CxxUtils/checker_macros.h"
 
#include <map>
#include <set>
#include <string>
#include <vector>
 
class Identifier;
class GeoAlignableTransform;
class GeoVAlignmentStore;
class StoreGateSvc;
class CondAttrListCollection;
 
namespace InDetDD {
 
  class ExtendedAlignableTransform;
  class TRT_BarrelDescriptor;
  class TRT_EndcapDescriptor;
  class TRT_Numerology;

 
  class TRT_DetectorManager
    : public InDetDetectorManager,
      public TRT_Conditions
  {
  public:
 
    // Constructor
    TRT_DetectorManager(StoreGateSvc * detStore);
 
    // Destructor
    ~TRT_DetectorManager();

    virtual unsigned int getNumTreeTops()           const override;                //
    virtual PVConstLink  getTreeTop(unsigned int i) const override;                //
    //-----------------------------------------------------------------------------//

    void setIdHelper(const TRT_ID * idHelper, bool owns=true);                     //
    //-----------------------------------------------------------------------------//
 
 
    //** Get and set information about gasType. -----------------------------------*/
    enum ActiveGasType { unknown, oldgas, newgas };                                //
    ActiveGasType gasType() const;                                                 //
    void setGasType(const ActiveGasType &);                                        //
    //-----------------------------------------------------------------------------//

    TRT_Numerology *getNumerology();                                               //
    const TRT_Numerology * getNumerology() const;                                  //
    //-----------------------------------------------------------------------------//
 

    const TRT_BaseElement *getElement(Identifier id) const;       // Slow          //
    const TRT_BaseElement *getElement(IdentifierHash id) const;   // Fast          //
    //-----------------------------------------------------------------------------//

    const TRT_DetElementContainer* getDetectorElementContainer() const;

    const TRT_DetElementCollection* getDetectorElementCollection() const;
    TRT_DetElementCollection::const_iterator getDetectorElementBegin() const;
    TRT_DetElementCollection::const_iterator getDetectorElementEnd() const;
    TRT_DetElementCollection::iterator getDetectorElementBegin();
    TRT_DetElementCollection::iterator getDetectorElementEnd();
 
 
    //-----------------------------------------------------------------------------//
 

    const TRT_BarrelElement *getBarrelElement(unsigned int positive                //
                          , unsigned int moduleIndex           //
                          , unsigned int phiIndex              //
                          , unsigned int strawLayerIndex) const;
 
    TRT_BarrelElement *getBarrelElement(unsigned int positive                      //
                    , unsigned int moduleIndex                 //
                    , unsigned int phiIndex                    //
                    , unsigned int strawLayerIndex);           //
    //                                                                             //
    //-----------------------------------------------------------------------------//
 

    const TRT_EndcapElement *getEndcapElement(unsigned int positive                 //
                          , unsigned int wheelIndex             //
                          , unsigned int strawLayerIndex        //
                          , unsigned int phiIndex) const;       //
    TRT_EndcapElement *getEndcapElement(unsigned int positive                       //
                    , unsigned int wheelIndex                   //
                    , unsigned int strawLayerIndex              //
                    , unsigned int phiIndex);                   //
    //                                                                              //
    //------------------------------------------------------------------------------//

    const TRT_Conditions * conditions() const;                                     //
    //-----------------------------------------------------------------------------//
 
 
    // This is for Detector Descriptors -------------------------------------------//
    void addTreeTop(const PVLink&);                                                //
    //                                                                             //
    // Manage the barrel elements:                                                 //
    void manageBarrelElement(TRT_BarrelElement *barrel);                           //
    //                                                                             //
    // Manage the endcap elements:                                                 //
    void manageEndcapElement(TRT_EndcapElement *endcap);                           //
    //                                                                             //
    //-----------------------------------------------------------------------------//
 
    //-----------------------------------------------------------------------------//
    //                                                                             //
    //  Transform of straw relative to module.  Stored as a tranformation field    //
    //  to make this compact.  Note , GeoXF::Function * is a pointer to a so-      //
    //  called "TRANSFUNCTION".  Functions are delete when the manager is deleted. //
    //                                                                             //
    void setBarrelTransformField(size_t i, const GeoXF::Function *field);          //
    const GeoXF::Function *barrelTransformField(size_t i) const;                   //
    void setEndcapTransformField(size_t i, const GeoXF::Function *field);          //
    const GeoXF::Function *endcapTransformField(size_t i) const;                   //
    //                                                                             //
    //-----------------------------------------------------------------------------//

    void addAlignableTransform (int level,
                const Identifier &id,
                GeoAlignableTransform *transform,
                const GeoVFullPhysVol * child = 0,
                const GeoVFullPhysVol * frameVol = 0);

    void addAlignableTransform (int level,
                const Identifier &id,
                GeoAlignableTransform *transform,
                const GeoVPhysVol * child = 0,
                const GeoVPhysVol * frameVol = 0);

    virtual bool setAlignableTransformDelta(int level,
                                            const Identifier & id,
                                            const Amg::Transform3D & delta,
                                            FrameType frame,
                                            GeoVAlignmentStore* alignStore) const override;

    bool setAlignableTransformAnyFrameDelta(ExtendedAlignableTransform * extXF,
                                            const Amg::Transform3D & delta,
                                            FrameType frame,
                                            GeoVAlignmentStore* alignStore) const;
 

    virtual void invalidateAll() override;

    virtual void updateAll() const override;
 

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

    StatusCode alignmentCallback();

    virtual
    bool processGlobalAlignment(const std::string &, int level, FrameType frame,
                                const CondAttrListCollection* obj,
                                GeoVAlignmentStore* alignStore) const override;
 
    bool processSpecialAlignment(const std::string & key, InDetDD::AlignFolderType dummy) override;
 
    bool processSpecialAlignment(const std::string& key,
                                 const CondAttrListCollection* obj=nullptr,
                                 GeoVAlignmentStore* alignStore=nullptr) const override;

    void setBarrelDescriptor(const TRT_BarrelDescriptor* barrelDescriptor);
    void setEndcapDescriptor(const TRT_EndcapDescriptor* endcapDescriptor);
 
 private:
 
 
    // Illegal operations:---------------------------------------------------------//
    //                                                                             //
    const TRT_DetectorManager & operator=(const TRT_DetectorManager &right);       //
    TRT_DetectorManager(const TRT_DetectorManager &right);                         //
    //                                                                             //
    //-----------------------------------------------------------------------------//
 
 
    // Private member data:--------------------------------------------------------//
    std::vector<PVLink> m_volume;                                                  //
    //                                                                             //
    TRT_DetElementContainer m_elementContainer;                                    //
    //                                                                             //
    TRT_Numerology  *m_numerology;                                                 //
    const TRT_ID    *m_idHelper;                                                   //
    bool             m_ownsIdHelper;                                               //
    const GeoXF::Function *m_barrelXF[3]{};                                        //
    const GeoXF::Function *m_endcapXF[3]{};                                        //
    //                                                                             //
    ActiveGasType m_gasType;                                                       //
    unsigned int m_digvers;                                                        //
    std::string m_digversname;                                                     //
 
    // Alignment stuff
    typedef std::map<Identifier, ExtendedAlignableTransform *> AlignableTransformMap;
    std::vector< AlignableTransformMap > m_alignableTransforms;
 
    // Descriptors are owned by TRT_DetectorManager.
    std::set<const TRT_BarrelDescriptor*> m_barrelDescriptors;
    std::set<const TRT_EndcapDescriptor*> m_endcapDescriptors;
 
    // here temporarily
    virtual const TRT_ID *getIdHelper() const override;                                    //
 
    //-----------------------------------------------------------------------------//
  };
}
#ifndef GAUDI_NEUTRAL
#include "AthenaKernel/CLASS_DEF.h"
CLASS_DEF(InDetDD::TRT_DetectorManager, 10907524, 1)
#endif
#endif