Add histograms and enlarge the Ntracklets range to 400 for pPb data analysis

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

diff --git a/PWGHF/vertexingHF/macros/AddTaskDvsMultiplicity.C b/PWGHF/vertexingHF/macros/AddTaskDvsMultiplicity.C
index f9415b06dbac9ba81db98ba4fdabda919f59cb1d..2a477e2aea90997df9c8907f388bef534689e942 100644 (file)
--- a/PWGHF/vertexingHF/macros/AddTaskDvsMultiplicity.C
+++ b/PWGHF/vertexingHF/macros/AddTaskDvsMultiplicity.C
@@ -9,7 +9,9 @@ AliAnalysisTaskSEDvsMultiplicity *AddTaskDvsMultiplicity(Int_t system=0,
                                                         Double_t refMult=9.26,
                                                         Bool_t subtractDau=kFALSE,
                                                         Bool_t NchWeight=kFALSE,
-                                                        Int_t configuration = AliAnalysisTaskSEDvsMultiplicity::kNtrk10)
+                                                        Int_t recoEstimator = AliAnalysisTaskSEDvsMultiplicity::kNtrk10,
+                                                        Int_t MCEstimator = AliAnalysisTaskSEDvsMultiplicity::kEta10,
+                                                        Bool_t isPPbData=kFALSE)
 {
   //
   // Test macro for the AliAnalysisTaskSE for D+ candidates
@@ -73,7 +75,10 @@ AliAnalysisTaskSEDvsMultiplicity *AddTaskDvsMultiplicity(Int_t system=0,
   dMultTask->SetUseBit(kTRUE);
   dMultTask->SetDoImpactParameterHistos(kFALSE);
   dMultTask->SetSubtractTrackletsFromDaughters(subtractDau);
-  dMultTask->SetMultiplicityEstimator(configuration);
+  dMultTask->SetMultiplicityEstimator(recoEstimator);
+  dMultTask->SetMCPrimariesEstimator(MCEstimator);
+  dMultTask->SetMCOption(MCOption);
+  if(isPPbData) dMultTask->SetIsPPbData();
 
   if(NchWeight){
     TH1F *hNchPrimaries = (TH1F*)filecuts->Get("hGenPrimaryParticlesInelGt0");