From fc6d3e14b0a878707c70b78a7c757f055b170aa0 Mon Sep 17 00:00:00 2001
From: cholm <cholm@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Tue, 26 Apr 2011 20:20:44 +0000
Subject: [PATCH] Scripts to add MC correction tasks

---
 PWG2/FORWARD/analysis2/AddTaskCentralMCCorr.C | 56 +++++++++++++++++++
 PWG2/FORWARD/analysis2/AddTaskForwardMCCorr.C | 55 ++++++++++++++++++
 2 files changed, 111 insertions(+)
 create mode 100644 PWG2/FORWARD/analysis2/AddTaskCentralMCCorr.C
 create mode 100644 PWG2/FORWARD/analysis2/AddTaskForwardMCCorr.C

diff --git a/PWG2/FORWARD/analysis2/AddTaskCentralMCCorr.C b/PWG2/FORWARD/analysis2/AddTaskCentralMCCorr.C
new file mode 100644
index 00000000000..70c0a837f0f
--- /dev/null
+++ b/PWG2/FORWARD/analysis2/AddTaskCentralMCCorr.C
@@ -0,0 +1,56 @@
+/**
+ * @file   AddTaskCentralMCCorr.C
+ * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
+ * @date   Tue Apr 26 09:55:29 2011
+ * 
+ * @brief  
+ * 
+ * @ingroup pwg2_forward_scripts_tasks
+ * 
+ */
+
+/** 
+ * Add a Central MC correction generator task to train 
+ * 
+ * 
+ * @return Added task 
+ *
+ * @ingroup pwg2_central_mc
+ */
+AliAnalysisTask*
+AddTaskCentralMCCorr()
+{
+  // --- Creating the manager and handlers ---------------------------
+  AliAnalysisManager *mgr  = AliAnalysisManager::GetAnalysisManager();
+  if (!mgr->GetMCtruthEventHandler()) { 
+    Error("AddTaskCentralMCCorr", 
+	  "No MC input handler defined - cannot continue");
+    return 0;
+  }
+
+  // --- Add our task ------------------------------------------------
+  AliCentralMCCorrectionsTask* task2 = new AliCentralMCCorrectionsTask("spd");
+  mgr->AddTask(task2);
+  task2->SetNPhiBins(40);
+  // task2->GetTrackDensity().SetDebug(false);
+  
+  // --- create containers for input/output --------------------------
+  AliAnalysisDataContainer *sums = 
+    mgr->CreateContainer("CentralSums", TList::Class(), 
+			 AliAnalysisManager::kOutputContainer, 
+			 AliAnalysisManager::GetCommonFileName());
+  AliAnalysisDataContainer *output = 
+    mgr->CreateContainer("CentralResults", 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);
+
+  return task;
+}
+//
+// EOF
+// 
diff --git a/PWG2/FORWARD/analysis2/AddTaskForwardMCCorr.C b/PWG2/FORWARD/analysis2/AddTaskForwardMCCorr.C
new file mode 100644
index 00000000000..d2fb187a6a0
--- /dev/null
+++ b/PWG2/FORWARD/analysis2/AddTaskForwardMCCorr.C
@@ -0,0 +1,55 @@
+/**
+ * @file   AddTaskForwardMCCorr.C
+ * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
+ * @date   Tue Apr 26 09:56:39 2011
+ * 
+ * @brief  
+ * 
+ * 
+ * @ingroup pwg2_forward_scripts_tasks
+ */
+/** 
+ * Add a Forward MC correction generator task to train 
+ * 
+ * 
+ * @return Added task 
+ *
+ * @ingroup pwg2_forward_mc
+ */
+AliAnalysisTask*
+AddTaskForwardMCCorr()
+{
+  // --- Creating the manager and handlers ---------------------------
+  AliAnalysisManager *mgr  = AliAnalysisManager::GetAnalysisManager();
+  if (!mgr->GetMCtruthEventHandler()) { 
+    Error("AddTaskCentralMCCorr", 
+	  "No MC input handler defined - cannot continue");
+    return 0;
+  }
+
+  // --- Add our task ------------------------------------------------
+  AliForwardMCCorrectionsTask* task = new AliForwardMCCorrectionsTask("fmd");
+  mgr->AddTask(task);
+  task->GetTrackDensity().SetDebug(false);
+  task->GetTrackDensity().SetMaxConsequtiveStrips(3);
+  
+  // --- 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);
+
+  return task;
+}
+//
+// EOF
+// 
-- 
2.43.0