TracccMeasurementConverterAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUEVENT_TRACCCMEASUREMENTCONVERTERALG_H
#define ACTSGPUEVENT_TRACCCMEASUREMENTCONVERTERALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "ActsGPUEvent/TracccMeasurementCollection.h"
#include "ActsGPUEvent/TracccSiliconClusterCollection.h"
#include "ActsGPUEvent/TracccSiliconCellCollection.h"
#include "ActsGPUEvent/GeometryIdMapping.h"
#include "traccc/edm/measurement_collection.hpp"
#include "AthDeviceInterfaces/IMemoryResourceTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
 
#include "xAODInDetMeasurement/PixelClusterContainer.h"
#include "xAODInDetMeasurement/PixelClusterAuxContainer.h"
#include "xAODInDetMeasurement/StripClusterContainer.h"
#include "xAODInDetMeasurement/StripClusterAuxContainer.h"
#include "xAODInDetMeasurement/SpacePointContainer.h"
#include "xAODInDetMeasurement/SpacePointAuxContainer.h"
 
#include "InDetIdentifier/PixelID.h"
#include "InDetIdentifier/SCT_ID.h"
#include "InDetReadoutGeometry/SiDetectorElement.h"
#include "PixelReadoutGeometry/PixelDetectorManager.h"
#include "SCT_ReadoutGeometry/SCT_DetectorManager.h"
 
#include "PixelReadoutGeometry/PixelModuleDesign.h"
#include "ReadoutGeometryBase/SiCellId.h"
#include "SCT_ReadoutGeometry/SCT_BarrelModuleSideDesign.h"
#include "SCT_ReadoutGeometry/SCT_ForwardModuleSideDesign.h"
#include "SCT_ReadoutGeometry/SCT_ModuleSideDesign.h"
#include "SCT_ReadoutGeometry/StripStereoAnnulusDesign.h"
 
#include "GaudiKernel/ToolHandle.h"
 
namespace ActsTrk {

class TracccMeasurementConverterAlg : public AthReentrantAlgorithm
{
public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;

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

  Gaudi::Property<bool> m_convertClustersWithCells{
      this, "ConvertClustersWithCells", true,
      "Whether to associate cells with the clusters."};

  SG::ReadHandleKey<traccc::edm::measurement_collection::buffer> m_inputMeasKey{
      this, "InputMeasurements", "TracccMeasurements",
      "Input traccc measurement collection buffer"};
  SG::ReadHandleKey<traccc::edm::silicon_cluster_collection::buffer> m_inputClusterKey{
      this, "InputClusters", "TracccClusters",
      "Input traccc cluster collection buffer"};
  SG::ReadHandleKey<traccc::edm::silicon_cell_collection::buffer> m_inputCellsKey{
      this, "InputCells", "TracccCells",
      "Input traccc cell collection buffer"};


  SG::WriteHandleKey<xAOD::PixelClusterContainer> m_outputPixelKey{
      this, "OutputPixelClusters", "ITkTracccPixelClusters",
      "Output xAOD pixel cluster container"};
  SG::WriteHandleKey<xAOD::SpacePointContainer> m_outputPixelSpacePointsKey{
      this, "OutputPixelSpacePoints", "ITkTracccPixelSpacepoints",
      "Output xAOD pixel space point container"};
  SG::WriteHandleKey<std::vector<unsigned int>> m_outputMeasToPixelSPKey{
      this, "OutputMeasToPixelSP", "ITkTracccMeasToPixelSP",
      "Output mapping from traccc measurement index to pixel spacepoint index"};
  SG::WriteHandleKey<std::vector<unsigned int>> m_outputMeasToStripClKey{
      this, "OutputMeasToStripCl", "ITkTracccMeasToStripCl",
      "Output mapping from traccc measurement index to strip cluster index"};             
  SG::WriteHandleKey<xAOD::StripClusterContainer> m_outputStripKey{
      this, "OutputStripClusters", "ITkTracccStripClusters",
      "Output xAOD strip cluster container"};


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

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

  Gaudi::Property<std::string> m_geoIdMappingObjectName{this, "GeoIdMapping", "", "ID mapping between the three detector description realms."};
  const ActsTrk::GeometryIdMapping* m_idMapping{nullptr};

  mutable std::atomic<int> m_nPix = 0;
  mutable std::atomic<int> m_nStrip = 0;
  mutable std::atomic<int> m_nMeas = 0;

  const PixelID* m_pixelID{nullptr};
  const SCT_ID*  m_stripID{nullptr};
  Gaudi::Property<std::string> m_idHelperName {this, "IDHelperName", "PixelID",
    "Pixel-like ID helper name to retrieve from DetectorStore"};
 
  const InDetDD::PixelDetectorManager* m_pixelManager{nullptr};
  const InDetDD::SCT_DetectorManager*  m_stripManager{nullptr};
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUEVENT_TRACCCMEASUREMENTCONVERTERALG_H