Restore centrEstimator configurable for FT0C and FT0M

Author: lorber98

PWGLF/Tasks/Strangeness/strangenessInJetsIons.cxx

@@ -82,9 +82,7 @@ using SelCollisions = soa::Join<aod::Collisions,
                                 aod::CentFT0Ms>;
 using SimCollisions = soa::Join<aod::McCollisionLabels,
                                 aod::Collisions,
-                                aod::EvSels,
-                                aod::CentFT0Cs,
-                                aod::CentFT0Ms>;
+                                aod::EvSels>;
 using DaughterTracks = soa::Join<aod::Tracks, aod::TracksIU, aod::TracksExtra, aod::TracksCovIU, aod::TracksDCA,
                                  aod::pidTPCFullPi, aod::pidTPCFullKa, aod::pidTPCFullPr,
                                  aod::pidTOFFullPi, aod::pidTOFFullKa, aod::pidTOFFullPr>;
@@ -127,7 +125,7 @@ struct StrangenessInJetsIons {
   Configurable<double> deltaEtaEdge{"deltaEtaEdge", 0.05, "eta gap from detector edge"};
   Configurable<bool> cfgSkimmedProcessing{"cfgSkimmedProcessing", false, "Enable processing of skimmed data"};
   Configurable<std::string> triggerName{"triggerName", "fOmega", "Software trigger name"};
-  // Configurable<int> centrEstimator{"centrEstimator", 1, "Select centrality estimator. Options: 0 = FT0C, 1 = FT0M. CCDB objects available only for FT0M."};
+  Configurable<int> centrEstimator{"centrEstimator", 1, "Select centrality estimator. Options: 0 = FT0C, 1 = FT0M. CCDB objects available only for FT0M."};
 
   // Event selection
   Configurable<bool> requireNoSameBunchPileup{"requireNoSameBunchPileup", true, "Require kNoSameBunchPileup selection"};
@@ -217,14 +215,15 @@ struct StrangenessInJetsIons {
     }
 
     // Define binning and axis specifications for multiplicity, eta, pT, PID, and invariant mass histograms
-    std::string multAxTitle = "FT0M percentile";
-    // if (centrEstimator == 0) {
-    //   multAxTitle = "FT0C percentile";
-    // } else if (centrEstimator == 1) {
-    //   multAxTitle = "FT0M percentile";
-    // } else {
-    //   LOG(fatal) << "Centrality estimator " << centrEstimator << " not available. Exit." << endl;
-    // }
+    std::string multAxTitle;
+    if (centrEstimator == 0) {
+      multAxTitle = "FT0C percentile";
+      LOG(warning) << "FT0C ccdb object not available yet." << endl;
+    } else if (centrEstimator == 1) {
+      multAxTitle = "FT0M percentile";
+    } else {
+      LOG(fatal) << "Centrality estimator " << centrEstimator << " not available. Exit." << endl;
+    }
     AxisSpec multAxis = {multBinning, multAxTitle};
 
     const AxisSpec ptAxis{100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"};
@@ -1169,12 +1168,12 @@ struct StrangenessInJetsIons {
     registryData.fill(HIST("number_of_events_data"), 7.5);
 
     // Event multiplicity
-    float multiplicity = collision.centFT0M();
-    // if (centrEstimator == 0) {
-    //   multiplicity = collision.centFT0C();
-    // } else {
-    //   multiplicity = collision.centFT0M();
-    // }
+    float multiplicity;
+    if (centrEstimator == 0) {
+      multiplicity = collision.centFT0C();
+    } else {
+      multiplicity = collision.centFT0M();
+    }
 
     // Fill event multiplicity
     registryData.fill(HIST("number_of_events_vsmultiplicity"), multiplicity);
@@ -1356,7 +1355,7 @@ struct StrangenessInJetsIons {
   Preslice<aod::McParticles> perMCCollision = o2::aod::mcparticle::mcCollisionId;
   Preslice<DaughterTracksMC> perCollisionTrk = o2::aod::track::collisionId;
 
-  void processMCgenerated(soa::Join<aod::McCollisions, aod::McCentFT0Ms> const& collisions,
+  void processMCgenerated(soa::Join<aod::McCollisions, aod::McCentFT0Ms, aod::McCentFT0Cs> const& collisions,
                           aod::McParticles const& mcParticles)
   {
     // Define per-event particle containers
@@ -1418,12 +1417,12 @@ struct StrangenessInJetsIons {
       // float genMultiplicity = multiplicity;
 
       // Multiplicity of generated event
-      double genMultiplicity = collision.centFT0M();
-      // if (centrEstimator == 0) {
-      //   genMultiplicity = collision.centFT0C();
-      // } else {
-      //   genMultiplicity = collision.centFT0M();
-      // }
+      float genMultiplicity;
+      if (centrEstimator == 0) {
+        genMultiplicity = collision.centFT0C();
+      } else {
+        genMultiplicity = collision.centFT0M();
+      }
 
       // MC particles per collision
       auto mcParticlesPerColl = mcParticles.sliceBy(perMCCollision, collision.globalIndex());
@@ -1660,7 +1659,7 @@ struct StrangenessInJetsIons {
 
   // Reconstructed MC events
   void processMCreconstructed(SimCollisions const& collisions,
-                              soa::Join<aod::McCollisions, aod::McCentFT0Ms> const&,
+                              soa::Join<aod::McCollisions, aod::McCentFT0Ms, aod::McCentFT0Cs> const&,
                               DaughterTracksMC const& mcTracks, aod::V0Datas const& fullV0s,
                               aod::CascDataExt const& Cascades, const aod::McParticles&)
   {
@@ -1680,7 +1679,7 @@ struct StrangenessInJetsIons {
         continue;
       }
 
-      const auto& mcCollision = collision.mcCollision_as<soa::Join<aod::McCollisions, aod::McCentFT0Ms>>();
+      const auto& mcCollision = collision.mcCollision_as<soa::Join<aod::McCollisions, aod::McCentFT0Ms, aod::McCentFT0Cs>>();
 
       // Clear containers at the start of the event loop
       fjParticles.clear();
@@ -1716,8 +1715,12 @@ struct StrangenessInJetsIons {
       registryMC.fill(HIST("number_of_events_mc_rec"), 4.5);
 
       // Event multiplicity
-      float multiplicity = mcCollision.centFT0M();
-
+      float multiplicity;
+      if (centrEstimator == 0) {
+        multiplicity = mcCollision.centFT0C();
+      } else {
+        multiplicity = mcCollision.centFT0M();
+      }
       // Number of V0 and cascades per collision
       auto v0sPerColl = fullV0s.sliceBy(perCollisionV0, collision.globalIndex());
       auto cascPerColl = Cascades.sliceBy(perCollisionCasc, collision.globalIndex());