DeviceTripletSeedingAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUPATTERNRECOGNITION_DEVICETRIPLETSEEDINGALG_H
#define ACTSGPUPATTERNRECOGNITION_DEVICETRIPLETSEEDINGALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "IDeviceSeedingAlgProviderTool.h"
#include "AthDeviceInterfaces/IMemoryResourceTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
#include "ActsGPUEvent/TracccDetectorGeometryDescription.h"
#include "ActsGPUEvent/TracccSpacepointCollection.h"
#include "ActsGPUEvent/TracccSeedCollection.h"
 
#include "vecmem/utils/cuda/copy.hpp"
 
template <typename scalar_t>
using unit = detray::unit<scalar_t>;
 
namespace ActsTrk {
class DeviceTripletSeedingAlg : public AthReentrantAlgorithm {
public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;

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

    ToolHandle<IDeviceSeedingAlgProviderTool> m_seedingAlgProviderTool{
        this, "SeedingAlgProviderTool", "",
        "Tool providing the appropriate backend device seeding algorithm"};

    ToolHandle<AthDevice::IMemoryResourceTool> m_deviceMR{
        this, "DeviceMR", "",
        "Device memory resource tool"};

    SG::ReadHandleKey<traccc::edm::spacepoint_collection::const_view> m_inputPixelSPKey{
        this, "InputTracccPixelSpacepoints", "",
        "Input traccc spacepoint collection buffer"};

    SG::WriteHandleKey<traccc::edm::seed_collection::buffer> m_outputPixelSeedsKey{
        this, "OutputTracccPixelSeeds", "",
        "Output traccc pixel seed collection buffer"};

    
    traccc::seedfinder_config m_seedfinder;
    traccc::seedfilter_config m_seedfilter;
 
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUPATTERNRECOGNITION_DEVICETRIPLETSEEDINGALG_H