xAODToTracccMeasurementConverterAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUEVENTCNV_XAODTOTRACCCMEASUREMENTCONVERTERALG_H
#define ACTSGPUEVENTCNV_XAODTOTRACCCMEASUREMENTCONVERTERALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "ActsGPUEvent/TracccMeasurementCollection.h"
#include "ActsGPUEvent/TracccDetectorConditionsDescription.h"
#include "ActsGPUEvent/TracccDetectorDesignDescription.h"
#include "ActsGPUEvent/GeometryIdMapping.h"
 
#include "AthDeviceInterfaces/IMemoryResourcesTool.h"
#include "AthDeviceInterfaces/ICopiesTool.h"
 
#include "xAODInDetMeasurement/PixelClusterContainer.h"
#include "xAODInDetMeasurement/StripClusterContainer.h"
#include "InDetIdentifier/PixelID.h"
#include "InDetIdentifier/SCT_ID.h"
#include "SGTools/StlVectorClids.h"
 
#include <array>
#include <atomic>
#include <memory_resource>
#include <cstdint>
#include <unordered_map>
#include <vector>
 
namespace ActsTrk {

class xAODToTracccMeasurementConverterAlg : public AthReentrantAlgorithm
{
public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;

  virtual StatusCode initialize() override;
  virtual StatusCode execute(const EventContext& ctx) const override;
  virtual StatusCode finalize() override;
 
private:
  struct MeasurementRecord {
    std::uint64_t geometryId{0};
    unsigned int condIndex{0};
    bool isPixel{true};
    unsigned int hostIndex{0};
    std::array<float, 2u> localPosition{0.f, 0.f};
    std::array<float, 2u> localVariance{0.f, 0.f};
    float diameter{0.f};
  };

  StatusCode condIndexFor(const Identifier& moduleId, unsigned int& condIndex) const;

  SG::ReadHandleKey<xAOD::PixelClusterContainer> m_inputPixelClustersKey{
      this, "InputPixelClusters", "ITkPixelClusters", "Input xAOD pixel clusters"};
  SG::ReadHandleKey<xAOD::StripClusterContainer> m_inputStripClustersKey{
      this, "InputStripClusters", "ITkStripClusters", "Input xAOD strip clusters"};


  SG::WriteHandleKey<traccc::edm::measurement_collection::buffer> m_outputMeasKey{
      this, "OutputTracccMeasurements", "TracccMeasurementCollection",
      "Output traccc measurement collection buffer"};


  SG::WriteHandleKey<std::vector<unsigned int>> m_outputMeasToPixelKey{
      this, "OutputMeasToPixelCluster", "TracccMeasToPixelCluster",
      "Output mapping from traccc measurement index to xAOD pixel cluster index"};
  SG::WriteHandleKey<std::vector<unsigned int>> m_outputMeasToStripKey{
      this, "OutputMeasToStripCluster", "TracccMeasToStripCluster",
      "Output mapping from traccc measurement index to xAOD strip cluster index"};


  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"};


  Gaudi::Property<std::string> m_geoIdMappingObjectName{
      this, "GeoIdMappingObjectName", "TracccGeometryIdMapping",
      "StoreGate name for the detray/acts/athena geo id mapping"};
  Gaudi::Property<std::string> m_hostCondObjectName{
      this, "HostConditionsObjectName", "TracccHostCondConfig",
      "Traccc host conditions object"};
  Gaudi::Property<std::string> m_hostDesignObjectName{
      this, "HostDigitizationObjectName", "TracccHostDigitizationConfig",
      "Traccc host digitization object"};
  const ActsTrk::GeometryIdMapping* m_geoIdMapping{nullptr};
  const traccc::detector_conditions_description::host* m_hostCond{nullptr};
  const traccc::detector_design_description::host* m_hostDesign{nullptr};
 
  const PixelID* m_pixelID{nullptr};
  const SCT_ID* m_stripID{nullptr};
 
  std::unordered_map<std::uint64_t, unsigned int> m_detrayIdToCondIndex{};

  mutable std::atomic<unsigned long> m_nPixel = 0;
  mutable std::atomic<unsigned long> m_nStrip = 0;
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUEVENTCNV_XAODTOTRACCCMEASUREMENTCONVERTERALG_H