RDOtoTracccCellConverterCommons.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUEVENT_RDOTOTRACCCCELLCONVERTERCOMMONS_H
#define ACTSGPUEVENT_RDOTOTRACCCCELLCONVERTERCOMMONS_H
 
#include "AthenaBaseComps/AthMessaging.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/ReadHandleKey.h"
#include "ActsGPUEvent/TracccSiliconCellCollection.h"
#include "ActsGPUEvent/TracccDetectorConditionsDescription.h"
#include "AthDeviceInterfaces/IMemoryResourceTool.h"
#include "AthDeviceInterfaces/ICopiesTool.h"
#include "InDetIdentifier/PixelID.h"
#include "InDetIdentifier/SCT_ID.h"
#include "Identifier/Identifier.h"
#include "GaudiKernel/ToolHandle.h"
#include "ActsGPUEvent/GeometryIdMapping.h"
 
#include <traccc/io/csv/cell.hpp>
 
#include <unordered_map>
#include <cstdint>
#include <bitset>
 
class AthReentrantAlgorithm;
 
namespace ActsTrk {

 
struct RDOtoTracccCellConverterCommons : public AthMessaging
{
  RDOtoTracccCellConverterCommons(AthReentrantAlgorithm& parent);
 
  StatusCode initialize();
  StatusCode finalize();
 
  AthReentrantAlgorithm& m_parent;
 
  SG::WriteHandleKey<traccc::edm::silicon_cell_collection::buffer> m_tracccCellsKey;
 
  mutable std::atomic<int> m_nPix = 0;
  mutable std::atomic<int> m_nStrip = 0;
  mutable std::atomic<int> m_nCells = 0;
 
  const PixelID* m_pixelID{nullptr};
  const SCT_ID*  m_stripID{nullptr};
 
  ToolHandle<AthDevice::IMemoryResourceTool> m_hostMR;
  ToolHandle<AthDevice::IMemoryResourceTool> m_deviceMR;
  ToolHandle<AthDevice::ICopiesTool> m_copiesTool;
 
  Gaudi::Property<std::string> m_geoIdMappingObjectName;
  const ActsTrk::GeometryIdMapping* m_geoIdMapping{nullptr};
  Gaudi::Property<std::string> m_hostCondObjectName;
  const traccc::detector_conditions_description::host* m_hostCond{nullptr};
 
  // Geometry conversion maps
  const std::unordered_map<Identifier, uint64_t>* m_athenaToDetray{nullptr};
  std::unordered_map<uint64_t, unsigned int> m_DetrayIdToDetDescrIndexMap{};
 
  Gaudi::Property<bool> m_CPUCellSorting;
  Gaudi::Property<bool> m_UsePixelToTForCellActivation;
 
  // Returns the input cells sorted in a object setup with host_copy
  traccc::edm::silicon_cell_collection::buffer sortCells(
      vecmem::copy const & host_copy
    , traccc::edm::silicon_cell_collection::device const & cells
  ) const;
  StatusCode copyToGpuAndRecordToSG(
      EventContext const & ctx
    , traccc::edm::silicon_cell_collection::buffer const & cells
  ) const;
 
  StringProperty m_stripRDOTimeBinStr;
  int m_stripRDOTimeBinBits[3]{-1, -1, -1};
  // decode the property string into bits
  StatusCode decodeTimeBins();
  // check if the time pattern matches the requirements of strip RDO "timeBins"
  bool passTiming(const std::bitset<3>& timePattern) const;
  StatusCode buildDetrayMaps() const;
};
 
void sort_traccc_soa(
    traccc::edm::silicon_cell_collection::device & sorted_cells
  , traccc::edm::silicon_cell_collection::device const & cells
  );
 
} // namespace ActsTrk
 
#endif