Skip to content

Commit d5bae59

Browse files
creetz16creetz16
authored
[PWGLF] Adding MC daughter momenta to KF 3-body candidate output tree (AliceO2Group#8528)
1 parent 5f54bab commit d5bae59

File tree

3 files changed

+56
-84
lines changed

3 files changed

+56
-84
lines changed

‎PWGLF/DataModel/Vtx3BodyTables.h‎

Lines changed: 19 additions & 69 deletions
Original file line numberDiff line numberDiff line change
@@ -386,9 +386,10 @@ DECLARE_SOA_COLUMN(TPCInnerParamTrack0, tpcinnerparamtrack0, float); //! mom
386386
DECLARE_SOA_COLUMN(TPCInnerParamTrack1, tpcinnerparamtrack1, float); //! momentum at inner wall of TPC of pion daughter
387387
DECLARE_SOA_COLUMN(TPCInnerParamTrack2, tpcinnerparamtrack2, float); //! momentum at inner wall of TPC of deuteron daughter
388388
// PID
389-
DECLARE_SOA_COLUMN(TPCNSigmaProton, tpcnsigmaproton, float); //! nsigma of TPC PID of the proton daughter
390-
DECLARE_SOA_COLUMN(TPCNSigmaPion, tpcnsigmapion, float); //! nsigma of TPC PID of the pion daughter
391-
DECLARE_SOA_COLUMN(TPCNSigmaDeuteron, tpcnsigmadeuteron, float); //! nsigma of TPC PID of the bachelor daughter
389+
DECLARE_SOA_COLUMN(TPCNSigmaProton, tpcnsigmaproton, float); //! nsigma proton of TPC PID of the proton daughter
390+
DECLARE_SOA_COLUMN(TPCNSigmaPion, tpcnsigmapion, float); //! nsigma pion of TPC PID of the pion daughter
391+
DECLARE_SOA_COLUMN(TPCNSigmaDeuteron, tpcnsigmadeuteron, float); //! nsigma deuteron of TPC PID of the bachelor daughter
392+
DECLARE_SOA_COLUMN(TPCNSigmaPionBach, tpcnsigmapionbach, float); //! nsigma pion of TPC PID of the bachelor daughter
392393
DECLARE_SOA_COLUMN(TPCdEdxProton, tpcdedxproton, float); //! TPC dEdx of the proton daughter
393394
DECLARE_SOA_COLUMN(TPCdEdxPion, tpcdedxpion, float); //! TPC dEdx of the pion daughter
394395
DECLARE_SOA_COLUMN(TPCdEdxDeuteron, tpcdedxdeuteron, float); //! TPC dEdx of the bachelor daughter
@@ -406,6 +407,12 @@ DECLARE_SOA_COLUMN(GenCtau, genctau, double); //! generated ctau
406407
DECLARE_SOA_COLUMN(GenPhi, genphi, float); //! generated phi
407408
DECLARE_SOA_COLUMN(GenEta, geneta, float); //! generated eta
408409
DECLARE_SOA_COLUMN(GenRapidity, genrapidity, float); //! generated rapidity
410+
DECLARE_SOA_COLUMN(GenPosP, genposp, float); //! generated momentum pos daughter particle
411+
DECLARE_SOA_COLUMN(GenPosPt, genpospt, float); //! generated transverse momentum pos daughter particle
412+
DECLARE_SOA_COLUMN(GenNegP, gennegp, float); //! generated momentum neg daughter particle
413+
DECLARE_SOA_COLUMN(GenNegPt, gennegpt, float); //! generated transverse momentum neg daughter particle
414+
DECLARE_SOA_COLUMN(GenBachP, genbachp, float); //! generated momentum bachelor daughter particle
415+
DECLARE_SOA_COLUMN(GenBachPt, genbachpt, float); //! generated transverse momentum bachelor daughter particle
409416
DECLARE_SOA_COLUMN(IsTrueH3L, istrueh3l, bool); //! flag for true hypertriton candidate
410417
DECLARE_SOA_COLUMN(IsTrueAntiH3L, istrueantih3l, bool); //! flag for true anti-hypertriton candidate
411418
DECLARE_SOA_COLUMN(PdgCode, pdgcode, int); //! MC particle PDG code
@@ -452,31 +459,11 @@ DECLARE_SOA_TABLE(KFVtx3BodyDatas, "AOD", "KFVTX3BODYDATA",
452459
kfvtx3body::DCAxyTrackPosToPV, kfvtx3body::DCAxyTrackNegToPV, kfvtx3body::DCAxyTrackBachToPV,
453460
kfvtx3body::DCATrackPosToPV, kfvtx3body::DCATrackNegToPV, kfvtx3body::DCATrackBachToPV,
454461
kfvtx3body::Track0Sign, kfvtx3body::Track1Sign, kfvtx3body::Track2Sign, // track sing: proton, pion, deuteron
455-
kfvtx3body::TPCNSigmaProton, kfvtx3body::TPCNSigmaPion, kfvtx3body::TPCNSigmaDeuteron,
462+
kfvtx3body::TPCNSigmaProton, kfvtx3body::TPCNSigmaPion, kfvtx3body::TPCNSigmaDeuteron, kfvtx3body::TPCNSigmaPionBach,
456463
kfvtx3body::TPCdEdxProton, kfvtx3body::TPCdEdxPion, kfvtx3body::TPCdEdxDeuteron,
457464
kfvtx3body::TOFNSigmaDeuteron,
458465
kfvtx3body::ITSClusSizeDeuteron,
459-
kfvtx3body::PIDTrackingDeuteron,
460-
461-
// dynamic columns
462-
vtx3body::VtxRadius<vtx3body::X, vtx3body::Y>,
463-
vtx3body::DistOverTotMom<vtx3body::X, vtx3body::Y, vtx3body::Z, kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
464-
vtx3body::VtxCosPA<vtx3body::X, vtx3body::Y, vtx3body::Z, kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
465-
vtx3body::DCAVtxToPV<vtx3body::X, vtx3body::Y, vtx3body::Z, kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
466-
467-
// Longitudinal
468-
vtx3body::YHypertriton<kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
469-
vtx3body::Eta<kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
470-
vtx3body::Phi<kfvtx3body::Px, kfvtx3body::Py>,
471-
vtx3body::Track0Pt<vtx3body::PxTrack0, vtx3body::PyTrack0>, // proton pT
472-
vtx3body::Track0Eta<vtx3body::PxTrack0, vtx3body::PyTrack0, vtx3body::PzTrack0>, // proton eta
473-
vtx3body::Track0Phi<vtx3body::PxTrack0, vtx3body::PyTrack0>, // proton phi
474-
vtx3body::Track1Pt<vtx3body::PxTrack1, vtx3body::PyTrack1>, // pion pT
475-
vtx3body::Track1Eta<vtx3body::PxTrack1, vtx3body::PyTrack1, vtx3body::PzTrack1>, // pion eta
476-
vtx3body::Track1Phi<vtx3body::PxTrack1, vtx3body::PyTrack1>, // pion phi
477-
vtx3body::Track2Pt<vtx3body::PxTrack2, vtx3body::PyTrack2>, // deuteron pT
478-
vtx3body::Track2Eta<vtx3body::PxTrack2, vtx3body::PyTrack2, vtx3body::PzTrack2>, // deuteron eta
479-
vtx3body::Track2Phi<vtx3body::PxTrack2, vtx3body::PyTrack2>); // deuteron phi
466+
kfvtx3body::PIDTrackingDeuteron);
480467

481468
using KFVtx3BodyData = KFVtx3BodyDatas::iterator;
482469
namespace kfvtx3body
@@ -518,31 +505,11 @@ DECLARE_SOA_TABLE(KFVtx3BodyDatasLite, "AOD", "KF3BODYLITE",
518505
kfvtx3body::DCAxyTrack0ToTrack1KF, kfvtx3body::DCAxyTrack0ToTrack2KF, kfvtx3body::DCAxyTrack1ToTrack2KF,
519506
kfvtx3body::DCAVtxDaughtersKF,
520507
kfvtx3body::Track0Sign, kfvtx3body::Track1Sign, kfvtx3body::Track2Sign, // track sing: proton, pion, deuteron
521-
kfvtx3body::TPCNSigmaProton, kfvtx3body::TPCNSigmaPion, kfvtx3body::TPCNSigmaDeuteron,
508+
kfvtx3body::TPCNSigmaProton, kfvtx3body::TPCNSigmaPion, kfvtx3body::TPCNSigmaDeuteron, kfvtx3body::TPCNSigmaPionBach,
522509
kfvtx3body::TPCdEdxProton, kfvtx3body::TPCdEdxPion, kfvtx3body::TPCdEdxDeuteron,
523510
kfvtx3body::TOFNSigmaDeuteron,
524511
kfvtx3body::ITSClusSizeDeuteron,
525-
kfvtx3body::PIDTrackingDeuteron,
526-
527-
// dynamic columns
528-
vtx3body::VtxRadius<vtx3body::X, vtx3body::Y>,
529-
vtx3body::DistOverTotMom<vtx3body::X, vtx3body::Y, vtx3body::Z, kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
530-
vtx3body::VtxCosPA<vtx3body::X, vtx3body::Y, vtx3body::Z, kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
531-
vtx3body::DCAVtxToPV<vtx3body::X, vtx3body::Y, vtx3body::Z, kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
532-
533-
// Longitudinal
534-
vtx3body::YHypertriton<kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
535-
vtx3body::Eta<kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
536-
vtx3body::Phi<kfvtx3body::Px, kfvtx3body::Py>,
537-
vtx3body::Track0Pt<vtx3body::PxTrack0, vtx3body::PyTrack0>, // proton pT
538-
vtx3body::Track0Eta<vtx3body::PxTrack0, vtx3body::PyTrack0, vtx3body::PzTrack0>, // proton eta
539-
vtx3body::Track0Phi<vtx3body::PxTrack0, vtx3body::PyTrack0>, // proton phi
540-
vtx3body::Track1Pt<vtx3body::PxTrack1, vtx3body::PyTrack1>, // pion pT
541-
vtx3body::Track1Eta<vtx3body::PxTrack1, vtx3body::PyTrack1, vtx3body::PzTrack1>, // pion eta
542-
vtx3body::Track1Phi<vtx3body::PxTrack1, vtx3body::PyTrack1>, // pion phi
543-
vtx3body::Track2Pt<vtx3body::PxTrack2, vtx3body::PyTrack2>, // deuteron pT
544-
vtx3body::Track2Eta<vtx3body::PxTrack2, vtx3body::PyTrack2, vtx3body::PzTrack2>, // deuteron eta
545-
vtx3body::Track2Phi<vtx3body::PxTrack2, vtx3body::PyTrack2>); // deuteron phi
512+
kfvtx3body::PIDTrackingDeuteron);
546513

547514
using KFVtx3BodyDataLite = KFVtx3BodyDatasLite::iterator;
548515

@@ -579,7 +546,7 @@ DECLARE_SOA_TABLE(McKFVtx3BodyDatas, "AOD", "MCKF3BODYDATAS",
579546
kfvtx3body::DCAxyTrackPosToPV, kfvtx3body::DCAxyTrackNegToPV, kfvtx3body::DCAxyTrackBachToPV,
580547
kfvtx3body::DCATrackPosToPV, kfvtx3body::DCATrackNegToPV, kfvtx3body::DCATrackBachToPV,
581548
kfvtx3body::Track0Sign, kfvtx3body::Track1Sign, kfvtx3body::Track2Sign, // track sing: proton, pion, deuteron
582-
kfvtx3body::TPCNSigmaProton, kfvtx3body::TPCNSigmaPion, kfvtx3body::TPCNSigmaDeuteron,
549+
kfvtx3body::TPCNSigmaProton, kfvtx3body::TPCNSigmaPion, kfvtx3body::TPCNSigmaDeuteron, kfvtx3body::TPCNSigmaPionBach,
583550
kfvtx3body::TPCdEdxProton, kfvtx3body::TPCdEdxPion, kfvtx3body::TPCdEdxDeuteron,
584551
kfvtx3body::TOFNSigmaDeuteron,
585552
kfvtx3body::ITSClusSizeDeuteron,
@@ -593,30 +560,13 @@ DECLARE_SOA_TABLE(McKFVtx3BodyDatas, "AOD", "MCKF3BODYDATAS",
593560
kfvtx3body::GenPhi,
594561
kfvtx3body::GenEta,
595562
kfvtx3body::GenRapidity,
563+
kfvtx3body::GenPosP, kfvtx3body::GenPosPt,
564+
kfvtx3body::GenNegP, kfvtx3body::GenNegPt,
565+
kfvtx3body::GenBachP, kfvtx3body::GenBachPt,
596566
kfvtx3body::IsTrueH3L, kfvtx3body::IsTrueAntiH3L,
597567
kfvtx3body::PdgCode,
598568
kfvtx3body::IsReco,
599-
kfvtx3body::SurvEvSel,
600-
601-
// dynamic columns
602-
vtx3body::VtxRadius<vtx3body::X, vtx3body::Y>,
603-
vtx3body::DistOverTotMom<vtx3body::X, vtx3body::Y, vtx3body::Z, kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
604-
vtx3body::VtxCosPA<vtx3body::X, vtx3body::Y, vtx3body::Z, kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
605-
vtx3body::DCAVtxToPV<vtx3body::X, vtx3body::Y, vtx3body::Z, kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
606-
607-
// Longitudinal
608-
vtx3body::YHypertriton<kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
609-
vtx3body::Eta<kfvtx3body::Px, kfvtx3body::Py, kfvtx3body::Pz>,
610-
vtx3body::Phi<kfvtx3body::Px, kfvtx3body::Py>,
611-
vtx3body::Track0Pt<vtx3body::PxTrack0, vtx3body::PyTrack0>, // proton pT
612-
vtx3body::Track0Eta<vtx3body::PxTrack0, vtx3body::PyTrack0, vtx3body::PzTrack0>, // proton eta
613-
vtx3body::Track0Phi<vtx3body::PxTrack0, vtx3body::PyTrack0>, // proton phi
614-
vtx3body::Track1Pt<vtx3body::PxTrack1, vtx3body::PyTrack1>, // pion pT
615-
vtx3body::Track1Eta<vtx3body::PxTrack1, vtx3body::PyTrack1, vtx3body::PzTrack1>, // pion eta
616-
vtx3body::Track1Phi<vtx3body::PxTrack1, vtx3body::PyTrack1>, // pion phi
617-
vtx3body::Track2Pt<vtx3body::PxTrack2, vtx3body::PyTrack2>, // deuteron pT
618-
vtx3body::Track2Eta<vtx3body::PxTrack2, vtx3body::PyTrack2, vtx3body::PzTrack2>, // deuteron eta
619-
vtx3body::Track2Phi<vtx3body::PxTrack2, vtx3body::PyTrack2>); // deuteron phi
569+
kfvtx3body::SurvEvSel);
620570

621571
// Definition of labels for KFVtx3BodyDatas
622572
namespace mckfvtx3bodylabel

‎PWGLF/TableProducer/Nuspex/decay3bodybuilder.cxx‎

Lines changed: 5 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -739,6 +739,7 @@ struct decay3bodyBuilder {
739739
float dEdxProton;
740740
float dEdxPion;
741741
float tpcNsigmaDeuteron = trackBach.tpcNSigmaDe();
742+
float tpcNsigmaPionBach = trackBach.tpcNSigmaPi();
742743
float dEdxDeuteron = trackBach.tpcSignal();
743744
if (isMatter) { // hypertriton (proton, pi-, deuteron)
744745
tpcNsigmaProton = trackPos.tpcNSigmaPr();
@@ -1047,12 +1048,8 @@ struct decay3bodyBuilder {
10471048
kfpPion.GetQ(),
10481049
trackBach.sign(),
10491050
// daughter PID
1050-
tpcNsigmaProton,
1051-
tpcNsigmaPion,
1052-
tpcNsigmaDeuteron,
1053-
dEdxProton,
1054-
dEdxPion,
1055-
dEdxDeuteron,
1051+
tpcNsigmaProton, tpcNsigmaPion, tpcNsigmaDeuteron, tpcNsigmaPionBach,
1052+
dEdxProton, dEdxPion, dEdxDeuteron,
10561053
tofNSigmaDeuteron,
10571054
averageClusterSizeDeuteron,
10581055
trackBach.pidForTracking());
@@ -1099,12 +1096,8 @@ struct decay3bodyBuilder {
10991096
kfpPion.GetQ(),
11001097
trackBach.sign(),
11011098
// daughter PID
1102-
tpcNsigmaProton,
1103-
tpcNsigmaPion,
1104-
tpcNsigmaDeuteron,
1105-
dEdxProton,
1106-
dEdxPion,
1107-
dEdxDeuteron,
1099+
tpcNsigmaProton, tpcNsigmaPion, tpcNsigmaDeuteron, tpcNsigmaPionBach,
1100+
dEdxProton, dEdxPion, dEdxDeuteron,
11081101
tofNSigmaDeuteron,
11091102
averageClusterSizeDeuteron,
11101103
trackBach.pidForTracking());
@@ -1380,7 +1373,6 @@ struct decay3bodyLabelBuilder {
13801373

13811374
struct decay3bodyInitializer {
13821375
Spawns<aod::Vtx3BodyDatas> vtx3bodydatas;
1383-
// Spawns<aod::KFVtx3BodyDatas> kfvtx3bodydatas;
13841376
void init(InitContext const&) {}
13851377
};
13861378

‎PWGLF/TableProducer/Nuspex/threebodyKFTask.cxx‎

Lines changed: 32 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,6 +186,12 @@ struct threebodyKFTask {
186186
std::array<float, 3> genDecVtx{-1.f};
187187
int vtx3bodyPDGcode = -1;
188188
double MCmassPrPi = -1.;
189+
float genPosP = -1.;
190+
float genPosPt = -1.;
191+
float genNegP = -1.;
192+
float genNegPt = -1.;
193+
float genBachP = -1.;
194+
float genBachPt = -1.;
189195

190196
auto track0 = vtx3bodydata.track0_as<MCLabeledTracksIU>();
191197
auto track1 = vtx3bodydata.track1_as<MCLabeledTracksIU>();
@@ -209,6 +215,12 @@ struct threebodyKFTask {
209215
genRap = MCvtx3body.y();
210216
genP = MCvtx3body.p();
211217
genPt = MCvtx3body.pt();
218+
genPosP = lMCTrack0.p();
219+
genPosPt = lMCTrack0.pt();
220+
genNegP = lMCTrack1.p();
221+
genNegPt = lMCTrack1.pt();
222+
genBachP = lMCTrack2.p();
223+
genBachPt = lMCTrack2.pt();
212224
filledMothers.push_back(MCvtx3body.globalIndex());
213225
} // end is H3L or Anti-H3L
214226
if (MCvtx3body.pdgCode() == motherPdgCode && lMCTrack0.pdgCode() == 2212 && lMCTrack1.pdgCode() == -211 && lMCTrack2.pdgCode() == bachelorPdgCode) {
@@ -261,7 +273,7 @@ struct threebodyKFTask {
261273
vtx3bodydata.dcaxytrackpostopv(), vtx3bodydata.dcaxytracknegtopv(), vtx3bodydata.dcaxytrackbachtopv(),
262274
vtx3bodydata.dcatrackpostopv(), vtx3bodydata.dcatracknegtopv(), vtx3bodydata.dcatrackbachtopv(),
263275
vtx3bodydata.track0sign(), vtx3bodydata.track1sign(), vtx3bodydata.track2sign(), // proton, pion, deuteron
264-
vtx3bodydata.tpcnsigmaproton(), vtx3bodydata.tpcnsigmapion(), vtx3bodydata.tpcnsigmadeuteron(),
276+
vtx3bodydata.tpcnsigmaproton(), vtx3bodydata.tpcnsigmapion(), vtx3bodydata.tpcnsigmadeuteron(), vtx3bodydata.tpcnsigmapionbach(),
265277
vtx3bodydata.tpcdedxproton(), vtx3bodydata.tpcdedxpion(), vtx3bodydata.tpcdedxdeuteron(),
266278
vtx3bodydata.tofnsigmadeuteron(),
267279
vtx3bodydata.itsclussizedeuteron(),
@@ -274,6 +286,7 @@ struct threebodyKFTask {
274286
genPhi,
275287
genEta,
276288
genRap,
289+
genPosP, genPosPt, genNegP, genNegPt, genBachP, genBachPt,
277290
isTrueH3L, isTrueAntiH3L,
278291
vtx3bodyPDGcode,
279292
true, // is reconstructed
@@ -288,6 +301,12 @@ struct threebodyKFTask {
288301
double genMCmassPrPi = -1.;
289302
bool isTrueGenH3L = false;
290303
bool isTrueGenAntiH3L = false;
304+
float genPBach = -1.;
305+
float genPtBach = -1.;
306+
float genPPos = -1.;
307+
float genPtPos = -1.;
308+
float genPNeg = -1.;
309+
float genPtNeg = -1.;
291310

292311
// check if mcparticle was reconstructed and already filled in the table
293312
if (std::find(filledMothers.begin(), filledMothers.end(), mcparticle.globalIndex()) != std::end(filledMothers)) {
@@ -327,8 +346,12 @@ struct threebodyKFTask {
327346
for (auto& mcparticleDaughter : mcparticle.template daughters_as<aod::McParticles>()) {
328347
if (mcparticleDaughter.pdgCode() == 2212) {
329348
protonMom = {mcparticleDaughter.px(), mcparticleDaughter.py(), mcparticleDaughter.pz()};
349+
genPPos = mcparticleDaughter.p();
350+
genPtPos = mcparticleDaughter.pt();
330351
} else if (mcparticleDaughter.pdgCode() == -211) {
331352
piMinusMom = {mcparticleDaughter.px(), mcparticleDaughter.py(), mcparticleDaughter.pz()};
353+
genPNeg = mcparticleDaughter.p();
354+
genPtNeg = mcparticleDaughter.pt();
332355
}
333356
}
334357
genMCmassPrPi = RecoDecay::m(array{protonMom, piMinusMom}, array{o2::constants::physics::MassProton, o2::constants::physics::MassPionCharged});
@@ -343,8 +366,12 @@ struct threebodyKFTask {
343366
for (auto& mcparticleDaughter : mcparticle.template daughters_as<aod::McParticles>()) {
344367
if (mcparticleDaughter.pdgCode() == -2212) {
345368
antiProtonMom = {mcparticleDaughter.px(), mcparticleDaughter.py(), mcparticleDaughter.pz()};
369+
genPNeg = mcparticleDaughter.p();
370+
genPtNeg = mcparticleDaughter.pt();
346371
} else if (mcparticleDaughter.pdgCode() == 211) {
347372
piPlusMom = {mcparticleDaughter.px(), mcparticleDaughter.py(), mcparticleDaughter.pz()};
373+
genPPos = mcparticleDaughter.p();
374+
genPtPos = mcparticleDaughter.pt();
348375
}
349376
}
350377
genMCmassPrPi = RecoDecay::m(array{antiProtonMom, piPlusMom}, array{o2::constants::physics::MassProton, o2::constants::physics::MassPionCharged});
@@ -360,6 +387,8 @@ struct threebodyKFTask {
360387
for (auto& mcDaughter : mcparticle.daughters_as<aod::McParticles>()) {
361388
if (std::abs(mcDaughter.pdgCode()) == bachelorPdgCode) {
362389
genDecayVtx = {mcDaughter.vx(), mcDaughter.vy(), mcDaughter.vz()};
390+
genPBach = mcDaughter.p();
391+
genPtBach = mcDaughter.pt();
363392
}
364393
}
365394
double genMClifetime = RecoDecay::sqrtSumOfSquares(genDecayVtx[0] - mcparticle.vx(), genDecayVtx[1] - mcparticle.vy(), genDecayVtx[2] - mcparticle.vz()) * o2::constants::physics::MassHyperTriton / mcparticle.p();
@@ -390,7 +419,7 @@ struct threebodyKFTask {
390419
-1, -1, -1,
391420
-1, -1, -1,
392421
-1, -1, -1,
393-
-1, -1, -1,
422+
-1, -1, -1, -1,
394423
-1, -1, -1,
395424
-1,
396425
-1,
@@ -403,6 +432,7 @@ struct threebodyKFTask {
403432
mcparticle.phi(),
404433
mcparticle.eta(),
405434
mcparticle.y(),
435+
genPPos, genPtPos, genPNeg, genPtNeg, genPBach, genPtBach,
406436
isTrueGenH3L, isTrueGenAntiH3L,
407437
mcparticle.pdgCode(),
408438
false, // is reconstructed

0 commit comments

Comments
 (0)