Treatment for the count of reconstructed tracks

Author: lucasjsilva

PWGLF/Tasks/GlobalEventProperties/studyPnch.cxx

@@ -12,7 +12,7 @@
 /// \file studyPnch.cxx
 ///
 /// \brief task for analysis of charged-particle multiplicity distributions
-/// \author Abhi Modak (abhi.modak@cern.ch)
+/// \author Abhi Modak (abhi.modak@cern.ch), Lucas José (lucas.jose.franco.da.silva@cern.ch)
 /// \since September 10, 2025
 
 #include "PWGLF/DataModel/LFStrangenessTables.h"
@@ -42,6 +42,7 @@
 
 #include <cmath>
 #include <cstdlib>
+#include <unordered_map>
 #include <vector>
 
 using namespace o2;
@@ -96,6 +97,7 @@ struct StudyPnch {
   ConfigurableAxis ft0aMultHistBin{"ft0aMultHistBin", {501, -0.5, 500.5}, ""};
   ConfigurableAxis ft0cMultHistBin{"ft0cMultHistBin", {501, -0.5, 500.5}, ""};
   ConfigurableAxis ptHistBin{"ptHistBin", {200, 0., 20.}, ""};
+  ConfigurableAxis countNumberTracks{"countNumberTracks", {10, -0.5, 9.5}, ""};
 
   Configurable<bool> isApplyTFcut{"isApplyTFcut", true, "Enable TimeFrameBorder cut"};
   Configurable<bool> isApplyITSROcut{"isApplyITSROcut", true, "Enable ITS ReadOutFrameBorder cut"};
@@ -112,6 +114,7 @@ struct StudyPnch {
     AxisSpec axisFt0aMult = {ft0aMultHistBin, "ft0a", "FT0AMultAxis"};
     AxisSpec axisFt0cMult = {ft0cMultHistBin, "ft0c", "FT0CMultAxis"};
     AxisSpec axisPt = {ptHistBin, "pT", "pTAxis"};
+    AxisSpec axisCountNumberTracks = {countNumberTracks, "Count", "CountAxis"};
 
     histos.add("EventHist", "EventHist", kTH1D, {axisEvent}, false);
     histos.add("VtxZHist", "VtxZHist", kTH1D, {axisVtxZ}, false);
@@ -143,6 +146,7 @@ struct StudyPnch {
       histos.add("hMultiplicityMCrec", "hMultiplicityMCrec", kTH1F, {axisMult}, true);
       histos.add("hMultiplicityMCgen", "hMultiplicityMCgen", kTH1F, {axisMult}, true);
       histos.add("hResponseMatrix", "hResponseMatrix", kTH2F, {axisMult, axisMult}, true);
+      histos.add("hCountNTracks", "hCountNTracks", kTH1F, {axisCountNumberTracks}, true);
     }
     if (doprocessEvtLossSigLossMC) {
       histos.add("MCEventHist", "MCEventHist", kTH1F, {axisEvent}, false);
@@ -261,6 +265,7 @@ struct StudyPnch {
   int countNTracksMcCol(countTrk const& tracks, McColType const& McCol)
   {
     auto nTrk = 0;
+    std::unordered_map<int, int> recoFrequencies; // Map that stores globalIndex and the times it appears
     for (const auto& track : tracks) {
       if (!isTrackSelected(track)) {
         continue;
@@ -271,18 +276,23 @@ struct StudyPnch {
         if (particle.mcCollisionId() != McCol.mcCollisionId()) {
           continue;
         }
+        auto globalIndex = particle.globalIndex();
+        recoFrequencies[globalIndex]++; // Increment the count for this globalIndex
       }
       histos.fill(HIST("PhiVsEtaHist"), track.phi(), track.eta());
-      nTrk++;
     }
+    // Once all the frequencies have been counted, a loop can be made to fill the histogram
+    for (const auto& [globalIndex, frequency] : recoFrequencies) {
+      histos.fill(HIST("hCountNTracks"), frequency);
+      // Fill histogram with not cloned tracks
+      if (frequency == 1) {
+        nTrk++;
+      }
+    }
+    // return recoFrequencies;
     return nTrk;
   }
 
-  bool isINELgt0(auto nTrk)
-  {
-    return nTrk > 0;
-  }
-
   Filter fTrackSelectionITS = ncheckbit(aod::track::v001::detectorMap, (uint8_t)o2::aod::track::ITS) &&
                               ncheckbit(aod::track::trackCutFlag, TrackSelectionIts);
   Filter fTrackSelectionTPC = ifnode(ncheckbit(aod::track::v001::detectorMap, (uint8_t)o2::aod::track::TPC),
@@ -296,7 +306,7 @@ struct StudyPnch {
       return;
     }
     auto mult = countNTracks(tracks);
-    if (isINELgt0(mult)) {
+    if (mult > 0) {
       histos.fill(HIST("hMultiplicityData"), mult);
     }
   }