GeometryBuilderCond.cxx

Go to the documentation of this file.

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

// GeometryBuilderCond.cxx, (c) ATLAS Detector software
 
// Trk include
#include "TrkDetDescrTools/GeometryBuilderCond.h"
#include "TrkVolumes/CylinderVolumeBounds.h"
#include "TrkGeometry/TrackingVolume.h"
#include "TrkGeometry/TrackingGeometry.h"
#include "TrkGeometry/GlueVolumesDescriptor.h"
 
#ifdef TRKDETDESCR_MEMUSAGE
#include <unistd.h>
#endif
 
// Amg
#include "GeoPrimitives/GeoPrimitives.h"
// STD
#include <map>
//Athena
#include "AthenaKernel/IOVInfiniteRange.h"
// Gaudi
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/SystemOfUnits.h"
 
 
// constructor
Trk::GeometryBuilderCond::GeometryBuilderCond(const std::string& t, const std::string& n, const IInterface* p)
: AthAlgTool(t,n,p),
  TrackingVolumeManipulator()
{
    declareInterface<IGeometryBuilderCond>(this);
}
 
 
// Athena standard methods
// initialize
StatusCode Trk::GeometryBuilderCond::initialize()
{
 
    // Retrieve the volume array creator  ----------------------------------------------------
    ATH_CHECK(m_trackingVolumeArrayCreator.retrieve());
 
    // Retrieve the tracking volume helper  --------------------------------------------------
    ATH_CHECK (m_trackingVolumeHelper.retrieve());
    // Geometries =============================================================================
    // (I) Inner Detector ---------------------------------------------------------------------
    if (!m_inDetGeometryBuilderCond.empty()) {
        ATH_CHECK(m_inDetGeometryBuilderCond.retrieve());
    }
    // (H) High Granularity Timing Detector ----------------------------------------------------
    if(!m_hgtdGeometryBuilderCond.empty()) {
        if (m_hgtdGeometryBuilderCond.retrieve().isFailure()) {
            ATH_MSG_FATAL("Failed to retrieve tool " << m_hgtdGeometryBuilderCond );
            return StatusCode::FAILURE;
        } else
            ATH_MSG_INFO( "Retrieved tool " << m_hgtdGeometryBuilderCond );
    }
    // (C) Calorimeter --------------------------------------------------------------------------
    if (!m_caloGeometryBuilderCond.empty()) {
        ATH_CHECK (m_caloGeometryBuilderCond.retrieve());
    }
    // (M) Muon System -------------------------------------------------------------------------
    if (!m_muonGeometryBuilderCond.empty()) {
        ATH_CHECK(m_muonGeometryBuilderCond.retrieve());
    }
 
    // if no world dimensions are declared, take default ones
    if (m_worldDimension.empty())
        m_worldDimension = std::vector<double>{0.*Gaudi::Units::meter, 10.*Gaudi::Units::meter, 15.*Gaudi::Units::meter};
 
    // if no world materials are declared, take default ones - set vacuum
    if (m_worldMaterialProperties.size() < 5)
        m_worldMaterialProperties = std::vector<double>{10.e10,10.e10,1e-10, 0., 0.};
 
    m_worldMaterial = Trk::Material(m_worldMaterialProperties[0],
                    m_worldMaterialProperties[1],
                    m_worldMaterialProperties[2],
                    m_worldMaterialProperties[3],
                    m_worldMaterialProperties[4]);
 
    ATH_MSG_DEBUG( " initialize() successful" );
 
    return StatusCode::SUCCESS;
}
 
std::unique_ptr< Trk::TrackingGeometry>
Trk::GeometryBuilderCond::trackingGeometry(const EventContext& ctx,
                                           Trk::TrackingVolume* /*tVol*/,
                                           SG::WriteCondHandle<TrackingGeometry>& whandle) const
{
 
    // the geometry to be constructed
    std::unique_ptr<Trk::TrackingGeometry> tGeometry;
    if ( m_inDetGeometryBuilderCond.empty() && m_hgtdGeometryBuilderCond.empty() && m_caloGeometryBuilderCond.empty() && m_muonGeometryBuilderCond.empty() ) {
 
        ATH_MSG_VERBOSE( "Configured to only create world TrackingVolume." );
 
        auto worldBounds = std::make_shared<Trk::CylinderVolumeBounds>(m_worldDimension[0],
                                                                       m_worldDimension[1],
                                                                       m_worldDimension[2]);
 
        Trk::TrackingVolume* worldVolume = new Trk::TrackingVolume(nullptr,
                                                                   worldBounds,
                                                                   m_worldMaterial,
                                                                   nullptr,
                                                                   nullptr,
                                                                   "EmptyWorldVolume");
        // create a new geometry
        tGeometry = std::make_unique<Trk::TrackingGeometry>(worldVolume);
    } else
        tGeometry = atlasTrackingGeometry(ctx, whandle);
    // sign it before you return anything
    tGeometry->sign(geometrySignature());
    return tGeometry;
}
 
 
std::unique_ptr<Trk::TrackingGeometry>
Trk::GeometryBuilderCond::atlasTrackingGeometry(const EventContext& ctx,
                                                SG::WriteCondHandle<TrackingGeometry>& whandle) const
{
    // the return geometry
    std::unique_ptr<Trk::TrackingGeometry> atlasTrackingGeometry;
 
    // A ------------- INNER DETECTOR SECTION --------------------------------------------------------------------------------
    // get the Inner Detector and/or HGTD and/or Calorimeter trackingGeometry
 
    // the volumes to be given to higher level tracking geometry builders
    Trk::TrackingVolume* inDetVolume    = nullptr;
    Trk::TrackingVolume* hgtdVolume     = nullptr;
    Trk::TrackingVolume* caloVolume     = nullptr;
 
    // mark the highest volume
    Trk::TrackingVolume* highestVolume  = nullptr;
 
#ifdef TRKDETDESCR_MEMUSAGE
    m_memoryLogger.refresh(getpid());
    ATH_MSG_INFO( "[ memory usage ] Start of TrackingGeometry building: "  );
    ATH_MSG_INFO( m_memoryLogger );
#endif
 
    // ========================== INNER DETECTOR PART =================================================
    if (!m_inDetGeometryBuilderCond.empty()) {
        // debug output
        ATH_MSG_VERBOSE( "ID Tracking Geometry is going to be built." );
        // build the geometry
        std::unique_ptr<Trk::TrackingGeometry> inDetTrackingGeometry =
          m_inDetGeometryBuilderCond->trackingGeometry(ctx, nullptr, whandle);
        // check
        if (inDetTrackingGeometry) {
            // sign it
            inDetTrackingGeometry->sign(m_inDetGeometryBuilderCond->geometrySignature());
            // check whether the world has to be created or not
            if (m_createWorld) {
                // checkout the highest InDet volume
                inDetVolume = inDetTrackingGeometry->checkoutHighestTrackingVolume();
                // assign it as the highest volume
                highestVolume = inDetVolume;
            } else // -> Take the exit and return ID stand alone
                atlasTrackingGeometry = std::move(inDetTrackingGeometry);
        }
 
#ifdef TRKDETDESCR_MEMUSAGE
        m_memoryLogger.refresh(getpid());
        ATH_MSG_INFO( "[ memory usage ] After InDet TrackingGeometry building: "  );
        ATH_MSG_INFO( m_memoryLogger );
#endif
 
    }
 
    // ========================== HGTD PART =================================================
    // if a HGTD Geometry Builder is present -> wrap it around the ID
    if (!m_hgtdGeometryBuilderCond.empty()) {
        if (inDetVolume)
            ATH_MSG_VERBOSE( "HGTD Tracking Geometry is going to be built with enclosed ID." );
        else
            ATH_MSG_VERBOSE( "HGTD Tracking Geometry is going to be built stand-alone." );
        // get the InnerDetector TrackingGeometry
        std::unique_ptr<Trk::TrackingGeometry> hgtdTrackingGeometry =
          m_hgtdGeometryBuilderCond->trackingGeometry(ctx, inDetVolume, whandle);
        // if you have to create world or there is a Calo/Muon geometry builder ...
        if (hgtdTrackingGeometry) {
            // sign it
            hgtdTrackingGeometry->sign(m_hgtdGeometryBuilderCond->geometrySignature());
            if (m_createWorld){
                // check out the highest Calo volume
                hgtdVolume = hgtdTrackingGeometry->checkoutHighestTrackingVolume();
                // assign it as the highest volume (overwrite ID)
                highestVolume = hgtdVolume;
            } else // -> Take the exit and return HGTD back
                atlasTrackingGeometry = std::move(hgtdTrackingGeometry);
        }
 
#ifdef TRKDETDESCR_MEMUSAGE
        m_memoryLogger.refresh(getpid());
        ATH_MSG_INFO( "[ memory usage ] After Calo TrackingGeometry building: "  );
        ATH_MSG_INFO( m_memoryLogger );
#endif
 
    }
 
    // ========================== CALORIMETER PART =================================================
    // if a Calo Geometry Builder is present -> wrap it around the ID or HGTD
    if (!m_caloGeometryBuilderCond.empty()) {
        std::string enclosed = "stand-alone.";
        if (inDetVolume and hgtdVolume)
            enclosed = "with encloded ID/HGTD.";
        else if (inDetVolume or hgtdVolume)
            enclosed = (inDetVolume) ? "with encloded ID." : "with encloded HGTD.";
        ATH_MSG_VERBOSE( "Calorimeter Tracking Geometry is going to be built "<< enclosed );
 
        // get the InnerDetector TrackingGeometry or the HGTD tracking geometry
        std::unique_ptr<Trk::TrackingGeometry> caloTrackingGeometry;
        if (inDetVolume and not hgtdVolume)
          caloTrackingGeometry = m_caloGeometryBuilderCond->trackingGeometry(ctx, inDetVolume, whandle);
        else
          caloTrackingGeometry = m_caloGeometryBuilderCond->trackingGeometry(ctx, hgtdVolume, whandle);
        // if you have to create world or there is a Muon geometry builder ...
        if (caloTrackingGeometry) {
            // sign it
            caloTrackingGeometry->sign(m_caloGeometryBuilderCond->geometrySignature());
            if (m_createWorld){
                // check out the highest Calo volume
                caloVolume = caloTrackingGeometry->checkoutHighestTrackingVolume();
                // assign it as the highest volume (overwrite ID)
                highestVolume = caloVolume;
            } else // -> Take the exit and return Calo back
                atlasTrackingGeometry = std::move(caloTrackingGeometry);
        }
 
#ifdef TRKDETDESCR_MEMUSAGE
        m_memoryLogger.refresh(getpid());
        ATH_MSG_INFO( "[ memory usage ] After Calo TrackingGeometry building: "  );
        ATH_MSG_INFO( m_memoryLogger );
#endif
 
    }
 
    // ========================== MUON SYSTEM PART =================================================
    // if Muon Geometry Builder is present -> wrap either ID or Calo
    if (!m_muonGeometryBuilderCond.empty()) {
        std::string enclosed = "stand-alone.";
        if (inDetVolume and hgtdVolume and caloVolume )
            enclosed = "with encloded ID/HGTD/Calo.";
        else if (inDetVolume or hgtdVolume or caloVolume) {
            if (inDetVolume) {
              if (hgtdVolume)
                enclosed = "with encloded ID/HGTD";
              else if (caloVolume)
                enclosed = "with encloded ID/Calo";
              else
                enclosed = "with encloded ID";
            } else if (hgtdVolume) {
              if (caloVolume)
                enclosed = "with encloded HGTD/Calo";
              else
                enclosed = "with encloded HGTD";
            } else {
              enclosed = "with encloded Calo";
            }
        }
        ATH_MSG_VERBOSE( "Muon System Tracking Geometry is going to be built "<< enclosed );
 
        // there's nothing outside the muons -- wrap the calo or the HGTD if one or both of them exist
        if (inDetVolume and not hgtdVolume and not caloVolume)
            atlasTrackingGeometry = m_muonGeometryBuilderCond->trackingGeometry(ctx, inDetVolume, whandle);
        else if (hgtdVolume and not caloVolume)
            atlasTrackingGeometry = m_muonGeometryBuilderCond->trackingGeometry(ctx, hgtdVolume, whandle);
        else
            atlasTrackingGeometry = m_muonGeometryBuilderCond->trackingGeometry(ctx, caloVolume, whandle);
 
        // sign it
        if (atlasTrackingGeometry)
            atlasTrackingGeometry->sign(m_muonGeometryBuilderCond->geometrySignature());
 
#ifdef TRKDETDESCR_MEMUSAGE
        m_memoryLogger.refresh(getpid());
        ATH_MSG_INFO( "[ memory usage ] After Muon TrackingGeometry building: "  );
        ATH_MSG_INFO( m_memoryLogger );
#endif
 
        // ========================== WRAPPING SECTION FOR ID/CALO ====================================
    } else if (m_createWorld && highestVolume) {
        // wrapping and world creation has been switched on
        ATH_MSG_VERBOSE( "Enclosing world is going to be built for: " << highestVolume->volumeName() );
        // get the glue volumes
        Trk::GlueVolumesDescriptor& innerGlueVolumes = highestVolume->glueVolumesDescriptor();
        // some screen output
        ATH_MSG_VERBOSE( "Retrieved with following glue volumes: " << innerGlueVolumes );
        // at negative face
        const std::vector<Trk::TrackingVolume*>& innerNegativeFaceVolumes = innerGlueVolumes.glueVolumes(Trk::negativeFaceXY);
        // at cylinder cover
        const std::vector<Trk::TrackingVolume*>& innerCentralFaceVolumes = innerGlueVolumes.glueVolumes(Trk::cylinderCover);
        // at positive face
        const std::vector<Trk::TrackingVolume*>& innerPositiveFaceVolumes = innerGlueVolumes.glueVolumes(Trk::positiveFaceXY);
 
        // get the dimensions
        // cast them to CylinderVolumeBounds
        const Trk::CylinderVolumeBounds* innerVolumeBounds = dynamic_cast<const Trk::CylinderVolumeBounds*>(&(highestVolume->volumeBounds()));
        if (!innerVolumeBounds) ATH_MSG_WARNING( "atlasTrackingGeometry() ... dynamic cast to innerVolumeBounds failed!" );
        double innerVolumeOuterR      = innerVolumeBounds ? innerVolumeBounds->outerRadius() : 0;
        double innerVolumeHalflengthZ = innerVolumeBounds ? innerVolumeBounds->halflengthZ() : 0;
        // Hierarchy after enclosing
        //
        // AtlasWorldVolume:
        //    AtlasInnerCylinder
        //       AtlasInnerNegativeSector
        //       InnerEnclosedVolume (can be ID/Calo)
        //       AtlasOuterNegativeSector
        //    AtlasOuterTube
        // B -------------- BUILD WORLD AROUND for ID stand alone applications
 
        double innerCylinderSectorHalflengthZ = 0.5*(m_worldDimension[2] - innerVolumeHalflengthZ);
        auto innerCylinderSectorBounds = std::make_shared<Trk::CylinderVolumeBounds>(0., innerVolumeOuterR, innerCylinderSectorHalflengthZ);
 
        double innerCylinderSectorPositionZ = fabs(m_worldDimension[2]-innerCylinderSectorHalflengthZ);
 
        // the AtlasInnerNegativeSector
        auto atlasInnerNegativeSectorTransf = std::make_unique<Amg::Transform3D>(Amg::Translation3D(0.,0.,-innerCylinderSectorPositionZ));
        Trk::TrackingVolume* atlasInnerNegativeSector = new Trk::TrackingVolume(
                               std::move(atlasInnerNegativeSectorTransf),
                               innerCylinderSectorBounds,
                               m_worldMaterial,
                               nullptr,
                               nullptr,
                               "AtlasInnerNegativeSector");
 
        // the AtlasInnerPositiveSector
        auto atlasInnerPositiveSectorTransf = std::make_unique<Amg::Transform3D>(Amg::Translation3D(0.,0.,innerCylinderSectorPositionZ));
        Trk::TrackingVolume* atlasInnerPositiveSector = new Trk::TrackingVolume(
                               std::move(atlasInnerPositiveSectorTransf),
                               std::make_shared<Trk::CylinderVolumeBounds>(*innerCylinderSectorBounds),
                               m_worldMaterial,
                               nullptr,
                               nullptr,
                               "AtlasInnerPositiveSector");
 
        ATH_MSG_VERBOSE( "Inner Negative/Positive Sectors built successfully." );
 
        // create the subvolume Array
        auto atlasInnerSectorVolumes = std::vector<Trk::TrackingVolume*>{atlasInnerNegativeSector,highestVolume,atlasInnerPositiveSector};
 
        ATH_MSG_VERBOSE( "Create the Atlas Inner Sector volumes. " );
        std::unique_ptr<Trk::BinnedArray<Trk::TrackingVolume>> atlasInnerSectorVolumeArray = m_trackingVolumeArrayCreator ?
                m_trackingVolumeArrayCreator->cylinderVolumesArrayInZ(atlasInnerSectorVolumes) : nullptr;
 
 
        // Atlas inner Sector bounds
        auto innerSectorBounds =
                std::make_shared<Trk::CylinderVolumeBounds>(0., innerVolumeOuterR, m_worldDimension[2]);
        // build the Tracking volumes
        Trk::TrackingVolume* atlasInnerSector = new Trk::TrackingVolume(nullptr,
                                                                        innerSectorBounds,
                                                                        m_worldMaterial,
                                                                        nullptr,
                                                                        std::move(atlasInnerSectorVolumeArray),
                                                                        "AtlasInnerSector");
 
        // Atlas outer Sector
        auto outerSectorBounds =
                std::make_shared<Trk::CylinderVolumeBounds>(innerVolumeOuterR, m_worldDimension[1], m_worldDimension[2]);
        Trk::TrackingVolume* atlasOuterSector = new Trk::TrackingVolume(nullptr,
                                                                        outerSectorBounds,
                                                                        m_worldMaterial,
                                                                        nullptr,
                                                                        nullptr,
                                                                        "AtlasOuterSector");
 
        ATH_MSG_VERBOSE( "Atlas Inner/Outer Sectors built successfully." );
 
        // create the array of Inner and Outer sector
        auto atlasVolumes =  std::vector<Trk::TrackingVolume*>{atlasInnerSector, atlasOuterSector};
 
        std::unique_ptr<Trk::BinnedArray<Trk::TrackingVolume>> atlasVolumeArray = m_trackingVolumeArrayCreator ?
                m_trackingVolumeArrayCreator->cylinderVolumesArrayInR(atlasVolumes) : nullptr;
 
        // create the Atlas volume bounds
        auto atlasBounds = std::make_shared<Trk::CylinderVolumeBounds>(0., m_worldDimension[1], m_worldDimension[2]);
 
        // create the Atlas TrackingVolume
        Trk::TrackingVolume* atlasVolume = new Trk::TrackingVolume(nullptr,
                                                                   atlasBounds,
                                                                   m_worldMaterial,
                                                                   nullptr,
                                                                   std::move(atlasVolumeArray),
                                                                   "Atlas");
 
        ATH_MSG_VERBOSE( "Atlas Tracking World volume built successfully." );
 
 
        // glue the inner sector to be complete
        m_trackingVolumeHelper->glueTrackingVolumes( *atlasInnerNegativeSector, Trk::positiveFaceXY,
                                                      innerNegativeFaceVolumes, Trk::negativeFaceXY );
 
        m_trackingVolumeHelper->glueTrackingVolumes( *atlasInnerPositiveSector, Trk::negativeFaceXY,
                                                      innerPositiveFaceVolumes, Trk::positiveFaceXY );
 
        ATH_MSG_VERBOSE( "Atlas Inner Sector glued successfully together." );
 
        // iterators to the face volumes
        auto volIter    = innerCentralFaceVolumes.begin();
        auto volIterEnd = innerCentralFaceVolumes.end();
 
        // glue outer and inner sector together
        std::vector<Trk::TrackingVolume*> atlasInnerOuterVolumes;
        atlasInnerOuterVolumes.push_back(atlasInnerNegativeSector);
        for ( ; volIter != volIterEnd; ++volIter)
            if (*volIter) atlasInnerOuterVolumes.push_back(*volIter);
        atlasInnerOuterVolumes.push_back(atlasInnerPositiveSector);
 
        m_trackingVolumeHelper->glueTrackingVolumes(*atlasOuterSector, Trk::tubeInnerCover,
                                                     atlasInnerOuterVolumes, Trk::tubeOuterCover);
 
        ATH_MSG_VERBOSE( "Atlas Inner/Outer Sector glued successfully together." );
 
        // job done -> create the TrackingGeometry
        atlasTrackingGeometry = std::make_unique<Trk::TrackingGeometry>(atlasVolume);
 
        // detailed information about this tracking geometry
        ATH_MSG_VERBOSE( "Atlas TrackingGeometry built with following parameters : ");
        //ATH_MSG_VERBOSE( " - TrackingVolume containers            : " << atlasTrackingGeometry->numberOfContainerVolumes() );
        //ATH_MSG_VERBOSE( " - TrackingVolume at navigation level   : " << atlasTrackingGeometry->numberOfContainerVolumes() );
        //ATH_MSG_VERBOSE( " - Contained static layers              : " << atlasTrackingGeometry->boundaryLayers().size());
        ATH_MSG_VERBOSE( " - Unique material layers on boundaries : " << atlasTrackingGeometry->boundaryLayers().size());
 
#ifdef TRKDETDESCR_MEMUSAGE
        m_memoryLogger.refresh(getpid());
        ATH_MSG_INFO( "[ memory usage ] After Outer Sector TrackingGeometry building: "  );
        ATH_MSG_INFO( m_memoryLogger );
#endif
 
    }
 
    if (atlasTrackingGeometry) {
        if (m_navigationLevel < 3)
      atlasTrackingGeometry->registerNavigationLevel( Trk::NavigationLevel(m_navigationLevel.value()));
    }
    else ATH_MSG_WARNING( "atlasTrackingGeometry() ... atlasTrackingGeometry = 0, could not call registerNavigationLevel and propagateMagneticFieldProperties" );
 
#ifdef TRKDETDESCR_MEMUSAGE
    m_memoryLogger.refresh(getpid());
    ATH_MSG_INFO( "[ memory usage ] End of TrackingGeometry building: "  );
    ATH_MSG_INFO( m_memoryLogger );
#endif
 
    // synchronize the layers
    if (atlasTrackingGeometry) {
      if (m_synchronizeLayers) atlasTrackingGeometry->synchronizeLayers(msg());
 
      // compactify if configured to do so
      if (m_compactify) atlasTrackingGeometry->compactify(msg());
    }
    return atlasTrackingGeometry;
}