Skip to content

Commit cfa4a7b

Browse files
committed
Feat: add ranged based PID selection and rejection
1 parent 4f6de0e commit cfa4a7b

File tree

2 files changed

+105
-169
lines changed

2 files changed

+105
-169
lines changed

PWGCF/Femto/Core/partitions.h

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -31,15 +31,17 @@
3131
// standard track partition
3232
#define MAKE_TRACK_PARTITION(selection) \
3333
ifnode(selection.chargeSign.node() != 0, ifnode(selection.chargeSign.node() > 0, o2::aod::femtobase::stored::signedPt > 0.f, o2::aod::femtobase::stored::signedPt < 0.f), true) && \
34-
(nabs(selection.chargeAbs.node() * o2::aod::femtobase::stored::signedPt) > selection.ptMin) && \
35-
(nabs(selection.chargeAbs.node() * o2::aod::femtobase::stored::signedPt) < selection.ptMax) && \
34+
(nabs(selection.chargeAbs * o2::aod::femtobase::stored::signedPt) > selection.ptMin) && \
35+
(nabs(selection.chargeAbs * o2::aod::femtobase::stored::signedPt) < selection.ptMax) && \
3636
(o2::aod::femtobase::stored::eta > selection.etaMin) && \
3737
(o2::aod::femtobase::stored::eta < selection.etaMax) && \
3838
(o2::aod::femtobase::stored::phi > selection.phiMin) && \
3939
(o2::aod::femtobase::stored::phi < selection.phiMax) && \
40-
ifnode(TRACK_MOMENTUM(selection.chargeAbs.node(), o2::aod::femtobase::stored::pt, o2::aod::femtobase::stored::eta) <= selection.pidThres, \
41-
ncheckbit(o2::aod::femtotracks::mask, selection.maskLowMomentum) && (o2::aod::femtotracks::mask & selection.rejectionMaskLowMomentum.node()) == 0u, \
42-
ncheckbit(o2::aod::femtotracks::mask, selection.maskHighMomentum) && (o2::aod::femtotracks::mask & selection.rejectionMaskHighMomentum.node()) == 0u)
40+
ifnode(TRACK_MOMENTUM(selection.chargeAbs, o2::aod::femtobase::stored::signedPt, o2::aod::femtobase::stored::eta) <= selection.pidThres, \
41+
ncheckbit(o2::aod::femtotracks::mask, selection.maskLowMomentum) && \
42+
(o2::aod::femtotracks::mask & selection.rejectionMaskLowMomentum) == static_cast<o2::aod::femtodatatypes::TrackMaskType>(0), \
43+
ncheckbit(o2::aod::femtotracks::mask, selection.maskHighMomentum) && \
44+
(o2::aod::femtotracks::mask & selection.rejectionMaskHighMomentum) == static_cast<o2::aod::femtodatatypes::TrackMaskType>(0))
4345

4446
// partition for phis and rhos, i.e. resonance that are their own antiparticle
4547
#define MAKE_RESONANCE_0_PARTITON(selection) \

0 commit comments

Comments
 (0)