G4-ExampleB0/main.cpp

45 lines
1.2 KiB
C++
Raw Normal View History

2022-04-28 20:18:48 +08:00
#include "QGSP_BERT.hh"
#include "G4MTRunManager.hh"
#include "G4VUserActionInitialization.hh"
#include "G4UImanager.hh"
#include "G4UIExecutive.hh"
#include "G4VisExecutive.hh"
#include "Geometry.h"
#include "ParticleSource.h"
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;
}