DeviceTrackFindingAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUPATTERNRECOGNITION_DEVICETRACKFINDINGALG_H
#define ACTSGPUPATTERNRECOGNITION_DEVICETRACKFINDINGALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "IDeviceTrackFindingAlgProviderTool.h"
#include "AthDeviceInterfaces/IMemoryResourceTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
#include "ActsGPUEvent/TracccMagField.h"
#include "ActsGPUEvent/TracccDetectorGeometryDescription.h"
#include "ActsGPUEvent/TracccDetectorGeometryDescription.h"
#include "ActsGPUEvent/TracccTrkParamCollection.h"
#include "ActsGPUEvent/TracccMeasurementCollection.h"
#include "ActsGPUEvent/TracccTrackContainer.h"
 
namespace ActsTrk {
class DeviceTrackFindingAlg    : public AthReentrantAlgorithm {
public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;

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

    ToolHandle<IDeviceTrackFindingAlgProviderTool> m_trkFindingAlgProviderTool{
        this, "TrackFindingAlgProviderTool", "",
        "Tool providing the appropriate backend device track finding algorithm"};

    SG::ReadHandleKey<traccc::edm::measurement_collection::const_view> m_inputMeasKey{
        this, "InputTracccMeasurements", "",
        "Input traccc measurements collection buffer"};
    
    Gaudi::Property<std::string> m_inputMagFieldKey{
        this, "InputTracccMagField", "",
        "Input traccc inhom. mag. field"};  

    Gaudi::Property<std::string> m_deviceDetectorObjectName{
        this, "InputTracccDetectorGeometry", "",
        "Input traccc detector geometry"};             

    SG::WriteHandleKey<traccc::bound_track_parameters_collection_types::buffer> m_inputTrkParamKey{
        this, "InputTracccTrackParameters", "",
        "Input traccc track parameter collection buffer"};


    SG::WriteHandleKey<traccc_track_container::buffer> m_outputTracksKey{
        this, "OutputTracccTracks", "",
        "Output traccc tracks collection buffer"};

    
    traccc::finding_config m_finding_cfg;
    const traccc::magnetic_field* m_deviceMagField{nullptr};
    const traccc::detector_buffer* m_deviceDetector{nullptr};
 
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUPATTERNRECOGNITION_DEVICETRACKFINDINGALG_H