Adding flag selection in the hadron tasks (#5218)

Co-authored-by: Luigi <ldellost@aliceml.cern.ch>

Author: DelloStritto

Analysis/Tasks/PWGHF/HFD0CandidateSelector.cxx

@@ -324,11 +324,17 @@ struct HFD0CandidateSelector {
 
     for (auto& hfCandProng2 : hfCandProng2s) { //looping over 2 prong candidates
 
+      statusD0 = 0;
+      statusD0bar = 0;
+
+      if (!(hfCandProng2.hfflag() & 1 << D0ToPiK)) {
+        hfSelD0Candidate(statusD0, statusD0bar);
+        continue;
+      }
+
       auto trackPos = hfCandProng2.index0_as<aod::BigTracksPID>(); //positive daughter
       auto trackNeg = hfCandProng2.index1_as<aod::BigTracksPID>(); //negative daughter
 
-      statusD0 = 0;
-      statusD0bar = 0;
       topolD0 = true;
       topolD0bar = true;
       pidD0 = -1;

Analysis/Tasks/PWGHF/HFLcCandidateSelector.cxx

@@ -287,9 +287,8 @@ struct HFLcCandidateSelector {
 
       statusLcpKpi = 0;
       statusLcpiKp = 0;
-      int LcFlag = hfCandProng3.hfflag();
 
-      if (!(LcFlag & 1 << 1)) {
+      if (!(hfCandProng3.hfflag() & 1 << LcToPKPi)) {
         hfSelLcCandidate(statusLcpKpi, statusLcpiKp);
         continue;
       }

Analysis/Tasks/PWGHF/taskD0.cxx

@@ -64,6 +64,9 @@ struct TaskD0 {
   {
     //Printf("Candidates: %d", candidates.size());
     for (auto& candidate : candidates) {
+      if (!(candidate.hfflag() & 1 << D0ToPiK)) {
+        continue;
+      }
       if (cutEtaCandMax >= 0. && std::abs(candidate.eta()) > cutEtaCandMax) {
         //Printf("Candidate: eta rejection: %g", candidate.eta());
         continue;
@@ -120,6 +123,9 @@ struct TaskD0MC {
     // MC rec.
     //Printf("MC Candidates: %d", candidates.size());
     for (auto& candidate : candidates) {
+      if (!(candidate.hfflag() & 1 << D0ToPiK)) {
+        continue;
+      }
       if (cutEtaCandMax >= 0. && std::abs(candidate.eta()) > cutEtaCandMax) {
         //Printf("MC Rec.: eta rejection: %g", candidate.eta());
         continue;

Analysis/Tasks/PWGHF/taskLc.cxx

@@ -63,6 +63,9 @@ struct TaskLc {
   void process(soa::Filtered<soa::Join<aod::HfCandProng3, aod::HFSelLcCandidate>> const& candidates)
   {
     for (auto& candidate : candidates) {
+      if (!(candidate.hfflag() & 1 << LcToPKPi)) {
+        continue;
+      }
       if (cutEtaCandMax >= 0. && std::abs(candidate.eta()) > cutEtaCandMax) {
         //Printf("Candidate: eta rejection: %g", candidate.eta());
         continue;
@@ -120,6 +123,9 @@ struct TaskLcMC {
     // MC rec.
     //Printf("MC Candidates: %d", candidates.size());
     for (auto& candidate : candidates) {
+      if (!(candidate.hfflag() & 1 << LcToPKPi)) {
+        continue;
+      }
       if (cutEtaCandMax >= 0. && std::abs(candidate.eta()) > cutEtaCandMax) {
         //Printf("MC Rec.: eta rejection: %g", candidate.eta());
         continue;