Adding flag selection in the hadron tasks (#5218)

Co-authored-by: Luigi <ldellost@aliceml.cern.ch>
AliceO2Group · Jan 20, 2021 · d5f8673 · d5f8673
1 parent 40bd7f9
commit d5f8673
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 4 deletions.
diff --git a/Analysis/Tasks/PWGHF/HFD0CandidateSelector.cxx b/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;

diff --git a/Analysis/Tasks/PWGHF/HFLcCandidateSelector.cxx b/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;
       }

diff --git a/Analysis/Tasks/PWGHF/taskD0.cxx b/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;

diff --git a/Analysis/Tasks/PWGHF/taskLc.cxx b/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;