TracccTritonTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TracccTritonTool_H
#define TracccTritonTool_H
 
// System include(s).
#include <iostream>
#include <list>
#include <memory>
 
#include "AthOnnxInterfaces/IAthInferenceTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "ITracccTritonTool.h"

 
class TracccTritonTool : public extends<AthAlgTool, ITracccTritonTool> {
public:
    TracccTritonTool(const std::string& type, const std::string& name, const IInterface* parent);
    virtual StatusCode initialize() override;

    // Main methods for remote track finding asked by the ITracccTritonTool

    virtual StatusCode getTracks(
        std::vector<uint8_t>& cellBytes,
        std::vector<TracccTrackParameters>& TracccTrackParameters,
        std::vector<LocalMeasurementInfoInTracks>& TracccMeasurementsInfoInTracks) const override;
 
 
private:
    ToolHandle<AthInfer::IAthInferenceTool> m_TracccTritonTool{
        this, "TritonTool", "AthInfer::TritonTool"};
 
};
 
#endif  // TracccTritonTool_H