remoll.cc

/*!
  remoll - 12 GeV Moller Simluation

  Seamus Riordan, et al.
  riordan@jlab.org

*/

#include "G4Types.hh"

#ifdef G4MULTITHREADED
#include "G4MTRunManager.hh"
typedef G4MTRunManager RunManager;
#else
#include "G4RunManager.hh"
typedef G4RunManager RunManager;
#endif

#include "G4Version.hh"
#include "G4UImanager.hh"

#include "remollRun.hh"
#include "remollRunData.hh"

#include "remollPhysicsList.hh"
#include "remollActionInitialization.hh"
#include "remollDetectorConstruction.hh"
#include "remollParallelConstruction.hh"

#include "G4VisExecutive.hh"
#include "G4UIExecutive.hh"

#include "TROOT.h"

#ifdef __APPLE__
#include <unistd.h>
#endif

#include <ctime>
#include <fstream>

namespace {
  void PrintUsage() {
    G4cerr << "Usage: " << G4endl;
    G4cerr << " remoll [-g geometry] [-m macro] [-u UIsession] [-r seed] ";
#ifdef G4MULTITHREADED
    G4cerr << "[-t nThreads] ";
#endif
    G4cerr << "[macro]" << G4endl;
  }
}

int main(int argc, char** argv) {

    // Running time measurement: start
    clock_t tStart = clock();

    #if ROOT_VERSION_CODE >= ROOT_VERSION(6,0,0)
    // Fix for #40: avoids LLVM/GL errors, but only for ROOT 6:
    // make sure gROOT is loaded before LLVM by using it first
    gROOT->Reset();
    #endif

    // Initialize the random seed
    G4long seed = time(0) + (int) getpid();
    // Open /dev/urandom
    std::ifstream urandom("/dev/urandom", std::ios::in | std::ios::binary);
    // If stream is open
    if (urandom) {
      urandom.read(reinterpret_cast<char*>(&seed), sizeof(seed));
      urandom.close();
      seed = labs(seed);
    } else G4cerr << "Can't read /dev/urandom." << G4endl;


    // Parse command line options
    G4String macro;
    G4String session;
    G4String geometry_gdmlfile;
    G4String parallel_gdmlfile;
#ifdef G4MULTITHREADED
    G4int threads = 0;
#endif
    //
    for (G4int i = 1; i < argc; ++i) {
      if      (G4String(argv[i]) == "-m") macro    = argv[++i];
      else if (G4String(argv[i]) == "-g") geometry_gdmlfile = argv[++i];
      else if (G4String(argv[i]) == "-p") parallel_gdmlfile = argv[++i];
      else if (G4String(argv[i]) == "-u") session  = argv[++i];
      else if (G4String(argv[i]) == "-r") seed     = atol(argv[++i]);
#ifdef G4MULTITHREADED
      else if (G4String(argv[i]) == "-t") threads  = atoi(argv[++i]);
#endif
      else if (argv[i][0] != '-') macro = argv[i];
      else {
        PrintUsage();
        return 1;
      }
    }


    //-------------------------------
    // Initialization of Run manager
    //-------------------------------
    RunManager* runManager = new RunManager;
    #ifdef G4MULTITHREADED
    if (threads > 0) runManager->SetNumberOfThreads(threads);
    #endif

    // Set the default random seed
    G4Random::setTheSeed(seed);

    // Detector geometry
    G4String material_name = "material";
    remollDetectorConstruction* detector = new remollDetectorConstruction(material_name, geometry_gdmlfile);
    // Parallel world geometry
    G4String parallel_name = "parallel"; // Note: name must correspond with name of G4ParallelWorldPhysics
    remollParallelConstruction* parallel = new remollParallelConstruction(parallel_name, parallel_gdmlfile);
    detector->RegisterParallelWorld(parallel);
    runManager->SetUserInitialization(detector);

    // Physics list
    remollPhysicsList* physlist = new remollPhysicsList();
    runManager->SetUserInitialization(physlist);

    // Run action
    remollActionInitialization* useraction = new remollActionInitialization();
    runManager->SetUserInitialization(useraction);

    //----------------
    // Visualization:
    //----------------
    // Initialize visualization
    //
    // Verbose level "warnings" (3) as is too noisy during initialization
    G4VisManager* visManager = new G4VisExecutive("quiet");
    visManager->Initialize();
    visManager->SetVerboseLevel("warnings");

    // Get the pointer to the User Interface manager
    G4UImanager* UImanager = G4UImanager::GetUIpointer();
    // use double precision
    #if G4VERSION_NUMBER >= 1031
    G4UImanager::UseDoublePrecisionStr(true);
    #else
    G4cout << "remoll: Warning: issue 130, double precision input may be truncated." << G4endl;
    G4cout << "remoll: see also https://github.com/JeffersonLab/remoll/issues/130" << G4endl;
    #endif

    // Define UI session for interactive mode
    if (macro.size())
    {
      // Run in batch mode
      // Copy contents of macro into buffer to be written out into ROOT file
      remollRun::GetRunData()->SetMacroFile(macro);
      UImanager->ExecuteMacroFile(macro);
    } else {
      // Define UI session for interactive mode
      G4UIExecutive* ui = new G4UIExecutive(argc,argv,session);
      if (ui->IsGUI())
        UImanager->ApplyCommand("/control/execute macros/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;


    // Running time measurement: end
    clock_t tEnd = clock();
    G4cout << " Running time[s]: "
        << double(tEnd - tStart) / double(CLOCKS_PER_SEC)
        << G4endl;

    // Return success
    return 0;
}