xAODToTracccSpacePointConverterAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUEVENTCNV_XAODTOTRACCCSPACEPOINTCONVERTERALG_H
#define ACTSGPUEVENTCNV_XAODTOTRACCCSPACEPOINTCONVERTERALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/ReadHandleKeyArray.h"
#include "StoreGate/WriteHandleKey.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "ActsGPUEvent/TracccSpacepointCollection.h"
 
#include "AthDeviceInterfaces/IMemoryResourcesTool.h"
#include "AthDeviceInterfaces/ICopiesTool.h"
 
#include "xAODInDetMeasurement/SpacePointContainer.h"
#include "SGTools/StlVectorClids.h"
 
#include <atomic>
#include <memory_resource>
#include <vector>
 
namespace ActsTrk {

class xAODToTracccSpacePointConverterAlg : public AthReentrantAlgorithm
{
public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;

  virtual StatusCode initialize() override;
  virtual StatusCode execute(const EventContext& ctx) const override;
  virtual StatusCode finalize() override;
 
private:
  SG::ReadHandleKeyArray<xAOD::SpacePointContainer> m_inputSpacePointKeys{
      this, "InputSpacePoints", {}, "Input xAOD space point containers"};
  SG::ReadHandleKey<std::vector<unsigned int>> m_inputMeasToClusterKey{
      this, "InputMeasToCluster", "",
      "Input mapping from traccc measurement index to xAOD cluster index"};


  SG::WriteHandleKey<traccc::edm::spacepoint_collection::buffer> m_outputSPKey{
      this, "OutputTracccSpacepoints", "",
      "Output traccc spacepoint collection buffer"};


  SG::WriteHandleKey<std::vector<unsigned int>> m_outputSPContainerKey{
      this, "OutputSPToHostContainer", "",
      "Output mapping from traccc spacepoint index to input container number"};
  SG::WriteHandleKey<std::vector<unsigned int>> m_outputSPIndexKey{
      this, "OutputSPToHostIndex", "",
      "Output mapping from traccc spacepoint index to index in the input container"};


  ToolHandle<AthDevice::IMemoryResourcesTool> m_MRs{
      this, "MemoryResourcesTool", "",
      "The memory resources tool to use for allocating memory on the host and the device"};
  ToolHandle<AthDevice::ICopiesTool> m_copiesTool{
      this, "CopiesTool", "", "Tool that provides host and device copy objects"};


  mutable std::atomic<unsigned long> m_nSpacePoints = 0;
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUEVENTCNV_XAODTOTRACCCSPACEPOINTCONVERTERALG_H