#include <ctime>
#include <cstdlib>		

#include <sys/types.h>
#include <sys/time.h>

// for atoi, At first, you should check if the standard C++ is provided. 
// If not, you have to change the left header file to <stdio.h> 

#include "G4RunManager.hh"
#include "G4UImanager.hh"
#include "G4UIterminal.hh"
#include "G4UItcsh.hh"
#include "G4ios.hh"

#ifdef G4UI_USE_XM
#include "G4UIXm.hh"
#endif

#ifdef G4UI_USE_WIN32
#include "G4UIWin32.hh"
#endif

#include "LGDetectorConstruction.hh"
#include "LGPhysicsList.hh"
#include "LGPhysicsListMessenger.hh"
#include "LGPrimaryGeneratorAction.hh"
#include "LGEventAction.hh"
#include "LGRunAction.hh"
#include "LGSteppingAction.hh"
#include "LGStackingAction.hh"
#include "LGTrackingAction.hh"
#include "LGTrackingMessenger.hh"

#include "TFile.h"
#include "TTree.h"
#include "TH1.h"

#include "Randomize.hh"

#ifdef G4VIS_USE
#include "G4VisExecutive.hh"
#endif

void Usage(){
  G4cout <<"Usage: ";
  G4cout <<"       LGGui [options] "<<G4endl;
  
  G4cout <<"        -roff    : turn off random gun direction"<<G4endl;
  G4cout <<"        -f foo   : non interactive macro file name"<<G4endl;
  G4cout <<"        -l #     : length of tappered in cm (default is 11.0cm)"<<G4endl;
  G4cout <<"        -r #     : length of round strait section in cm (default is 7.0cm)"<<G4endl;
  G4cout <<"        -b #     : bend radius in cm (default is 100.0 cm)"<<G4endl;
  G4cout <<"        -a #     : bend angle in deg (default is 10.0 deg)"<<G4endl;
  G4cout <<"        -s       : bend section is before the tapered section (default is after) "<<G4endl;
  G4cout <<"        -n #     : number of events to generate"<<G4endl;
  G4cout <<"        -o foo   : root output file name"<<G4endl;
  G4cout <<"        -nqe     : set quantum efficiency of photo cathode to unity(default FEU-84-3)"<<G4endl; 

  return;
}

void makeHistos(TH1F**, TH2F**, G4int);

int main(int argc, char** argv)
{
  G4int n_Events=0;
  
  G4double zLG = 11.0;      // default length of tappered section of light guide in cm
  G4double BendR = 100.;    // default bending radius of torus section
  G4double BendA = 10.;     // default bend angle of torus section
  G4String file_Name;

  G4double Lround = 7.0;    // default length of round strait section where PMT couples to
  G4int BendFirst = 0;
  G4int Interactive = 1;
  G4int RandomDir = 1;
  G4int LGs = 0;
  G4int QE = 1;
  
  char output_FileName[128] = "lg.root"; // 'ROOT' output file name in Bach mode
  
  for (int i=1;i<argc;i++){
    
    if (!strcmp(argv[i],"-n")){
      i++;
      n_Events	= atoi(argv[i]);
      Interactive = 0; 
    }
    if (!strcmp(argv[i],"-r")){
      i++;
      Lround = atoi(argv[i]);
    }
    if (!strcmp(argv[i],"-s")){
      BendFirst = 1; 
    }
    if (!strcmp(argv[i],"-roff")){
      RandomDir = 0; 
    }
    if (!strcmp(argv[i],"-l")){
      i++;
      zLG       = atof(argv[i]);
    }
    if (!strcmp(argv[i],"-b")){
      i++;
      BendR       = atof(argv[i]);
    }
    if (!strcmp(argv[i],"-a")){
      i++;
      BendA       = atof(argv[i]);
    }
    if (!strcmp(argv[i],"-f")){
      i++;
      file_Name = argv[i];
      Interactive = 0;
    }
    if (!strcmp(argv[i],"-o")){
      i++;
      sprintf(output_FileName,"%s",argv[i]);
    }
    if (!strcmp(argv[i],"-nqe")){
      QE = 0;
    }
    if (!strcmp(argv[i],"-h"    ) ||
	!strcmp(argv[i],"-help" ) ||
	!strcmp(argv[i],"--h"   ) ||
	!strcmp(argv[i],"--help")){
      Usage();
      return 1;
    }
  }


  //----------------------------------------------------------------------
  //-------- 0. Seed the random number generator manually
  //----------------------------------------------------------------------
  //  unsigned long myseed = time(NULL);
  struct timeval t1;
  struct timezone tz;
  gettimeofday(&t1, &tz);
  G4cout<<t1.tv_usec<<G4endl;
  unsigned long myseed = ((unsigned long) t1.tv_usec)+time(NULL);
  G4cout<<"seed = "<<myseed<<G4endl;
  CLHEP::HepRandom::setTheSeed(myseed);
  
  //----------------------------------------------------------------------
  //-------- 1. Construction of the default run manager
  //----------------------------------------------------------------------
  G4RunManager* run_Manager = new G4RunManager;	
  
  //----------------------------------------------------------------------
  //-------- 2. Mandatory initialization class set-up
  //----------------------------------------------------------------------

  LGDetectorConstruction* detector_Construction = new 
    LGDetectorConstruction(zLG, Lround, BendR, BendA, BendFirst); 

  run_Manager->SetUserInitialization(detector_Construction);
  run_Manager->SetUserInitialization(new LGPhysicsList);
  
  //----------------------------------------------------------------------
  //-------- 3. Instantiation and Initialization of the Visualization Manager
  //----------------------------------------------------------------------
  
  
  G4UIsession* session = NULL;
  G4VisManager* vis_Manager = NULL;
  if (Interactive){   // Define UI session for interactive mode.
    // G4UIterminal is a (dumb) terminal.
#if defined(G4UI_USE_XM)
    session = new G4UIXm(argc, argv);
#elif defined(G4UI_USE_WIN32)
    session = new G4UIWin32();
#elif defined(G4UI_USE_TCSH)
    session = new G4UIterminal(new G4UItcsh);      
#else
    session = new G4UIterminal();
#endif
    
#ifdef G4VIS_USE
    G4cout << "Instantiating Visualization Manager!" << G4endl;
    vis_Manager = new G4VisExecutive;
    vis_Manager->Initialize();
#endif
  }
  
  //----------------------------------------------------------------------
  //-------- 4. User action class set-up
  //----------------------------------------------------------------------
  LGPrimaryGeneratorAction* generator_Action = new LGPrimaryGeneratorAction(RandomDir);
  run_Manager->SetUserAction(generator_Action);
  
  TFile* f = NULL;
  TTree* LG_Tree = NULL;
  if (!Interactive){
    f = new TFile(output_FileName, "RECREATE");
    LG_Tree = new TTree("LGTree", "LGTree"); // create a tree in Bach mode
  }
  
  // setup histogram to record Cerenkov photon wavelength
  TH1F *spectra[20];
  TH2F *spectra2d[6];

  makeHistos(spectra, spectra2d, BendFirst);

  LGEventAction* event_Action = new LGEventAction(detector_Construction, 
						  generator_Action,
						  Interactive, LG_Tree);
  run_Manager->SetUserAction(event_Action);
  run_Manager->SetUserAction(new LGRunAction(detector_Construction, 
					     event_Action));
  run_Manager->SetUserAction(new LGSteppingAction(event_Action, zLG, QE, 
						  spectra,spectra2d, BendFirst));
  run_Manager->SetUserAction(new LGStackingAction(event_Action));
  run_Manager->SetUserAction(new LGTrackingAction);
  
  //			----------------------------------------------------------------------
  //			-------- 5. Initialization of G4 kernel
  //			----------------------------------------------------------------------
  run_Manager->Initialize();
  
  //			----------------------------------------------------------------------
  //			-------- 6. UI Session
  //			----------------------------------------------------------------------

  G4UImanager* UI = G4UImanager::GetUIpointer();		// Getting the pointer to UI manager

  if (session){   // Define UI session for interactive mode.
    
    // G4UIterminal is a (dumb) terminal.
    UI->ApplyCommand("/control/execute vis.mac");    
#if defined(G4UI_USE_XM) || defined(G4UI_USE_WIN32)
    // Customize the G4UIXm,Win32 menubar with a macro file :
    UI->ApplyCommand("/control/execute gui.mac");
#endif
    session->SessionStart();
    delete session;
  }
  else{		// Batch mode
    G4cout << "IN BATCH MODE!" << G4endl;
    UI->ApplyCommand("/control/saveHistory history.txt");
    UI->ApplyCommand("/event/verbose 0");
    UI->ApplyCommand("/tracking/verbose 0");
    UI->ApplyCommand("/phys/verbose 0");
    UI->ApplyCommand("/control/verbose 2");
    UI->ApplyCommand("/run/verbose 2");
    
    G4String command = "/control/execute ";
    UI->ApplyCommand(command+file_Name);

    G4cout<<"Generate "<<n_Events<< " events"<<G4endl;
    run_Manager->BeamOn(n_Events);
  }
  
  //			----------------------------------------------------------------------
  //			-------- 7. Job Termination	
  //			----------------------------------------------------------------------
  if(Interactive){
#ifdef G4VIS_USE
    delete vis_Manager;
#endif
  }
  delete run_Manager;
  
  if (!Interactive){
    G4double tin   = spectra2d[1]->GetEntries();
    G4double tout  = spectra2d[2]->GetEntries();
    G4double tout1 = spectra2d[3]->GetEntries();
    G4double tout2 = spectra2d[4]->GetEntries();
    G4double tout3 = spectra2d[5]->GetEntries();
    if (BendFirst){
      G4cout<<"Loss in bending section: "<< (1.-tout/tin)*100. <<"%"<<G4endl;
      G4cout<<"Loss in tapered section: "<< (1.-tout1/tout)*100. <<"%"<<G4endl;
    } else {
      G4cout<<"Loss in tapered section: "<< (1.-tout/tin)*100. <<"%"<<G4endl;
      G4cout<<"Loss in bending section: "<< (1.-tout1/tout)*100. <<"%"<<G4endl;
    }
    G4cout<<"Loss in round section    : "<< (1.-tout2/tout1)*100. <<"%"<<G4endl;
    G4cout<<"Loss TOTAL               : "<< (1.-tout3/tin)*100. <<"%"<<G4endl;
    LG_Tree->Write();
    for (int i=0;i<6;i++){
      spectra2d[i]->Write();
    }
    f->Close();
  }


  G4cout << "END OF PROGRAM: Success!" << G4endl;

  return 0;
}


void makeHistos(TH1F** hist1d, TH2F** hist2d, G4int BF){

  hist2d[0] = new TH2F("hist2d0","photons hit entrance of light guide",
                          80,-40.,40.,80,-40.,40.);
  hist2d[0]->GetXaxis()->SetTitle("Horizontal distance [mm]");
  hist2d[0]->GetYaxis()->SetTitle("Vertical distance [mm]");
  if (BF){
    hist2d[1] = new TH2F("hist2d1","photons hit start bending square section",
			 80,-40.,40.,80,-40.,40.);
  } else {
    hist2d[1] = new TH2F("hist2d1","photons hit start of Tappered section",
			 80,-40.,40.,80,-40.,40.);
  }
  hist2d[1]->GetXaxis()->SetTitle("Horizontal distance [mm]");
  hist2d[1]->GetYaxis()->SetTitle("Vertical distance [mm]");
  if (BF){
    hist2d[2] = new TH2F("hist2d2","photons hit start of tapered section",
			 80,-40.,40.,80,-40.,40.);
  } else {
    hist2d[2] = new TH2F("hist2d2","photons hit start of round bend section",
			 80,-40.,40.,80,-40.,40.);
  }
  hist2d[2]->GetXaxis()->SetTitle("Horizontal distance [mm]");
  hist2d[2]->GetYaxis()->SetTitle("Vertical distance [mm]");

  hist2d[3] = new TH2F("hist2d3","photons hit start of round strait section",
		       80,-40.,40.,80,-40.,40.);
  hist2d[3]->GetXaxis()->SetTitle("Horizontal distance [mm]");
  hist2d[3]->GetYaxis()->SetTitle("Vertical distance [mm]");

  hist2d[4] = new TH2F("hist2d4","photons hit PMT glass",
                          80,-40.,40.,80,-40.,40.);
  hist2d[4]->GetXaxis()->SetTitle("Horizontal distance [mm]");
  hist2d[4]->GetYaxis()->SetTitle("Vertical distance [mm]");

  hist2d[5] = new TH2F("hist2d5","photons hit PMT cathode",
                          80,-40.,40.,80,-40.,40.);
  hist2d[5]->GetXaxis()->SetTitle("Horizontal distance [mm]");
  hist2d[5]->GetYaxis()->SetTitle("Vertical distance [mm]");


}