Update HadNucleiFemto.cxx

Author: blacwovie

PWGCF/Femto/FemtoNuclei/TableProducer/HadNucleiFemto.cxx

@@ -84,6 +84,10 @@ using CollBracket = o2::math_utils::Bracket<int>;
 using HyperCandidates = aod::DataHypCandsWColl;
 using HyperCandidatesMC = aod::MCHypCandsWColl;
 using CollisionsFull = soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs, aod::FT0Mults>;
+// The standard Femto producer stores multNTracksPV as its collision
+// multiplicity. Subscribe to PVMults only in the mixed-event process so the
+// same vertex--multiplicity pool definition can optionally be reproduced.
+using CollisionsFullWithPVMult = soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs, aod::FT0Mults, aod::PVMults>;
 using CollisionsFullMC = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels, aod::CentFT0Cs, aod::FT0Mults>;
 using HadHyperCollisionsFull = soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0As, aod::CentFT0Cs, aod::CentFT0Ms, aod::FT0Mults>;
 using HadHyperCollisionsFullMC = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels, aod::CentFT0As, aod::CentFT0Cs, aod::CentFT0Ms, aod::FT0Mults>;
@@ -333,6 +337,7 @@ struct HadNucleiFemto {
     // Event selection and mixing configuration
     Configurable<float> settingCutVertex{"settingCutVertex", 10.0f, "Accepted z-vertex range"};
     Configurable<int> settingNoMixedEvents{"settingNoMixedEvents", 5, "Number of mixed events per event"};
+    Configurable<bool> settingUseMultiplicityBinning{"settingUseMultiplicityBinning", false, "Use vertex and multNTracksPV for event mixing instead of vertex and centrality"};
     Configurable<bool> settingRequireBothSpeciesForMixing{"settingRequireBothSpeciesForMixing", false, "Use only events containing at least one selected nucleus and one selected hadron for mixed-event pairing"};
     Configurable<bool> settingEnableBkgUS{"settingEnableBkgUS", false, "Enable US background"};
     Configurable<bool> settingSaveUSandLS{"settingSaveUSandLS", true, "Save All Pairs"};
@@ -627,7 +632,9 @@ struct HadNucleiFemto {
   // binning for EM background
   ConfigurableAxis axisVertex{"axisVertex", {30, -10, 10}, "Binning for vtxz"};
   ConfigurableAxis axisCentrality{"axisCentrality", {40, 0, 100}, "Binning for centrality"};
-  using BinningType = ColumnBinningPolicy<aod::collision::PosZ, aod::cent::CentFT0C>;
+  ConfigurableAxis axisMultiplicity{"axisMultiplicity", {VARIABLE_WIDTH, 0., 4., 8., 12., 16., 20., 24., 28., 32., 36., 40., 44., 48., 52., 56., 60., 64., 68., 72., 76., 80., 84., 88., 92., 96., 100., 200.}, "Variable-width multNTracksPV bins for event mixing"};
+  using CentralityBinningType = ColumnBinningPolicy<aod::collision::PosZ, aod::cent::CentFT0C>;
+  using MultiplicityBinningType = ColumnBinningPolicy<aod::collision::PosZ, aod::mult::MultNTracksPV>;
 
   std::array<float, 6> mBBparamsNucleus{};
   float mMassHad{0.f};
@@ -3084,7 +3091,7 @@ struct HadNucleiFemto {
                          std::unordered_map<int, std::deque<HadHyperEvent>>& mixingPools,
                          int& mixingRunNumber)
   {
-    const BinningType configuredBinningPolicy{{axisVertex, axisCentrality}, true};
+    const CentralityBinningType configuredBinningPolicy{{axisVertex, axisCentrality}, true};
     for (const auto& collision : collisions) {
       if (!selectCollision<isMC>(collision, bcs)) {
         continue;
@@ -3236,10 +3243,11 @@ struct HadNucleiFemto {
   }
   PROCESS_SWITCH(HadNucleiFemto, processSameEvent, "Process Same event", false);
 
-  void processMixedEvent(const CollisionsFull& collisions, const TrackCandidates& tracks, const aod::BCsWithTimestamps&)
+  void processMixedEvent(const CollisionsFullWithPVMult& collisions, const TrackCandidates& tracks, const aod::BCsWithTimestamps&)
   {
     LOG(debug) << "Processing mixed event";
-    const BinningType configuredBinningPolicy{{axisVertex, axisCentrality}, true};
+    const CentralityBinningType centralityBinningPolicy{{axisVertex, axisCentrality}, true};
+    const MultiplicityBinningType multiplicityBinningPolicy{{axisVertex, axisMultiplicity}, true};
 
     for (const auto& collision : collisions) {
       mQaRegistry.fill(HIST("hMixedEventSelections"), 0);
@@ -3285,7 +3293,12 @@ struct HadNucleiFemto {
         continue;
       }
 
-      const int poolBin = configuredBinningPolicy.getBin(std::make_tuple(collision.posZ(), collision.centFT0C()));
+      const int poolBin = eventMixing.settingUseMultiplicityBinning.value
+                            ? multiplicityBinningPolicy.getBin(std::make_tuple(collision.posZ(), collision.multNTracksPV()))
+                            : centralityBinningPolicy.getBin(std::make_tuple(collision.posZ(), collision.centFT0C()));
+      if (poolBin < 0) {
+        continue;
+      }
       auto& pool = mMixingPools[poolBin];
       mQaRegistry.fill(HIST("hMixingPoolOccupancy"), pool.size());