File ExaTrkXTrackFinding.hpp¶

namespace Acts

Note

This file is foreseen for the Geometry module to replace Extent

class ExaTrkXTrackFinding

#include <Acts/Plugins/ExaTrkX/ExaTrkXTrackFinding.hpp>

Class implementing the Exa.TrkX track finding algorithm.

It holds the required ONNX objects.

Public Functions

ExaTrkXTrackFinding(const Config &config)

Constructor of the track finding module.

inline const Config &config() const: Return the configuration object of the track finding module.

void getTracks(std::vector<float> &input_values, std::vector<uint32_t> &spacepointIDs, std::vector<std::vector<uint32_t>> &trackCandidates) const

Do the track finding.

Note

The input values are not const, because the underlying ONNX API takes only non-const pointers.

Parameters

input_values – [in] Packed spacepoints in the form [ r1, phi1, z1, r2, phi2, z2, … ]
spacepointIDs – [in] corresponding spacepoint IDs to the input_values.
trackCandidates – [out] nested vector containing the spacepoint ids of the found tracks

Private Functions

void buildEdges(std::vector<float> &embedFeatures, std::vector<int64_t> &edgeList, int64_t numSpacepoints) const¶

void runSessionWithIoBinding(Ort::Session &sess, std::vector<const char*> &inputNames, std::vector<Ort::Value> &inputData, std::vector<const char*> &outputNames, std::vector<Ort::Value> &outputData) const¶

Private Members

struct Config

#include <Acts/Plugins/ExaTrkX/ExaTrkXTrackFinding.hpp>

Configuration struct for the track finding.

Public Members