[PWGLF] now checking mcprocess to flag primaries and secondaries (#13878)

Author: GiorgioAlbertoLucia

PWGLF/DataModel/LFSlimNucleiTables.h

@@ -54,6 +54,7 @@ DECLARE_SOA_COLUMN(PDGcode, pdgCode, int);
 DECLARE_SOA_COLUMN(MotherPDGcode, MotherpdgCode, int);
 DECLARE_SOA_COLUMN(MotherDecRad, motherDecRad, float);
 DECLARE_SOA_COLUMN(AbsoDecL, absoDecL, float);
+DECLARE_SOA_COLUMN(McProcess, mcProcess, uint64_t);
 
 } // namespace NucleiTableNS
 
@@ -197,6 +198,7 @@ DECLARE_SOA_TABLE(NucleiTableRed, "AOD", "NUCLEITABLERED",
                   NucleiTableNS::DCAxy,
                   NucleiTableNS::DCAz,
                   NucleiTableNS::Flags,
+                  NucleiTableNS::McProcess,
                   NucleiTableNS::PDGcode,
                   NucleiTableNS::MotherPDGcode);
 

PWGLF/TableProducer/QC/nucleiQC.cxx

@@ -51,6 +51,7 @@
 #include "ReconstructionDataFormats/Track.h"
 
 #include "Math/Vector4D.h"
+#include "TMCProcess.h"
 #include "TRandom3.h"
 
 #include <algorithm>
@@ -222,27 +223,31 @@ struct nucleiQC {
   void fillNucleusFlagsPdgsMc(const Tparticle& particle, nuclei::SlimCandidate& candidate)
   {
     candidate.pdgCode = particle.pdgCode();
+    candidate.mcProcess = particle.getProcess();
 
-    if (particle.isPhysicalPrimary()) {
-      candidate.flags |= nuclei::Flags::kIsPhysicalPrimary;
-
-      // heavy flavour mother
-      // if (particle.has_mothers()) {
-      //  for (const auto& motherparticle : particle.mothers_as<aod::McParticles>()) {
-      //    if (std::find(nuclei::hfMothCodes.begin(), nuclei::hfMothCodes.end(), std::abs(motherparticle.pdgCode())) != nuclei::hfMothCodes.end()) {
-      //      flags |= kIsSecondaryFromWeakDecay;
-      //      motherPdgCode = motherparticle.pdgCode();
-      //      break;
-      //    }
-      //  }
-      //}
-
-    } else if (particle.has_mothers()) {
-      candidate.flags |= nuclei::Flags::kIsSecondaryFromWeakDecay;
+    if (particle.has_mothers()) {
       for (const auto& motherparticle : particle.template mothers_as<aod::McParticles>()) {
         candidate.motherPdgCode = motherparticle.pdgCode();
       }
+    }
 
+    if (particle.isPhysicalPrimary()) {
+      candidate.flags |= nuclei::Flags::kIsPhysicalPrimary;
+
+      ///<  heavy flavour mother
+      /*if (particle.has_mothers()) {
+        for (const auto& motherparticle : particle.mothers_as<aod::McParticles>()) {
+          if (std::find(nuclei::hfMothCodes.begin(), nuclei::hfMothCodes.end(), std::abs(motherparticle.pdgCode())) != nuclei::hfMothCodes.end()) {
+            flags |= kIsSecondaryFromWeakDecay;
+            motherPdgCode = motherparticle.pdgCode();
+            break;
+          }
+        }
+      }*/
+
+    } else if (particle.getProcess() == TMCProcess::kPDecay) {
+      ///<  assuming that strong decays are included in the previous step
+      candidate.flags |= nuclei::Flags::kIsSecondaryFromWeakDecay;
     } else {
       candidate.flags |= nuclei::Flags::kIsSecondaryFromMaterial;
     }
@@ -329,7 +334,8 @@ struct nucleiQC {
                                        .ptGenerated = 0.f, // to be filled for mc
                                        .etaGenerated = 0.f,
                                        .phiGenerated = 0.f,
-                                       .centrality = nuclei::getCentrality(collision, cfgCentralityEstimator)};
+                                       .centrality = nuclei::getCentrality(collision, cfgCentralityEstimator),
+                                       .mcProcess = TMCProcess::kPNoProcess};
 
     fillDcaInformation(collision, track, candidate);
     fillNucleusFlagsPdgs(iSpecies, collision, track, candidate);
@@ -486,6 +492,7 @@ struct nucleiQC {
         candidate.DCAxy,
         candidate.DCAz,
         candidate.flags,
+        candidate.mcProcess,
         candidate.pdgCode,
         candidate.motherPdgCode);
     }

PWGLF/Utils/nucleiUtils.h

@@ -24,6 +24,8 @@
 #include "Framework/HistogramRegistry.h"
 #include "Framework/HistogramSpec.h"
 
+#include "TMCProcess.h"
+
 #include <algorithm>
 #include <string>
 #include <vector>
@@ -103,6 +105,7 @@ struct SlimCandidate {
   float etaGenerated = -999.f;
   float phiGenerated = -999.f;
   float centrality = -1.f;
+  uint64_t mcProcess = TMCProcess::kPNoProcess;
 };
 
 enum Species {