Jan Kieseler
285 lines
8.3 KiB
C++
285 lines
8.3 KiB
C++
//
|
|
// ********************************************************************
|
|
// * License and Disclaimer *
|
|
// * *
|
|
// * The Geant4 software is copyright of the Copyright Holders of *
|
|
// * the Geant4 Collaboration. It is provided under the terms and *
|
|
// * conditions of the Geant4 Software License, included in the file *
|
|
// * LICENSE and available at http://cern.ch/geant4/license . These *
|
|
// * include a list of copyright holders. *
|
|
// * *
|
|
// * Neither the authors of this software system, nor their employing *
|
|
// * institutes,nor the agencies providing financial support for this *
|
|
// * work make any representation or warranty, express or implied, *
|
|
// * regarding this software system or assume any liability for its *
|
|
// * use. Please see the license in the file LICENSE and URL above *
|
|
// * for the full disclaimer and the limitation of liability. *
|
|
// * *
|
|
// * This code implementation is the result of the scientific and *
|
|
// * technical work of the GEANT4 collaboration. *
|
|
// * By using, copying, modifying or distributing the software (or *
|
|
// * any work based on the software) you agree to acknowledge its *
|
|
// * use in resulting scientific publications, and indicate your *
|
|
// * acceptance of all terms of the Geant4 Software license. *
|
|
// ********************************************************************
|
|
//
|
|
//
|
|
/// \file exampleB4a.cc
|
|
/// \brief Main program of the B4a example
|
|
|
|
#include "DetectorConstruction.hh"
|
|
#include "ActionInitialization.hh"
|
|
|
|
#include "G4RunManagerFactory.hh"
|
|
#include "G4SteppingVerbose.hh"
|
|
#include "G4UIcommand.hh"
|
|
#include "G4UImanager.hh"
|
|
#include "G4UIExecutive.hh"
|
|
#include "G4VisExecutive.hh"
|
|
#include "FTFP_BERT.hh"
|
|
#include "Randomize.hh"
|
|
|
|
#include "ConstructionWrapper.hh"
|
|
|
|
#include "SystemBuilder.hh"
|
|
|
|
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
|
|
|
|
|
|
|
|
|
|
|
|
|
|
namespace {
|
|
void PrintUsage() {
|
|
G4cerr << " Usage: " << G4endl;
|
|
G4cerr << " exampleB4a [-m macro ] [-u UIsession] [-t nThreads] [-vDefault]"
|
|
<< G4endl;
|
|
G4cerr << " note: -t option is available only for multi-threaded mode."
|
|
<< G4endl;
|
|
}
|
|
}
|
|
|
|
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
|
|
|
|
int main2(int argc,char** argv)
|
|
{
|
|
// Evaluate arguments
|
|
//
|
|
if ( argc > 7 ) {
|
|
PrintUsage();
|
|
return 1;
|
|
}
|
|
|
|
G4String macro;
|
|
G4String session;
|
|
G4bool verboseBestUnits = true;
|
|
#ifdef G4MULTITHREADED
|
|
G4int nThreads = 0;
|
|
#endif
|
|
for ( G4int i=1; i<argc; i=i+2 ) {
|
|
if ( G4String(argv[i]) == "-m" ) macro = argv[i+1];
|
|
else if ( G4String(argv[i]) == "-u" ) session = argv[i+1];
|
|
#ifdef G4MULTITHREADED
|
|
else if ( G4String(argv[i]) == "-t" ) {
|
|
nThreads = G4UIcommand::ConvertToInt(argv[i+1]);
|
|
}
|
|
#endif
|
|
else if ( G4String(argv[i]) == "-vDefault" ) {
|
|
verboseBestUnits = false;
|
|
--i; // this option is not followed with a parameter
|
|
}
|
|
else {
|
|
PrintUsage();
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
// Detect interactive mode (if no macro provided) and define UI session
|
|
//
|
|
G4UIExecutive* ui = nullptr;
|
|
if ( ! macro.size() ) {
|
|
ui = new G4UIExecutive(argc, argv, session);
|
|
}
|
|
|
|
// Optionally: choose a different Random engine...
|
|
// G4Random::setTheEngine(new CLHEP::MTwistEngine);
|
|
|
|
// Use G4SteppingVerboseWithUnits
|
|
if ( verboseBestUnits ) {
|
|
G4int precision = 4;
|
|
G4SteppingVerbose::UseBestUnit(precision);
|
|
}
|
|
|
|
// Construct the default run manager
|
|
//
|
|
auto* runManager =
|
|
G4RunManagerFactory::CreateRunManager(G4RunManagerType::Default);
|
|
#ifdef G4MULTITHREADED
|
|
if ( nThreads > 0 ) {
|
|
runManager->SetNumberOfThreads(nThreads);
|
|
}
|
|
#endif
|
|
|
|
ConstructionWrapper * cw = new ConstructionWrapper();
|
|
cw->addLayer(1, "G4_Pb", false);
|
|
cw->addLayer(10, "G4_Si", true, 9);
|
|
cw->addLayer(1, "G4_Pb", false);
|
|
cw->addLayer(20, "G4_Si", true, 15);
|
|
cw->addLayer(4, "G4_Pb", false);
|
|
cw->addLayer(3, "G4_Si", true, 23);
|
|
|
|
|
|
// Set mandatory initialization classes
|
|
//
|
|
auto detConstruction = new B4::DetectorConstruction(cw);
|
|
|
|
runManager->SetUserInitialization(detConstruction);
|
|
|
|
auto physicsList = new FTFP_BERT;
|
|
runManager->SetUserInitialization(physicsList);
|
|
|
|
auto actionInitialization = new B4a::ActionInitialization(detConstruction);
|
|
runManager->SetUserInitialization(actionInitialization);
|
|
|
|
actionInitialization->setGeneratorProperties(100, 1000, "e-");
|
|
|
|
// Initialize visualization
|
|
//
|
|
auto visManager = new G4VisExecutive;
|
|
// G4VisExecutive can take a verbosity argument - see /vis/verbose guidance.
|
|
// G4VisManager* visManager = new G4VisExecutive("Quiet");
|
|
visManager->Initialize();
|
|
|
|
// Get the pointer to the User Interface manager
|
|
auto UImanager = G4UImanager::GetUIpointer();
|
|
|
|
// Process macro or start UI session
|
|
//
|
|
if ( macro.size() ) {
|
|
// batch mode
|
|
G4String command = "/control/execute ";
|
|
UImanager->ApplyCommand(command+macro);
|
|
}
|
|
else {
|
|
// interactive mode : define UI session
|
|
UImanager->ApplyCommand("/control/execute init_vis.mac");
|
|
if (ui->IsGUI()) {
|
|
UImanager->ApplyCommand("/control/execute gui.mac");
|
|
}
|
|
ui->SessionStart();
|
|
delete ui;
|
|
}
|
|
|
|
// Job termination
|
|
// Free the store: user actions, physics_list and detector_description are
|
|
// owned and deleted by the run manager, so they should not be deleted
|
|
// in the main() program !
|
|
|
|
delete visManager;
|
|
delete runManager;
|
|
delete cw;
|
|
return 0;
|
|
}
|
|
|
|
//later
|
|
//class Runner {
|
|
//public:
|
|
// void initialize(ConstructionWrapper CW, bool gui=false);
|
|
// void run(int nEvents, std::string partSpecies, double minEnergy, double maxEnergy);
|
|
//}
|
|
|
|
void run(ConstructionWrapper CW, int nEvents, std::string partSpecies, double minEnergy, double maxEnergy, bool gui=false){
|
|
|
|
char* argv[]={(char*)"dummy"};
|
|
|
|
// Construct the default run manager
|
|
//
|
|
auto* runManager =
|
|
G4RunManagerFactory::CreateRunManager(G4RunManagerType::Default);
|
|
runManager->SetNumberOfThreads(1);
|
|
|
|
ConstructionWrapper * cw = &CW;
|
|
G4String session;
|
|
G4UIExecutive* ui = nullptr;
|
|
if ( gui ) {
|
|
ui = new G4UIExecutive((int)1, argv, session);
|
|
}
|
|
|
|
// Set mandatory initialization classes
|
|
//
|
|
auto detConstruction = new B4::DetectorConstruction(cw);
|
|
|
|
runManager->SetUserInitialization(detConstruction);
|
|
|
|
auto physicsList = new FTFP_BERT;
|
|
runManager->SetUserInitialization(physicsList);
|
|
|
|
auto actionInitialization = new B4a::ActionInitialization(detConstruction);
|
|
runManager->SetUserInitialization(actionInitialization);
|
|
|
|
actionInitialization->setGeneratorProperties(minEnergy, maxEnergy, partSpecies);
|
|
|
|
// Initialize visualization
|
|
//
|
|
auto visManager = new G4VisExecutive;
|
|
// G4VisExecutive can take a verbosity argument - see /vis/verbose guidance.
|
|
// G4VisManager* visManager = new G4VisExecutive("Quiet");
|
|
visManager->Initialize();
|
|
|
|
// Get the pointer to the User Interface manager
|
|
auto UImanager = G4UImanager::GetUIpointer();
|
|
|
|
// Process macro or start UI session
|
|
//
|
|
|
|
|
|
if ( gui ) {
|
|
// interactive mode : define UI session
|
|
UImanager->ApplyCommand("/control/execute init_vis.mac");
|
|
if (ui->IsGUI()) {
|
|
UImanager->ApplyCommand("/control/execute gui.mac");
|
|
}
|
|
ui->SessionStart();
|
|
delete ui;
|
|
}
|
|
else {
|
|
G4String command = "/control/execute run1.mac";
|
|
UImanager->ApplyCommand("/run/initialize");
|
|
UImanager->ApplyCommand("/run/beamOn "+ std::to_string(nEvents));
|
|
}
|
|
|
|
|
|
// Job termination
|
|
// Free the store: user actions, physics_list and detector_description are
|
|
// owned and deleted by the run manager, so they should not be deleted
|
|
// in the main() program !
|
|
delete visManager;
|
|
delete runManager;
|
|
|
|
}
|
|
|
|
int main(){
|
|
|
|
ConstructionWrapper cw;
|
|
//to be moved
|
|
cw.addLayer(1, "G4_Pb", false);
|
|
cw.addLayer(10, "G4_Si", true, 9);
|
|
cw.addLayer(1, "G4_Pb", false);
|
|
cw.addLayer(20, "G4_Si", true, 15);
|
|
cw.addLayer(4, "G4_Pb", false);
|
|
cw.addLayer(3, "G4_Si", true, 23);
|
|
|
|
SystemBuilder builder;
|
|
builder.init(cw, true);
|
|
//builder.run_gui();
|
|
builder.run_batch(10000, ((std::string)"e-").data(), 1, 100);
|
|
|
|
//run(cw, 1000, "e-", 1, 100, true);
|
|
|
|
return 0;
|
|
}
|
|
|
|
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
|
|
|