From 8e441b00b6faa73f7b19b9e7f9f88795d0a6558e Mon Sep 17 00:00:00 2001
From: Marcello Di Costanzo <mdicosta@aliceml.cern.ch>
Date: Thu, 1 Oct 2026 15:09:16 +0200
Subject: [PATCH] Correlated bkgs in derived data creators

---
 PWGHF/TableProducer/derivedDataCreatorD0ToKPi.cxx      | 2 +-
 PWGHF/TableProducer/derivedDataCreatorDplusToPiKPi.cxx | 2 +-
 PWGHF/TableProducer/derivedDataCreatorDsToKKPi.cxx     | 2 +-
 PWGHF/TableProducer/derivedDataCreatorDstarToD0Pi.cxx  | 2 +-
 PWGHF/TableProducer/derivedDataCreatorLcToPKPi.cxx     | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/PWGHF/TableProducer/derivedDataCreatorD0ToKPi.cxx b/PWGHF/TableProducer/derivedDataCreatorD0ToKPi.cxx
index 6c8dd98f3c9..ad8e52744e1 100644
--- a/PWGHF/TableProducer/derivedDataCreatorD0ToKPi.cxx
+++ b/PWGHF/TableProducer/derivedDataCreatorD0ToKPi.cxx
@@ -299,7 +299,7 @@ struct HfDerivedDataCreatorD0ToKPi {
             }
           }
           if constexpr (OnlySig) {
-            if (std::abs(flagMcRec) != o2::hf_decay::hf_cand_2prong::DecayChannelMain::D0ToPiK) {
+            if (!acceptCorrelatedBkgs && (std::abs(flagMcRec) != o2::hf_decay::hf_cand_2prong::DecayChannelMain::D0ToPiK)) {
               continue;
             }
           }
diff --git a/PWGHF/TableProducer/derivedDataCreatorDplusToPiKPi.cxx b/PWGHF/TableProducer/derivedDataCreatorDplusToPiKPi.cxx
index b1cad301ba6..bf66abd0420 100644
--- a/PWGHF/TableProducer/derivedDataCreatorDplusToPiKPi.cxx
+++ b/PWGHF/TableProducer/derivedDataCreatorDplusToPiKPi.cxx
@@ -304,7 +304,7 @@ struct HfDerivedDataCreatorDplusToPiKPi {
             }
           }
           if constexpr (OnlySig) {
-            if (std::abs(flagMcRec) != hf_decay::hf_cand_3prong::DecayChannelMain::DplusToPiKPi) {
+            if (!acceptCorrelatedBkgs && (std::abs(flagMcRec) != hf_decay::hf_cand_3prong::DecayChannelMain::DplusToPiKPi)) {
               continue;
             }
           }
diff --git a/PWGHF/TableProducer/derivedDataCreatorDsToKKPi.cxx b/PWGHF/TableProducer/derivedDataCreatorDsToKKPi.cxx
index b1a2562c47a..c30f037d976 100644
--- a/PWGHF/TableProducer/derivedDataCreatorDsToKKPi.cxx
+++ b/PWGHF/TableProducer/derivedDataCreatorDsToKKPi.cxx
@@ -287,7 +287,7 @@ struct HfDerivedDataCreatorDsToKKPi {
             }
           }
           if constexpr (OnlySig) {
-            if (std::abs(flagMcRec) != hf_decay::hf_cand_3prong::DecayChannelMain::DsToPiKK) {
+            if (!acceptCorrelatedBkgs && (std::abs(flagMcRec) != hf_decay::hf_cand_3prong::DecayChannelMain::DsToPiKK)) {
               continue;
             }
           }
diff --git a/PWGHF/TableProducer/derivedDataCreatorDstarToD0Pi.cxx b/PWGHF/TableProducer/derivedDataCreatorDstarToD0Pi.cxx
index 5f20c81e4a5..c73262f6572 100644
--- a/PWGHF/TableProducer/derivedDataCreatorDstarToD0Pi.cxx
+++ b/PWGHF/TableProducer/derivedDataCreatorDstarToD0Pi.cxx
@@ -279,7 +279,7 @@ struct HfDerivedDataCreatorDstarToD0Pi {
             }
           }
           if constexpr (OnlySig) {
-            if (std::abs(flagMcRec) != hf_decay::hf_cand_dstar::DecayChannelMain::DstarToPiKPi) {
+            if (!acceptCorrelatedBkgs && (std::abs(flagMcRec) != hf_decay::hf_cand_dstar::DecayChannelMain::DstarToPiKPi)) {
               continue;
             }
           }
diff --git a/PWGHF/TableProducer/derivedDataCreatorLcToPKPi.cxx b/PWGHF/TableProducer/derivedDataCreatorLcToPKPi.cxx
index d543b4605f0..c308143988c 100644
--- a/PWGHF/TableProducer/derivedDataCreatorLcToPKPi.cxx
+++ b/PWGHF/TableProducer/derivedDataCreatorLcToPKPi.cxx
@@ -280,7 +280,7 @@ struct HfDerivedDataCreatorLcToPKPi {
             }
           }
           if constexpr (OnlySig) {
-            if (std::abs(flagMcRec) != hf_decay::hf_cand_3prong::DecayChannelMain::LcToPKPi) {
+            if (!acceptCorrelatedBkgs && (std::abs(flagMcRec) != hf_decay::hf_cand_3prong::DecayChannelMain::LcToPKPi)) {
               continue;
             }
           }