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/IMemoryResourceTool.h"
 
// xAOD include(s).
#include "xAODTracking/TrackParticleContainer.h"
 
// Traccc include(s).
#include <traccc/edm/track_collection.hpp>
 
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::IMemoryResourceTool> m_hostMR{
      this, "HostMR", "", "The host memory resource tool to use"};

  ToolHandle<AthDevice::IMemoryResourceTool> m_deviceMR{
      this, "DeviceMR", "", "The device memory resource tool 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