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
 
// Local include(s).
#include "TrackParticleContainer.h"
 
// Framework include(s).
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
// Device include(s).
#include "AthDeviceInterfaces/IMemoryResourceTool.h"
 
// xAOD include(s).
#include "xAODTracking/TrackParticleContainer.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::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"};

 
};  // class LinearTransformTaskExampleAlg

StatusCode calibrateOnGPU(const TrackParticleContainer::const_view& input,
                          TrackParticleContainer::view& output);
 
}  // namespace AthCUDAExamples
 
#endif  // ATHEXCUDA_TRACKPARTICLECALIBRATOREXAMPLEALG_H