Skip to content

Commit 9beb96e

Browse files
fchinufchinu
authored
[PWGHF] Add B mother information to tree creators (AliceO2Group#15863)
1 parent 7e10286 commit 9beb96e

3 files changed

Lines changed: 32 additions & 13 deletions

File tree

‎PWGHF/TableProducer/treeCreatorD0ToKPi.cxx‎

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
#include "PWGHF/Core/HfHelper.h"
2222
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
2323
#include "PWGHF/DataModel/CandidateSelectionTables.h"
24+
#include "PWGHF/Utils/utilsAnalysis.h"
2425

2526
#include "Common/Core/RecoDecay.h"
2627

@@ -139,7 +140,8 @@ DECLARE_SOA_TABLE(HfCandD0Lites, "AOD", "HFCANDD0LITE",
139140
full::Y,
140141
full::FlagMc,
141142
full::FlagMcDecayChanRec,
142-
full::OriginMcRec)
143+
full::OriginMcRec,
144+
hf_cand_mc_flag::PdgBhadMotherPart)
143145

144146
DECLARE_SOA_TABLE(HfCandD0Fulls, "AOD", "HFCANDD0FULL",
145147
collision::PosX,
@@ -201,7 +203,8 @@ DECLARE_SOA_TABLE(HfCandD0Fulls, "AOD", "HFCANDD0FULL",
201203
full::E,
202204
full::FlagMc,
203205
full::FlagMcDecayChanRec,
204-
full::OriginMcRec);
206+
full::OriginMcRec,
207+
hf_cand_mc_flag::PdgBhadMotherPart);
205208

206209
DECLARE_SOA_TABLE(HfCandD0FullEvs, "AOD", "HFCANDD0FULLEV",
207210
collision::NumContrib,
@@ -286,7 +289,7 @@ struct HfTreeCreatorD0ToKPi {
286289

287290
template <bool ApplyMl, typename T>
288291
auto fillTable(const T& candidate, int candFlag, double invMass, double topoChi2,
289-
double ct, double y, double e, int8_t flagMc, int8_t flagMcDecay, int8_t origin)
292+
double ct, double y, double e, int8_t flagMc, int8_t flagMcDecay, int8_t origin, int8_t flagBmother)
290293
{
291294
if (fillCandidateLiteTable) {
292295
rowCandidateLite(
@@ -325,7 +328,8 @@ struct HfTreeCreatorD0ToKPi {
325328
y,
326329
flagMc,
327330
flagMcDecay,
328-
origin);
331+
origin,
332+
flagBmother);
329333
} else {
330334
double cosThetaStar = candFlag == 0 ? HfHelper::cosThetaStarD0(candidate) : HfHelper::cosThetaStarD0bar(candidate);
331335
rowCandidateFull(
@@ -388,7 +392,8 @@ struct HfTreeCreatorD0ToKPi {
388392
e,
389393
flagMc,
390394
flagMcDecay,
391-
origin);
395+
origin,
396+
flagBmother);
392397
}
393398
if constexpr (ApplyMl) {
394399
if (candFlag == 0) {
@@ -446,10 +451,10 @@ struct HfTreeCreatorD0ToKPi {
446451
massD0bar = HfHelper::invMassD0barToKPi(candidate);
447452
}
448453
if (candidate.isSelD0()) {
449-
fillTable<ApplyMl>(candidate, 0, massD0, topolChi2PerNdf, ctD, yD, eD, 0, 0, 0);
454+
fillTable<ApplyMl>(candidate, 0, massD0, topolChi2PerNdf, ctD, yD, eD, 0, 0, 0, -1);
450455
}
451456
if (candidate.isSelD0bar()) {
452-
fillTable<ApplyMl>(candidate, 1, massD0bar, topolChi2PerNdf, ctD, yD, eD, 0, 0, 0);
457+
fillTable<ApplyMl>(candidate, 1, massD0bar, topolChi2PerNdf, ctD, yD, eD, 0, 0, 0, -1);
453458
}
454459
}
455460
}
@@ -547,10 +552,10 @@ struct HfTreeCreatorD0ToKPi {
547552
massD0bar = HfHelper::invMassD0barToKPi(candidate);
548553
}
549554
if (candidate.isSelD0()) {
550-
fillTable<ApplyMl>(candidate, 0, massD0, topolChi2PerNdf, ctD, yD, eD, candidate.flagMcMatchRec(), candidate.flagMcDecayChanRec(), candidate.originMcRec());
555+
fillTable<ApplyMl>(candidate, 0, massD0, topolChi2PerNdf, ctD, yD, eD, candidate.flagMcMatchRec(), candidate.flagMcDecayChanRec(), candidate.originMcRec(), o2::analysis::getBHadMotherFlag(candidate.pdgBhadMotherPart()));
551556
}
552557
if (candidate.isSelD0bar()) {
553-
fillTable<ApplyMl>(candidate, 1, massD0bar, topolChi2PerNdf, ctD, yD, eD, candidate.flagMcMatchRec(), candidate.flagMcDecayChanRec(), candidate.originMcRec());
558+
fillTable<ApplyMl>(candidate, 1, massD0bar, topolChi2PerNdf, ctD, yD, eD, candidate.flagMcMatchRec(), candidate.flagMcDecayChanRec(), candidate.originMcRec(), o2::analysis::getBHadMotherFlag(candidate.pdgBhadMotherPart()));
554559
}
555560
}
556561

‎PWGHF/TableProducer/treeCreatorDplusToPiKPi.cxx‎

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
#include "PWGHF/DataModel/AliasTables.h"
2323
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
2424
#include "PWGHF/DataModel/CandidateSelectionTables.h"
25+
#include "PWGHF/Utils/utilsAnalysis.h"
2526

2627
#include "Common/Core/RecoDecay.h"
2728
#include "Common/DataModel/Centrality.h"
@@ -149,7 +150,8 @@ DECLARE_SOA_TABLE(HfCandDpLites, "AOD", "HFCANDDPLITE",
149150
collision::NumContrib,
150151
hf_cand_mc_flag::FlagMcMatchRec,
151152
hf_cand_mc_flag::OriginMcRec,
152-
hf_cand_mc_flag::FlagMcDecayChanRec)
153+
hf_cand_mc_flag::FlagMcDecayChanRec,
154+
hf_cand_mc_flag::PdgBhadMotherPart)
153155

154156
DECLARE_SOA_TABLE(HfCandDpFulls, "AOD", "HFCANDDPFULL",
155157
collision::NumContrib,
@@ -230,7 +232,8 @@ DECLARE_SOA_TABLE(HfCandDpFulls, "AOD", "HFCANDDPFULL",
230232
full::Centrality,
231233
hf_cand_mc_flag::FlagMcMatchRec,
232234
hf_cand_mc_flag::OriginMcRec,
233-
hf_cand_mc_flag::FlagMcDecayChanRec);
235+
hf_cand_mc_flag::FlagMcDecayChanRec,
236+
hf_cand_mc_flag::PdgBhadMotherPart);
234237

235238
DECLARE_SOA_TABLE(HfCandDpFullEvs, "AOD", "HFCANDDPFULLEV",
236239
collision::NumContrib,
@@ -305,10 +308,12 @@ struct HfTreeCreatorDplusToPiKPi {
305308
int8_t flagMc = 0;
306309
int8_t originMc = 0;
307310
int8_t channelMc = 0;
311+
int8_t bMotherFlag = -1;
308312
if constexpr (DoMc) {
309313
flagMc = candidate.flagMcMatchRec();
310314
originMc = candidate.originMcRec();
311315
channelMc = candidate.flagMcDecayChanRec();
316+
bMotherFlag = o2::analysis::getBHadMotherFlag(candidate.pdgBhadMotherPart());
312317
}
313318

314319
std::vector<float> outputMl = {-999., -999.};
@@ -374,7 +379,8 @@ struct HfTreeCreatorDplusToPiKPi {
374379
coll.numContrib(),
375380
flagMc,
376381
originMc,
377-
channelMc);
382+
channelMc,
383+
bMotherFlag);
378384
} else {
379385
rowCandidateFull(
380386
coll.numContrib(),
@@ -455,7 +461,8 @@ struct HfTreeCreatorDplusToPiKPi {
455461
cent,
456462
flagMc,
457463
originMc,
458-
channelMc);
464+
channelMc,
465+
bMotherFlag);
459466
}
460467
}
461468

‎PWGHF/TableProducer/treeCreatorDsToKKPi.cxx‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
#include "PWGHF/DataModel/AliasTables.h"
2424
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
2525
#include "PWGHF/DataModel/CandidateSelectionTables.h"
26+
#include "PWGHF/Utils/utilsAnalysis.h"
2627

2728
#include "Common/Core/RecoDecay.h"
2829
#include "Common/DataModel/Centrality.h"
@@ -149,6 +150,7 @@ DECLARE_SOA_TABLE(HfCandDsLites, "AOD", "HFCANDDSLITE",
149150
hf_cand_mc_flag::FlagMcMatchRec,
150151
hf_cand_mc_flag::OriginMcRec,
151152
hf_cand_mc_flag::FlagMcDecayChanRec,
153+
hf_cand_mc_flag::PdgBhadMotherPart,
152154
hf_cand_mc_flag::IsCandidateSwapped,
153155
full::Sign);
154156

@@ -222,6 +224,7 @@ DECLARE_SOA_TABLE(HfCandDsFulls, "AOD", "HFCANDDSFULL",
222224
hf_cand_mc_flag::FlagMcMatchRec,
223225
hf_cand_mc_flag::OriginMcRec,
224226
hf_cand_mc_flag::FlagMcDecayChanRec,
227+
hf_cand_mc_flag::PdgBhadMotherPart,
225228
hf_cand_mc_flag::IsCandidateSwapped,
226229
full::Sign);
227230

@@ -346,6 +349,7 @@ struct HfTreeCreatorDsToKKPi {
346349
int8_t flagMc{0};
347350
int8_t originMc{0};
348351
int8_t channelMc{0};
352+
int8_t bMotherFlag{-1};
349353
int8_t isSwapped{MassHypo}; // 0 if KKPi, 1 if PiKK
350354
float eCand{0.f};
351355
float ctCand{0.f};
@@ -355,6 +359,7 @@ struct HfTreeCreatorDsToKKPi {
355359
originMc = candidate.originMcRec();
356360
channelMc = candidate.flagMcDecayChanRec();
357361
isSwapped = candidate.isCandidateSwapped();
362+
bMotherFlag = o2::analysis::getBHadMotherFlag(candidate.pdgBhadMotherPart());
358363
if (fillDplusMc && candidate.flagMcDecayChanRec() == channelsResonant[Mother::Dplus][decayChannel]) {
359364
eCand = HfHelper::eDplus(candidate);
360365
ctCand = HfHelper::ctDplus(candidate);
@@ -420,6 +425,7 @@ struct HfTreeCreatorDsToKKPi {
420425
flagMc,
421426
originMc,
422427
channelMc,
428+
bMotherFlag,
423429
isSwapped,
424430
prong0.sign() + prong1.sign() + prong2.sign());
425431
} else {
@@ -493,6 +499,7 @@ struct HfTreeCreatorDsToKKPi {
493499
flagMc,
494500
originMc,
495501
channelMc,
502+
bMotherFlag,
496503
isSwapped,
497504
prong0.sign() + prong1.sign() + prong2.sign());
498505
}

0 commit comments

Comments
 (0)