update for transporting event info along the TRD train

[u/mrichter/AliRoot.git] / PWG1 / TRD / macros / AddTRDcheckPID.C

diff --git a/PWG1/TRD/macros/AddTRDcheckPID.C b/PWG1/TRD/macros/AddTRDcheckPID.C
index e3109813bb4d2b9f840111f90b2c1ab6dd0393d4..80a7b05f153eef69c287a9541b301167c9496e29 100644 (file)
--- a/PWG1/TRD/macros/AddTRDcheckPID.C
+++ b/PWG1/TRD/macros/AddTRDcheckPID.C
@@ -4,7 +4,7 @@
 #include "AliLog.h"
 #include "AliAnalysisManager.h"
 #include "AliAnalysisDataContainer.h"
-#include "PWG1/TRD/macros/AliTRDperformanceTrain.h"
+#include "PWG1/TRD/AliTRDpwg1Helper.h"
 #include "PWG1/TRD/AliTRDcheckPID.h"
 #include "PWG1/TRD/AliTRDpidRefMaker.h"
 #include "PWG1/TRD/AliTRDpidRefMakerNN.h"
@@ -13,7 +13,7 @@
 
 void AddTRDcheckPID(AliAnalysisManager *mgr, Int_t map, AliAnalysisDataContainer **ci, AliAnalysisDataContainer **co)
 {
-  Info("AddTRDcheckPID", Form("[0]=\"%s\" [1]=\"%s\"", ci[0]->GetName(), ci[1]->GetName()));
+  Info("AddTRDcheckPID", Form("[0]=\"%s\" [1]=\"%s\" [2]=\"%s\"", ci[0]->GetName(), ci[1]->GetName(), ci[2]->GetName()));
 
   AliTRDcheckPID *pid(NULL);
   mgr->AddTask(pid = new AliTRDcheckPID((char*)"TRDcheckPID"));
@@ -23,13 +23,14 @@ void AddTRDcheckPID(AliAnalysisManager *mgr, Int_t map, AliAnalysisDataContainer
 
   // define PID exchange container
   co[0] = mgr->CreateContainer("InfoPID", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
-  mgr->ConnectInput (pid, 0, mgr->GetCommonInputContainer());
-  mgr->ConnectInput (pid, 1, ci[0]);
-  mgr->ConnectInput (pid, 2, ci[1]);
+  mgr->ConnectInput (pid, 0, mgr->GetCommonInputContainer()); // connect main (ESD) container
+  mgr->ConnectInput (pid, 1, ci[0]);                          // connect barrel tracks container
+  mgr->ConnectInput (pid, 2, ci[1]);                          // connect event info container
+  mgr->ConnectInput (pid, 3, ci[2]);                          // connect V0s container
   mgr->ConnectOutput(pid, 1, mgr->CreateContainer(pid->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Performance",mgr->GetCommonFileName())));
   mgr->ConnectOutput(pid, 2, co[0]);
 
-  if(TSTBIT(map, kPIDRefMaker)){
+  if(TESTBIT(map, AliTRDpwg1Helper::kPIDRefMaker)){
 
     //AliLog::SetClassDebugLevel("AliTRDpidRefMaker", 3);
     //AliLog::SetClassDebugLevel("AliTRDpidRefMakerNN", 3);
@@ -41,25 +42,27 @@ void AddTRDcheckPID(AliAnalysisManager *mgr, Int_t map, AliAnalysisDataContainer
     ref->SetDebugLevel(3);
     ref->SetMCdata(mgr->GetMCtruthEventHandler());
     ref->SetFriends(kTRUE);
-    mgr->ConnectInput( ref, 0, mgr->GetCommonInputContainer());
-    mgr->ConnectInput( ref, 1, ci[0]);
-    mgr->ConnectInput( ref, 2, ci[1]);
-    mgr->ConnectInput( ref, 3, co[0]);
-    mgr->ConnectOutput(ref, 1, mgr->CreateContainer("MonitorNN", TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD.CalibPIDrefMaker",mgr->GetCommonFileName())));
-    mgr->ConnectOutput(ref, 2, mgr->CreateContainer(ref->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD.CalibPIDrefMaker", mgr->GetCommonFileName())));
+    mgr->ConnectInput( ref, 0, mgr->GetCommonInputContainer()); // connect main (ESD) container
+    mgr->ConnectInput( ref, 1, ci[0]);                          // connect barrel tracks container
+    mgr->ConnectInput( ref, 2, ci[1]);                          // connect event info container
+    mgr->ConnectInput( ref, 3, ci[2]);                          // connect V0s container
+    mgr->ConnectInput( ref, 4, co[0]);                          // connect pid Info container
+    mgr->ConnectOutput(ref, 1, mgr->CreateContainer("MonitorNN", TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration",mgr->GetCommonFileName())));
+    mgr->ConnectOutput(ref, 2, mgr->CreateContainer(ref->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
   
     // TRD pid reference maker LQ 
     mgr->AddTask(ref = new AliTRDpidRefMakerLQ((char*)"TRDrefMakerLQ"));
     ref->SetDebugLevel(3);
     ref->SetMCdata(mgr->GetMCtruthEventHandler());
     ref->SetFriends(kTRUE);
-    mgr->ConnectInput(ref, 0, mgr->GetCommonInputContainer());
-    mgr->ConnectInput(ref, 1, ci[0]);
-    mgr->ConnectInput(ref, 2, ci[1]);
-    mgr->ConnectInput(ref, 3, co[0]);
-    mgr->ConnectOutput(ref, 1, mgr->CreateContainer("MonitorLQ", TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD.CalibPIDrefMaker", mgr->GetCommonFileName())));
-    mgr->ConnectOutput(ref, 2, mgr->CreateContainer(ref->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD.CalibPIDrefMaker", mgr->GetCommonFileName())));
-    mgr->ConnectOutput(ref, 3, mgr->CreateContainer("PDF", TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD.CalibPIDrefMakerLQ", mgr->GetCommonFileName())));
+    mgr->ConnectInput(ref, 0, mgr->GetCommonInputContainer()); // connect main (ESD) container
+    mgr->ConnectInput(ref, 1, ci[0]);                          // connect barrel tracks container
+    mgr->ConnectInput(ref, 2, ci[1]);                          // connect event info container
+    mgr->ConnectInput(ref, 3, ci[2]);                          // connect V0s container
+    mgr->ConnectInput(ref, 4, co[0]);                          // connect pid Info container
+    mgr->ConnectOutput(ref, 1, mgr->CreateContainer("MonitorLQ", TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
+    mgr->ConnectOutput(ref, 2, mgr->CreateContainer(ref->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
+    mgr->ConnectOutput(ref, 3, mgr->CreateContainer("PDF", TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
   }
 }