make scene able to run parallel simulations
This commit is contained in:
84
src/HEP/Geant/testing/SkyPlaneEmitterTest.cpp
Normal file
84
src/HEP/Geant/testing/SkyPlaneEmitterTest.cpp
Normal file
@@ -0,0 +1,84 @@
|
||||
#include "Geant/Solid.h"
|
||||
#include "HEP/Geant/GeantEvent.h"
|
||||
#include "HEP/Geant/Scene.h"
|
||||
#include "HEP/Geant/EmitterPrimary.hh"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Math/Units.h"
|
||||
#include "HEP/Detectors/DetectorChamber.h"
|
||||
|
||||
#include <Geant4/G4SystemOfUnits.hh>
|
||||
|
||||
#include <iostream>
|
||||
|
||||
using namespace uLib;
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
|
||||
int nEvents = 10000;
|
||||
if (argc > 1) {
|
||||
nEvents = std::stoi(argv[1]);
|
||||
}
|
||||
|
||||
// 1. Setup Geant4 Scene
|
||||
Geant::Scene scene;
|
||||
scene.ConstructWorldBox(Vector3f(30_m, 30_m, 30_m), "G4_AIR");
|
||||
|
||||
ContainerBox iron_box;
|
||||
iron_box.Scale(Vector3f(18_m, 18_m, 18_m));
|
||||
iron_box.SetPosition(Vector3f(-9_m, -9_m, -9_m));
|
||||
Geant::BoxSolid* iron_cube = new Geant::BoxSolid("IronCube", &iron_box);
|
||||
iron_cube->SetNistMaterial("G4_Fe");
|
||||
iron_cube->Update();
|
||||
scene.AddSolid(iron_cube);
|
||||
|
||||
// Top Detector Chamber (along Y axis)
|
||||
DetectorChamber* top_chamber_box = new DetectorChamber();
|
||||
top_chamber_box->Scale(Vector3f(20_m, 40_cm, 20_m));
|
||||
top_chamber_box->Rotate(90_deg, Vector3f(1, 0, 0));
|
||||
top_chamber_box->SetPosition(Vector3f(-10_m, 12_m, -10_m));
|
||||
Geant::BoxSolid* top_chamber = new Geant::BoxSolid("TopChamber", top_chamber_box);
|
||||
top_chamber->SetNistMaterial("G4_AIR");
|
||||
top_chamber->Update();
|
||||
scene.AddSolid(top_chamber);
|
||||
|
||||
// Bottom Detector Chamber (along Y axis)
|
||||
DetectorChamber* bottom_chamber_box = new DetectorChamber();
|
||||
bottom_chamber_box->Scale(Vector3f(20_m, 40_cm, 20_m));
|
||||
bottom_chamber_box->Rotate(90_deg, Vector3f(1, 0, 0));
|
||||
bottom_chamber_box->SetPosition(Vector3f(-10_m, -12_m, -10_m));
|
||||
Geant::BoxSolid* bottom_chamber = new Geant::BoxSolid("BottomChamber", bottom_chamber_box);
|
||||
bottom_chamber->SetNistMaterial("G4_AIR");
|
||||
bottom_chamber->Update();
|
||||
scene.AddSolid(bottom_chamber);
|
||||
|
||||
// Setup SkyPlaneEmitterPrimary
|
||||
Geant::SkyPlaneEmitterPrimary* emitter = new Geant::SkyPlaneEmitterPrimary();
|
||||
emitter->SetPosition(Vector3f(0, 14.9_m, 0));
|
||||
emitter->Rotate(-90_deg, Vector3f(1, 0, 0));
|
||||
emitter->SetSkySize(Vector2f(20_m, 20_m));
|
||||
|
||||
scene.SetEmitter(emitter);
|
||||
scene.SetVerbosity(1);
|
||||
scene.Initialize();
|
||||
|
||||
std::cout << "Starting simulation of " << nEvents << " events..." << std::endl;
|
||||
Vector<Geant::GeantEvent> results;
|
||||
scene.RunSimulation(nEvents, results);
|
||||
|
||||
std::cout << "Simulation finished. Collected " << results.size() << " events." << std::endl;
|
||||
|
||||
// Sample output to verify data collection
|
||||
if (!results.empty()) {
|
||||
std::cout << "Summary: " << std::endl;
|
||||
std::cout << " Total events generated: " << results.size() << std::endl;
|
||||
size_t total_steps = 0;
|
||||
for (const auto& event : results) {
|
||||
total_steps += event.Path().size();
|
||||
}
|
||||
std::cout << " Total simulation steps: " << total_steps << std::endl;
|
||||
std::cout << " Average steps per event: " << static_cast<double>(total_steps) / results.size() << std::endl;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user