PWG1/TRD/macros/AddTRDresolution.C

   1 #if ! defined (__CINT__) || defined (__MAKECINT__)
   2 #include "TTree.h"
   3 #include "TError.h"
   4 #include "AliLog.h"
   5 #include "AliAnalysisManager.h"
   6 #include "AliAnalysisDataContainer.h"
   7 #include "TRD/AliTRDgeometry.h"
   8 #include "PWG1/TRD/AliTRDpwg1Helper.h"
   9 #include "PWG1/TRD/AliTRDresolution.h"
  10 #include "PWG1/TRD/AliTRDclusterResolution.h"
  11 #include "PWG1/TRD/AliTRDalignmentTask.h"
  12 #endif
  13
  14 void AddTRDresolution(AliAnalysisManager *mgr, Int_t map, AliAnalysisDataContainer **ci)
  15 {
  16   Info("AddTRDresolution", Form("[0]=\"%s\" [1]=\"%s\" [2]=\"%s\" [3]=\"%s\"", ci[0]->GetName(), ci[1]->GetName(), ci[2]->GetName(), ci[3]->GetName()));
  17   AliAnalysisDataContainer *evInfoContainer = ci[3];
  18
  19   //AliLog::SetClassDebugLevel("AliTRDrecoTask", 2);
  20   //AliLog::SetClassDebugLevel("AliTRDresolution", 2);
  21   AliTRDresolution *res(NULL);
  22   const Char_t *suffix[]={"", "SA", "K"};
  23   for(Int_t itq=0; itq<1/*3*/; itq++){
  24     mgr->AddTask(res = new AliTRDresolution(Form("TRDresolution%s", suffix[itq])));
  25     res->SetMCdata(mgr->GetMCtruthEventHandler());
  26     res->SetPostProcess(kFALSE);
  27     res->SetDebugLevel(0);
  28     //if(itq==0) res->SetSegmentationLevel(1);
  29     // use these settings if you know what you are doing !
  30     //res->SetTrackRefit();
  31     //res->SetPtThreshold(0.);
  32     res->SetNameId(suffix[itq]);
  33     mgr->ConnectInput(res, 0, mgr->GetCommonInputContainer()); // connect main (ESD) container
  34     mgr->ConnectInput(res, 1, ci[itq]);                        // conect track info container
  35     mgr->ConnectInput(res, 2, evInfoContainer);                // conect event info container
  36     mgr->ConnectOutput(res,1, mgr->CreateContainer(res->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Performance", mgr->GetCommonFileName())));
  37
  38     // Create output containers for calibration tasks
  39     AliAnalysisDataContainer *co(NULL);
  40     co = mgr->CreateContainer(Form("%sCl2Trk%s", res->GetName(), suffix[itq]), TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
  41     mgr->ConnectOutput(res, AliTRDresolution::kClToTrk, co);
  42     co = mgr->CreateContainer(Form("%sCl2MC%s", res->GetName(), suffix[itq]), TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
  43     mgr->ConnectOutput(res, AliTRDresolution::kClToMC, co);
  44
  45     TObjArray *coa = mgr->GetContainers();
  46     // Cluster Error Parameterization
  47     if(TESTBIT(map, AliTRDpwg1Helper::kClErrParam)){
  48       AliTRDclusterResolution *taskCl(NULL);
  49       AliLog::SetClassDebugLevel("AliTRDclusterResolution", 2);
  50       for(Int_t idet(10); idet<11/*AliTRDgeometry::kNdet*/; idet++){
  51         mgr->AddTask(taskCl = new AliTRDclusterResolution(Form("ClErrCalib%03d", idet)));
  52         taskCl->SetCalibrationRegion(idet);
  53         taskCl->SetDebugLevel(0);
  54         mgr->ConnectInput(taskCl,  0, mgr->GetCommonInputContainer());  // connect main (ESD) container
  55         mgr->ConnectInput(taskCl,  1, (AliAnalysisDataContainer*)coa->FindObject(Form("%sCl2Trk%s", res->GetName(), suffix[itq])));
  56         mgr->ConnectInput(taskCl,  2, evInfoContainer);
  57         mgr->ConnectOutput(taskCl, 1, mgr->CreateContainer(taskCl->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
  58         if(mgr->GetMCtruthEventHandler()){
  59           mgr->AddTask(taskCl = new AliTRDclusterResolution(Form("ClErrCalibMC%03d", idet)));
  60           taskCl->SetCalibrationRegion(idet);
  61           taskCl->SetDebugLevel(0);
  62           mgr->ConnectInput(taskCl,  0, mgr->GetCommonInputContainer());
  63           mgr->ConnectInput(taskCl,  1, (AliAnalysisDataContainer*)coa->FindObject(Form("%sCl2MC%s", res->GetName(), suffix[itq])));
  64           mgr->ConnectInput(taskCl,  2, evInfoContainer);
  65           mgr->ConnectOutput(taskCl, 1, mgr->CreateContainer(taskCl->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
  66         }
  67       }
  68     }
  69   }
  70
  71   // TRD alignment
  72   if(TESTBIT(map, AliTRDpwg1Helper::kAlignment)){
  73     AliTRDalignmentTask *taskAlign(NULL);
  74     mgr->AddTask(taskAlign = new AliTRDalignmentTask((char*)"TRDalignment"));
  75     taskAlign->SetDebugLevel(0);
  76     //AliLog::SetClassDebugLevel("AliTRDalignmentTask", 5);
  77     mgr->ConnectInput(taskAlign,  0, mgr->GetCommonInputContainer());
  78     mgr->ConnectInput(taskAlign,  1, ci[0]);
  79     mgr->ConnectInput(taskAlign,  2, evInfoContainer);
  80     mgr->ConnectOutput(taskAlign, 1, mgr->CreateContainer(Form("h%s", taskAlign->GetName()), TObjArray::Class(), AliAnalysisManager::kExchangeContainer));
  81     mgr->ConnectOutput(taskAlign, 2, mgr->CreateContainer(taskAlign->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Alignment",mgr->GetCommonFileName())));
  82   }
  83 }
  84