TracccTritonInitializer.h

Go to the documentation of this file.

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

 
#ifndef TRACCC_TRITON_INITIALIZER_H
#define TRACCC_TRITON_INITIALIZER_H
 
#include <cstddef>
#include <memory>
#include <string>
 
class ISvcLocator;
 
namespace triton { namespace backend { namespace traccc {

class TracccTritonInitializer {
public:
    struct Config {
        int deviceId = 0;
        std::size_t nSlots = 1;
        std::string sequenceName = "TracccDeviceRecoSeq";
        std::string condSequenceName = "AthCondSeq";
        std::string streamSvcName = "AthCUDA::PerEventStreamSvc";
        std::string cellsKey;
        std::string measurementsKey;
        std::string clustersKey;
        std::string tracksKey;
    };

    static TracccTritonInitializer& instance();

    void initialize(const Config& config);

    void finalize();

    bool isReady() const;

    const Config& config() const;

    int deviceId() const;

    ISvcLocator& serviceLocator() const;

    std::size_t acquireSlot();
 
    ~TracccTritonInitializer();
 
    TracccTritonInitializer(const TracccTritonInitializer&) = delete;
    TracccTritonInitializer& operator=(const TracccTritonInitializer&) = delete;
 
private:
    TracccTritonInitializer();
 
    struct Impl;
    std::unique_ptr<Impl> m_impl;
};
 
}}}  // namespace triton::backend::traccc
 
#endif  // TRACCC_TRITON_INITIALIZER_H