TRD/qaRec/macros/AddTRDcheckPID.C

   1 #if ! defined (__CINT__) || defined (__MAKECINT__)
   2 #include "TTree.h"
   3 #include "AliLog.h"
   4 #include "AliAnalysisManager.h"
   5 #include "AliAnalysisDataContainer.h"
   6 #include "TRD/qaRec/macros/AliTRDperformanceTrain.h"
   7 #include "TRD/qaRec/AliTRDcheckPID.h"
   8 #include "TRD/qaRec/AliTRDpidRefMaker.h"
   9 #endif
  10
  11 #include "TRD/qaRec/macros/helper.C"
  12 void AddTRDcheckPID(AliAnalysisManager *mgr, Char_t *trd, AliAnalysisDataContainer **ci/*, AliAnalysisDataContainer **co*/)
  13 {
  14   Int_t map = ParseOptions(trd);
  15   if(TSTBIT(map, kCheckPID)){
  16     AliTRDcheckPID *pid = 0x0;
  17     mgr->AddTask(pid = new AliTRDcheckPID());
  18     pid->SetDebugLevel(0);
  19     pid->SetMCdata(mgr->GetMCtruthEventHandler());
  20     mgr->ConnectInput(pid, 0, ci[0]);
  21     mgr->ConnectOutput(pid, 0, mgr->CreateContainer(pid->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.Performance.root"));
  22   }
  23
  24   if(TSTBIT(map, kPIDRefMaker)){
  25     // TRD pid reference maker
  26     AliTRDpidRefMaker *ref = new AliTRDpidRefMaker();
  27     mgr->AddTask(ref);
  28     ref->SetDebugLevel(3);
  29     AliLog::SetClassDebugLevel("AliTRDpidRefMaker", 3);
  30     ref->SetMCdata(mgr->GetMCtruthEventHandler());
  31     ref->SetFriends(kTRUE);
  32     mgr->ConnectInput( ref, 0, ci[0]);
  33     mgr->ConnectInput( ref, 1, ci[2]);
  34     mgr->ConnectOutput(ref, 0, mgr->CreateContainer(Form("Moni%s", ref->GetName()), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Calib%s.root", ref->GetName())));
  35     mgr->ConnectOutput(ref, 1, mgr->CreateContainer(ref->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Calib%s.root", ref->GetName())));
  36   }
  37 }
  38