Skip to content

Commit 58b0292

Browse files
DelloStrittoDelloStritto
authored
PWGHF: Adding candidate/particle origin MC infos in the tree creators (#2391)
1 parent 58cdb72 commit 58b0292

5 files changed

Lines changed: 44 additions & 16 deletions

File tree

‎PWGHF/TableProducer/treeCreatorChicToJpsiGamma.cxx‎

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,8 @@ DECLARE_SOA_COLUMN(Phi, phi, float);
5656
DECLARE_SOA_COLUMN(Y, y, float);
5757
DECLARE_SOA_COLUMN(E, e, float);
5858
DECLARE_SOA_COLUMN(MCflag, mcflag, int8_t);
59+
DECLARE_SOA_COLUMN(OriginMcRec, originMcRec, int8_t);
60+
DECLARE_SOA_COLUMN(OriginMcGen, originMcGen, int8_t);
5961
// Events
6062
DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int);
6163
DECLARE_SOA_COLUMN(RunNumber, runNumber, int);
@@ -90,7 +92,8 @@ DECLARE_SOA_TABLE(HfCandChicFull, "AOD", "HFCANDChicFull",
9092
full::Phi,
9193
full::Y,
9294
hf_cand_chic::JpsiToMuMuMass,
93-
full::MCflag);
95+
full::MCflag,
96+
full::OriginMcRec);
9497

9598
DECLARE_SOA_TABLE(HfCandChicFullEvents, "AOD", "HFCANDChicFullE",
9699
collision::BCId,
@@ -108,7 +111,8 @@ DECLARE_SOA_TABLE(HfCandChicFullParticles, "AOD", "HFCANDChicFullP",
108111
full::Phi,
109112
full::Y,
110113
hf_cand_chic::JpsiToMuMuMass,
111-
full::MCflag);
114+
full::MCflag,
115+
full::OriginMcGen);
112116

113117
} // namespace o2::aod
114118

@@ -193,7 +197,8 @@ struct HfTreeCreatorChicToJpsiGamma {
193197
candidate.phi(),
194198
FunctionY,
195199
candidate.jpsiToMuMuMass(),
196-
candidate.flagMcMatchRec());
200+
candidate.flagMcMatchRec(),
201+
candidate.originMcRec());
197202
}
198203
};
199204
fillTable(0, candidate.isSelChicToJpsiToMuMuGamma(), invMassChicToJpsiGamma(candidate), ctChic(candidate), yChic(candidate));
@@ -212,7 +217,8 @@ struct HfTreeCreatorChicToJpsiGamma {
212217
particle.phi(),
213218
RecoDecay::y(array{particle.px(), particle.py(), particle.pz()}, massChic),
214219
0., // put here the jpsi mass
215-
particle.flagMcMatchGen());
220+
particle.flagMcMatchGen(),
221+
particle.originMcGen());
216222
}
217223
}
218224
}

‎PWGHF/TableProducer/treeCreatorLbToLcPi.cxx‎

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ DECLARE_SOA_COLUMN(CPA, cpa, float);
6060
DECLARE_SOA_COLUMN(CPAXY, cpaXY, float);
6161
DECLARE_SOA_COLUMN(Ct, ct, float);
6262
DECLARE_SOA_COLUMN(MCflag, mcflag, int8_t);
63+
DECLARE_SOA_COLUMN(OriginMcRec, originMcRec, int8_t);
6364
DECLARE_SOA_COLUMN(NSigRICHTrk0Pi, nsigRICHTrk0Pi, float);
6465
DECLARE_SOA_COLUMN(NSigfRICHTrk0Pi, nsigfRICHTrk0Pi, float);
6566
DECLARE_SOA_COLUMN(NSigTOFTrk0Pi, nsigTOFTrk0Pi, float);
@@ -161,7 +162,8 @@ DECLARE_SOA_TABLE(HfCandLbFull, "AOD", "HFCANDLbFull",
161162
full::Eta,
162163
full::Phi,
163164
full::Y,
164-
full::MCflag);
165+
full::MCflag,
166+
full::OriginMcRec);
165167

166168
} // namespace o2::aod
167169

@@ -318,7 +320,8 @@ struct HfTreeCreatorLbToLcPi {
318320
candidate.eta(),
319321
candidate.phi(),
320322
FunctionY,
321-
candidate.flagMcMatchRec());
323+
candidate.flagMcMatchRec(),
324+
candidate.originMcRec());
322325
}
323326
};
324327
fillTable(candidate.isSelLbToLcPi(), invMassLbToLcPi(candidate), ctLb(candidate), yLb(candidate));

‎PWGHF/TableProducer/treeCreatorLcToPKPi.cxx‎

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,8 @@ DECLARE_SOA_COLUMN(CPA, cpa, float);
7575
DECLARE_SOA_COLUMN(CPAXY, cpaXY, float);
7676
DECLARE_SOA_COLUMN(Ct, ct, float);
7777
DECLARE_SOA_COLUMN(MCflag, mcflag, int8_t);
78+
DECLARE_SOA_COLUMN(OriginMcRec, originMcRec, int8_t);
79+
DECLARE_SOA_COLUMN(OriginMcGen, originMcGen, int8_t);
7880
DECLARE_SOA_COLUMN(IsCandidateSwapped, isCandidateSwapped, int8_t);
7981
// Events
8082
DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int);
@@ -152,6 +154,7 @@ DECLARE_SOA_TABLE(HfCand3ProngFull, "AOD", "HFCAND3PFull",
152154
full::Y,
153155
full::E,
154156
full::MCflag,
157+
full::OriginMcRec,
155158
full::IsCandidateSwapped);
156159

157160
DECLARE_SOA_TABLE(HfCand3ProngFullEvents, "AOD", "HFCAND3PFullE",
@@ -169,7 +172,8 @@ DECLARE_SOA_TABLE(HfCand3ProngFullParticles, "AOD", "HFCAND3PFullP",
169172
full::Eta,
170173
full::Phi,
171174
full::Y,
172-
full::MCflag);
175+
full::MCflag,
176+
full::OriginMcGen);
173177

174178
} // namespace o2::aod
175179

@@ -290,6 +294,7 @@ struct HfTreeCreatorLcToPKPi {
290294
FunctionY,
291295
FunctionE,
292296
candidate.flagMcMatchRec(),
297+
candidate.originMcRec(),
293298
candidate.isCandidateSwapped());
294299
}
295300
};
@@ -308,7 +313,8 @@ struct HfTreeCreatorLcToPKPi {
308313
particle.eta(),
309314
particle.phi(),
310315
RecoDecay::y(array{particle.px(), particle.py(), particle.pz()}, RecoDecay::getMassPDG(particle.pdgCode())),
311-
particle.flagMcMatchGen());
316+
particle.flagMcMatchGen(),
317+
particle.originMcGen());
312318
}
313319
}
314320
}
@@ -417,6 +423,7 @@ struct HfTreeCreatorLcToPKPi {
417423
FunctionY,
418424
FunctionE,
419425
0.,
426+
0.,
420427
0.);
421428
}
422429
};

‎PWGHF/TableProducer/treeCreatorXToJpsiPiPi.cxx‎

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,8 @@ DECLARE_SOA_COLUMN(CPA, cpa, float);
6868
DECLARE_SOA_COLUMN(CPAXY, cpaXY, float);
6969
DECLARE_SOA_COLUMN(Ct, ct, float);
7070
DECLARE_SOA_COLUMN(MCflag, mcflag, int8_t);
71+
DECLARE_SOA_COLUMN(OriginMcRec, originMcRec, int8_t);
72+
DECLARE_SOA_COLUMN(OriginMcGen, originMcGen, int8_t);
7173
// Events
7274
DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int);
7375
DECLARE_SOA_COLUMN(RunNumber, runNumber, int);
@@ -114,7 +116,8 @@ DECLARE_SOA_TABLE(HfCandXFull, "AOD", "HFCANDXFull",
114116
full::DR1,
115117
full::DR2,
116118
full::PiBalance,
117-
full::MCflag);
119+
full::MCflag,
120+
full::OriginMcRec);
118121

119122
DECLARE_SOA_TABLE(HfCandXFullEvents, "AOD", "HFCANDXFullE",
120123
collision::BCId,
@@ -131,7 +134,8 @@ DECLARE_SOA_TABLE(HfCandXFullParticles, "AOD", "HFCANDXFullP",
131134
full::Eta,
132135
full::Phi,
133136
full::Y,
134-
full::MCflag);
137+
full::MCflag,
138+
full::OriginMcGen);
135139

136140
} // namespace o2::aod
137141

@@ -224,7 +228,8 @@ struct HfTreeCreatorXToJpsiPiPi {
224228
FunctionDR1,
225229
FunctionDR2,
226230
FunctionPiBalance,
227-
candidate.flagMcMatchRec());
231+
candidate.flagMcMatchRec(),
232+
candidate.originMcRec());
228233
}
229234
};
230235

@@ -242,7 +247,8 @@ struct HfTreeCreatorXToJpsiPiPi {
242247
particle.eta(),
243248
particle.phi(),
244249
RecoDecay::y(array{particle.px(), particle.py(), particle.pz()}, massX),
245-
particle.flagMcMatchGen());
250+
particle.flagMcMatchGen(),
251+
particle.originMcGen());
246252
}
247253
}
248254
}

‎PWGHF/TableProducer/treeCreatorXiccToPKPiPi.cxx‎

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,8 @@ DECLARE_SOA_COLUMN(CPA, cpa, float);
5858
DECLARE_SOA_COLUMN(CPAXY, cpaXY, float);
5959
DECLARE_SOA_COLUMN(Ct, ct, float);
6060
DECLARE_SOA_COLUMN(MCflag, mcflag, int8_t);
61+
DECLARE_SOA_COLUMN(OriginMcRec, originMcRec, int8_t);
62+
DECLARE_SOA_COLUMN(OriginMcGen, originMcGen, int8_t);
6163
// Xic selection variable
6264
DECLARE_SOA_COLUMN(XicM, xicM, float);
6365
DECLARE_SOA_COLUMN(XicCt, xicCt, float);
@@ -132,7 +134,8 @@ DECLARE_SOA_TABLE(HfCandXiccFull, "AOD", "HFCANDXiccFull",
132134
full::Eta,
133135
full::Phi,
134136
full::Y,
135-
full::MCflag);
137+
full::MCflag,
138+
full::OriginMcRec);
136139

137140
DECLARE_SOA_TABLE(HfCandXiccFullEvents, "AOD", "HFCANDXiccFullE",
138141
collision::BCId,
@@ -149,7 +152,8 @@ DECLARE_SOA_TABLE(HfCandXiccFullParticles, "AOD", "HFCANDXiccFullP",
149152
full::Eta,
150153
full::Phi,
151154
full::Y,
152-
full::MCflag);
155+
full::MCflag,
156+
full::OriginMcGen);
153157

154158
} // namespace o2::aod
155159

@@ -246,7 +250,8 @@ struct HfTreeCreatorXiccToPKPiPi {
246250
candidate.eta(),
247251
candidate.phi(),
248252
FunctionY,
249-
candidate.flagMcMatchRec());
253+
candidate.flagMcMatchRec(),
254+
candidate.originMcRec());
250255
}
251256
};
252257

@@ -263,7 +268,8 @@ struct HfTreeCreatorXiccToPKPiPi {
263268
particle.eta(),
264269
particle.phi(),
265270
RecoDecay::y(array{particle.px(), particle.py(), particle.pz()}, RecoDecay::getMassPDG(particle.pdgCode())),
266-
particle.flagMcMatchGen());
271+
particle.flagMcMatchGen(),
272+
particle.originMcGen());
267273
}
268274
}
269275
}

0 commit comments

Comments
 (0)