[PWGLF] Add Midrapidity hist In H-Scorrelation task (#16194)

Author: kcui1

PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

@@ -161,6 +161,7 @@ struct HStrangeCorrelation {
     ConfigurableAxis axisPtQA{"axisPtQA", {VARIABLE_WIDTH, 0.0f, 0.1f, 0.2f, 0.3f, 0.4f, 0.5f, 0.6f, 0.7f, 0.8f, 0.9f, 1.0f, 1.1f, 1.2f, 1.3f, 1.4f, 1.5f, 1.6f, 1.7f, 1.8f, 1.9f, 2.0f, 2.2f, 2.4f, 2.6f, 2.8f, 3.0f, 3.2f, 3.4f, 3.6f, 3.8f, 4.0f, 4.4f, 4.8f, 5.2f, 5.6f, 6.0f, 6.5f, 7.0f, 7.5f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 17.0f, 19.0f, 21.0f, 23.0f, 25.0f, 30.0f, 35.0f, 40.0f, 50.0f}, "pt axis for QA histograms"};
     ConfigurableAxis axisMassNSigma{"axisMassNSigma", {40, -2, 2}, "Axis for mass Nsigma"};
     ConfigurableAxis axisMultiplicity{"axisMultiplicity", {VARIABLE_WIDTH, 0, 20, 40, 60, 100, 120, 140, 160, 180, 200, 220, 240, 260, 280, 300}, "Binning of the Multiplicity axis in model prediction process"};
+    ConfigurableAxis axisMidrapidityMultiplicity{"axisMidrapidityMultiplicity", {VARIABLE_WIDTH, 0, 20, 40, 60, 100, 120, 140, 160, 180, 200, 220, 240, 260, 280, 300}, "Binning of the Midrapidity Multiplicity axis in model prediction process"};
 
   } axesConfigurations;
 
@@ -1803,7 +1804,7 @@ struct HStrangeCorrelation {
 
     bool hStrange = false;
     for (int i = 0; i < AssocParticleTypes; i++) {
-      if (TESTBIT(doCorrelation, i)) {
+      if (TESTBIT(doCorrelation, i) && (doprocessSameEventHV0s || doprocessSameEventHCascades)) {
         if (masterConfigurations.doFullCorrelationStudy)
           histos.add(fmt::format("sameEvent/Signal/{}", Particlenames[i]).c_str(), "", kTHnF, {axisDeltaPhiNDim, axisDeltaEtaNDim, axisPtAssocNDim, axisPtTriggerNDim, axisVtxZNDim, axisMultNDim});
         if (doDeltaPhiStarCheck && masterConfigurations.doFullCorrelationStudy) {
@@ -1829,13 +1830,13 @@ struct HStrangeCorrelation {
         }
       }
     }
-    if (TESTBIT(doCorrelation, 7)) {
+    if (TESTBIT(doCorrelation, 7) && doprocessSameEventHPions) {
       histos.add("hPionEtaVsPtAllSelected", "", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMult});
       histos.add("hPionEtaVsPt", "", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMult});
       histos.add("hPositivePionEtaVsPt", "", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMult});
       histos.add("hNegativePionEtaVsPt", "", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMult});
     }
-    if (TESTBIT(doCorrelation, 8)) {
+    if (TESTBIT(doCorrelation, 8) && doprocessSameEventHHadrons) {
       histos.add("hAsssocTrackEtaVsPtVsPhi", "", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisPhi});
       histos.add("hAssocPrimaryEtaVsPt", "", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMult});
       histos.add("hAssocHadronsAllSelectedEtaVsPt", "", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMult});
@@ -1931,9 +1932,15 @@ struct HStrangeCorrelation {
       } else {
         if (masterConfigurations.doSeparateFT0Prediction) {
           histos.add("Prediction/hTriggerFT0A", "Trigger Tracks FT0A", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMultiplicity});
+          histos.add("Prediction/hFT0AvsNchEta08", "Nch in 0.8 vs FT0A multiplicity", kTH2F, {axesConfigurations.axisMultiplicity, axesConfigurations.axisMidrapidityMultiplicity});
+          histos.add("Prediction/hFT0AvsNchEta05", "Nch in 0.5 vs FT0A multiplicity", kTH2F, {axesConfigurations.axisMultiplicity, axesConfigurations.axisMidrapidityMultiplicity});
           histos.add("Prediction/hTriggerFT0C", "Trigger Tracks FT0C", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMultiplicity});
+          histos.add("Prediction/hFT0CvsNchEta08", "Nch in 0.8 vs FT0C multiplicity", kTH2F, {axesConfigurations.axisMultiplicity, axesConfigurations.axisMidrapidityMultiplicity});
+          histos.add("Prediction/hFT0CvsNchEta05", "Nch in 0.5 vs FT0C multiplicity", kTH2F, {axesConfigurations.axisMultiplicity, axesConfigurations.axisMidrapidityMultiplicity});
         }
         histos.add("Prediction/hTrigger", "Trigger Tracks", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMultiplicity});
+        histos.add("Prediction/hFT0MvsNchEta08", "Nch in 0.8 vs FT0M multiplicity", kTH2F, {axesConfigurations.axisMultiplicity, axesConfigurations.axisMidrapidityMultiplicity});
+        histos.add("Prediction/hFT0MvsNchEta05", "Nch in 0.5 vs FT0M multiplicity", kTH2F, {axesConfigurations.axisMultiplicity, axesConfigurations.axisMidrapidityMultiplicity});
       }
       for (int i = 0; i < AssocParticleTypes; i++) {
         if (TESTBIT(doCorrelation, i))
@@ -3317,6 +3324,8 @@ struct HStrangeCorrelation {
     float multFT0M = -1;
     float multFT0A = -1;
     float multFT0C = -1;
+    float multEta08 = -1;
+    float multEta05 = -1;
     histos.fill(HIST("Prediction/hEventSelection"), 0.5);
     if (masterConfigurations.selectINELgtZERO && !o2::pwglf::isINELgt0mc(mcParticles, pdgDB)) {
       return;
@@ -3334,6 +3343,18 @@ struct HStrangeCorrelation {
       multFT0M = mCounter.countFT0A(mcParticles) + mCounter.countFT0C(mcParticles);
       multFT0A = mCounter.countFT0A(mcParticles);
       multFT0C = mCounter.countFT0C(mcParticles);
+      multEta08 = mCounter.countEta08(mcParticles);
+      multEta05 = mCounter.countEta05(mcParticles);
+    }
+    if (!masterConfigurations.useCentralityinPrediction) {
+      if (masterConfigurations.doSeparateFT0Prediction) {
+        histos.fill(HIST("Prediction/hFT0AvsNchEta08"), multFT0A, multEta08);
+        histos.fill(HIST("Prediction/hFT0AvsNchEta05"), multFT0A, multEta05);
+        histos.fill(HIST("Prediction/hFT0CvsNchEta08"), multFT0C, multEta08);
+        histos.fill(HIST("Prediction/hFT0CvsNchEta05"), multFT0C, multEta05);
+      }
+      histos.fill(HIST("Prediction/hFT0MvsNchEta08"), multFT0M, multEta08);
+      histos.fill(HIST("Prediction/hFT0MvsNchEta05"), multFT0M, multEta05);
     }
     int iteratorNum = -1;
     for (auto const& mcParticle : mcParticles) {