Feat: add rapidity histogram for mother

Author: ariedel-cern

PWGCF/Femto/Core/mcParticleHistManager.h

@@ -16,6 +16,7 @@
 #ifndef PWGCF_FEMTO_CORE_MCPARTICLEHISTMANAGER_H_
 #define PWGCF_FEMTO_CORE_MCPARTICLEHISTMANAGER_H_
 
+#include "PWGCF/Femto/Core/femtoUtils.h"
 #include "PWGCF/Femto/Core/histManager.h"
 #include "PWGCF/Femto/Core/modes.h"
 
@@ -52,6 +53,7 @@ enum McParticleHist {
   kMotherEta,
   kMotherPhi,
   kMotherSign,
+  kMotherY,
   // origin (kept in sync with TrackHist, even though some categories
   // are not expected to be populated for pure MC truth particles)
   kOrigin,
@@ -105,6 +107,7 @@ constexpr std::array<histmanager::HistInfo<McParticleHist>, kMcParticleHistLast>
       {kMotherEta, o2::framework::HistType::kTH1F, "hMotherEta", "Mother pseudorapidity; #eta; Entries"},
       {kMotherPhi, o2::framework::HistType::kTH1F, "hMotherPhi", "Mother azimuthal angle; #varphi; Entries"},
       {kMotherSign, o2::framework::HistType::kTH1F, "hMotherSign", "Sign of mother charge; Sign; Entries"},
+      {kMotherY, o2::framework::HistType::kTH1F, "hMotherY", "Mother rapidity; Y; Entries"},
       {kOrigin, o2::framework::HistType::kTH1F, "hOrigin", "Status Codes (=Origin); Status Code; Entries"},
       {kNoMcParticle, o2::framework::HistType::kTH1F, "hNoMcParticle", "Particles with no associated MC information; p_{T} (GeV/#it{c}); Entries"},
       {kPrimary, o2::framework::HistType::kTH1F, "hPrimary", "Primary particles; p_{T} (GeV/#it{c}); Entries"},
@@ -132,6 +135,7 @@ auto makeMcParticleHistSpecMap(const T& confBinning)
     {kMotherEta, {confBinning.eta}},
     {kMotherPhi, {confBinning.phi}},
     {kMotherSign, {confBinning.sign}},
+    {kMotherY, {confBinning.eta}},
     {kNoMcParticle, {confBinning.pt}},
     {kPrimary, {confBinning.pt}},
     {kFromWrongCollision, {confBinning.pt}},
@@ -182,6 +186,7 @@ class McParticleHistManager
     mHistogramRegistry->add(mcDir + getHistNameV2(kMotherEta, HistTable), getHistDesc(kMotherEta, HistTable), getHistType(kMotherEta, HistTable), {Specs.at(kMotherEta)});
     mHistogramRegistry->add(mcDir + getHistNameV2(kMotherPhi, HistTable), getHistDesc(kMotherPhi, HistTable), getHistType(kMotherPhi, HistTable), {Specs.at(kMotherPhi)});
     mHistogramRegistry->add(mcDir + getHistNameV2(kMotherSign, HistTable), getHistDesc(kMotherSign, HistTable), getHistType(kMotherSign, HistTable), {Specs.at(kMotherSign)});
+    mHistogramRegistry->add(mcDir + getHistNameV2(kMotherY, HistTable), getHistDesc(kMotherY, HistTable), getHistType(kMotherY, HistTable), {Specs.at(kMotherY)});
 
     this->enableOptionalHistograms(confBinning);
     this->initOrigin(Specs);
@@ -204,6 +209,7 @@ class McParticleHistManager
       mHistogramRegistry->fill(HIST(prefix) + HIST(McDir) + HIST(getHistName(kMotherEta, HistTable)), mother.eta());
       mHistogramRegistry->fill(HIST(prefix) + HIST(McDir) + HIST(getHistName(kMotherPhi, HistTable)), mother.phi());
       mHistogramRegistry->fill(HIST(prefix) + HIST(McDir) + HIST(getHistName(kMotherSign, HistTable)), mother.sign());
+      mHistogramRegistry->fill(HIST(prefix) + HIST(McDir) + HIST(getHistName(kMotherY, HistTable)), mother.y(utils::getPdgMass(mother.pdgCode())));
     } else {
       mHistogramRegistry->fill(HIST(prefix) + HIST(McDir) + HIST(getHistName(kPdgMother, HistTable)), 0);
     }

PWGCF/Femto/Core/particleCleaner.h

@@ -18,8 +18,6 @@
 
 #include "PWGCF/Femto/Core/femtoUtils.h"
 
-#include "Common/Core/RecoDecay.h"
-
 #include <CommonConstants/MathConstants.h>
 #include <Framework/Configurable.h>
 
@@ -251,10 +249,11 @@ class ParticleCleaner
         motherKinematicsOk = false;
       }
 
-      // rapidity needs the mother's mass, which FMcMothers does not store directly -- look it up
-      // from its pdg code (same mechanism as the kstar/Q3 mass lookup in pairHistManager.h)
+      // rapidity needs the mother's mass, which FMcMothers does not store directly -- its Y()
+      // dynamic column takes mass as a free argument, looked up from the pdg code (same mechanism
+      // as the kstar/Q3 mass lookup in pairHistManager.h)
       if (mMotherRapidityMin > -1e6f || mMotherRapidityMax < 1e6f) {
-        double motherRapidity = RecoDecay::y(mother.pt(), mother.eta(), utils::getPdgMass(mother.pdgCode()));
+        double motherRapidity = mother.y(utils::getPdgMass(mother.pdgCode()));
         if (!(motherRapidity > mMotherRapidityMin && motherRapidity < mMotherRapidityMax)) {
           motherKinematicsOk = false;
         }

PWGCF/Femto/DataModel/FemtoTables.h

@@ -246,6 +246,11 @@ DECLARE_SOA_DYNAMIC_COLUMN(Theta, theta, //! theta
                            [](float eta) -> float {
                              return 2.f * std::atan(std::exp(-eta));
                            });
+DECLARE_SOA_DYNAMIC_COLUMN(Y, y, //! rapidity -- mass is not a stored column, so it is passed in as an
+                                 //! extra free argument at the call site, e.g. mother.y(mass)
+                           [](float signedPt, float eta, float mass) -> float {
+                             return RecoDecay::y(std::fabs(signedPt), eta, mass);
+                           });
 } // namespace dynamic
 
 namespace lite
@@ -1761,7 +1766,8 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FMcMothers_001, "FMCMOTHER", 1, //! first dir
                                    femtobase::dynamic::Px<femtobase::stored::SignedPt, femtobase::stored::Phi>,
                                    femtobase::dynamic::Py<femtobase::stored::SignedPt, femtobase::stored::Phi>,
                                    femtobase::dynamic::Pz<femtobase::stored::SignedPt, femtobase::stored::Eta>,
-                                   femtobase::dynamic::Theta<femtobase::stored::Eta>);
+                                   femtobase::dynamic::Theta<femtobase::stored::Eta>,
+                                   femtobase::dynamic::Y<femtobase::stored::SignedPt, femtobase::stored::Eta>); // mass is a free argument, e.g. mother.y(mass)
 
 using FMcMothers = FMcMothers_001;
 using FMcMother = FMcMothers::iterator;