|
18 | 18 | #ifndef PWGHF_DATAMODEL_CANDIDATERECONSTRUCTIONTABLES_H_ |
19 | 19 | #define PWGHF_DATAMODEL_CANDIDATERECONSTRUCTIONTABLES_H_ |
20 | 20 |
|
21 | | -#include "PWGHF/Utils/utilsEvSelHf.h" |
| 21 | +#include "PWGHF/DataModel/CandidateSkimmingTables.h" |
22 | 22 | #include "PWGHF/Utils/utilsPid.h" |
23 | 23 | // |
24 | 24 | #include "PWGLF/DataModel/LFStrangenessTables.h" |
|
36 | 36 |
|
37 | 37 | #include <array> |
38 | 38 | #include <cstdint> |
39 | | -#include <vector> |
40 | 39 |
|
41 | 40 | namespace o2::aod |
42 | 41 | { |
@@ -200,238 +199,6 @@ DECLARE_SOA_TABLE(PidTpcTofTinyPr, "AOD", "PIDTPCTOFTINYPR", //! Table of the TP |
200 | 199 | DECLARE_SOA_TABLE(PidTpcTofTinyDe, "AOD", "PIDTPCTOFTINYDE", //! Table of the TPC & TOF Combined NSigma for deuteron |
201 | 200 | pid_tpc_tof_static_tiny::TpcTofNSigmaDe); |
202 | 201 |
|
203 | | -namespace hf_sel_collision |
204 | | -{ |
205 | | -DECLARE_SOA_COLUMN(WhyRejectColl, whyRejectColl, o2::hf_evsel::HfCollisionRejectionMask); //! |
206 | | -} // namespace hf_sel_collision |
207 | | - |
208 | | -DECLARE_SOA_TABLE(HfSelCollision, "AOD", "HFSELCOLLISION", //! |
209 | | - hf_sel_collision::WhyRejectColl); |
210 | | - |
211 | | -namespace hf_sel_track |
212 | | -{ |
213 | | -DECLARE_SOA_COLUMN(IsSelProng, isSelProng, uint32_t); //! |
214 | | -DECLARE_SOA_COLUMN(IsIdentifiedPid, isIdentifiedPid, uint32_t); //! |
215 | | -DECLARE_SOA_COLUMN(IsPositive, isPositive, bool); //! |
216 | | -} // namespace hf_sel_track |
217 | | - |
218 | | -DECLARE_SOA_TABLE(HfSelTrack, "AOD", "HFSELTRACK", //! |
219 | | - hf_sel_track::IsSelProng, |
220 | | - hf_sel_track::IsIdentifiedPid, |
221 | | - hf_sel_track::IsPositive); |
222 | | - |
223 | | -namespace hf_pv_refit_track |
224 | | -{ |
225 | | -DECLARE_SOA_COLUMN(PvRefitX, pvRefitX, float); //! |
226 | | -DECLARE_SOA_COLUMN(PvRefitY, pvRefitY, float); //! |
227 | | -DECLARE_SOA_COLUMN(PvRefitZ, pvRefitZ, float); //! |
228 | | -DECLARE_SOA_COLUMN(PvRefitSigmaX2, pvRefitSigmaX2, float); //! |
229 | | -DECLARE_SOA_COLUMN(PvRefitSigmaXY, pvRefitSigmaXY, float); //! |
230 | | -DECLARE_SOA_COLUMN(PvRefitSigmaY2, pvRefitSigmaY2, float); //! |
231 | | -DECLARE_SOA_COLUMN(PvRefitSigmaXZ, pvRefitSigmaXZ, float); //! |
232 | | -DECLARE_SOA_COLUMN(PvRefitSigmaYZ, pvRefitSigmaYZ, float); //! |
233 | | -DECLARE_SOA_COLUMN(PvRefitSigmaZ2, pvRefitSigmaZ2, float); //! |
234 | | -DECLARE_SOA_COLUMN(PvRefitDcaXY, pvRefitDcaXY, float); //! |
235 | | -DECLARE_SOA_COLUMN(PvRefitDcaZ, pvRefitDcaZ, float); //! |
236 | | -} // namespace hf_pv_refit_track |
237 | | - |
238 | | -DECLARE_SOA_TABLE(HfPvRefitTrack, "AOD", "HFPVREFITTRACK", //! |
239 | | - hf_pv_refit_track::PvRefitX, |
240 | | - hf_pv_refit_track::PvRefitY, |
241 | | - hf_pv_refit_track::PvRefitZ, |
242 | | - hf_pv_refit_track::PvRefitSigmaX2, |
243 | | - hf_pv_refit_track::PvRefitSigmaXY, |
244 | | - hf_pv_refit_track::PvRefitSigmaY2, |
245 | | - hf_pv_refit_track::PvRefitSigmaXZ, |
246 | | - hf_pv_refit_track::PvRefitSigmaYZ, |
247 | | - hf_pv_refit_track::PvRefitSigmaZ2, |
248 | | - hf_pv_refit_track::PvRefitDcaXY, |
249 | | - hf_pv_refit_track::PvRefitDcaZ); |
250 | | - |
251 | | -namespace hf_track_index |
252 | | -{ |
253 | | -DECLARE_SOA_INDEX_COLUMN(Collision, collision); //! Collision index |
254 | | -DECLARE_SOA_INDEX_COLUMN_FULL(Prong0, prong0, int, Tracks, "_0"); //! Index to first prong |
255 | | -DECLARE_SOA_INDEX_COLUMN_FULL(Prong1, prong1, int, Tracks, "_1"); //! Index to second prong |
256 | | -DECLARE_SOA_INDEX_COLUMN_FULL(Prong2, prong2, int, Tracks, "_2"); //! Index to third prong |
257 | | -DECLARE_SOA_INDEX_COLUMN_FULL(Prong3, prong3, int, Tracks, "_3"); //! Index to fourth prong |
258 | | -DECLARE_SOA_INDEX_COLUMN_FULL(Prong4, prong4, int, Tracks, "_4"); //! Index to fifth prong |
259 | | -DECLARE_SOA_INDEX_COLUMN(V0, v0); //! Index to V0 prong |
260 | | -DECLARE_SOA_INDEX_COLUMN(Cascade, cascade); //! Index to cascade prong |
261 | | -DECLARE_SOA_COLUMN(HFflag, hfflag, uint8_t); //! Bitmap to store selection results, o2-linter: disable=name/o2-column (written to disk) |
262 | | - |
263 | | -DECLARE_SOA_COLUMN(FlagD0ToKPi, flagD0ToKPi, uint8_t); //! |
264 | | -DECLARE_SOA_COLUMN(FlagJpsiToEE, flagJpsiToEE, uint8_t); //! |
265 | | -DECLARE_SOA_COLUMN(FlagJpsiToMuMu, flagJpsiToMuMu, uint8_t); //! |
266 | | - |
267 | | -DECLARE_SOA_COLUMN(FlagDplusToPiKPi, flagDplusToPiKPi, uint8_t); //! |
268 | | -DECLARE_SOA_COLUMN(FlagLcToPKPi, flagLcToPKPi, uint8_t); //! |
269 | | -DECLARE_SOA_COLUMN(FlagDsToKKPi, flagDsToKKPi, uint8_t); //! |
270 | | -DECLARE_SOA_COLUMN(FlagXicToPKPi, flagXicToPKPi, uint8_t); //! |
271 | | - |
272 | | -DECLARE_SOA_COLUMN(FlagDstarToD0Pi, flagDstarToD0Pi, uint8_t); //! |
273 | | - |
274 | | -DECLARE_SOA_COLUMN(MlProbSkimD0ToKPi, mlProbSkimD0ToKPi, std::vector<float>); //! ML probabilities (background, prompt, non-prompt) for D0->Kpi |
275 | | -DECLARE_SOA_COLUMN(MlProbSkimDplusToPiKPi, mlProbSkimDplusToPiKPi, std::vector<float>); //! ML probabilities (background, prompt, non-prompt) for D+->Kpipi |
276 | | -DECLARE_SOA_COLUMN(MlProbSkimDsToKKPi, mlProbSkimDsToKKPi, std::vector<float>); //! ML probabilities (background, prompt, non-prompt) for Ds->KKpi |
277 | | -DECLARE_SOA_COLUMN(MlProbSkimLcToPKPi, mlProbSkimLcToPKPi, std::vector<float>); //! ML probabilities (background, prompt, non-prompt) for Lc->pKpi |
278 | | -DECLARE_SOA_COLUMN(MlProbSkimXicToPKPi, mlProbSkimXicToPKPi, std::vector<float>); //! ML probabilities (background, prompt, non-prompt) for Xic->pKpi |
279 | | -} // namespace hf_track_index |
280 | | - |
281 | | -DECLARE_SOA_TABLE(Hf2Prongs_000, "AOD", "HF2PRONG", //! Table for HF 2 prong candidates (Run 2 converted format) |
282 | | - o2::soa::Index<>, |
283 | | - hf_track_index::Prong0Id, |
284 | | - hf_track_index::Prong1Id, |
285 | | - hf_track_index::HFflag); |
286 | | - |
287 | | -DECLARE_SOA_TABLE_VERSIONED(Hf2Prongs_001, "AOD", "HF2PRONG", 1, //! Table for HF 2 prong candidates (Run 3 format) |
288 | | - o2::soa::Index<>, |
289 | | - hf_track_index::CollisionId, |
290 | | - hf_track_index::Prong0Id, |
291 | | - hf_track_index::Prong1Id, |
292 | | - hf_track_index::HFflag); |
293 | | - |
294 | | -using Hf2Prongs = Hf2Prongs_001; |
295 | | -using Hf2Prong = Hf2Prongs::iterator; |
296 | | - |
297 | | -DECLARE_SOA_TABLE(HfCascades_000, "AOD", "HFCASCADE", //! Table for HF candidates with a V0 (Run 2 converted format) |
298 | | - o2::soa::Index<>, |
299 | | - hf_track_index::Prong0Id, |
300 | | - hf_track_index::V0Id); |
301 | | - |
302 | | -DECLARE_SOA_TABLE_VERSIONED(HfCascades_001, "AOD", "HFCASCADE", 1, //! Table for HF candidates with a V0 (Run 3 format) |
303 | | - o2::soa::Index<>, |
304 | | - hf_track_index::CollisionId, |
305 | | - hf_track_index::Prong0Id, |
306 | | - hf_track_index::V0Id); |
307 | | - |
308 | | -using HfCascades = HfCascades_001; |
309 | | -using HfCascade = HfCascades::iterator; |
310 | | - |
311 | | -DECLARE_SOA_TABLE(Hf3Prongs_000, "AOD", "HF3PRONG", //! Table for HF 3 prong candidates (Run 2 converted format) |
312 | | - o2::soa::Index<>, |
313 | | - hf_track_index::Prong0Id, |
314 | | - hf_track_index::Prong1Id, |
315 | | - hf_track_index::Prong2Id, |
316 | | - hf_track_index::HFflag); |
317 | | - |
318 | | -DECLARE_SOA_TABLE_VERSIONED(Hf3Prongs_001, "AOD", "HF3PRONG", 1, //! Table for HF 3 prong candidates (Run 3 format) |
319 | | - o2::soa::Index<>, |
320 | | - hf_track_index::CollisionId, |
321 | | - hf_track_index::Prong0Id, |
322 | | - hf_track_index::Prong1Id, |
323 | | - hf_track_index::Prong2Id, |
324 | | - hf_track_index::HFflag); |
325 | | - |
326 | | -using Hf3Prongs = Hf3Prongs_001; |
327 | | -using Hf3Prong = Hf3Prongs::iterator; |
328 | | - |
329 | | -DECLARE_SOA_TABLE(HfCascLf2Prongs, "AOD", "HFCASCLF2PRONG", //! Table for HF 2 prong candidates with a Cascade |
330 | | - o2::soa::Index<>, |
331 | | - hf_track_index::CollisionId, |
332 | | - hf_track_index::CascadeId, |
333 | | - hf_track_index::Prong0Id, |
334 | | - hf_track_index::HFflag); |
335 | | -using HfCascLf2Prong = HfCascLf2Prongs::iterator; |
336 | | - |
337 | | -DECLARE_SOA_TABLE(HfCascLf3Prongs, "AOD", "HFCASCLF3PRONG", //! Table for HF 3 prong candidates with a Cascade |
338 | | - o2::soa::Index<>, |
339 | | - hf_track_index::CollisionId, |
340 | | - hf_track_index::CascadeId, |
341 | | - hf_track_index::Prong0Id, |
342 | | - hf_track_index::Prong1Id); |
343 | | -using HfCascLf3Prong = HfCascLf3Prongs::iterator; |
344 | | - |
345 | | -namespace hf_track_index |
346 | | -{ |
347 | | -DECLARE_SOA_INDEX_COLUMN_FULL(ProngD0, prongD0, int, Hf2Prongs, ""); //! Index to a D0 prong |
348 | | -} // namespace hf_track_index |
349 | | - |
350 | | -DECLARE_SOA_TABLE(HfDstars_000, "AOD", "HFDSTAR", //! D* -> D0pi candidates (Run 2 converted format) |
351 | | - o2::soa::Index<>, |
352 | | - hf_track_index::Prong0Id, |
353 | | - hf_track_index::ProngD0Id); |
354 | | - |
355 | | -DECLARE_SOA_TABLE_VERSIONED(HfDstars_001, "AOD", "HFDSTAR", 1, //! D* -> D0pi candidates (Run 3 format) |
356 | | - o2::soa::Index<>, |
357 | | - hf_track_index::CollisionId, |
358 | | - hf_track_index::Prong0Id, |
359 | | - hf_track_index::ProngD0Id); |
360 | | - |
361 | | -using HfDstars = HfDstars_001; |
362 | | -using HfDstar = HfDstars::iterator; |
363 | | - |
364 | | -DECLARE_SOA_TABLE(HfCutStatus2Prong, "AOD", "HFCUTSTATUS2P", //! |
365 | | - hf_track_index::FlagD0ToKPi, |
366 | | - hf_track_index::FlagJpsiToEE, |
367 | | - hf_track_index::FlagJpsiToMuMu); |
368 | | - |
369 | | -DECLARE_SOA_TABLE(HfCutStatus3Prong, "AOD", "HFCUTSTATUS3P", //! |
370 | | - hf_track_index::FlagDplusToPiKPi, |
371 | | - hf_track_index::FlagLcToPKPi, |
372 | | - hf_track_index::FlagDsToKKPi, |
373 | | - hf_track_index::FlagXicToPKPi); |
374 | | - |
375 | | -DECLARE_SOA_TABLE(HfCutStatusDstar, "AOD", "HFCUTSTATUSDST", //! |
376 | | - hf_track_index::FlagDstarToD0Pi); |
377 | | - |
378 | | -DECLARE_SOA_TABLE(Hf2ProngMlProbs, "AOD", "HF2PRONGMLPROB", //! Table for ML scores of HF 2 prong candidates |
379 | | - hf_track_index::MlProbSkimD0ToKPi); |
380 | | - |
381 | | -DECLARE_SOA_TABLE(Hf3ProngMlProbs, "AOD", "HF3PRONGMLPROB", //! Table for ML scores of HF 3 prong candidates |
382 | | - hf_track_index::MlProbSkimDplusToPiKPi, |
383 | | - hf_track_index::MlProbSkimLcToPKPi, |
384 | | - hf_track_index::MlProbSkimDsToKKPi, |
385 | | - hf_track_index::MlProbSkimXicToPKPi); |
386 | | - |
387 | | -namespace hf_pv_refit |
388 | | -{ |
389 | | -DECLARE_SOA_COLUMN(PvRefitX, pvRefitX, float); //! |
390 | | -DECLARE_SOA_COLUMN(PvRefitY, pvRefitY, float); //! |
391 | | -DECLARE_SOA_COLUMN(PvRefitZ, pvRefitZ, float); //! |
392 | | -DECLARE_SOA_COLUMN(PvRefitSigmaX2, pvRefitSigmaX2, float); //! |
393 | | -DECLARE_SOA_COLUMN(PvRefitSigmaXY, pvRefitSigmaXY, float); //! |
394 | | -DECLARE_SOA_COLUMN(PvRefitSigmaY2, pvRefitSigmaY2, float); //! |
395 | | -DECLARE_SOA_COLUMN(PvRefitSigmaXZ, pvRefitSigmaXZ, float); //! |
396 | | -DECLARE_SOA_COLUMN(PvRefitSigmaYZ, pvRefitSigmaYZ, float); //! |
397 | | -DECLARE_SOA_COLUMN(PvRefitSigmaZ2, pvRefitSigmaZ2, float); //! |
398 | | -} // namespace hf_pv_refit |
399 | | - |
400 | | -DECLARE_SOA_TABLE(HfPvRefit2Prong, "AOD", "HFPVREFIT2PRONG", //! |
401 | | - hf_pv_refit::PvRefitX, |
402 | | - hf_pv_refit::PvRefitY, |
403 | | - hf_pv_refit::PvRefitZ, |
404 | | - hf_pv_refit::PvRefitSigmaX2, |
405 | | - hf_pv_refit::PvRefitSigmaXY, |
406 | | - hf_pv_refit::PvRefitSigmaY2, |
407 | | - hf_pv_refit::PvRefitSigmaXZ, |
408 | | - hf_pv_refit::PvRefitSigmaYZ, |
409 | | - hf_pv_refit::PvRefitSigmaZ2); |
410 | | - |
411 | | -DECLARE_SOA_TABLE(HfPvRefit3Prong, "AOD", "HFPVREFIT3PRONG", //! |
412 | | - hf_pv_refit::PvRefitX, |
413 | | - hf_pv_refit::PvRefitY, |
414 | | - hf_pv_refit::PvRefitZ, |
415 | | - hf_pv_refit::PvRefitSigmaX2, |
416 | | - hf_pv_refit::PvRefitSigmaXY, |
417 | | - hf_pv_refit::PvRefitSigmaY2, |
418 | | - hf_pv_refit::PvRefitSigmaXZ, |
419 | | - hf_pv_refit::PvRefitSigmaYZ, |
420 | | - hf_pv_refit::PvRefitSigmaZ2, |
421 | | - o2::soa::Marker<1>); |
422 | | - |
423 | | -DECLARE_SOA_TABLE(HfPvRefitDstar, "AOD", "HFPVREFITDSTAR", //! |
424 | | - hf_pv_refit::PvRefitX, |
425 | | - hf_pv_refit::PvRefitY, |
426 | | - hf_pv_refit::PvRefitZ, |
427 | | - hf_pv_refit::PvRefitSigmaX2, |
428 | | - hf_pv_refit::PvRefitSigmaXY, |
429 | | - hf_pv_refit::PvRefitSigmaY2, |
430 | | - hf_pv_refit::PvRefitSigmaXZ, |
431 | | - hf_pv_refit::PvRefitSigmaYZ, |
432 | | - hf_pv_refit::PvRefitSigmaZ2, |
433 | | - o2::soa::Marker<2>); |
434 | | - |
435 | 202 | // general decay properties |
436 | 203 | namespace hf_cand |
437 | 204 | { |
|
0 commit comments