TRD/qaRec/macros/AddTRDpidChecker.C

   1 #if ! defined (__CINT__) || defined (__MAKECINT__)
   2 #include "TTree.h"
   3 #include "AliAnalysisManager.h"
   4 #include "AliAnalysisDataContainer.h"
   5 #include "TRD/qaRec/run.h"
   6 #include "TRD/qaRec/AliTRDpidChecker.h"
   7 #include "TRD/qaRec/AliTRDpidRefMaker.h"
   8 #endif
   9
  10
  11 void AddTRDpidChecker(AliAnalysisManager *mgr, AliAnalysisDataContainer **ci, AliAnalysisDataContainer **co, Int_t map)
  12 {
  13   AliTRDpidChecker *pid = 0x0;
  14   mgr->AddTask(pid = new AliTRDpidChecker());
  15   pid->SetDebugLevel(0);
  16   pid->SetMCdata(mgr->GetMCtruthEventHandler());
  17   mgr->ConnectInput(pid, 0, ci[0]);
  18   mgr->ConnectOutput(pid, 0, mgr->CreateContainer(pid->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.Performance.root"));
  19
  20   // TRD pid reference
  21   if(!(TSTBIT(map, kPIDRefMaker))) return;
  22   AliTRDpidRefMaker *ref = 0x0;
  23   mgr->AddTask(ref = new AliTRDpidRefMaker());
  24   ref->SetDebugLevel(0);
  25   ref->SetMCdata(mgr->GetMCtruthEventHandler());
  26
  27   // Create containers for input/output
  28   mgr->ConnectInput( ref, 0, ci[0]);
  29   mgr->ConnectOutput(ref, 0, mgr->CreateContainer(ref->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", ref->GetName())));
  30
  31   // network container
  32   co[0] = mgr->CreateContainer(Form("%sNN", ref->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sNN.root", ref->GetName()));
  33   // likelihood container
  34   co[1] = mgr->CreateContainer(Form("%sLQ", ref->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sLQ.root", ref->GetName()));
  35   mgr->ConnectOutput(ref, 1, co[0]);
  36   mgr->ConnectOutput(ref, 2, co[1]);
  37 }
  38