CaloBlueprintNodeBuilder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSGEOMETRY_CALOBLUEPRINTNODEBUILDER_H
#define ACTSGEOMETRY_CALOBLUEPRINTNODEBUILDER_H
 
#include "ActsGeometryInterfaces/IBlueprintNodeBuilder.h"
#include "CaloDetDescr/CaloDetDescrManager.h"
 
#include "Acts/Surfaces/CylinderSurface.hpp"
#include "Acts/Surfaces/DiscSurface.hpp"
#include "Acts/Utilities/Helpers.hpp"
 
#include "CaloIdentifier/CaloCell_ID.h"
#include "CaloDetDescr/CaloDetDescrElement.h"
 
#include <array>
#include <memory>
#include <span>
 
 
namespace ActsTrk {
 

    class CaloBlueprintNodeBuilder : public extends<AthAlgTool, IBlueprintNodeBuilder> {
    public:
        
        enum class caloRegion {DiscNegativeZ, 
                               DiscPositiveZ, 
                               BarrelCylinder, 
                               nRegions};
        friend std::ostream& operator<<(std::ostream& ostr, const caloRegion region) {
            using enum caloRegion;
            switch(region) {
              case DiscNegativeZ: ostr<<"DiscNegativeZ"; break;
              case DiscPositiveZ: ostr<<"DiscPositiveZ"; break;
              case BarrelCylinder: ostr<<"BarrelCylinder"; break;
              case nRegions: ostr<<"nRegions"; break;
            }
            return ostr;
        }
        
        using DetElVec_t = std::vector<const CaloDetDescrElement*>;
        using DetElementMap_t = std::array<DetElVec_t, CaloSampling::getNumberOfSamplings()>;
        
        using SurfaceMap_t = std::array<std::vector<std::shared_ptr<Acts::Surface>>,
                                              Acts::toUnderlying(caloRegion::nRegions)>;
 
        using base_class::base_class;
        std::shared_ptr<Acts::BlueprintNode> buildBlueprintNode(const Acts::GeometryContext& gctx,
                                                                std::shared_ptr<Acts::BlueprintNode>&& childNode) override;
        virtual StatusCode initialize() override final;
 
    private:
        DetElementMap_t fillDetectorElements(const CaloDetDescrManager& detDecrMgr) const;
        SurfaceMap_t translateToSurfaces(const DetElementMap_t& detElements) const;

        std::shared_ptr<Acts::Surface> createCylinderSurface(std::span<const CaloDetDescrElement* const> detElements) const;

        std::shared_ptr<Acts::Surface> createDiscSurface(std::span<const CaloDetDescrElement* const> detElements) const;
        std::unique_ptr<Acts::TrackingVolume> envelopeVolume(const Acts::GeometryContext& tgContext,
                                                             const SurfaceMap_t& surfacesPerRegion) const;
        std::vector<std::shared_ptr<Acts::Surface>> fillLayer(const Acts::GeometryContext& tgContext,
                                                              std::shared_ptr<Acts::Surface>&& centralSurf) const;
        
        Gaudi::Property<std::vector<std::uint32_t>> m_surfaceDuplicates{this, "insertExtraSurfaces",
            std::vector<std::uint32_t>(CaloSampling::Unknown, 0) };
                              
        Gaudi::Property<double> m_radiusTolerance { this
        , "RadiusTolerance"
        , 2.0
        , "Tolerance for determining if a ring of cells in phi has changed the radius w.r.t to the previous ring in phi" };
 
        Gaudi::Property<double> m_zTolerance { this
        , "ZTolerance"
        , 2.0
        , "Tolerance for determining if a ring of cells in phi has changed the z w.r.t to the previous ring in phi" };
  };
}
ACTS_OSTREAM_FORMATTER(ActsTrk::CaloBlueprintNodeBuilder::caloRegion);
#endif