PWGLF: add mask to save only certain candidates (#5609)

* PWGLF: add mask to save only certain candidates

* Please consider the following formatting changes

---------

Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: ddobrigk

PWGLF/TableProducer/Strangeness/lambdakzeroMLSelectionTreeCreator.cxx

@@ -108,6 +108,9 @@ struct lambdakzeroMLSelectionTreeCreator {
   // Axis:
   ConfigurableAxis centralityAxis{"centralityAxis", {100, 0.0f, 100.0f}, ""};
 
+  // Save certain candidates only
+  Configurable<uint8_t> saveCandidateMask{"saveCandidateMask", 15, "Mask to decide saving candidates"};
+
   void init(InitContext const&)
   {
     histos.add("hEventCentrality", "hEventCentrality", kTH1F, {centralityAxis});
@@ -242,6 +245,9 @@ struct lambdakzeroMLSelectionTreeCreator {
     // 13: Consistent with Lambda, Gamma, and K0Short, 14: Consistent with Anti-Lambda, Gamma, and K0Short
     // 15: Consistent with Lambda, Anti-Lambda, Gamma, and K0Short
 
+    if ((Candidate.SelHypothesis & saveCandidateMask) != saveCandidateMask)
+      return;
+
     // MC flags
     Candidate.isLambda = false;
     Candidate.isAntiLambda = false;