[Common] add extra MFT counters to mult table (#8980)

Co-authored-by: ALICE Builder <alibuild@users.noreply.github.com>
AliceO2Group · Dec 13, 2024 · 7826dc6 · 7826dc6
1 parent f60e2b9
commit 7826dc6
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 26 deletions.
diff --git a/Common/DataModel/Multiplicity.h b/Common/DataModel/Multiplicity.h
@@ -47,7 +47,8 @@ DECLARE_SOA_DYNAMIC_COLUMN(IsInelGt1, isInelGt1, //! is INEL > 1
                            [](int multPveta1) -> bool { return multPveta1 > 1; });
 
 // forward track counters
-DECLARE_SOA_COLUMN(MFTNtracks, mftNtracks, int); //!
+DECLARE_SOA_COLUMN(MFTNalltracks, mftNalltracks, int); //! overall counter, uses AO2D coll assoc
+DECLARE_SOA_COLUMN(MFTNtracks, mftNtracks, int);       //! reassigned, uses mult group software
 
 // MC
 DECLARE_SOA_COLUMN(MultMCFT0A, multMCFT0A, int);                       //!
@@ -113,7 +114,7 @@ DECLARE_SOA_TABLE(PVMults, "AOD", "PVMULT", //! Multiplicity from the PV contrib
                   mult::IsInelGt0<mult::MultNTracksPVeta1>,
                   mult::IsInelGt1<mult::MultNTracksPVeta1>);
 DECLARE_SOA_TABLE(MFTMults, "AOD", "MFTMULT", //! Multiplicity with MFT
-                  mult::MFTNtracks);
+                  mult::MFTNalltracks, mult::MFTNtracks);
 using BarrelMults = soa::Join<TrackletMults, TPCMults, PVMults>;
 using Mults = soa::Join<BarrelMults, FV0Mults, FT0Mults, FDDMults, ZDCMults>;
 using FT0Mult = FT0Mults::iterator;

diff --git a/Common/TableProducer/multiplicityTable.cxx b/Common/TableProducer/multiplicityTable.cxx
@@ -28,6 +28,7 @@
 #include "TableHelper.h"
 #include "MetadataHelper.h"
 #include "TList.h"
+#include "PWGMM/Mult/DataModel/bestCollisionTable.h"
 
 using namespace o2;
 using namespace o2::framework;
@@ -49,8 +50,7 @@ static constexpr int kFT0MultZeqs = 10;
 static constexpr int kFDDMultZeqs = 11;
 static constexpr int kPVMultZeqs = 12;
 static constexpr int kMultMCExtras = 13;
-static constexpr int kMFTMults = 14;
-static constexpr int nTables = 15;
+static constexpr int nTables = 14;
 
 // Checking that the Zeq tables are after the normal ones
 static_assert(kFV0Mults < kFV0MultZeqs);
@@ -72,10 +72,9 @@ static const std::vector<std::string> tableNames{"FV0Mults",       // 0
                                                  "FT0MultZeqs",    // 10
                                                  "FDDMultZeqs",    // 11
                                                  "PVMultZeqs",     // 12
-                                                 "MultMCExtras",   // 13
-                                                 "MFTMults"};      // 14
+                                                 "MultMCExtras"};  // 13
 static const std::vector<std::string> parameterNames{"Enable"};
-static const int defaultParameters[nTables][nParameters]{{-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}};
+static const int defaultParameters[nTables][nParameters]{{-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}, {-1}};
 
 struct MultiplicityTable {
   SliceCache cache;
@@ -94,7 +93,7 @@ struct MultiplicityTable {
   Produces<aod::PVMultZeqs> tablePVZeqs;        // 12
   Produces<aod::MultMCExtras> tableExtraMc;     // 13
   Produces<aod::Mult2MCExtras> tableExtraMult2MCExtras;
-  Produces<aod::MFTMults> mftMults;             // 14
+  Produces<aod::MFTMults> mftMults;             // Not accounted for, produced using custom process function to avoid dependencies
   Produces<aod::MultsGlobal> multsGlobal;       // Not accounted for, produced based on process function processGlobalTrackingCounters
 
   // For vertex-Z corrections in calibration
@@ -301,8 +300,7 @@ struct MultiplicityTable {
                    aod::Zdcs const&,
                    aod::FV0As const&,
                    aod::FT0s const&,
-                   aod::FDDs const&,
-                   aod::MFTTracks const& mftTracks)
+                   aod::FDDs const&)
   {
     // reserve memory
     for (auto i : mEnabledTables) {
@@ -348,9 +346,6 @@ struct MultiplicityTable {
           break;
         case kMultMCExtras: // MC extra information (nothing to do in the data)
           break;
-        case kMFTMults: // Equalized multiplicity for PV
-          mftMults.reserve(collisions.size());
-          break;
         default:
           LOG(fatal) << "Unknown table requested: " << i;
           break;
@@ -629,19 +624,6 @@ struct MultiplicityTable {
           case kMultMCExtras: // MC only (nothing to do)
           {
           } break;
-          case kMFTMults: {
-            // for centrality estimation with the MFT if desired
-            // step 1: produce proper grouping
-            const uint64_t collIdx = collision.globalIndex();
-            auto mftTracksGrouped = mftTracks.sliceBy(perCollisionMFT, collIdx);
-            int nTracks = 0;
-            for (auto& track : mftTracksGrouped) {
-              if (track.nClusters() >= 5) { // hardcoded on purpose to avoid trouble
-                nTracks++;
-              }
-            }
-            mftMults(nTracks);
-          } break;
           default: // Default
           {
             LOG(fatal) << "Unknown table requested: " << i;
@@ -754,12 +736,44 @@ struct MultiplicityTable {
     multsGlobal(nGlobalTracks, multNContribsEta08_kGlobalTrackWoDCA, multNContribsEta10_kGlobalTrackWoDCA, multNContribsEta05_kGlobalTrackWoDCA);
   }
 
+  void processRun3MFT(soa::Join<aod::Collisions, aod::EvSels>::iterator const&,
+                      o2::aod::MFTTracks const& mftTracks,
+                      soa::SmallGroups<aod::BestCollisionsFwd> const& retracks)
+  {
+    int nAllTracks = 0;
+    int nTracks = 0;
+
+    for (auto& track : mftTracks) {
+      if (track.nClusters() >= 5) { // hardcoded for now
+        nAllTracks++;
+      }
+    }
+
+    if (retracks.size() > 0) {
+      for (auto& retrack : retracks) {
+        auto track = retrack.mfttrack();
+        if (track.nClusters() < 5) {
+          continue; // min cluster requirement
+        }
+        if ((track.eta() > -2.0f) && (track.eta() < -3.9f)) {
+          continue; // too far to be of true interest
+        }
+        if (std::abs(retrack.bestDCAXY()) > 2.0f) {
+          continue; // does not point to PV properly
+        }
+        nTracks++;
+      }
+    }
+    mftMults(nAllTracks, nTracks);
+  }
+
   // Process switches
   PROCESS_SWITCH(MultiplicityTable, processRun2, "Produce Run 2 multiplicity tables", false);
   PROCESS_SWITCH(MultiplicityTable, processRun3, "Produce Run 3 multiplicity tables", true);
   PROCESS_SWITCH(MultiplicityTable, processGlobalTrackingCounters, "Produce Run 3 global counters", false);
   PROCESS_SWITCH(MultiplicityTable, processMC, "Produce MC multiplicity tables", false);
   PROCESS_SWITCH(MultiplicityTable, processMC2Mults, "Produce MC -> Mult map", false);
+  PROCESS_SWITCH(MultiplicityTable, processRun3MFT, "Produce MFT mult tables", false);
 };
 
 WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)