// --------------------------------------------------------------------------
//
// Macro for reconstruction in STS from MC data
//
// Tasks:  CbmStsDigitise
//         CbmStsFindHits
//         CbmStsMatchHits
//
// V. Friese   12/09/2006
// Version     01/12/2006
//
// --------------------------------------------------------------------------


void sts_reco(Int_t nEvents = 5)
{

  // ========================================================================
  //          Adjust this part according to your requirements

  gStyle->SetPalette(1);  
  // Verbosity level (0=quiet, 1=event level, 2=track level, 3=debug)
  Int_t iVerbose = 0;
  
  TString inFile = Form("urqmd.auau.25gev.centr.mc_sts_standard_n5.root");
  
  // Background file (for MAPS pileup)
  TString bgFile = inFile;
  
  // Parameter file
  TString parFile = Form("urqmd.auau.25gev.centr.params_sts_standard_n5.root");

  // Output file
  TString outFile = Form("urqmd.auau.25gev.centr.reco_sts_standard_n5_standardDigi_n%d.root",nEvents);

  cout << "***********************************" << endl;
  cout << "** STS RECONSTRUCTION SCRIPT ******" << endl;
  cout << "** settings: **********************" << endl;
  cout << "** number of events: " << nEvents << endl;
  cout << "** input file name:" << endl;
  cout << "                    " << inFile.Data() << endl;
  cout << "** parameter file name:" << endl;
  cout << "                        " << parFile.Data() << endl;
  cout << "** output file name:" << endl;
  cout << "                     " << outFile.Data() << endl;
  cout << "***********************************" << endl;
  
  // Digitisation file
  TString digiFile = "sts_standard.gsi.digi.par";

  // ----  Load libraries   -------------------------------------------------
  gROOT->LoadMacro("$VMCWORKDIR/gconfig/basiclibs.C");
  basiclibs();
  gSystem->Load("libGeoBase");
  gSystem->Load("libParBase");
  gSystem->Load("libBase");
  gSystem->Load("libMCStack");
  gSystem->Load("libField");
  gSystem->Load("libPassive");
  gSystem->Load("libSts");
  gSystem->Load("libGen");
  gSystem->Load("libRich");
  gSystem->Load("libTrd");
  gSystem->Load("libTof");
  gSystem->Load("libGlobal");
  gSystem->Load("libKF");
  gSystem->Load("libL1");

  // ------------------------------------------------------------------------
  CbmKF* kalman= new CbmKF();
  CbmL1 *l1 = new CbmL1();
  //  l1->SetPerformanceFileName(perL1File.Data());
  //  CbmStsTrackFinder* stsTrackFinder    = new CbmStsTrackFinderIdeal();
  CbmStsTrackFinder* stsTrackFinder    = new CbmL1StsTrackFinder();
  CbmStsTrackFitter* stsTrackFitter    = new CbmStsKFTrackFitter();
  CbmPrimaryVertexFinder* pvFinder     = new CbmPVFinderKF(); 
 
  // -----   Timer   --------------------------------------------------------
  TStopwatch timer;
  timer.Start();
  // ------------------------------------------------------------------------

  // -----   Reconstruction run   -------------------------------------------
  CbmRunAna *fRun= new CbmRunAna();
  fRun->SetInputFile(inFile);
  fRun->SetOutputFile(outFile);
  // ------------------------------------------------------------------------
  
  // -----  Parameter database   --------------------------------------------
  TString stsDigiFile = gSystem->Getenv("VMCWORKDIR");
  stsDigiFile += "/parameters/sts/";
  stsDigiFile += digiFile;
  CbmRuntimeDb* rtdb = fRun->GetRuntimeDb();
  CbmParRootFileIo*  parIo1 = new CbmParRootFileIo();
  CbmParAsciiFileIo* parIo2 = new CbmParAsciiFileIo();
  parIo1->open(parFile.Data());
  parIo2->open(stsDigiFile.Data(),"in");
  rtdb->setFirstInput(parIo1);
  rtdb->setSecondInput(parIo2);
  rtdb->setOutput(parIo1);
  rtdb->saveOutput();
  fRun->LoadGeometry();
  // ------------------------------------------------------------------------

  // -----   STS digitizer   ------------------------------------------------
  CbmStsDigitize* stsDigitize = new CbmStsDigitize("STSDigitize", iVerbose);
  fRun->AddTask(stsDigitize);
  // ------------------------------------------------------------------------
  
  // ---  STS hit finding   -------------------------------------------------
  CbmStsFindHits* findHits = new CbmStsFindHits("STSFindHits", iVerbose);
  fRun->AddTask(findHits);
  // ------------------------------------------------------------------------
  
  // ---  STS hit matching   ------------------------------------------------
  CbmStsMatchHits* matchHits = new CbmStsMatchHits("STSMatchHits", 
 						   iVerbose);
  fRun->AddTask(matchHits);
  // ------------------------------------------------------------------------
  
  // -----   STS track finding   --------------------------------------------
  CbmStsFindTracks* stsFindTracks = new CbmStsFindTracks("Track Finder",
							 stsTrackFinder,
							 iVerbose);
  fRun->AddTask(kalman);
  fRun->AddTask(l1);
  fRun->AddTask(stsFindTracks);
  // ------------------------------------------------------------------------
  
  // -----   STS track matching   -------------------------------------------
  CbmStsMatchTracks* stsMatchTracks = new CbmStsMatchTracks("Match tracks",
							    iVerbose);
  fRun->AddTask(stsMatchTracks);
  // ------------------------------------------------------------------------
  
  
  // -----   STS track fitting   --------------------------------------------
  CbmStsFitTracks* stsFitTracks = new CbmStsFitTracks("STS Track Fitter",
						      stsTrackFitter,
						      iVerbose);
  fRun->AddTask(stsFitTracks);
  // ------------------------------------------------------------------------
  
  
  // -----   Primary vertex finding   ---------------------------------------
  CbmFindPrimaryVertex* pvFindTask = new CbmFindPrimaryVertex(pvFinder);
  fRun->AddTask(pvFindTask);
  // ------------------------------------------------------------------------
  
  // ----- STS find tracks quality check ------------------------------------
  CbmStsReconstructionQa* stsRecoQa = new CbmStsReconstructionQa(kTRUE,4,0.7,1);
  fRun->AddTask(stsRecoQa);
  
  
  // -----   Initialize simulation run   ----------------------------------------
  fRun->Init();
  // ------------------------------------------------------------------------

 
  // -----   Start run   ----------------------------------------------------
  fRun->Run(0,nEvents);
  // ------------------------------------------------------------------------


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


}