Updates in plottinmg macro (ChiaraB)

[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / macros / AddTaskDplus.C

diff --git a/PWGHF/vertexingHF/macros/AddTaskDplus.C b/PWGHF/vertexingHF/macros/AddTaskDplus.C
index 495a6b48fd1c669c92d1ed42fe78e5ed3b99203a..2ca68a6ee7b351675a6add7b5daeb6c3544ffd96 100644 (file)
--- a/PWGHF/vertexingHF/macros/AddTaskDplus.C
+++ b/PWGHF/vertexingHF/macros/AddTaskDplus.C
@@ -5,7 +5,8 @@ AliAnalysisTaskSEDplus *AddTaskDplus(Int_t system=0/*0=pp,1=PbPb*/,
                                     Bool_t readMC=kFALSE,
                                     TString finDirname="Loose",
                                     TString filename="",
-                                    TString finAnObjname="AnalysisCuts")
+                                    TString finAnObjname="AnalysisCuts",
+                                    Int_t etaRange=0)
 {
   //                                                                                                                                    
   // Test macro for the AliAnalysisTaskSE for D+ candidates 
@@ -56,8 +57,10 @@ AliAnalysisTaskSEDplus *AddTaskDplus(Int_t system=0/*0=pp,1=PbPb*/,
   dplusTask->SetDebugLevel(0);
   dplusTask->SetMassLimits(0.2);
   dplusTask->SetUseBit(kTRUE);
-
+  dplusTask->SetSystem(system);
   if (doSparse) dplusTask->SetDoImpactParameterHistos(kTRUE);
+  if(etaRange==1) dplusTask->SetUseOnlyPositiveEta();
+  if(etaRange==-1) dplusTask->SetUseOnlyNegativeEta();
 
   mgr->AddTask(dplusTask);
   
@@ -100,7 +103,7 @@ AliAnalysisTaskSEDplus *AddTaskDplus(Int_t system=0/*0=pp,1=PbPb*/,
   if(storeNtuple){
     AliAnalysisDataContainer *coutputDplus2 = mgr->CreateContainer(ntuplename,TNtuple::Class(),
                                                                   AliAnalysisManager::kOutputContainer,
-                                                                  "InvMassDplus_nt1.root");
+                                                                  outputfile.Data());
     
     coutputDplus2->SetSpecialOutput();
   }