Implementation of ONNX model inference for PALLAS plasma beam generation. More...
#include "PlasmaMLPALLASOnnxInference.hh"#include "Randomize.hh"#include "CLHEP/Units/SystemOfUnits.h"#include "CLHEP/Units/PhysicalConstants.h"#include <cmath>
Include dependency graph for PlasmaMLPALLASOnnxInference.cc:
Detailed Description
Implementation of ONNX model inference for PALLAS plasma beam generation.
This file implements the PlasmaMLPALLASOnnxInference class, which provides a mechanism to generate plasma beam parameters by performing inference on a pre-trained ONNX model. The class handles:
- Initialization of the ONNX runtime environment
- Loading the specified ONNX model file
- Preparing input tensors from physical simulation parameters
- Running inference and converting outputs to physical units
- Sampling transverse beam coordinates (x, z) and momenta (xp, zp)
The main purpose is to generate physically realistic beam parameters for use in Geant4 simulations of the PALLAS plasma beamline.
Input parameters for inference include:
- Laser focal position (fXof)
- Laser normalized amplitude (fA0)
- Plasma density (fCN2)
- Gas pressure (fPressure)
Output parameters encapsulated in the BeamParameters structure include:
- Kinetic energy (Ekin) and spread (dEkin)
- Beam charge (Q)
- Normalized emittance (epsb)
- Transverse positions (x, z) and angles (xp, zp)
The class uses CLHEP units for all physical quantities.
- Date
- 2025
- Copyright
- PALLAS Project - GEANT4 Collaboration
