TracccSeedConverterAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUEVENT_TRACCCSEEDCONVERTERALG_H
#define ACTSGPUEVENT_TRACCCSEEDCONVERTERALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "ActsGPUEvent/TracccSeedCollection.h"
#include "ActsGPUEvent/TracccSpacepointCollection.h"
#include "ActsEvent/SeedContainer.h"
 
#include "AthDeviceInterfaces/IMemoryResourceTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
 
#include "xAODInDetMeasurement/SpacePointContainer.h"
#include "xAODInDetMeasurement/SpacePointAuxContainer.h"
 
#include "GaudiKernel/ToolHandle.h"
 
namespace ActsTrk {

class TracccSeedConverterAlg : public AthReentrantAlgorithm
{
public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;

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

  SG::ReadHandleKey<xAOD::SpacePointContainer> m_inputSPKey{
      this, "InputSpacepoints", "TracccSpacepoints",
      "Input traccc SP collection buffer"};
  SG::ReadHandleKey<traccc::edm::spacepoint_collection::buffer> m_inputSPDeviceKey{
      this, "InputSpacepointsDevice", "TracccSpacepointsDevice",
      "Input traccc SP collection device buffer"};
  SG::ReadHandleKey<std::vector<unsigned int>> m_inputMeasToPixelSPKey{
      this, "InputMeasToPixelSP", "TracccMeasToPixelSP",
      "Input mapping from traccc measurement index to pixel spacepoint index"};        
  SG::ReadHandleKey<traccc::edm::seed_collection::buffer> m_inputSeedsKey{
      this, "InputSeeds", "TracccSeeds",
      "Input traccc seed collection buffer"};


  SG::WriteHandleKey<ActsTrk::SeedContainer> m_outputSeedsKey{
      this, "OutputSeeds", "ITkTracccSeeds",
      "Output ACTS seeds container"};


  ToolHandle<AthDevice::IMemoryResourceTool> m_hostMR{
    this, "HostMR", "", "Host memory resource tool"};

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

  mutable std::atomic<int> m_nSP = 0;
  mutable std::atomic<int> m_nSeeds = 0;
  
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUEVENT_TRACCCSEEDCONVERTERALG_H