AthExCUDA/src/TrackParticleCalibratorExampleAlg.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
//
// Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHEXCUDA_TRACKPARTICLECALIBRATOREXAMPLEALG_H
#define ATHEXCUDA_TRACKPARTICLECALIBRATOREXAMPLEALG_H
 
// Framework include(s).
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
// Device include(s).
#include "AthCUDAInterfaces/IStreamTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
#include "AthDeviceInterfaces/IMemoryResourcesTool.h"
 
// xAOD include(s).
#include "xAODTracking/TrackParticleContainer.h"
 
// Traccc include(s).
#include <traccc/edm/track_collection.hpp>
 
// CUDA include(s).
#include <cuda_runtime_api.h>
 
namespace AthCUDAExamples {

class TrackParticleCalibratorExampleAlg : public AthReentrantAlgorithm {
 
 public:
  // Inherit the base class's constructor(s).
  using AthReentrantAlgorithm::AthReentrantAlgorithm;


  virtual StatusCode initialize() override;

  virtual StatusCode execute(const EventContext& ctx) const override;

 
 private:

  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_inputKey{
      this, "InputContainer", "InDetTrackParticles",
      "The input track particle container"};

  SG::WriteHandleKey<xAOD::TrackParticleContainer> m_outputKey{
      this, "OutputContainer", "CalibratedInDetTrackParticles",
      "The output track particle container"};

  ToolHandle<AthDevice::IMemoryResourcesTool> m_mrTool{
      this, "MemoryResourcesTool", "",
      "Tool providing the memory resource(s) to use"};

  ToolHandle<AthDevice::ICopyTool> m_hostCopyTool{this, "HostCopyTool", "",
                                                  "The host copy tool to use"};

  ToolHandle<AthDevice::ICopyTool> m_deviceCopyTool{
      this, "DeviceCopyTool", "", "The device copy tool to use"};

  ToolHandle<AthCUDA::IStreamTool> m_streamTool{this, "StreamTool", "",
                                                "The stream tool to use"};

 
};  // class LinearTransformTaskExampleAlg

StatusCode calibrateOnGPU(
    cudaStream_t stream,
    const traccc::edm::track_collection<traccc::default_algebra>::const_view&
        input,
    traccc::edm::track_collection<traccc::default_algebra>::view& output);
 
}  // namespace AthCUDAExamples
 
#endif  // ATHEXCUDA_TRACKPARTICLECALIBRATOREXAMPLEALG_H