DeviceTrkParamEstimationAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUPATTERNRECOGNITION_DEVICETRKPARAMESTIMATIONALG_H
#define ACTSGPUPATTERNRECOGNITION_DEVICETRKPARAMESTIMATIONALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "IDeviceTrkParamAlgProviderTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
#include "ActsGPUEvent/TracccMagField.h"
#include "ActsGPUEvent/TracccTrkParamCollection.h"
#include "ActsGPUEvent/TracccSeedCollection.h"
#include "ActsGPUEvent/TracccSpacepointCollection.h"
#include "ActsGPUEvent/TracccMeasurementCollection.h"
 
template <typename scalar_t>
using unit = detray::unit<scalar_t>;
 
namespace ActsTrk {
class DeviceTrkParamEstimationAlg    : public AthReentrantAlgorithm {
public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;

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

    ToolHandle<IDeviceTrkParamAlgProviderTool> m_trkParamAlgProviderTool{
        this, "TrkParamAlgProviderTool", "",
        "Tool providing the appropriate backend device track parameter estimation algorithm"};

    SG::ReadHandleKey<traccc::edm::spacepoint_collection::const_view> m_inputSpacePointsKey{
        this, "InputTracccSpacepoints", "",
        "Input traccc space points collection buffer"};

    SG::ReadHandleKey<traccc::edm::measurement_collection::const_view> m_inputMeasKey{
        this, "InputTracccMeasurements", "",
        "Input traccc measurements collection buffer"};

    SG::ReadHandleKey<traccc::edm::seed_collection::const_view> m_inputSeedsKey{
        this, "InputTracccSeeds", "",
        "Input traccc seeds collection buffer"};

    Gaudi::Property<std::string> m_inputMagFieldKey{
        this, "InputTracccMagField", "",
        "Input traccc inhom. mag. field"};

    SG::WriteHandleKey<traccc::bound_track_parameters_collection_types::buffer> m_outputTrkParamKey{
        this, "OutputTracccTrackParameters", "",
        "Output traccc track parameter collection buffer"};

 
    traccc::track_params_estimation_config m_trkparam_config;
    const traccc::magnetic_field* m_deviceMagField{nullptr};
 
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUPATTERNRECOGNITION_DEVICETRKPARAMESTIMATIONALG_H