G4-ExampleB0/main.cpp

43 lines
1.2 KiB
C++

#include "G4MTRunManager.hh"
#include "G4ScoringManager.hh"
#include "G4UIExecutive.hh"
#include "G4UImanager.hh"
#include "G4VUserActionInitialization.hh"
#include "G4VisExecutive.hh"
#include "Geometry.h"
#include "ParticleSource.h"
#include "QGSP_BERT.hh"
class ActionInitialzation : public G4VUserActionInitialization {
public:
virtual void Build() const { SetUserAction(new GeneratorAction); }
};
int main(int argc, char** argv) {
G4UIExecutive* ui = nullptr;
if (argc == 1) ui = new G4UIExecutive(argc, argv);
auto runManager = new G4MTRunManager;
auto visManager = new G4VisExecutive;
auto UIManager = G4UImanager::GetUIpointer();
runManager->SetUserInitialization(new DetectorConstruction());
runManager->SetUserInitialization(new QGSP_BERT());
runManager->SetUserInitialization(new ActionInitialzation());
visManager->Initialize();
if (!ui) {
G4String command = "/control/execute";
G4String fileName = argv[1];
UIManager->ApplyCommand(command + fileName);
} else {
UIManager->ApplyCommand("/control/execute vis.mac");
ui->SessionStart();
delete ui;
}
delete runManager;
delete visManager;
return 0;
}