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


void AddTRDresolution(AliAnalysisManager *mgr, AliAnalysisDataContainer **ci, AliAnalysisDataContainer **co, Int_t map)
{
  AliTRDresolution *task = 0x0;
  mgr->AddTask(task = new AliTRDresolution());
  task->SetMCdata(mgr->GetMCtruthEventHandler());
  task->SetPostProcess(kFALSE);
  task->SetDebugLevel(0);
  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"}; 
  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());
    taskCl->SetExB();
    mgr->ConnectInput(taskCl, 0, co[0]);
    mgr->ConnectOutput(taskCl, 0, mgr->CreateContainer(taskCl->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", taskCl->GetName())));

    mgr->AddTask(taskCl = new AliTRDclusterResolution("ClErrParamMC"));
    taskCl->SetExB();
    mgr->ConnectInput(taskCl, 0, co[2]);
    mgr->ConnectOutput(taskCl, 0, mgr->CreateContainer(taskCl->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", taskCl->GetName())));
  }

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