Add generated Xic+ mass to the Gen tree

Author: JaeYoonCHO

PWGHF/DataModel/CandidateReconstructionTables.h

@@ -1500,6 +1500,7 @@ DECLARE_SOA_COLUMN(NSigTofPiFromLambda, nSigTofPiFromLambda, float);
 DECLARE_SOA_COLUMN(NSigTofPrFromLambda, nSigTofPrFromLambda, float);
 // MC
 DECLARE_SOA_COLUMN(DecayLengthMcGen, decayLengthMcGen, float);
+DECLARE_SOA_COLUMN(MassMcGen, massMcGen, float);
 // Residuals and pulls
 DECLARE_SOA_COLUMN(PtResidual, ptResidual, float);
 DECLARE_SOA_COLUMN(PResidual, pResidual, float);
@@ -1605,7 +1606,8 @@ DECLARE_SOA_TABLE(HfCandXicMcGen, "AOD", "HFCANDXICMCGEN",
                   hf_cand_mc_flag::FlagMcMatchGen,
                   hf_cand_mc_flag::OriginMcGen,
                   hf_cand_mc_flag::PdgBhadMotherPart,
-                  hf_cand_xic_to_xi_pi_pi::DecayLengthMcGen);
+                  hf_cand_xic_to_xi_pi_pi::DecayLengthMcGen,
+                  hf_cand_xic_to_xi_pi_pi::MassMcGen);
 
 // table with residuals and pulls of PV
 DECLARE_SOA_TABLE(HfCandXicResid, "AOD", "HFCANDXICRESID",

PWGHF/TableProducer/candidateCreatorXicToXiPiPi.cxx

@@ -1301,12 +1301,13 @@ struct HfCandidateCreatorXicToXiPiPiExpressions {
       if (rejectionMask != 0) {
         // at least one event selection not satisfied --> reject all particles from this collision
         for (unsigned int i = 0; i < mcParticlesPerMcColl.size(); ++i) {
-          rowMcMatchGen(-99, -99, -99, decayLengthGen);
+          rowMcMatchGen(-99, -99, -99, decayLengthGen, -1.f);
         }
         continue;
       }
 
       for (const auto& particle : mcParticlesPerMcColl) {
+        const float massGen = RecoDecay::m(particle.pVector(), particle.e());
         sign = 0;
         flag = 0;
         origin = RecoDecay::OriginType::None;
@@ -1359,9 +1360,9 @@ struct HfCandidateCreatorXicToXiPiPiExpressions {
         // Fill table
         if (origin == RecoDecay::OriginType::NonPrompt) {
           auto bHadMother = mcParticles.rawIteratorAt(idxBhadMothers[0]);
-          rowMcMatchGen(flag, origin, bHadMother.pdgCode(), decayLengthGen);
+          rowMcMatchGen(flag, origin, bHadMother.pdgCode(), decayLengthGen, massGen);
         } else {
-          rowMcMatchGen(flag, origin, 0, decayLengthGen);
+          rowMcMatchGen(flag, origin, 0, decayLengthGen, massGen);
         }
       } // close loop over generated particles
     } // close loop over McCollisions

PWGHF/TableProducer/treeCreatorXicToXiPiPi.cxx

@@ -364,6 +364,7 @@ DECLARE_SOA_TABLE(HfCandXicToXiPiPiFullPs, "AOD", "HFXICXI2PIFULLP",
                   hf_cand_mc_flag::OriginMcGen,
                   hf_cand_mc_flag::PdgBhadMotherPart,
                   full::Pt,
+                  full::M,
                   full::Eta,
                   full::Phi,
                   full::Y,
@@ -812,6 +813,7 @@ struct HfTreeCreatorXicToXiPiPi {
           particle.originMcGen(),
           particle.pdgBhadMotherPart(),
           particle.pt(),
+          particle.massMcGen(),
           particle.eta(),
           particle.phi(),
           RecoDecay::y(particle.pVector(), o2::constants::physics::MassXiCPlus),
@@ -867,6 +869,7 @@ struct HfTreeCreatorXicToXiPiPi {
           particle.originMcGen(),
           particle.pdgBhadMotherPart(),
           particle.pt(),
+          particle.massMcGen(),
           particle.eta(),
           particle.phi(),
           RecoDecay::y(particle.pVector(), o2::constants::physics::MassXiCPlus),
@@ -913,6 +916,7 @@ struct HfTreeCreatorXicToXiPiPi {
           particle.originMcGen(),
           particle.pdgBhadMotherPart(),
           particle.pt(),
+          particle.massMcGen(),
           particle.eta(),
           particle.phi(),
           RecoDecay::y(particle.pVector(), o2::constants::physics::MassXiCPlus),