DeviceSPFormationAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUDATAPREPARATION_DEVICESPFORMATIONALG_H
#define ACTSGPUDATAPREPARATION_DEVICESPFORMATIONALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "IDeviceSPFormationAlgProviderTool.h"
#include "AthDeviceInterfaces/IMemoryResourceTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
#include "ActsGPUEvent/TracccDetectorGeometryDescription.h"
#include "ActsGPUEvent/TracccMeasurementCollection.h"
#include "ActsGPUEvent/TracccSpacepointCollection.h"
 
 
 
#include "vecmem/utils/cuda/copy.hpp"
 
namespace ActsTrk {
class DeviceSPFormationAlg : public AthReentrantAlgorithm {
public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;

  virtual StatusCode initialize() override;
  virtual StatusCode execute(const EventContext& ctx) const override;
 
private:

    Gaudi::Property<std::string> m_deviceDetectorName{
        this, "DeviceDetectorName", "DetrayDeviceDetector",
        "Detray device detector object"};


    ToolHandle<IDeviceSPFormationAlgProviderTool> m_spAlgProviderTool{
        this, "SPFormationAlgProviderTool", "",
        "Tool providing the appropriate backend device spacepoint formation algorithm"};

    ToolHandle<AthDevice::IMemoryResourceTool> m_deviceMR{
        this, "DeviceMR", "",
        "Device memory resource tool"};

    SG::ReadHandleKey<traccc::edm::measurement_collection::const_view> m_inputMeasKey{
        this, "InputTracccMeasurements", "",
        "Input traccc measurement collection buffer"};

    SG::WriteHandleKey<traccc::edm::spacepoint_collection::buffer> m_outputPixelSPKey{
        this, "OutputTracccPixelSpacepoints", "",
        "Output traccc pixel spacepoint collection buffer"};

 
    // Device buffers — retrieved from detStore
    const traccc::detector_buffer* m_deviceDetector{nullptr};
 
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUDATAPREPARATION_DEVICESPFORMATIONALG_H