TracccTritonRunner.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef TRACCC_TRITON_RUNNER_H
#define TRACCC_TRITON_RUNNER_H
 
#include <cstddef>
#include <cstdint>
#include <memory>
 
#include "ActsGPUEvent/TracccMeasurementCollection.h"
#include "ActsGPUEvent/TracccSiliconClusterCollection.h"
#include "ActsGPUEvent/TracccTrackContainer.h"
 
#include "GaudiKernel/EventIDBase.h"
#include "vecmem/memory/host_memory_resource.hpp"
 
namespace triton { namespace backend { namespace traccc {
 
class TracccTritonInitializer;

 
class TracccTritonRunner {
public:
    struct Output {
        Output() = default;
        Output(const Output&) = delete;
        Output& operator=(const Output&) = delete;
 
        vecmem::host_memory_resource mr;
        ::traccc::edm::measurement_collection::host measurements{mr};
        ::traccc::edm::silicon_cluster_collection::host clusters{mr};
        traccc_track_container::host tracks{mr};
    };

    TracccTritonRunner(TracccTritonInitializer& initializer, std::size_t slot);
 
    ~TracccTritonRunner();
 
    TracccTritonRunner(const TracccTritonRunner&) = delete;
    TracccTritonRunner& operator=(const TracccTritonRunner&) = delete;

    void run(const uint8_t* buffer, std::size_t byteSize,
             const EventIDBase& eventID, EventIDBase::number_type conditionsRun,
             bool printStats, Output& output);
 
private:
    struct Impl;
    std::unique_ptr<Impl> m_impl;
};
 
}}}  // namespace triton::backend::traccc
 
#endif  // TRACCC_TRITON_RUNNER_H