Adding support for extra centrality estimators e.g. ZNA, used for the pPb running...

diff --git a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.cxx b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.cxx
index 7a615da6b80c79b0faa4269e318063fede0b1bc3..41a40b45f8be2253b4ad356411baaafcd0f99d51 100755 (executable)
--- a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.cxx
+++ b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.cxx
@@ -70,6 +70,7 @@ AliAnalysisTaskBFPsi::AliAnalysisTaskBFPsi(const char *name)
   fHistListPIDQA(0),\r
   fHistEventStats(0),\r
   fHistCentStats(0),\r
+  fHistCentStatsUsed(0),    //++++++++++++++++++++++++++\r
   fHistTriggerStats(0),\r
   fHistTrackStats(0),\r
   fHistVx(0),\r
@@ -282,14 +283,20 @@ void AliAnalysisTaskBFPsi::UserCreateOutputObjects() {
     fHistEventStats->GetXaxis()->SetBinLabel(i,gCutName[i-1].Data());\r
   fList->Add(fHistEventStats);\r
 \r
-  TString gCentName[9] = {"V0M","FMD","TRK","TKL","CL0","CL1","V0MvsFMD","TKLvsV0M","ZEMvsZDC"};\r
+  TString gCentName[13] = {"V0M","V0A","V0C","FMD","TRK","TKL","CL0","CL1","ZNA","ZPA","V0MvsFMD","TKLvsV0M","ZEMvsZDC"};\r
   fHistCentStats = new TH2F("fHistCentStats",\r
                              "Centrality statistics;;Cent percentile",\r
-                           9,-0.5,8.5,220,-5,105);\r
-  for(Int_t i = 1; i <= 9; i++)\r
+                           13,-0.5,12.5,220,-5,105);\r
+  for(Int_t i = 1; i <= 13; i++){\r
     fHistCentStats->GetXaxis()->SetBinLabel(i,gCentName[i-1].Data());\r
+    //fHistCentStatsUsed->GetXaxis()->SetBinLabel(i,gCentName[i-1].Data());  //++++++++++++++++++++++\r
+  }\r
   fList->Add(fHistCentStats);\r
 \r
+  fHistCentStatsUsed = new TH2F("fHistCentStatsUsed","Centrality statistics;;Cent percentile", 1,-0.5,0.5,220,-5,105); //++++++++++++++++++++++\r
+  fHistCentStatsUsed->GetXaxis()->SetBinLabel(1,fCentralityEstimator.Data());  //++++++++++++++++++++++\r
+  fList->Add(fHistCentStatsUsed); //++++++++++++++++++++++\r
+\r
   fHistTriggerStats = new TH1F("fHistTriggerStats","Trigger statistics;TriggerBit;N_{events}",1025,0,1025);\r
   fList->Add(fHistTriggerStats);\r
 \r
@@ -755,14 +762,21 @@ Double_t AliAnalysisTaskBFPsi::IsEventAccepted(AliVEvent *event){
 \r
          // QA for centrality estimators\r
          fHistCentStats->Fill(0.,header->GetCentralityP()->GetCentralityPercentile("V0M"));\r
-         fHistCentStats->Fill(1.,header->GetCentralityP()->GetCentralityPercentile("FMD"));\r
-         fHistCentStats->Fill(2.,header->GetCentralityP()->GetCentralityPercentile("TRK"));\r
-         fHistCentStats->Fill(3.,header->GetCentralityP()->GetCentralityPercentile("TKL"));\r
-         fHistCentStats->Fill(4.,header->GetCentralityP()->GetCentralityPercentile("CL0"));\r
-         fHistCentStats->Fill(5.,header->GetCentralityP()->GetCentralityPercentile("CL1"));\r
-         fHistCentStats->Fill(6.,header->GetCentralityP()->GetCentralityPercentile("V0MvsFMD"));\r
-         fHistCentStats->Fill(7.,header->GetCentralityP()->GetCentralityPercentile("TKLvsV0M"));\r
-         fHistCentStats->Fill(8.,header->GetCentralityP()->GetCentralityPercentile("ZEMvsZDC"));\r
+         fHistCentStats->Fill(1.,header->GetCentralityP()->GetCentralityPercentile("V0A"));\r
+         fHistCentStats->Fill(2.,header->GetCentralityP()->GetCentralityPercentile("V0C"));\r
+         fHistCentStats->Fill(3.,header->GetCentralityP()->GetCentralityPercentile("FMD"));\r
+         fHistCentStats->Fill(4.,header->GetCentralityP()->GetCentralityPercentile("TRK"));\r
+         fHistCentStats->Fill(5.,header->GetCentralityP()->GetCentralityPercentile("TKL")); \r
+         fHistCentStats->Fill(6.,header->GetCentralityP()->GetCentralityPercentile("CL0"));\r
+         fHistCentStats->Fill(7.,header->GetCentralityP()->GetCentralityPercentile("CL1"));\r
+         fHistCentStats->Fill(8.,header->GetCentralityP()->GetCentralityPercentile("ZNA"));\r
+         fHistCentStats->Fill(9.,header->GetCentralityP()->GetCentralityPercentile("ZPA"));\r
+         fHistCentStats->Fill(10.,header->GetCentralityP()->GetCentralityPercentile("V0MvsFMD"));\r
+         fHistCentStats->Fill(11.,header->GetCentralityP()->GetCentralityPercentile("TKLvsV0M"));\r
+         fHistCentStats->Fill(12.,header->GetCentralityP()->GetCentralityPercentile("ZEMvsZDC"));\r
+\r
+         // Centrality estimator USED   ++++++++++++++++++++++++++++++\r
+         fHistCentStatsUsed->Fill(0.,header->GetCentralityP()->GetCentralityPercentile(fCentralityEstimator.Data()));\r
          \r
          // centrality QA (V0M)\r
          fHistV0M->Fill(event->GetVZEROData()->GetMTotV0A(), event->GetVZEROData()->GetMTotV0C());\r
@@ -786,14 +800,21 @@ Double_t AliAnalysisTaskBFPsi::IsEventAccepted(AliVEvent *event){
 \r
        // QA for centrality estimators\r
        fHistCentStats->Fill(0.,centrality->GetCentralityPercentile("V0M"));\r
-       fHistCentStats->Fill(1.,centrality->GetCentralityPercentile("FMD"));\r
-       fHistCentStats->Fill(2.,centrality->GetCentralityPercentile("TRK"));\r
-       fHistCentStats->Fill(3.,centrality->GetCentralityPercentile("TKL"));\r
-       fHistCentStats->Fill(4.,centrality->GetCentralityPercentile("CL0"));\r
-       fHistCentStats->Fill(5.,centrality->GetCentralityPercentile("CL1"));\r
-       fHistCentStats->Fill(6.,centrality->GetCentralityPercentile("V0MvsFMD"));\r
-       fHistCentStats->Fill(7.,centrality->GetCentralityPercentile("TKLvsV0M"));\r
-       fHistCentStats->Fill(8.,centrality->GetCentralityPercentile("ZEMvsZDC"));\r
+       fHistCentStats->Fill(1.,centrality->GetCentralityPercentile("V0A"));\r
+       fHistCentStats->Fill(2.,centrality->GetCentralityPercentile("V0C"));\r
+       fHistCentStats->Fill(3.,centrality->GetCentralityPercentile("FMD"));\r
+       fHistCentStats->Fill(4.,centrality->GetCentralityPercentile("TRK"));\r
+       fHistCentStats->Fill(5.,centrality->GetCentralityPercentile("TKL"));\r
+       fHistCentStats->Fill(6.,centrality->GetCentralityPercentile("CL0"));\r
+       fHistCentStats->Fill(7.,centrality->GetCentralityPercentile("CL1"));\r
+       fHistCentStats->Fill(8.,centrality->GetCentralityPercentile("ZNA"));\r
+       fHistCentStats->Fill(9.,centrality->GetCentralityPercentile("ZPA"));\r
+       fHistCentStats->Fill(10.,centrality->GetCentralityPercentile("V0MvsFMD"));\r
+       fHistCentStats->Fill(11.,centrality->GetCentralityPercentile("TKLvsV0M"));\r
+       fHistCentStats->Fill(12.,centrality->GetCentralityPercentile("ZEMvsZDC"));\r
+\r
+       // Centrality estimator USED   ++++++++++++++++++++++++++++++\r
+       fHistCentStatsUsed->Fill(0.,centrality->GetCentralityPercentile(fCentralityEstimator.Data()));\r
 \r
        // centrality QA (V0M)\r
        fHistV0M->Fill(event->GetVZEROData()->GetMTotV0A(), event->GetVZEROData()->GetMTotV0C());\r

diff --git a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h
index eefd2b0a6244407add36a381c15b568427c4568f..716d3333951f11aae1e87050ae54a203210d1fc1 100755 (executable)
--- a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h
+++ b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h
@@ -190,6 +190,7 @@ class AliAnalysisTaskBFPsi : public AliAnalysisTaskSE {
 \r
   TH2F *fHistEventStats; //event stats\r
   TH2F *fHistCentStats; //centrality stats\r
+  TH2F *fHistCentStatsUsed; //centrality stats USED +++++++++++++++++++++++\r
   TH1F *fHistTriggerStats; //trigger stats\r
   TH1F *fHistTrackStats; //Track filter bit stats\r
   TH1F *fHistVx; //x coordinate of the primary vertex\r