PlasmaMLPALLASRunAction.hh

Go to the documentation of this file.

#ifndef PlasmaMLPALLASRunAction_h
#define PlasmaMLPALLASRunAction_h 1
 
// Include base classes and Geant4 utilities
#include "G4UserRunAction.hh"       // Base class for user-defined run actions
#include "G4Run.hh"                 // Run object for event accumulation
#include "G4UImanager.hh"           // UI manager (for commands)
#include "G4VVisManager.hh"         // Visualization manager
#include "G4RunManager.hh"
#include "TFile.h"                  // ROOT file I/O
#include "TTree.h"
#include "TBranch.h"
#include <mutex>
#include "PlasmaMLPALLASPrimaryGeneratorAction.hh"
#include "PlasmaMLPALLASGeometryConstruction.hh"
#include "PlasmaMLPALLASEventAction.hh" 
 
 
// Forward declarations
class PlasmaMLPALLASPrimaryGeneratorAction;
class PlasmaMLPALLASGeometryConstruction;
 
struct RunTallyGlobalInput {
  // --- General run settings ---
  int NEvents = 0;                 
  int Display_Geometry = 0;             
  int Display_Collimators = 0;     
  int Display_Quadrupoles = 0;     
 
  // --- Physical parameters ---
  float Xoff = 0.0;               
  float p = 0.0;                  
  float cN2 = 0.0;                
  float A0 = 0.0;                 
  float Ekin = 0.0;               
  float dEkin = 0.0;              
  float Q = 0.0;                  
  float epsb = 0.0;               
  float Q1_Grad = 0.0;            
  float Q2_Grad = 0.0;            
  float Q3_Grad = 0.0;            
  float Q4_Grad = 0.0;            
  float Q1_Length = 0.0;          
  float Q2_Length = 0.0;          
  float Q3_Length = 0.0;          
  float Q4_Length = 0.0;          
  float SourceQ1Distance = 0.0;   
  float Q1Q2Distance = 0.0;       
  float Q2Q3Distance = 0.0;       
  float Q3Q4Distance = 0.0;       
  float B_Dipole = 0.0;           
  int B_Dipole_Map = 0;            
 
  void FillFrom(const PlasmaMLPALLASPrimaryGeneratorAction* gen, 
                const PlasmaMLPALLASGeometryConstruction* geo,
                int nEvents);
};
 
class PlasmaMLPALLASRunAction : public G4UserRunAction
{
public:
  PlasmaMLPALLASRunAction(const char* fileName, size_t nEventsGenerated, G4bool multithreaded);
 
  ~PlasmaMLPALLASRunAction();
 
  void BeginOfRunAction(const G4Run* run) override;
 
  void EndOfRunAction(const G4Run* run) override;
 
  template<typename T>
  static void UpdateStatistics(T& stats, const T& newStats, TTree* tree);
 
  // --- Specific statistics update methods ---
  void UpdateStatisticsGlobalInput(RunTallyGlobalInput);
  void UpdateStatisticsInput(RunTallyInput);
  void UpdateStatisticsQuadrupoles(RunTallyQuadrupoles);
  void UpdateStatisticsHorizontalColl(RunTallyCollimators);
  void UpdateStatisticsVerticalColl(RunTallyCollimators);
  void UpdateStatisticsBSYAG(RunTallyYAG);
  void UpdateStatisticsBSPECYAG(RunTallyYAG);
 
  void SetPrimaryGenerator(PlasmaMLPALLASPrimaryGeneratorAction* gen);
 
  void SetGeometry(PlasmaMLPALLASGeometryConstruction* geom);
 
private:
  // --- Output configuration ---
  G4String suffixe;     
  G4String fileName;    
 
  // --- Stored run statistics ---
  RunTallyGlobalInput StatsGlobalInput;
  RunTallyInput StatsInput;
  RunTallyQuadrupoles StatsQuadrupoles;
  RunTallyCollimators StatsHorizontalColl;
  RunTallyCollimators StatsVerticalColl;
  RunTallyYAG StatsBSYAG;
  RunTallyYAG StatsBSPECYAG;
 
  size_t NEventsGenerated; 
  G4bool flag_MT;          
 
  // --- ROOT file and trees ---
  TFile *f = nullptr;
  TTree *Tree_GlobalInput = nullptr;
  TTree *Tree_Input = nullptr;
  TTree *Tree_Quadrupoles = nullptr;
  TTree *Tree_HorizontalColl = nullptr;
  TTree *Tree_VerticalColl = nullptr;
  TTree *Tree_BSYAG = nullptr;
  TTree *Tree_BSPECYAG = nullptr;
  TBranch *RunBranch = nullptr;
 
  time_t start; 
 
  // --- Thread-safety ---
  static std::atomic<int> activeThreads;
  static G4Mutex fileMutex;
 
protected:
  PlasmaMLPALLASPrimaryGeneratorAction* fPrimaryGenerator = nullptr; 
  PlasmaMLPALLASGeometryConstruction* fGeometry = nullptr; 
};
 
#endif // PlasmaMLPALLASRunAction_h