Update hStrangeCorrelation.cxx

Author: kcui1

PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

@@ -1916,10 +1916,19 @@ struct HStrangeCorrelation {
       for (int i = 1; i <= histos.get<TH1>(HIST("Prediction/hEventSelection"))->GetNbinsX(); i++) {
         histos.get<TH1>(HIST("Prediction/hEventSelection"))->GetXaxis()->SetBinLabel(i, eventSelLabel[i - 1]);
       }
-      if (masterConfigurations.useCentralityinPrediction)
+      if (masterConfigurations.useCentralityinPrediction) {
+        if (masterConfigurations.doSeparateFT0Prediction) {
+          histos.add("Prediction/hTriggerFT0A", "Trigger Tracks FT0A", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMult});
+          histos.add("Prediction/hTriggerFT0C", "Trigger Tracks FT0C", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMult});
+        }
         histos.add("Prediction/hTrigger", "Trigger Tracks", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMult});
-      else
+      } else {
+        if (masterConfigurations.doSeparateFT0Prediction) {
+          histos.add("Prediction/hTriggerFT0A", "Trigger Tracks FT0A", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMultiplicity});
+          histos.add("Prediction/hTriggerFT0C", "Trigger Tracks FT0C", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMultiplicity});
+        }
         histos.add("Prediction/hTrigger", "Trigger Tracks", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisMultiplicity});
+      }
       for (int i = 0; i < AssocParticleTypes; i++) {
         if (TESTBIT(doCorrelation, i))
           histos.add(fmt::format("Prediction/h{}", Particlenames[i]).c_str(), "", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisPhi});
@@ -3328,7 +3337,19 @@ struct HStrangeCorrelation {
       if (std::abs(mcParticle.pdgCode()) == PDG_t::kPiPlus || std::abs(mcParticle.pdgCode()) == PDG_t::kKPlus || std::abs(mcParticle.pdgCode()) == PDG_t::kProton || std::abs(mcParticle.pdgCode()) == PDG_t::kElectron || std::abs(mcParticle.pdgCode()) == PDG_t::kMuonMinus) {
         if (!masterConfigurations.doTriggPhysicalPrimary || mcParticle.isPhysicalPrimary()) {
           triggerIndices.emplace_back(iteratorNum);
-          histos.fill(HIST("Prediction/hTrigger"), gpt, geta, gphi);
+          if (masterConfigurations.useCentralityinPrediction) {
+            if (masterConfigurations.doSeparateFT0Prediction) {
+              histos.fill(HIST("Prediction/hTriggerFT0A"), gpt, geta, centMultFT0A);
+              histos.fill(HIST("Prediction/hTriggerFT0C"), gpt, geta, centMultFT0C);
+            }
+            histos.fill(HIST("Prediction/hTrigger"), gpt, geta, centMultFT0M);
+          } else {
+            if (masterConfigurations.doSeparateFT0Prediction) {
+              histos.fill(HIST("Prediction/hTriggerFT0A"), gpt, geta, multFT0A);
+              histos.fill(HIST("Prediction/hTriggerFT0C"), gpt, geta, multFT0C);
+            }
+            histos.fill(HIST("Prediction/hTrigger"), gpt, geta, multFT0M);
+          }
         }
         if (masterConfigurations.doCorrelationHadron) {
           if (!doAssocPhysicalPrimary || mcParticle.isPhysicalPrimary()) {