X-Git-Url: http://git.uio.no/git/?p=u%2Fmrichter%2FAliRoot.git;a=blobdiff_plain;f=PWG3%2FvertexingHF%2FAddTaskDplus.C;h=fe237feae4196701affc5c329efe1c9415fddbd9;hp=db8d0585a7091e28fc36c67ba0d134c4f13d222f;hb=1f4e9722d8d3a038d0cc535e82a503597ade0b37;hpb=41ff2555293ae9a3b5da571b4a285aacddec2e7f

diff --git a/PWG3/vertexingHF/AddTaskDplus.C b/PWG3/vertexingHF/AddTaskDplus.C
index db8d0585a70..fe237feae41 100644
--- a/PWG3/vertexingHF/AddTaskDplus.C
+++ b/PWG3/vertexingHF/AddTaskDplus.C
@@ -1,4 +1,4 @@
-AliAnalysisTaskSEDplus *AddTaskDplus()
+AliAnalysisTaskSEDplus *AddTaskDplus(Bool_t storeNtuple=kFALSE)
 {
   //                                                                                                                                    
   // Test macro for the AliAnalysisTaskSE for heavy-flavour candidates                                                                  
@@ -17,8 +17,8 @@ AliAnalysisTaskSEDplus *AddTaskDplus()
 
 
   // Aanalysis task                                                                                                                     
-  AliAnalysisTaskSEDplus *dplusTask = new AliAnalysisTaskSEDplus("DplusAnalysis");
-  dplusTask->SetDebugLevel(2);
+  AliAnalysisTaskSEDplus *dplusTask = new AliAnalysisTaskSEDplus("DplusAnalysis",storeNtuple);
+  dplusTask->SetDebugLevel(0);
   mgr->AddTask(dplusTask);
 
   //                                                                                                                                    
@@ -28,9 +28,23 @@ AliAnalysisTaskSEDplus *AddTaskDplus()
   AliAnalysisDataContainer *coutputDplus = mgr->CreateContainer("coutputDplus",TList::Class(),
                                                            AliAnalysisManager::kOutputContainer,
                                                            "InvMassDplus.root");
+  if(storeNtuple){
+    AliAnalysisDataContainer *coutputDplus2 = mgr->CreateContainer("coutputDplus2",TNtuple::Class(),
+                                                           AliAnalysisManager::kOutputContainer,
+								 "InvMassDplus_nt1.root");
+    AliAnalysisDataContainer *coutputDplus3 = mgr->CreateContainer("coutputDplus3",TNtuple::Class(),
+                                                           AliAnalysisManager::kOutputContainer,
+								 "InvMassDplus_nt2.root");
+    coutputDplus2->SetSpecialOutput();
+    coutputDplus3->SetSpecialOutput();
+  }
   mgr->ConnectInput(dplusTask,0,mgr->GetCommonInputContainer());
 
   mgr->ConnectOutput(dplusTask,1,coutputDplus);
-
+  
+  if(storeNtuple){
+    mgr->ConnectOutput(dplusTask,2,coutputDplus2);
+    mgr->ConnectOutput(dplusTask,3,coutputDplus3);
+  }
   return dplusTask;
 }