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::ReadHandleKey<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"};

    
    Gaudi::Property<unsigned int> m_maxNumBranchesPerSeed{this, "maxNumBranchesPerSeed", 3,
        "maximum number of branches per seed"};
    Gaudi::Property<unsigned int> m_maxNumBranchesPerSurface{this, "maxNumBranchesPerSurface", 1,
        "maximum number of branches per surface"};
    Gaudi::Property<unsigned int> m_minTrackCandidatesPerTrack{this, "minTrackCandidatesPerTrack", 7,
        "minimum number of measurements per track"};
    Gaudi::Property<unsigned int> m_maxTrackCandidatesPerTrack{this, "maxTrackCandidatesPerTrack", 20,
        "maximum number of measurements per track"};
    Gaudi::Property<unsigned int> m_maxNumSkippingPerCand{this, "maxNumSkippingPerCand", 2,
        "maximum number of holes per track candidate"};
    Gaudi::Property<unsigned int> m_maxNumConsecutiveSkipped{this, "maxNumConsecutiveSkipped", 1,
        "maximum number of consecutive holes per track candidate"};
    Gaudi::Property<float> m_chi2Max{this, "chi2Max", 30.f,
        "maximum chi2 for a measurement to be added to a track candidate"};
    Gaudi::Property<float> m_minPt{this, "minPt", 600.f,
        "minimum transverse momentum of a track candidate [MeV]"};
    Gaudi::Property<float> m_minP{this, "minP", 100.f,
        "minimum momentum of a track candidate [MeV]"};
    Gaudi::Property<unsigned int> m_maxNumTracksPerMeasurement{this, "maxNumTracksPerMeasurement", 1,
        "maximum number of tracks sharing a measurement, 0 for no limit"};
    Gaudi::Property<unsigned int> m_initialLinksPerSeed{this, "initialLinksPerSeed", 20,
        "initial number of links reserved per seed"};
    Gaudi::Property<float> m_minStepLengthForNextSurface{this, "minStepLengthForNextSurface", 0.5f,
        "minimum step length to the next surface [mm]"};
    Gaudi::Property<unsigned int> m_maxStepCountsForNextSurface{this, "maxStepCountsForNextSurface", 100,
        "maximum number of steps to the next surface"};

 
    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