JSONDeviceDetectorDescriptionProviderSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUDATAPREPARATION_JSONDEVICEDETECTORDESCRIPTIONPROVIDERSVC_H
#define ACTSGPUDATAPREPARATION_JSONDEVICEDETECTORDESCRIPTIONPROVIDERSVC_H
 
#include "AthenaBaseComps/AthService.h"
#include "GaudiKernel/ToolHandle.h"
#include "PathResolver/PathResolver.h"
#include "ActsGPUInterfaces/IDeviceDetectorDescriptionProviderSvc.h"
 
#include "StoreGate/ReadCondHandleKey.h"
#include "AthDeviceInterfaces/IMemoryResourcesTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
#include "InDetIdentifier/PixelID.h"
#include "InDetIdentifier/SCT_ID.h"
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
#include "PixelReadoutGeometry/PixelDetectorManager.h"
#include "SCT_ReadoutGeometry/SCT_DetectorManager.h"
 
#include "traccc/geometry/detector_design_description.hpp"
#include "traccc/geometry/detector_conditions_description.hpp"
 
#include "detray/core/detail/container_views.hpp"
#include "detray/core/detector.hpp"
#include "detray/detectors/itk_metadata.hpp"
 
 
#include "vecmem/utils/cuda/copy.hpp"
 
#include <memory>
#include <string>
#include <unordered_map>
 
namespace ActsTrk {

class JSONDeviceDetectorDescriptionProviderSvc
    : public extends<AthService, ActsTrk::IDeviceDetectorDescriptionProviderSvc>
{
public:
 
    using extends::extends;

    virtual StatusCode initialize() override;

    virtual const std::unordered_map<uint64_t, Identifier>&
        detrayToAthenaMap() const override;
    virtual const std::unordered_map<Identifier, uint64_t>&
        athenaToDetrayMap() const override;
 
private:
 
    ServiceHandle<StoreGateSvc> m_detStore{this, "DetectorStore", "StoreGateSvc/DetectorStore"};
    ToolHandle<AthDevice::IMemoryResourcesTool> m_MRs{
        this, "MemoryResourcesTool", "",
        "The memory resources tool to use for allocating memory on the device"};

    ToolHandle<AthDevice::ICopyTool> m_copy{
        this, "CopyProviderTool", "", "Vecmem copy provider tool"};

    Gaudi::Property<std::string> m_geometryFile{
        this, "GeometryFile", "",
        "Detray geometry JSON file"};
    Gaudi::Property<std::string> m_digitizationFile{
        this, "DigitizationFile", "",
        "Traccc digitization config JSON file"};
    Gaudi::Property<std::string> m_conditionsFile{
        this, "ConditionsFile", "",
        "Traccc conditions config JSON file"};
    Gaudi::Property<std::string> m_mapFile{
        this, "MapFile", "",
        "Path to the athena<->detray ID map CSV file"};   


    Gaudi::Property<std::string> m_deviceDesignObjectName{
        this, "DeviceDigitizationObjectName", "",
        "Traccc device digitization object"};
    Gaudi::Property<std::string> m_deviceCondObjectName{
        this, "DeviceConditionsObjectName", "",
        "Traccc device conditions object"};
    Gaudi::Property<std::string> m_hostDesignObjectName{
        this, "HostDigitizationObjectName", "",
        "Traccc host digitization object"};
    Gaudi::Property<std::string> m_hostCondObjectName{
        this, "HostConditionsObjectName", "",
        "Traccc host conditions object"};
    Gaudi::Property<std::string> m_deviceDetectorName{
        this, "DeviceDetectorName", "",
        "Detray device detector object"};
    Gaudi::Property<std::string> m_hostDetectorName{
        this, "HostDetectorName", "",
        "Detray host detector object"};    


    StatusCode loadIdMaps();
 
    std::unordered_map<uint64_t, Identifier> m_detrayToAthena;
    std::unordered_map<Identifier, uint64_t> m_athenaToDetray;
 
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUDATAPREPARATION_JSONDEVICEDETECTORDESCRIPTIONPROVIDERSVC_H