Skip to content

Commit f84a313

Browse files
committed
Fixed the comment from vit
1 parent dc42992 commit f84a313

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

PWGHF/TableProducer/trackIndexSkimCreator.cxx

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,11 +121,26 @@ enum ChannelsProtonPid {
121121
LcToPK0S,
122122
NChannelsProtonPid
123123
};
124+
125+
enum ChannelsKaonPid {
126+
Charm3ProngToKaon = 0,
127+
NChannelsKaonPid
128+
};
129+
130+
enum ChannelsLightNucleiPid {
131+
CdToDeKPi = 0,
132+
CtToTrKPi,
133+
ChToHeKPi,
134+
NChannelsLightNucleiPid
135+
};
136+
137+
124138
// kaon PID (opposite-sign track in 3-prong decays)
125139
constexpr int ChannelKaonPid = ChannelsProtonPid::NChannelsProtonPid;
126140
constexpr int ChannelsDeuteronPid = ChannelsProtonPid::NChannelsProtonPid + 1;
127141
constexpr int ChannelsTritonPid = ChannelsProtonPid::NChannelsProtonPid + 2;
128142
constexpr int ChannelsHeliumPid = ChannelsProtonPid::NChannelsProtonPid + 3;
143+
constexpr int NChannelsPidFor3Prong = ChannelsProtonPid::NChannelsProtonPid + ChannelsKaonPid::NChannelsKaonPid + ChannelsLightNucleiPid::NChannelsLightNucleiPid;
129144

130145
enum class ChannelsNucleiQA : int {
131146
Deuteron = 0,
@@ -539,8 +554,8 @@ struct HfTrackIndexSkimCreatorTagSelTracks {
539554
statusPid[ChannelsHeliumPid] = selectorHelium.statusTpcAndTof(hfTrack);
540555
}
541556

542-
int8_t flag = BIT(ChannelsProtonPid::NChannelsProtonPid + 4) - 1; // all bits on (including the kaon one)
543-
for (auto iChannel{0u}; iChannel < ChannelsProtonPid::NChannelsProtonPid + 4; ++iChannel) {
557+
int8_t flag = BIT(NChannelsPidFor3Prong) - 1; // all bits on (including the kaon one)
558+
for (auto iChannel{0u}; iChannel < NChannelsPidFor3Prong; ++iChannel) {
544559
if (statusPid[iChannel] == TrackSelectorPID::Rejected) {
545560
CLRBIT(flag, iChannel);
546561
}

0 commit comments

Comments
 (0)