PlasmaMLPALLASPrimaryGeneratorAction.hh

Go to the documentation of this file.

#ifndef PlasmaMLPALLASPrimaryGeneratorAction_h
#define PlasmaMLPALLASPrimaryGeneratorAction_h 1
 
#include "G4VUserPrimaryGeneratorAction.hh"
#include "PlasmaMLPALLASRunAction.hh"
#include "PlasmaMLPALLASPrimaryGeneratorMessenger.hh"
#include "G4ParticleGun.hh"
#include "G4GeneralParticleSource.hh"
#include "PlasmaMLPALLASOnnxInference.hh"
#include "PlasmaMLPALLASOnnxParameters.hh"
 
// Forward declarations
class G4ParticleGun;
class G4Event;
 
class PlasmaMLPALLASPrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction
{
public:
    PlasmaMLPALLASOnnxInference* GetOnnxInference() const { return onnxInference.get(); }
    
    PlasmaMLPALLASPrimaryGeneratorAction(size_t N, size_t numThreads, bool pMT);
 
    ~PlasmaMLPALLASPrimaryGeneratorAction();
 
    void GeneratePrimaries(G4Event *anEvent) override;
 
    void UpdateParticleDefinition();
 
    double GetML_Xoff() const { return OnnxParameters::Instance().GetXoff(); }       
    double GetML_A0() const { return OnnxParameters::Instance().GetA0(); }           
    double GetML_CN2() const { return OnnxParameters::Instance().GetCN2(); }         
    double GetML_Pressure() const { return OnnxParameters::Instance().GetPressure(); } 
 
    double GetEkin() const { return Ekin; }                 
    double GetdEkin() const { return dEkin * 100.; }        
    double GetQ() const { return Q * 1e12; }                
    double GetEPSB() const { return epsb * 1e6; }           
 
 
private:
    PlasmaMLPALLASPrimaryGeneratorMessenger* fPrimaryGeneratorMessenger = nullptr; 
    std::unique_ptr<PlasmaMLPALLASOnnxInference> onnxInference; 
    G4ParticleGun* particleGun = nullptr;                       
    G4GeneralParticleSource* particleSource = nullptr;          
    G4ParticleDefinition* particleDefinition = nullptr;         
    static void ShowProgress(double progress, std::chrono::high_resolution_clock::time_point startTime);
 
    size_t NEventsGenerated = 0;   
    size_t currentEvent = 0;       
    size_t numThreads = 0;         
    bool flag_MT = false;          
    int threadID = 0;              
    int eventID = 0;               
    int nEvent = 0;                
    std::chrono::duration<double> elapsedTime; 
    bool isStartTimeInitialized = false;       
    std::chrono::high_resolution_clock::time_point startTime; 
    double progressRemaining = 0.;             
    double averageTimePerProgress = 0.;        
    double estimatedRemainingTime = 0.;        
    // Beam physical parameters
    double Ekin = 1.;   
    double dEkin = 1.;  
    double Q = 1.;      
    double epsb = 1.;   
    // ONNX session
    std::unique_ptr<Ort::Session> fSession;          
    const OrtMemoryInfo* fMemory_info = nullptr;     
    std::vector<const char*> fInput_node_names = {"input"};   
    std::vector<const char*> fOutput_node_names = {"output"}; 
};
 
#endif