From a13d770cb0e33d50d09035b79a74a86cd85d5b90 Mon Sep 17 00:00:00 2001
From: dainese <dainese@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Tue, 17 Nov 2009 14:22:45 +0000
Subject: [PATCH] Added flags for each task

---
 PWG3/vertexingHF/AddD2HTrain.C | 87 ++++++++++++++++++++++------------
 1 file changed, 58 insertions(+), 29 deletions(-)

diff --git a/PWG3/vertexingHF/AddD2HTrain.C b/PWG3/vertexingHF/AddD2HTrain.C
index ea9adc6f154..8cf53ccd8f8 100644
--- a/PWG3/vertexingHF/AddD2HTrain.C
+++ b/PWG3/vertexingHF/AddD2HTrain.C
@@ -1,4 +1,10 @@
-void AddD2HTrain() {
+Int_t AddD2HTrain(Int_t addD0Mass=1,
+		  Int_t addD0MassLS=1,
+		  Int_t addDplus=1,
+		  Int_t addLSD0=1,
+		  Int_t addLSJpsi=1,
+		  Int_t addCFD0=1,
+		  Int_t addPromptD0=1) {
   // 
   // Task of the D2H subgroup of PWG3 that can run in the Official Train
   //
@@ -13,45 +19,68 @@ void AddD2HTrain() {
   //
 
   TString taskName="",loadMacroPath="$ALICE_ROOT/PWG3/vertexingHF/";
-  
+  Int_t ntasks=0;
+
   //taskName="AddTaskCompareHF.C"; taskName.Prepend(loadMacroPath.Data());
   //gROOT->LoadMacro(taskName.Data());
   //AliAnalysisTaskSECompareHF *cmpTask = AddTaskCompareHF();
   
-  taskName="AddTaskD0Mass.C"; taskName.Prepend(loadMacroPath.Data());
-  gROOT->LoadMacro(taskName.Data());
-  AliAnalysisTaskSED0Mass *d0massTask = AddTaskD0Mass();
-  AliAnalysisTaskSED0Mass *d0massLikeSignTask = AddTaskD0Mass(1); 
-  
-  taskName="AddTaskDplus.C"; taskName.Prepend(loadMacroPath.Data());
-  gROOT->LoadMacro(taskName.Data());
-  AliAnalysisTaskSEDplus *dplusTask = AddTaskDplus();
-  
+  if(addD0Mass || addD0MassLS) {
+    taskName="AddTaskD0Mass.C"; taskName.Prepend(loadMacroPath.Data());
+    gROOT->LoadMacro(taskName.Data());
+    if(addD0Mass) {
+      AliAnalysisTaskSED0Mass *d0massTask = AddTaskD0Mass();
+      ntasks++;
+    }
+    if(addD0MassLS) {
+      AliAnalysisTaskSED0Mass *d0massLikeSignTask = AddTaskD0Mass(1); 
+      ntasks++;
+    }
+  }
+
+  if(addDplus) {
+    taskName="AddTaskDplus.C"; taskName.Prepend(loadMacroPath.Data());
+    gROOT->LoadMacro(taskName.Data());
+    AliAnalysisTaskSEDplus *dplusTask = AddTaskDplus();
+    ntasks++;
+  }  
+
   //taskName="AddTaskSelectHF.C"; taskName.Prepend(loadMacroPath.Data());
   //gROOT->LoadMacro(taskName.Data());
   //AliAnalysisTaskSESelectHF *seleTask = AddTaskSelectHF();
   
-  taskName="AddTaskBkgLikeSignD0.C"; taskName.Prepend(loadMacroPath.Data());
-  gROOT->LoadMacro(taskName.Data());
-  AliAnalysisTaskSEBkgLikeSignD0 *lsD0Task = AddTaskBkgLikeSignD0();
-  
-  taskName="AddTaskBkgLikeSignJPSI.C"; taskName.Prepend(loadMacroPath.Data());
-  gROOT->LoadMacro(taskName.Data());
-  AliAnalysisTaskSEBkgLikeSignJPSI *lsJPSITask = AddTaskBkgLikeSignJPSI();
-  
+  if(addLSD0) {
+    taskName="AddTaskBkgLikeSignD0.C"; taskName.Prepend(loadMacroPath.Data());
+    gROOT->LoadMacro(taskName.Data());
+    AliAnalysisTaskSEBkgLikeSignD0 *lsD0Task = AddTaskBkgLikeSignD0();
+    ntasks++;
+  }
+
+  if(addLSJpsi) {
+    taskName="AddTaskBkgLikeSignJPSI.C"; taskName.Prepend(loadMacroPath.Data());
+    gROOT->LoadMacro(taskName.Data());
+    AliAnalysisTaskSEBkgLikeSignJPSI *lsJPSITask = AddTaskBkgLikeSignJPSI();
+    ntasks++;
+  }
+
   //taskName="AddTaskBtoJPSItoEle.C"; taskName.Prepend(loadMacroPath.Data());
   //gROOT->LoadMacro(taskName.Data());
   //AliAnalysisTaskSEBtoJPSItoEle *jpsiTask = AddTaskBtoJPSItoEle();
   
-  taskName="AddTaskCFMultiVarMultiStep.C"; taskName.Prepend(loadMacroPath.Data());
-  gROOT->LoadMacro(taskName.Data());
-  AliCFHeavyFlavourTaskMultiVarMultiStep *cfmvmsTask = AddTaskCFMultiVarMultiStep();
-  
-  taskName="AddTaskCharmFraction.C"; taskName.Prepend(loadMacroPath.Data());
-  gROOT->LoadMacro(taskName.Data());
-  Int_t switchMC[5]={1,1,1,1,1};
-  AliAnalysisTaskSECharmFraction *cFractTask = AddTaskCharmFraction("d0D0.root",switchMC);
-  
+  if(addCFD0) {
+    taskName="AddTaskCFMultiVarMultiStep.C"; taskName.Prepend(loadMacroPath.Data());
+    gROOT->LoadMacro(taskName.Data());
+    AliCFHeavyFlavourTaskMultiVarMultiStep *cfmvmsTask = AddTaskCFMultiVarMultiStep();
+    ntasks++;
+  }
+
+  if(addPromptD0) {
+    taskName="AddTaskCharmFraction.C"; taskName.Prepend(loadMacroPath.Data());
+    gROOT->LoadMacro(taskName.Data());
+    Int_t switchMC[5]={1,1,1,1,1};
+    AliAnalysisTaskSECharmFraction *cFractTask = AddTaskCharmFraction("d0D0.root",switchMC);
+    ntasks++;
+  }
 
-  return;
+  return ntasks;
 }
-- 
2.43.0