Adding Tab values for QpPb analysis

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

diff --git a/PWGHF/vertexingHF/macros/AddTaskDvsMultiplicity.C b/PWGHF/vertexingHF/macros/AddTaskDvsMultiplicity.C
old mode 100644 (file)
new mode 100755 (executable)
index 4ee4e81..b9f2eb5
--- a/PWGHF/vertexingHF/macros/AddTaskDvsMultiplicity.C
+++ b/PWGHF/vertexingHF/macros/AddTaskDvsMultiplicity.C
@@ -80,7 +80,9 @@ AliAnalysisTaskSEDvsMultiplicity *AddTaskDvsMultiplicity(Int_t system=0,
   if(isPPbData) dMultTask->SetIsPPbData();
 
   if(NchWeight){
-    TH1F *hNchPrimaries = (TH1F*)filecuts->Get("hGenPrimaryParticlesInelGt0");
+    TH1F *hNchPrimaries = NULL; 
+    if(isPPbData) hNchPrimaries = (TH1F*)filecuts->Get("hNtrUnCorrEvWithDWeight");
+    else hNchPrimaries = (TH1F*)filecuts->Get("hGenPrimaryParticlesInelGt0");
     if(hNchPrimaries) {
       dMultTask->UseMCNchWeight(true);
       dMultTask->SetHistoNchWeight(hNchPrimaries);
@@ -98,41 +100,43 @@ AliAnalysisTaskSEDvsMultiplicity *AddTaskDvsMultiplicity(Int_t system=0,
   if(estimatorFilename.EqualTo("") ) {
     printf("Estimator file not provided, multiplcity corrected histograms will not be filled\n");
   } else{
-         if ( isPPbData) {    //Only use two profiles if pPb
-            const Char_t* periodNames[2] = {"LHC13b", "LHC13c"};
-            TProfile* multEstimatorAvg[2];                       
-            TFile* fileEstimator=TFile::Open(estimatorFilename.Data());
-            if(!fileEstimator)  {
-               AliFatal("File with multiplicity estimator not found\n"); 
-               return;
-            }
-            
-            for(Int_t ip=0; ip<2; ip++) {
-               multEstimatorAvg[ip] = (TProfile*)(fileEstimator->Get(Form("SPDmult10_%s",periodNames[ip]))->Clone(Form("SPDmult10_%s_clone",periodNames[ip])));  
-            }
-            dMultTask->SetMultiplVsZProfileLHC13b(multEstimatorAvg[0]);
-            dMultTask->SetMultiplVsZProfileLHC13c(multEstimatorAvg[1]);
-            dMultTask->SetReferenceMultiplcity(refMult);
-            
-         }
-         else {
-            const Char_t* periodNames[4] = {"LHC10b", "LHC10c", "LHC10d", "LHC10e"};
-            TProfile* multEstimatorAvg[4];                       
-            TFile* fileEstimator=TFile::Open(estimatorFilename.Data());
-            if(!fileEstimator)  {
-               AliFatal("File with multiplicity estimator not found\n"); 
-               return;
-            }
-            
-            for(Int_t ip=0; ip<4; ip++) {
-               multEstimatorAvg[ip] = (TProfile*)(fileEstimator->Get(Form("SPDmult10_%s",periodNames[ip]))->Clone(Form("SPDmult10_%s_clone",periodNames[ip])));  
-            }
-            dMultTask->SetMultiplVsZProfileLHC10b(multEstimatorAvg[0]);
-            dMultTask->SetMultiplVsZProfileLHC10c(multEstimatorAvg[1]);
-            dMultTask->SetMultiplVsZProfileLHC10d(multEstimatorAvg[2]);
-            dMultTask->SetMultiplVsZProfileLHC10e(multEstimatorAvg[3]);
-            dMultTask->SetReferenceMultiplcity(refMult);
-   }
+                     
+    TFile* fileEstimator=TFile::Open(estimatorFilename.Data());
+    if(!fileEstimator)  {
+      AliFatal("File with multiplicity estimator not found\n");
+      return;
+    }
+
+    dMultTask->SetReferenceMultiplcity(refMult);
+
+    if (isPPbData) {    //Only use two profiles if pPb
+      const Char_t* periodNames[2] = {"LHC13b", "LHC13c"};
+      TProfile* multEstimatorAvg[2];
+      for(Int_t ip=0; ip<2; ip++) {
+       multEstimatorAvg[ip] = (TProfile*)(fileEstimator->Get(Form("SPDmult10_%s",periodNames[ip]))->Clone(Form("SPDmult10_%s_clone",periodNames[ip])));
+       if (!multEstimatorAvg[ip]) {
+         AliFatal(Form("Multiplicity estimator for %s not found! Please check your estimator file",periodNames[ip]));
+         return;
+       }
+      }
+      dMultTask->SetMultiplVsZProfileLHC13b(multEstimatorAvg[0]);
+      dMultTask->SetMultiplVsZProfileLHC13c(multEstimatorAvg[1]);
+    }
+    else {
+      const Char_t* periodNames[4] = {"LHC10b", "LHC10c", "LHC10d", "LHC10e"};
+      TProfile* multEstimatorAvg[4];                       
+      for(Int_t ip=0; ip<4; ip++) {
+       multEstimatorAvg[ip] = (TProfile*)(fileEstimator->Get(Form("SPDmult10_%s",periodNames[ip]))->Clone(Form("SPDmult10_%s_clone",periodNames[ip])));
+       if (!multEstimatorAvg[ip]) {
+         AliFatal(Form("Multiplicity estimator for %s not found! Please check your estimator file",periodNames[ip]));
+         return;
+       }
+      }
+      dMultTask->SetMultiplVsZProfileLHC10b(multEstimatorAvg[0]);
+      dMultTask->SetMultiplVsZProfileLHC10c(multEstimatorAvg[1]);
+      dMultTask->SetMultiplVsZProfileLHC10d(multEstimatorAvg[2]);
+      dMultTask->SetMultiplVsZProfileLHC10e(multEstimatorAvg[3]);
+    }
   }
   mgr->AddTask(dMultTask);