Skip to content

Commit 73b0df9

Browse files
authored
Refactor selection flags and improve variable usage
1 parent ca902ea commit 73b0df9

File tree

1 file changed

+10
-12
lines changed

1 file changed

+10
-12
lines changed

ALICE3/TableProducer/alice3HfSelector3Prong.cxx

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -313,32 +313,30 @@ struct Alice3HfSelector3Prong {
313313
template <CharmHadAlice3 CharmHad, typename CandType>
314314
void runSelect3Prong(CandType const& cands)
315315
{
316-
bool isSelMassHypo0{false};
317-
bool isSelMassHypo1{false};
318316
std::vector<float> outputMl{-1.f, -1.f, -1.f};
319317
uint32_t pidMask = 0;
320-
318+
321319
// looping over 3-prong cands
322320
for (const auto& cand : cands) {
323321
registry.fill(HIST("hSelections"), 1, cand.pt());
324322
outputMl = {-1.f, -1.f, -1.f};
325323
pidMask = 0;
326324

327-
auto ptCand = cand.pt();
328-
int const ptBin = findBin(binsPt, ptCand);
325+
const float ptCand = cand.pt();
326+
const int ptBin = findBin(binsPt, ptCand);
329327
if (ptBin == -1) {
330-
candSelFlags(isSelMassHypo0, isSelMassHypo1, pidMask);
328+
candSelFlags(false, false, pidMask);
331329
if (applyMl) {
332330
candMlScores(outputMl[0], outputMl[1], outputMl[2]);
333331
}
334332
continue;
335333
}
336334

337335
// Here all cands pass the cut on the mass selection
338-
bool const selMassHypo0 = selectionCandidateMass<CharmHad, false>(ptBin, cand);
339-
bool const selMassHypo1 = selectionCandidateMass<CharmHad, true>(ptBin, cand);
336+
const bool selMassHypo0 = selectionCandidateMass<CharmHad, false>(ptBin, cand);
337+
const bool selMassHypo1 = selectionCandidateMass<CharmHad, true>(ptBin, cand);
340338
if (!selMassHypo0 && !selMassHypo1) {
341-
candSelFlags(isSelMassHypo0, isSelMassHypo1, pidMask);
339+
candSelFlags(false, false, pidMask);
342340
if (applyMl) {
343341
candMlScores(outputMl[0], outputMl[1], outputMl[2]);
344342
}
@@ -348,7 +346,7 @@ struct Alice3HfSelector3Prong {
348346

349347
// Topological selection (TODO: track quality selection)
350348
if (!selectionTopol(cand, ptCand)) {
351-
candSelFlags(isSelMassHypo0, isSelMassHypo1, pidMask);
349+
candSelFlags(false, false, pidMask);
352350
if (applyMl) {
353351
candMlScores(outputMl[0], outputMl[1], outputMl[2]);
354352
}
@@ -359,7 +357,7 @@ struct Alice3HfSelector3Prong {
359357
// PID selection
360358
configurePidMask<CharmHad>(cand, pidMask);
361359
if (pidMask == 0) {
362-
candSelFlags(isSelMassHypo0, isSelMassHypo1, pidMask);
360+
candSelFlags(false, false, pidMask);
363361
if (applyMl) {
364362
candMlScores(outputMl[0], outputMl[1], outputMl[2]);
365363
}
@@ -375,7 +373,7 @@ struct Alice3HfSelector3Prong {
375373
isSelectedMl = mlResponse.isSelectedMl(inputFeaturesMassHypo0, ptCand, outputMl);
376374
candMlScores(outputMl[0], outputMl[1], outputMl[2]);
377375
if (!isSelectedMl) {
378-
candSelFlags(isSelMassHypo0, isSelMassHypo1, pidMask);
376+
candSelFlags(false, false, pidMask);
379377
continue;
380378
}
381379

0 commit comments

Comments
 (0)