PlasmaMLPALLASOnnxParameters.hh

Go to the documentation of this file.

#ifndef PlasmaMLPALLASOnnxParameters_h
#define PlasmaMLPALLASOnnxParameters_h
 
#include <atomic>
 
class OnnxParameters {
public:
    static OnnxParameters& Instance() {
        static OnnxParameters instance;
        return instance;
    }
 
    void SetStatusONNX(int status) {
        std::lock_guard<std::mutex> lock(updateMutex);
        StatusONNX.store(status, std::memory_order_release);
    }
 
    void SetXoff(double x) {
        std::lock_guard<std::mutex> lock(updateMutex);
        Xoff.store(x, std::memory_order_release);
    }
 
    void SetA0(double a0) {
        std::lock_guard<std::mutex> lock(updateMutex);
        A0.store(a0, std::memory_order_release);
    }
 
    void SetCN2(double cn2) {
        std::lock_guard<std::mutex> lock(updateMutex);
        CN2.store(cn2, std::memory_order_release);
    }
 
    void SetPressure(double p) {
        std::lock_guard<std::mutex> lock(updateMutex);
        Pressure.store(p, std::memory_order_release);
    }
 
    void SetParticleName(const std::string& name) {
        std::lock_guard<std::mutex> lock(particleNameMutex);
        ParticleName = name;
    }
 
    int GetStatusONNX() const {
        return StatusONNX.load(std::memory_order_acquire);
    }
 
    double GetXoff() const {
        return Xoff.load(std::memory_order_acquire);
    }
 
    double GetA0() const {
        return A0.load(std::memory_order_acquire);
    }
 
    double GetCN2() const {
        return CN2.load(std::memory_order_acquire);
    }
 
    double GetPressure() const {
        return Pressure.load(std::memory_order_acquire);
    }
 
    std::string GetParticleName() const {
        std::lock_guard<std::mutex> lock(particleNameMutex);
        return ParticleName;
    }
 
private:
    std::atomic<int> StatusONNX{0};      
    std::atomic<double> Xoff{0.0};       
    std::atomic<double> A0{0.0};         
    std::atomic<double> CN2{0.0};        
    std::atomic<double> Pressure{0.0};   
    mutable std::mutex particleNameMutex; 
    std::string ParticleName;             
    mutable std::mutex updateMutex;       
    OnnxParameters() = default;
 
    ~OnnxParameters() = default;
 
    OnnxParameters(const OnnxParameters&) = delete;            
    OnnxParameters& operator=(const OnnxParameters&) = delete; 
};
 
#endif