// Macro for running reconstruction

#if !defined(__CINT__) || defined(__MAKECINT__)
// ROOT includes
#include "TString.h"
#include "TStopwatch.h"
#include "TSystem.h"
#include "TChain.h"

// Fair includes
#include "FairRunAna.h"
#include "FairRuntimeDb.h"
#include "FairParRootFileIo.h"
#include "FairTask.h"
#include "FairField.h"
#include "FairTrackParP.h"

// MPD includes
#include "MpdTpcHitProducer.h"
#include "TpcClearerTask.h"
#include "TpcClusterizerTask.h"
//#include "TpcClusterizerTaskQA.h"
#include "TpcDriftTask.h"
//#include "TpcDriftTaskQA.h"
#include "TpcMWPCTask.h"
//#include "TpcMWPCTaskQA.h"
#include "TpcPadResponseTask.h"
//#include "TpcPadResponseTaskQA.h"
#include "TpcADCTask.h"
#include "MpdTpcClusterFinderTask.h"
//#include "MpdTpcClusterFinderTaskQA.h"
#include "TpcDistributor.h"
#include "TpcHitFinderTask.h"
#include "MpdKalmanFilter.h"
#include "TpcLheHitsMaker.h"
#include "MpdVertexZfinder.h"
#include "MpdTpcKalmanFilter.h"
#include "MpdKfPrimaryVertexFinder.h"
#include "MpdTofHitProducer.h"
#include "MpdEtofHitProducer.h"
#include "MpdEctTrackFinderTpc.h"
#include "MpdEctTrackFinderTof.h"
#include "MpdTofMatching.h"
#include "MpdEtofMatching.h"
#include "MpdFillDstTask.h"
#include "MpdGetNumEvents.h"

#include <iostream>
using namespace std;
#endif

// inFile - input file with MC data, default: evetest.root
// nStartEvent - number (start with zero) of first event to process, default: 0
// nEvents - number of events to process, 0 - all events of given file will be proccessed, default: 1
// outFile - output file with reconstructed data, default: mpddst.root
// run_type - proof execution, default "local". e.g.:
//      "proof" - run on proof-lite with "CPU" count workers,
//      "proof:workers=3" - run on proof-lite with 3 workers
//      "proof:user@proof.server:21001" - to run on the PROOF cluster created with PoD (under user 'MPD', default port - 21001)
//      "proof:user@proof.server:21001:workers=10" - to run on the PROOF cluster created with PoD with 10 workers (under USER, default port - 21001)
//	nc-farm : proof:mpd@nc8.jinr.ru:21001
void reco(TString inFile = "$VMCWORKDIR/macro/mpd/evetest.root", TString outFile = "mpddst.root", Int_t nStartEvent = 0, Int_t nEvents = 10000, TString run_type="local")
{
  // ========================================================================
  // Verbosity level (0=quiet, 1=event level, 2=track level, 3=debug)
  Int_t iVerbose = 1;
 nEvents =100;
  // Parameter file
  TString parFile = inFile;

  // ----  Load libraries   -------------------------------------------------
  gROOT->LoadMacro("$VMCWORKDIR/macro/mpd/mpdloadlibs.C");
  mpdloadlibs(kTRUE);       // load full set of main libraries

  gSystem->Load("libXMLIO");

  gROOT->LoadMacro("$VMCWORKDIR/macro/mpd/geometry_v2.C");
  geometry_v2(0x0, kFALSE);
  // ------------------------------------------------------------------------

  // -----   Timer   --------------------------------------------------------
  TStopwatch timer;
  timer.Start();
  // ------------------------------------------------------------------------

  // -----   Digitization run   -------------------------------------------
  int ind = run_type.Index(':');
  TString proof_name = "";
  if (ind >= 0){
      proof_name = run_type(ind+1,run_type.Length()-ind-1);
      run_type = run_type(0, ind);
  }

  FairRunAna *fRun= new FairRunAna(run_type, proof_name);
  fRun->SetInputFile(inFile);
  //fRun->AddFriend(inFile);
  fRun->SetOutputFile(outFile);
  fRun->SetProofParName("$VMCWORKDIR/gconfig/libMpdRoot.par");
  // ------------------------------------------------------------------------

  // -----  Parameter database   --------------------------------------------
  FairRuntimeDb* rtdb = fRun->GetRuntimeDb();
  FairParRootFileIo* parInput1 = new FairParRootFileIo();
  parInput1->open(parFile.Data());
  rtdb->setFirstInput(parInput1);
  rtdb->saveOutput();
  rtdb->print();
  // fRun->LoadGeometry();  // EL

  // ------------------------------------------------------------------------

  MpdKalmanFilter *kalman = MpdKalmanFilter::Instance("KF");
  fRun->AddTask(kalman);

  //FairTask* trackMS = new TpcLheHitsMaker("Hit producer");
  //fRun->AddTask(trackMS);

  //TpcDistributor* tpcDistributor = new TpcDistributor(10000, kTRUE, kTRUE);
  //fRun->AddTask(tpcDistributor);

  //MpdTpcClusterFinderTask *tpcClusterFinder = new MpdTpcClusterFinderTask();
  //fRun->AddTask( tpcClusterFinder );

  MpdTpcHitProducer* hitPr = new MpdTpcHitProducer();
  hitPr->SetModular(1);
  fRun->AddTask(hitPr);

 // FairTask* reco = new MpdEctTrackFinderTpc();
 // fRun->AddTask(reco);

  //FairTask* vertZ = new MpdVertexZfinder();
  //fRun->AddTask(vertZ);

  FairTask* recoKF = new MpdTpcKalmanFilter("Kalman filter");
  fRun->AddTask(recoKF);

 // FairTask* findVtx = new MpdKfPrimaryVertexFinder("Vertex finder");
  //fRun->AddTask(findVtx);

  // TOF hit producers
 // MpdTofHitProducer* tofHit = new MpdTofHitProducer("Hit producer");
  //fRun->AddTask(tofHit);

  // TOF matching
 // MpdTofMatching* tofMatch = new MpdTofMatching("TOF matching");
  //fRun->AddTask(tofMatch);

//  FairTask *tdigi= new MpdZdcDigiProducer("MpdZdcDigiProducer");
 // fRun->AddTask(tdigi);

  MpdFillDstTask* fillDST = new MpdFillDstTask("MpdDst task");
  fRun->AddTask(fillDST);

  // -----   Intialise   ----------------------------------------------------
  fRun->Init();
  if (run_type != "proof") cout << "Field: " << fRun->GetField()->GetBz(0.,0.,0.) << endl;
  else{
    TProof* pProof = fRun->GetProof();
    pProof->SetParameter("PROOF_PacketizerStrategy", (Int_t)0);
    ind = proof_name.Index(":workers=");
    if (ind >= 0){
	TString worker_count = proof_name(ind+9,proof_name.Length()-ind-9);
	if (worker_count.IsDigit())
	   pProof->SetParallel(worker_count.Atoi());
    }
  }

  // if nEvents is equal 0 then all events of the given file starting with "nStartEvent" should be processed
  if (nEvents == 0)
      nEvents = MpdGetNumEvents::GetNumROOTEvents(inFile.Data()) - nStartEvent;

  // -----   Run   ______________--------------------------------------------
  fRun->Run(nStartEvent, nStartEvent+nEvents);
  // ------------------------------------------------------------------------

  // -----   Finish   -------------------------------------------------------

  delete fRun;

  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;
  // ------------------------------------------------------------------------
}