From 2777c417fb4d735da1ba4fe82481b35c14774e9f Mon Sep 17 00:00:00 2001
From: Katarzyna Gwizdziel <katarzyna.gwizdziel@cern.ch>
Date: Mon, 28 Sep 2026 13:07:36 +0200
Subject: [PATCH] FU D0 task: adding y histo and non-prompt cut for reco D0s

---
 PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx b/PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx
index c0cd656cab1..0e54930d4df 100644
--- a/PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx
+++ b/PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx
@@ -314,6 +314,7 @@ struct FemtoUniversePairTaskTrackD0 {
                               {"hEtaD0D0bar", ";#eta ;counts", {HistType::kTH1F, {{200, -1., 1.}}}},
                               {"hEtaD0", ";#eta ;counts", {HistType::kTH1F, {{200, -1., 1.}}}},
                               {"hEtaD0bar", ";#eta ;counts", {HistType::kTH1F, {{200, -1., 1.}}}},
+                              {"hYD0D0bar", ";y ;counts", {HistType::kTH1F, {{200, -1., 1.}}}},
                               {"hPtDaughters", ";#it{p}_{T} (GeV/#it{c});counts", {HistType::kTH1F, {{300, 0., 12.}}}},
                               {"hSignDaughters", ";sign ;counts", {HistType::kTH1F, {{10, -2.5, 2.5}}}}}};
 
@@ -675,6 +676,7 @@ struct FemtoUniversePairTaskTrackD0 {
       registry.fill(HIST("hPtD0D0bar"), d0d0bar.pt());
       registry.fill(HIST("hPhiD0D0bar"), d0d0bar.phi());
       registry.fill(HIST("hEtaD0D0bar"), d0d0bar.eta());
+      registry.fill(HIST("hYD0D0bar"), d0d0bar.transRadius());
       // BDT score classes
       registry.fill(HIST("DebugBdt/hBdtScore0"), d0d0bar.decayVtxX());
       registry.fill(HIST("DebugBdt/hBdtScore1"), d0d0bar.decayVtxY());
@@ -1388,7 +1390,7 @@ struct FemtoUniversePairTaskTrackD0 {
   {
     for (auto const& part : recoParts) {
       // filling the histograms for identified hadrons
-      if ((part.partType() == aod::femtouniverseparticle::ParticleType::kD0) && (part.pt() > ConfDmesons.confMinPtD0D0barReco) && (part.pt() < ConfDmesons.confMaxPtD0D0barReco)) {
+      if ((part.partType() == aod::femtouniverseparticle::ParticleType::kD0) && (part.pt() > ConfDmesons.confMinPtD0D0barReco) && (part.pt() < ConfDmesons.confMaxPtD0D0barReco) && (part.decayVtxZ() < ConfMlProb.confMlProbNonPromptMax)) {
         // getting the efficiency value
         if (doEfficiencyCorr) {
           weight = efficiencyCalculator.getWeight(ParticleNo::TWO, part.pt());