File CugraphTrackBuilding.hpp

namespace Acts

Note

This file is foreseen for the Geometry module to replace Extent

class CugraphTrackBuilding : public Acts::TrackBuildingBase
#include <Acts/Plugins/ExaTrkX/CugraphTrackBuilding.hpp>

Public Functions

inline CugraphTrackBuilding(std::unique_ptr<const Logger> logger)
virtual std::vector<std::vector<int>> operator()(std::any nodes, std::any edges, std::any edge_weights, std::vector<int> &spacepointIDs) override

Perform track building.

Parameters

  • nodes – Node tensor with shape (n_nodes, n_node_features)

  • edges – Edge-index tensor with shape (2, n_edges)

  • edgeWeights – Edge-weights of the previous edge classification phase

  • spacepointIDs – IDs of the nodes (must have size=n_nodes)

Returns

tracks (as vectors of node-IDs)

Private Functions

inline const auto &logger() const

Private Members

std::unique_ptr<const Acts::Logger> m_logger