Updates

[u/mrichter/AliRoot.git] / PWGLF / FORWARD / analysis2 / AddTaskForwardMCCorr.C

diff --git a/PWGLF/FORWARD/analysis2/AddTaskForwardMCCorr.C b/PWGLF/FORWARD/analysis2/AddTaskForwardMCCorr.C
index f3d1a7e62d43f5c134181944030d9dc293b8b252..1366f1987d905c063da0b4a039b8a3860faa255a 100644 (file)
--- a/PWGLF/FORWARD/analysis2/AddTaskForwardMCCorr.C
+++ b/PWGLF/FORWARD/analysis2/AddTaskForwardMCCorr.C
@@ -6,7 +6,7 @@
  * @brief  
  * 
  * 
- * @ingroup pwg2_forward_scripts_tasks
+ * @ingroup pwglf_forward_scripts_tasks
  */
 /** 
  * Add a Forward MC correction generator task to train 
@@ -14,13 +14,13 @@
  * 
  * @return Added task 
  *
- * @ingroup pwg2_forward_mc
+ * @ingroup pwglf_forward_mc
  */
 AliAnalysisTask*
 AddTaskForwardMCCorr()
 {
   // --- Load libraries ----------------------------------------------
-  gROOT->LoadClass("AliAODForwardMult", "libPWG2forward2");
+  gROOT->LoadClass("AliAODForwardMult", "libPWGLFforward2");
 
   // --- Creating the manager and handlers ---------------------------
   AliAnalysisManager *mgr  = AliAnalysisManager::GetAnalysisManager();
@@ -31,26 +31,16 @@ AddTaskForwardMCCorr()
   }
 
   // --- Add our task ------------------------------------------------
-  AliForwardMCCorrectionsTask* task = new AliForwardMCCorrectionsTask("fmd");
-  mgr->AddTask(task);
+  AliForwardMCCorrectionsTask* task = 
+    new AliForwardMCCorrectionsTask("ForwardCorr");
   task->GetTrackDensity().SetDebug(false);
-  task->GetTrackDensity().SetMaxConsequtiveStrips(3);
-//  task->SetVertexAxis(40, -20., 20.);
+  AliFMDMCTrackDensity& dn = 
+    static_cast<AliFMDMCTrackDensity&>(task->GetTrackDensity());
+  dn.SetMaxConsequtiveStrips(3);
+  //  task->SetVertexAxis(40, -20., 20.);
   
-  // --- create containers for input/output --------------------------
-  AliAnalysisDataContainer *sums = 
-    mgr->CreateContainer("ForwardSums", TList::Class(), 
-                        AliAnalysisManager::kOutputContainer, 
-                        AliAnalysisManager::GetCommonFileName());
-  AliAnalysisDataContainer *output = 
-    mgr->CreateContainer("ForwardResults", TList::Class(), 
-                        AliAnalysisManager::kParamContainer, 
-                        AliAnalysisManager::GetCommonFileName());
-
   // --- connect input/output ----------------------------------------
-  mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
-  mgr->ConnectOutput(task, 1, sums);
-  mgr->ConnectOutput(task, 2, output);
+  task->Connect(0, 0);
 
   return task;
 }