- bug fixes (increase classdef)

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

diff --git a/PWG1/TRD/macros/AddTRDefficiency.C b/PWG1/TRD/macros/AddTRDefficiency.C
index f1ecdc2ecf0508325e22c0b6f24431878d26dbda..c434f7dd70458031dc02161b612e9481f9bc2f37 100644 (file)
--- a/PWG1/TRD/macros/AddTRDefficiency.C
+++ b/PWG1/TRD/macros/AddTRDefficiency.C
@@ -10,14 +10,19 @@
 
 void AddTRDefficiency(AliAnalysisManager *mgr, Int_t map, AliAnalysisDataContainer **ci/*, AliAnalysisDataContainer **co*/)
 {
-  Info("AddTRDefficiency", Form("[0]=\"%s\" [1]=\"%s\" [2]=\"%s\"", ci[0]->GetName(), ci[1]->GetName(), ci[2]->GetName()));
+  Info("AddTRDefficiency", Form("[0]=\"%s\" [1]=\"%s\" [2]=\"%s\" [3]=\"%s\"", ci[0]->GetName(), ci[1]->GetName(), ci[2]->GetName(), ci[3]->GetName()));
 
+  AliAnalysisDataContainer *evInfoContainer = ci[3];
   AliTRDrecoTask *eff(NULL);
   mgr->AddTask(eff = new AliTRDefficiency((char*)"TRDefficiency"));
   eff->SetDebugLevel(0);
   //AliLog::SetClassDebugLevel("AliTRDefficiency", 5);  
-  mgr->ConnectInput(eff, 0, mgr->GetCommonInputContainer());  
-  mgr->ConnectInput(eff, 1, ci[0]);
+  Int_t trackStatus = 0; // barrel tracks
+//                    = 1; // kink tracks
+//                    = 2; // SA tracks
+  mgr->ConnectInput(eff, 0, mgr->GetCommonInputContainer());  // connect main (ESD) container
+  mgr->ConnectInput(eff, 1, ci[trackStatus]);                 // conect track info container
+  mgr->ConnectInput(eff, 2, evInfoContainer);                 // conect event info container
   mgr->ConnectOutput(eff,1, mgr->CreateContainer(eff->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Performance", mgr->GetCommonFileName())));
     
 
@@ -29,8 +34,9 @@ void AddTRDefficiency(AliAnalysisManager *mgr, Int_t map, AliAnalysisDataContain
 
     // Create containers for input/output
     mgr->ConnectInput(eff, 0, mgr->GetCommonInputContainer());  
-    mgr->ConnectInput(eff, 1, ci[0]);
-    mgr->ConnectOutput(eff, 1, mgr->CreateContainer(eff->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Performance", mgr->GetCommonFileName(),eff->GetName())));
+    mgr->ConnectInput(eff, 1, ci[trackStatus]);
+    mgr->ConnectInput(eff, 2, evInfoContainer);
+    mgr->ConnectOutput(eff, 1, mgr->CreateContainer(eff->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Performance", mgr->GetCommonFileName())));
   }
 
   // TRD single track selection
@@ -41,6 +47,6 @@ void AddTRDefficiency(AliAnalysisManager *mgr, Int_t map, AliAnalysisDataContain
   //AliLog::SetClassDebugLevel("AliTRDmultiplicity", 5);  
   mgr->ConnectInput(eff, 0, mgr->GetCommonInputContainer());  
   mgr->ConnectInput(eff, 1, ci[0]);
-  mgr->ConnectOutput(eff, 1, mgr->CreateContainer(eff->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD.Calib%s", mgr->GetCommonFileName(),eff->GetName())));
+  mgr->ConnectOutput(eff, 1, mgr->CreateContainer(eff->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
 }