126 lines
4.5 KiB
C++
126 lines
4.5 KiB
C++
#include "musrDetectorConstruction.hh"
|
|
#include "musrPhysicsList.hh"
|
|
#include "musrPrimaryGeneratorAction.hh"
|
|
#include "musrRunAction.hh"
|
|
#include "musrEventAction.hh"
|
|
#include "musrStackingAction.hh"
|
|
#include "musrSteppingAction.hh"
|
|
#include "musrSteppingVerbose.hh"
|
|
|
|
#include "Randomize.hh"
|
|
|
|
#include "G4RunManager.hh"
|
|
#include "G4UImanager.hh"
|
|
#include "G4UIterminal.hh"
|
|
#include "G4UItcsh.hh" //JSL: should be commented on windows ?
|
|
|
|
// The following two lines are needed to cope with the problem of
|
|
// "Error in <TPluginManager::FindHandler>: Cannot find plugin handler for TVirtualStreamerInfo!
|
|
// Does $ROOTSYS/etc/plugins/TVirtualStreamerInfo exist?"
|
|
#include "TROOT.h"
|
|
#include "TPluginManager.h"
|
|
|
|
#include "G4VisExecutive.hh"
|
|
#include "G4TrajectoryDrawByCharge.hh" // TS Trajectory drawing by ID or charge
|
|
|
|
#include "musrRootOutput.hh"
|
|
#include "musrParameters.hh"
|
|
#include "musrErrorMessage.hh"
|
|
|
|
#include <cstdlib>
|
|
#include <memory>
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
G4String steeringFileName;
|
|
if (argc > 1) {
|
|
steeringFileName = argv[1];
|
|
} else {
|
|
G4cerr << "Usage: " << argv[0] << " <steering-file> [idle]" << G4endl;
|
|
return 1;
|
|
}
|
|
|
|
G4cout << "\n\n*************************************************************" << G4endl;
|
|
G4cout << " musrSim version 1.0.5 for Geant 11.4" << G4endl;
|
|
G4cout << " WWW: https://www.psi.ch/lmu/geant4-simulations" << G4endl;
|
|
// choose the Random engine
|
|
G4Random::setTheEngine(new CLHEP::HepJamesRandom);
|
|
|
|
// my Verbose output class
|
|
G4VSteppingVerbose::SetInstance(new musrSteppingVerbose);
|
|
|
|
// Run manager
|
|
auto runManager = std::make_unique<G4RunManager>();
|
|
|
|
// Create class "myParameters", which is a collection of many different parameters
|
|
auto myParameters = std::make_unique<musrParameters>(steeringFileName);
|
|
|
|
// Create class "musrErrorMessage"
|
|
auto myErrorMessage = std::make_unique<musrErrorMessage>();
|
|
|
|
// Create Root class for storing the output of the Geant simulation
|
|
auto myRootOutput = std::make_unique<musrRootOutput>();
|
|
|
|
// The following command is needed to cope with the problem of
|
|
// "Error in <TPluginManager::FindHandler>: Cannot find plugin handler for TVirtualStreamerInfo!
|
|
// Does $ROOTSYS/etc/plugins/TVirtualStreamerInfo exist?"
|
|
// /* magic line from Rene - for future reference! */
|
|
gROOT->GetPluginManager()->AddHandler("TVirtualStreamerInfo", "*", "TStreamerInfo", "RIO", "TStreamerInfo()");
|
|
|
|
// UserInitialization classes (mandatory)
|
|
if (argc > 1) {
|
|
G4int myRunNr = std::atoi(argv[1]); // Get the run number from the name of the
|
|
// parameter file, if it starts with a number.
|
|
if (myRunNr > 0) {
|
|
runManager->SetRunIDCounter(myRunNr);
|
|
}
|
|
}
|
|
// Keep raw pointer for cross-object writing but in principle object should be owned by runManager
|
|
// after the next line
|
|
auto musrdetector = new musrDetectorConstruction(steeringFileName);
|
|
runManager->SetUserInitialization(musrdetector);
|
|
runManager->SetUserInitialization(new musrPhysicsList);
|
|
|
|
// Visualization, if you choose to have it!
|
|
auto visManager = std::make_unique<G4VisExecutive>();
|
|
visManager->Initialize();
|
|
|
|
// UserAction classes
|
|
runManager->SetUserAction(new musrPrimaryGeneratorAction(musrdetector));
|
|
runManager->SetUserAction(new musrRunAction);
|
|
runManager->SetUserAction(new musrEventAction);
|
|
// Initiate musrStackingAction only if optical photons are required (otherwise not needed)
|
|
if (musrParameters::boolG4OpticalPhotons)
|
|
runManager->SetUserAction(new musrStackingAction);
|
|
|
|
runManager->SetUserAction(new musrSteppingAction);
|
|
|
|
// Initialize G4 kernel
|
|
runManager->Initialize();
|
|
|
|
// get the pointer to the User Interface manager
|
|
G4UImanager *UI = G4UImanager::GetUIpointer();
|
|
|
|
// Define (G)UI terminal for interactive mode
|
|
if (argc == 1) {
|
|
auto session = std::make_unique<G4UIterminal>();
|
|
UI->ApplyCommand("/control/execute vis.mac");
|
|
session->SessionStart();
|
|
} else {
|
|
G4String command = "/control/execute ";
|
|
G4String fileName = argv[1];
|
|
UI->ApplyCommand(command + fileName);
|
|
|
|
if (argc > 2) {
|
|
G4String SecondArgument = argv[2];
|
|
if (SecondArgument == "idle") {
|
|
auto session = std::make_unique<G4UIterminal>();
|
|
G4cout << "Go to idle state now:" << G4endl;
|
|
session->SessionStart();
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|