// -------------------------------------------------------------------------
// -----                    PndMasterRecoIdealTask source file              -----
// -----                   Created 31/01/16  by S. Spataro			 	------
// -----				   Edited 28/07/16 	 by J. Puetz
// -----                   Wrapper for  recotizing tasks               -----
// -------------------------------------------------------------------------


#include "PndMasterRecoIdealTask.h"
#include "PndMasterTask.h"

#include "FairGeane.h"
#include "PndTrkTracking2.h"
#include "PndSttMvdGemTrackingIdeal.h"
#include "PndMCTrackAssociator.h"
#include "PndRecoKalmanTask.h"
#include "PndFtsTrackerIdeal.h"

/**
 * @brief Default Constructor
 * @details # Reconstruction task list
 * Here all the reconstruction tasks are added to the task, with the standard settings. A check is done after each task if the tasklist enum is broken or not. At the end the event counter is added (each 100 events), and the verbosity is set to 0 to all the tasks (it can be changed afterwards with SetVerbosity() functions.
 **/
// -----   Default constructor   -------------------------------------------
PndMasterRecoIdealTask::PndMasterRecoIdealTask() :
  PndMasterTask("Master Reconstruction Task")
{
  
  // -----   Geane   ---------------------------------------
  this->Add(new FairGeane()); // 0
  if ((this->GetListOfTasks()->GetSize()-1) != kFairGeane) Error("PndMasterDigiTask","Error in task #%i", (this->GetListOfTasks()->GetSize()-1));


  // ----- MVD + STT + GEM Pattern Recognition --------------
  PndSttMvdGemTrackingIdeal *SttMvdGemTracking = NULL;
  this->Add(SttMvdGemTracking = new PndSttMvdGemTrackingIdeal()); // 2
  if ((this->GetListOfTasks()->GetSize()-1) != kPndSttMvdGemTracking) Error("PndMasterDigiTask","Error in task #%i", (this->GetListOfTasks()->GetSize()-1));
  SttMvdGemTracking->SetTrackOutput("SttMvdGemIdealTrack");
  SttMvdGemTracking->SetPersistence(kFALSE);


  // ----- Barrel Kalman Task     ----------------------------
  PndRecoKalmanTask* recoKalman = NULL;
  this->Add(recoKalman = new PndRecoKalmanTask()); // 4
  if ((this->GetListOfTasks()->GetSize()-1) != kPndRecoKalmanTask1) Error("PndMasterDigiTask","Error in task #%i", (this->GetListOfTasks()->GetSize()-1));
  recoKalman->SetTrackInBranchName("SttMvdGemIdealTrack");
//  recoKalman->SetTrackInIDBranchName("SttMvdGemIdealTrackID");
  recoKalman->SetTrackOutBranchName("SttMvdGemGenTrack");
  recoKalman->SetBusyCut(50); // CHECK to be tuned
  //recoKalman->SetIdealHyp(kTRUE);
  //recoKalman->SetNumIterations(3);
  recoKalman->SetTrackRep(0); // 0 Geane (default), 1 RK
  //recoKalman->SetPropagateToIP(kFALSE);
  
  // ----- MC Association #2 ---------------------------------
  PndMCTrackAssociator* trackMC2 = NULL;
  this->Add(trackMC2 = new PndMCTrackAssociator()); // 5
  if ((this->GetListOfTasks()->GetSize()-1) != kPndMCTrackAssociator2) Error("PndMasterDigiTask","Error in task #%i", (this->GetListOfTasks()->GetSize()-1));
  trackMC2->SetTrackInBranchName("SttMvdGemGenTrack"); 
  trackMC2->SetTrackOutBranchName("SttMvdGemGenTrackID");
  
  // -----  FTS Ideal Tracking    ----------------------------
  PndFtsTrackerIdeal* trackFts = NULL;
  this->Add(trackFts = new PndFtsTrackerIdeal()); // 6
  if ((this->GetListOfTasks()->GetSize()-1) != kPndFtsTrackerIdeal) Error("PndMasterDigiTask","Error in task #%i", (this->GetListOfTasks()->GetSize()-1));
  trackFts->SetRelativeMomentumSmearing(0.05);
  trackFts->SetVertexSmearing(0.05, 0.05, 0.05);
  trackFts->SetTrackingEfficiency(1.);
  trackFts->SetTrackOutput("FtsIdealTrack");
  trackFts->SetPersistence(kFALSE);

  // ----- MC Association #3 ---------------------------------
  // Useful only if you want to use ideal hypothesis in fwd kalman
  PndMCTrackAssociator* trackMCfwd = NULL;
  this->Add(trackMCfwd = new PndMCTrackAssociator()); // 7
  if ((this->GetListOfTasks()->GetSize()-1) != kPndMCTrackAssociator3) Error("PndMasterDigiTask","Error in task #%i", (this->GetListOfTasks()->GetSize()-1));
  trackMCfwd->SetTrackInBranchName("FtsIdealTrack");
  trackMCfwd->SetTrackOutBranchName("FtsIdealTrackID");

  // ----- Forward Kalman Task     ---------------------------
  PndRecoKalmanTask* recoKalmanFwd = NULL;
  this->Add(recoKalmanFwd = new PndRecoKalmanTask()); // 8
  if ((this->GetListOfTasks()->GetSize()-1) != kPndRecoKalmanTask2) Error("PndMasterDigiTask","Error in task #%i", (this->GetListOfTasks()->GetSize()-1));
  recoKalmanFwd->SetTrackInBranchName("FtsIdealTrack");
  //recoKalmanFwd->SetTrackInIDBranchName("FtsIdealTrackID");
  recoKalmanFwd->SetTrackOutBranchName("FtsIdealGenTrack");
  recoKalmanFwd->SetBusyCut(50); // CHECK to be tuned
  //recoKalmanFwd->SetIdealHyp(kTRUE);
  //recoKalmanFwd->SetNumIterations(3);
  recoKalmanFwd->SetTrackRep(0); // 0 Geane (default), 1 RK
  //recoKalmanFwd->SetPropagateToIP(kFALSE);

  // ----- MC Association #4 ---------------------------------
  PndMCTrackAssociator* trackMC3 = NULL;
  this->Add(trackMC3 = new PndMCTrackAssociator()); // 9
  if ((this->GetListOfTasks()->GetSize()-1) != kPndMCTrackAssociator4) Error("PndMasterDigiTask","Error in task #%i", (this->GetListOfTasks()->GetSize()-1));
  trackMC3->SetTrackInBranchName("FtsIdealGenTrack");
  trackMC3->SetTrackOutBranchName("FtsIdealGenTrackID");
 
  SetVerbose(0);
}
// -------------------------------------------------------------------------

/** Set the Persistency of all the tasks in the same way **/
void PndMasterRecoIdealTask::SetPersistency(Bool_t pers)
{
    // -----  MVD + STT Pattern Recognition -----------------------------------
  ((PndTrkTracking2*)GetListOfTasks()->At(kPndTrkTracking2))->SetPersistence(pers);

  // ----- MVD + STT + GEM Pattern Recognition --------------
  ((PndSttMvdGemTrackingIdeal*)GetListOfTasks()->At(kPndSttMvdGemTracking))->SetPersistence(pers);
  
  // ----- MC Association #1 ---------------------------------
  ((PndMCTrackAssociator*)GetListOfTasks()->At(kPndMCTrackAssociator1))->SetPersistence(pers);
  
  // ----- Barrel Kalman Task     ----------------------------
  ((PndRecoKalmanTask*)GetListOfTasks()->At(kPndRecoKalmanTask1))->SetPersistence(pers);
  
  // ----- MC Association #2 ---------------------------------
  ((PndMCTrackAssociator*)GetListOfTasks()->At(kPndMCTrackAssociator2))->SetPersistence(pers);
  
  // -----  FTS Ideal Tracking    ----------------------------
  ((PndFtsTrackerIdeal*)GetListOfTasks()->At(kPndFtsTrackerIdeal))->SetPersistence(pers);

  // ----- MC Association #3 ---------------------------------
  ((PndMCTrackAssociator*)GetListOfTasks()->At(kPndMCTrackAssociator3))->SetPersistence(pers);

  // ----- Forward Kalman Task     ---------------------------
  ((PndRecoKalmanTask*)GetListOfTasks()->At(kPndRecoKalmanTask2))->SetPersistence(pers);

  // ----- MC Association #4 ---------------------------------
  ((PndMCTrackAssociator*)GetListOfTasks()->At(kPndMCTrackAssociator4))->SetPersistence(pers);

  return;
}


// -----   Destructor   ----------------------------------------------------
PndMasterRecoIdealTask::~PndMasterRecoIdealTask()
{
}
// -------------------------------------------------------------------------

/** @cond CLASSIMP */
ClassImp(PndMasterRecoIdealTask);
/** @endcond */