diff --git a/PWGHF/DataModel/DerivedTables.h b/PWGHF/DataModel/DerivedTables.h index 3e9b374898a..63a23e2f052 100644 --- a/PWGHF/DataModel/DerivedTables.h +++ b/PWGHF/DataModel/DerivedTables.h @@ -859,6 +859,18 @@ DECLARE_SOA_TABLE_STAGED(HfDplusParEs, "HFDPPARE", //! Table with additional can hf_cand_par::Ct, o2::soa::Marker); +DECLARE_SOA_TABLE_STAGED(HfDplusDaugs, "HFDPDAUG", //! Table to study daughter properties + hf_cand_base::Pt, + hf_cand::Chi2PCA, + hf_cand_par::DecayLength, + hf_cand_par::PProng0, + hf_cand_par::PProng1, + hf_cand_par::PProng2, + hf_cand_par::NSigTpcTofPi0, + hf_cand_par::NSigTpcTofKa1, + hf_cand_par::NSigTpcTofPi2, + o2::soa::Marker); + DECLARE_SOA_TABLE_STAGED(HfDplusMls, "HFDPML", //! Table with candidate selection ML scores hf_cand_mc::MlScores, o2::soa::Marker); diff --git a/PWGHF/TableProducer/derivedDataCreatorDplusToPiKPi.cxx b/PWGHF/TableProducer/derivedDataCreatorDplusToPiKPi.cxx index ce15964cade..c113ad02d83 100644 --- a/PWGHF/TableProducer/derivedDataCreatorDplusToPiKPi.cxx +++ b/PWGHF/TableProducer/derivedDataCreatorDplusToPiKPi.cxx @@ -70,6 +70,7 @@ struct HfDerivedDataCreatorDplusToPiKPi { Produces rowCandidatePar; Produces rowCandidateParE; Produces rowCandidateSel; + Produces rowCandidateDaugs; Produces rowCandidateMl; Produces rowCandidateId; Produces rowCandidateMc; @@ -79,6 +80,7 @@ struct HfDerivedDataCreatorDplusToPiKPi { Configurable fillCandidatePar{"fillCandidatePar", true, "Fill candidate parameters"}; Configurable fillCandidateParE{"fillCandidateParE", true, "Fill candidate extended parameters"}; Configurable fillCandidateSel{"fillCandidateSel", true, "Fill candidate selection flags"}; + Configurable fillCandidateDaugs{"fillCandidateDaugs", false, "Fill candidate daughter parameters"}; Configurable fillCandidateMl{"fillCandidateMl", true, "Fill candidate selection ML scores"}; Configurable fillCandidateId{"fillCandidateId", true, "Fill original indices from the candidate table"}; Configurable fillCandidateMc{"fillCandidateMc", true, "Fill candidate MC info"}; @@ -191,6 +193,18 @@ struct HfDerivedDataCreatorDplusToPiKPi { rowCandidateSel( BIT(candFlag)); } + if (fillCandidateDaugs) { + rowCandidateDaugs( + candidate.pt(), + candidate.chi2PCA(), + candidate.decayLength(), + RecoDecay::p(candidate.pxProng0(), candidate.pyProng0(), candidate.pzProng0()), + RecoDecay::p(candidate.pxProng1(), candidate.pyProng1(), candidate.pzProng1()), + RecoDecay::p(candidate.pxProng2(), candidate.pyProng2(), candidate.pzProng2()), + candidate.tpcTofNSigmaPi0(), + candidate.tpcTofNSigmaKa1(), + candidate.tpcTofNSigmaPi2()); + } if (fillCandidateMl) { rowCandidateMl( mlScores);