Files
musrsim/musrSim.cc

130 lines
4.6 KiB
C++

#include "version.h"
#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;
G4cout << "\n\n*************************************************************" << G4endl;
G4cout << " musrSim version " << MUSRSIM_VERSION << " for Geant 11.4" << G4endl;
G4cout << " WWW: https://www.psi.ch/lmu/geant4-simulations\n" << G4endl;
if (argc > 1) {
steeringFileName = argv[1];
} else {
G4cerr << "Usage: " << argv[0] << " <steering-file> [idle]" << G4endl;
return 1;
}
// 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;
}