#if ! defined (__CINT__) || defined (__MAKECINT__)
#include "TTree.h"
#include "AliAnalysisManager.h"
#include "AliAnalysisDataContainer.h"
#include "TRD/qaRec/macros/AliTRDperformanceTrain.h"
#include "TRD/qaRec/AliTRDresolution.h"
#include "TRD/qaRec/AliTRDclusterResolution.h"
#include "TRD/qaRec/AliTRDalignmentTask.h"
#endif

#include "TRD/qaRec/macros/helper.C"
void AddTRDresolution(AliAnalysisManager *mgr, Char_t *trd, AliAnalysisDataContainer **ci/*, AliAnalysisDataContainer **co*/)
{
  Int_t map = ParseOptions(trd);
  AliTRDresolution *task = 0x0;
  if(!TSTBIT(map, kResolution)) return;
  mgr->AddTask(task = new AliTRDresolution());
  task->SetMCdata(mgr->GetMCtruthEventHandler());
  task->SetPostProcess(kFALSE);
  task->SetDebugLevel(1);
  mgr->ConnectInput( task, 0, ci[0]);
  mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.Performance.root"));

  // Create output containers for calibration tasks
  const Int_t nc = 4;
  const Char_t *cn[nc] = {"Cl", "Trklt", "MC_Cl", "MC_Trklt"}; 
  AliAnalysisDataContainer *co[] = {0x0, 0x0, 0x0, 0x0};
  for(Int_t ic = 0; ic<nc; ic++){
    co[ic] = mgr->CreateContainer(Form("%s%s", task->GetName(), cn[ic]), TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
    mgr->ConnectOutput(task, 1+ic, co[ic]);
  }

  // Cluster Error Parameterization
  if(TSTBIT(map, kClErrParam)){
    AliTRDclusterResolution *taskCl = 0x0;
    mgr->AddTask(taskCl = new AliTRDclusterResolution("ESD", "ESD Cluster error parameterization"));
    taskCl->SetExB();
    mgr->ConnectInput(taskCl, 0, co[0]);
    mgr->ConnectOutput(taskCl, 0, mgr->CreateContainer(taskCl->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.CalibClErrParam.root"));

    mgr->AddTask(taskCl = new AliTRDclusterResolution("MC", "MC Cluster error parameterization"));
    taskCl->SetExB();
    mgr->ConnectInput(taskCl, 0, co[2]);
    mgr->ConnectOutput(taskCl, 0, mgr->CreateContainer(taskCl->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.CalibClErrParam.root"));
  }

  // TRD alignment
  if(TSTBIT(map, kAlignment)){
    AliTRDalignmentTask *taskAl = 0x0;
    mgr->AddTask(taskAl = new AliTRDalignmentTask());
    taskAl->SetDebugLevel(0);
    mgr->ConnectInput(taskAl, 0, ci[0]);  
    mgr->ConnectOutput(taskAl, 0, mgr->CreateContainer(Form("h%s", taskAl->GetName()), TObjArray::Class(), AliAnalysisManager::kExchangeContainer));
    mgr->ConnectOutput(taskAl, 1, mgr->CreateContainer(task->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Calib%s.root", task->GetName())));
  }
}