PlasmaMLPALLASOnnxInference.hh

Go to the documentation of this file.

#ifndef PlasmaMLPALLASOnnxInference_H
#define PlasmaMLPALLASOnnxInference_H
 
#include <onnxruntime_cxx_api.h>
#include "globals.hh"
 
struct BeamParameters {
    G4double Ekin;   
    G4double dEkin;  
    G4double Q;      
    G4double epsb;   
    G4double x;      
    G4double xp;     
    G4double z;      
    G4double zp;     
};
 
class PlasmaMLPALLASOnnxInference {
public:
    PlasmaMLPALLASOnnxInference(const std::string& modelPath);
 
    BeamParameters GenerateBeam(G4double fXof, G4double fA0, G4double fCN2, G4double fPressure);
 
private:
    Ort::Env env;                           
    Ort::SessionOptions sessionOptions;     
    std::unique_ptr<Ort::Session> session;  
    Ort::MemoryInfo memoryInfo;             
    std::vector<const char*> inputNodeNames;
    std::vector<const char*> outputNodeNames;
};
 
#endif