[PWGHF] Add alternative FT0c centrality estimator in HF centrality and flow task

Author: stefanopolitano

PWGHF/Core/CentralityEstimation.h

@@ -30,6 +30,7 @@ enum CentralityEstimator {
   FT0M,
   FV0A,
   NTracksPV,
+  FT0CVariant1,
   NCentralityEstimators
 };
 
@@ -43,6 +44,11 @@ concept HasFT0CCent = requires(T collision) {
   collision.centFT0C();
 };
 
+template <typename T>
+concept HasFT0CVariant1Cent = requires(T collision) {
+  collision.centFT0CVariant1();
+};
+
 template <typename T>
 concept HasFT0MCent = requires(T collision) {
   collision.centFT0M();
@@ -76,6 +82,15 @@ float getCentralityColl(const TCollision& collision)
   return collision.centFT0C();
 }
 
+/// Evaluate centrality/multiplicity percentile using FT0C variant 1 estimator
+/// \param candidate is candidate
+/// \return centrality/multiplicity percentile of the collision
+template <HasFT0CVariant1Cent TCollision>
+float getCentralityColl(const TCollision& collision)
+{
+  return collision.centFT0CVariant1();
+}
+
 /// Evaluate centrality/multiplicity percentile using FT0M estimator
 /// \param candidate is candidate
 /// \return centrality/multiplicity percentile of the collision
@@ -144,6 +159,18 @@ float getCentralityColl(const TCollision& collision, const int centEstimator)
       }
       LOG(fatal) << "Collision does not have centFV0A().";
       break;
+    case CentralityEstimator::NTracksPV:
+      if constexpr (HasNTracksPvCent<TCollision>) {
+        return collision.centNTPV();
+      }
+      LOG(fatal) << "Collision does not have centNTPV().";
+      break;
+    case CentralityEstimator::FT0CVariant1:
+      if constexpr (HasFT0CVariant1Cent<TCollision>) {
+        return collision.centFT0CVariant1();
+      }
+      LOG(fatal) << "Collision does not have centFT0CVariant1().";
+      break;
     default:
       LOG(fatal) << "Centrality estimator not valid. See CentralityEstimator for valid values.";
       break;

PWGHF/D2H/Tasks/taskFlowCharmHadrons.cxx

@@ -109,7 +109,7 @@ struct HfTaskFlowCharmHadrons {
 
   Configurable<int> harmonic{"harmonic", 2, "harmonic number"};
   Configurable<int> qVecDetector{"qVecDetector", 3, "Detector for Q vector estimation (FV0A: 0, FT0M: 1, FT0A: 2, FT0C: 3, TPC Pos: 4, TPC Neg: 5, TPC Tot: 6)"};
-  Configurable<int> centEstimator{"centEstimator", 2, "Centrality estimation (FT0A: 1, FT0C: 2, FT0M: 3, FV0A: 4)"};
+  Configurable<int> centEstimator{"centEstimator", 2, "Centrality estimation (FT0A: 1, FT0C: 2, FT0M: 3, FV0A: 4, NTracksPV: 5, FT0CVariant1: 6)"};
   Configurable<int> selectionFlag{"selectionFlag", 1, "Selection Flag for hadron (e.g. 1 for skimming, 3 for topo. and kine., 7 for PID)"};
   Configurable<float> centralityMin{"centralityMin", 0., "Minimum centrality accepted in SP/EP computation (not applied in resolution process)"};
   Configurable<float> centralityMax{"centralityMax", 100., "Maximum centrality accepted in SP/EP computation (not applied in resolution process)"};
@@ -146,7 +146,7 @@ struct HfTaskFlowCharmHadrons {
   using CandXic0DataWMl = soa::Filtered<soa::Join<aod::HfCandToXiPiKf, aod::HfSelToXiPiKf, aod::HfMlToXiPi>>;
   using CandD0DataWMl = soa::Filtered<soa::Join<aod::HfCand2Prong, aod::HfSelD0, aod::HfMlD0>>;
   using CandD0Data = soa::Filtered<soa::Join<aod::HfCand2Prong, aod::HfSelD0>>;
-  using CollsWithQvecs = soa::Join<aod::Collisions, aod::EvSels, aod::QvectorFT0Cs, aod::QvectorFT0As, aod::QvectorFT0Ms, aod::QvectorFV0As, aod::QvectorBPoss, aod::QvectorBNegs, aod::QvectorBTots, aod::CentFV0As, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs>;
+  using CollsWithQvecs = soa::Join<aod::Collisions, aod::EvSels, aod::QvectorFT0Cs, aod::QvectorFT0As, aod::QvectorFT0Ms, aod::QvectorFV0As, aod::QvectorBPoss, aod::QvectorBNegs, aod::QvectorBTots, aod::CentFV0As, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs, aod::centFT0CVariant1s>;
   using TracksWithExtra = soa::Join<aod::Tracks, aod::TracksExtra>;
 
   Filter filterSelectDsCandidates = aod::hf_sel_candidate_ds::isSelDsToKKPi >= selectionFlag || aod::hf_sel_candidate_ds::isSelDsToPiKK >= selectionFlag;
@@ -878,7 +878,7 @@ struct HfTaskFlowCharmHadrons {
     float const xQVecBTot = collision.qvecBTotRe();
     float const yQVecBTot = collision.qvecBTotIm();
 
-    centrality = o2::hf_centrality::getCentralityColl(collision, o2::hf_centrality::CentralityEstimator::FT0C);
+    centrality = o2::hf_centrality::getCentralityColl(collision, CentEstimator);
     if (storeResoOccu) {
       const auto occupancy = o2::hf_occupancy::getOccupancyColl(collision, occEstimator);
       registry.fill(HIST("trackOccVsFT0COcc"), collision.trackOccupancyInTimeRange(), collision.ft0cOccupancyInTimeRange());