//
// hdgeant4.cc : main program for hdgeant4 physics simulation
//
// author: richard.t.jones at uconn.edu
// version: august 15, 2013
//

#include <GlueXUserOptions.hh>
#include <GlueXDetectorConstruction.hh>
#include <GlueXUserActionInitialization.hh>
#include <GlueXPhysicsList.hh>

#include <DANA/DApplication.h>
#include <unistd.h>

#include <G4MTRunManager.hh>
#include <G4RunManager.hh>
#include <G4UImanager.hh>

#ifdef G4VIS_USE
#include <G4VisExecutive.hh>
#endif

#ifdef G4UI_USE
#include <G4UIExecutive.hh>
#include <G4UIterminal.hh>
#include <G4UItcsh.hh>
#endif

int run_number = 0;

void usage()
{
   G4cout << G4endl
          << "Usage: hdgeant4 [options] [<batch.mac>]" << G4endl
          << " where options include:" << G4endl
          << "    -v : open a graphics window for visualization" << G4endl
          << "    -tN : start N worker threads, default 1" << G4endl
          << "    -rN : set run to N, default taken from control.in" << G4endl
          << G4endl;
   exit(9);
}

int main(int argc,char** argv)
{
   // Initialize the jana framework
   DApplication dapp(argc, argv);
   dapp.Init();

   // Interpret special command-line arguments
   int use_visualization = 0;
   int worker_threads = 1;
   int argi=1;
   int c;
   while ((c = getopt(argc, argv, "vt:r:")) != -1) {
      if (c == 'v') {
         use_visualization = 1;
         argi += 1;
      }
      else if (c == 't') {
         worker_threads = atoi(optarg);
         argi += 2;
      }
      else if (c == 'r') {
         run_number = atoi(optarg);
         argi += 2;
      }
      else {
         usage();
      }
   }

   // Read user options from file control.in
   GlueXUserOptions opts;
   if (! opts.ReadControl_in("control.in")) {
      exit(3);
   }
   if (run_number == 0) {
      std::map<int, int> runno_opts;
      if (opts.Find("RUNNO", runno_opts) || opts.Find("RUNG", runno_opts))
         run_number = runno_opts[1];
   }
   else {
      G4cerr << "Warning - "
             << "no run number specified in control.in, "
             << "default value of 9000 assumed." << G4endl;
      run_number = 9000;
   }

   // Declare our G4VSteppingVerbose implementation
   G4VSteppingVerbose::SetInstance(new GlueXSteppingVerbose());

   // Run manager handles the rest of the initialization
#ifdef G4MULTITHREADED
   G4MTRunManager runManager;
   runManager.SetNumberOfThreads(worker_threads);
#else
   G4RunManager runManager;
#endif

   // Geometry initialization
   GlueXDetectorConstruction *geometry = new GlueXDetectorConstruction();
   int Npara = geometry->GetParallelWorldCount();
   for (int para=1; para <= Npara; ++para) {
      G4String name = geometry->GetParallelWorldName(para);
      G4LogicalVolume *topvol = geometry->GetParallelWorldVolume(para);
      GlueXParallelWorld *parallelWorld = new GlueXParallelWorld(name,topvol);
      geometry->RegisterParallelWorld(parallelWorld);
   }
   runManager.SetUserInitialization(geometry);

   // Physics process initialization
   runManager.SetUserInitialization(new GlueXPhysicsList());
    
   // User actions initialization
   runManager.SetUserInitialization(new GlueXUserActionInitialization());

   // Initialize G4 kernel
   runManager.Initialize();
       
   // Initialize graphics (option -v)
   G4VisManager* visManager = 0;
   if (use_visualization) {
#ifdef G4VIS_USE
      visManager = new G4VisExecutive();
      visManager->Initialize();
#else
      G4cerr << "Visualization system not available,"
             << " please rebuild hdgeant4 with visualization enabled."
             << G4endl;
      exit(1);
#endif
   }

   // Start the user interface
   G4UImanager * UImanager = G4UImanager::GetUIpointer();  
   if (argc > argi) {   // batch mode  
      G4String command = "/control/execute ";
      G4String fileName = argv[argi];
      UImanager->ApplyCommand(command + fileName);
   }
   else if (visManager) {    // interactive mode with visualization
#ifdef G4UI_USE_EXECUTIVE
      G4UIExecutive UIexec(argc, argv, "qt");
      UImanager->ApplyCommand("/control/execute vis.mac");     
      UIexec.SessionStart();
#else
      G4UIterminal UIterm(new G4UItcsh());
      UImanager->ApplyCommand("/control/execute vis.mac");     
      UIterm.SessionStart();
#endif
   }
   else {    // interactive mode without visualization
      G4UIterminal UIterm(new G4UItcsh());
      UIterm.SessionStart();
   }
 
   // Clean up and exit
   if (visManager)
      delete visManager;
   return 0;
}