#include <iostream>
using std::cout;
using std::endl;

void radlength_sim(Int_t nEvents = 1000000)
{
   TString script = TString(gSystem->Getenv("LIT_SCRIPT"));

   TString dir  = "events/radlen_trd_v13c/";
   TString mcFile = dir + "radlength.mc.0000.root";
   TString parFile = dir + "radlength.param.0000.root";

   TString caveGeom = "cave.geo";
   TString mvdGeom = "";//"mvd/mvd_v07a.geo";
   TString stsGeom = "";//"sts/sts_v12b.geo.root";
   TString richGeom = "";//"rich/rich_v08a.geo";
   TString trdGeom = "trd/trd_v13c.root";
   TString muchGeom = "";//"much/much_v12b.geo";
   TString tofGeom = "";//"tof/tof_v13b.root";

   if (script == "yes") {
      mcFile = TString(gSystem->Getenv("LIT_MC_FILE"));
      parFile = TString(gSystem->Getenv("LIT_PAR_FILE"));

      caveGeom = TString(gSystem->Getenv("LIT_CAVE_GEOM"));
      //stsGeom = TString(gSystem->Getenv("LIT_STS_GEOM"));
      trdGeom = TString(gSystem->Getenv("LIT_TRD_GEOM"));
   }

   TStopwatch timer;
   timer.Start();

   gROOT->LoadMacro("$VMCWORKDIR/macro/littrack/loadlibs.C");
   loadlibs();

   FairRunSim* run = new FairRunSim();
   run->SetName("TGeant3");              // Transport engine
   run->SetOutputFile(mcFile);          // Output file
   FairRuntimeDb* rtdb = run->GetRuntimeDb();

   run->SetMaterials("media.geo");       // Materials

   // -----   Create detectors and passive volumes   -------------------------
   if (caveGeom != "") {
      FairModule* cave = new CbmCave("CAVE");
      cave->SetGeometryFileName(caveGeom);
      run->AddModule(cave);
   }

   if ( mvdGeom != "" ) {
      FairDetector* mvd = new CbmMvd("MVD", kTRUE);
      mvd->SetGeometryFileName(mvdGeom);
      run->AddModule(mvd);
   }

   if ( stsGeom != "" ) {
      FairDetector* sts = new CbmSts("STS", kTRUE);
      sts->SetGeometryFileName(stsGeom);
      run->AddModule(sts);
      cout << "    --- " << stsGeom << endl;
   }

   if ( richGeom != "" ) {
      FairDetector* rich = new CbmRich("RICH", kTRUE);
      rich->SetGeometryFileName(richGeom);
      run->AddModule(rich);
   }

   if ( muchGeom != "" ) {
      FairDetector* much = new CbmMuch("MUCH", kTRUE);
      much->SetGeometryFileName(muchGeom);
      run->AddModule(much);
   }

   if (trdGeom != "") {
      FairDetector* trd = new CbmTrd("TRD",kTRUE );
      trd->SetGeometryFileName(trdGeom);
      run->AddModule(trd);
   }

   if ( tofGeom != "" ) {
      FairDetector* tof = new CbmTof("TOF", kTRUE);
      tof->SetGeometryFileName(tofGeom);
      run->AddModule(tof);
   }

   // -----   Create PrimaryGenerator   --------------------------------------
   FairPrimaryGenerator* primGen = new FairPrimaryGenerator();
   run->SetGenerator(primGen);

   const int RMax = 700; // Maximum radius of the station
   FairBoxGenerator* box = new FairBoxGenerator(0, 1);
   box->SetBoxXYZ(-RMax, -RMax, RMax, RMax, 0.);
   box->SetPRange(0.1, 10);
   box->SetThetaRange(0., 0.);
   box->SetPhiRange(0., 0.);
   primGen->AddGenerator(box);

/*
   FairBoxGenerator* box = new FairBoxGenerator(0, 1);
   box->SetPRange(0.1, 10);
   box->SetXYZ(0., 0., 0.);
   box->SetPhiRange(0., 360.);
   box->SetThetaRange(0., 50.);
   primGen->AddGenerator(box);
*/
   run->SetStoreTraj(kFALSE);
   run->SetRadLenRegister(kTRUE);
   // ------------------------------------------------------------------------

   run->Init();

   // -----   Runtime database   ---------------------------------------------
   Bool_t kParameterMerged = kTRUE;
   FairParRootFileIo* parOut = new FairParRootFileIo(kParameterMerged);
   parOut->open(parFile.Data());
   rtdb->setOutput(parOut);
   rtdb->saveOutput();
   rtdb->print();
   // ------------------------------------------------------------------------

   run->Run(nEvents);

   // -----   Finish   -------------------------------------------------------
   timer.Stop();
   cout << "Macro finished succesfully." << endl;
   cout << "Output file is " << mcFile << endl;
   cout << "Parameter file is " << parFile << endl;
   cout << "Real time " << timer.RealTime() << " s, CPU time " << timer.CpuTime() << "s" << endl;
   // ------------------------------------------------------------------------
}