From e92e3edf0afac9b07354b4ca4a1be14b1df2042e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?BiaoZhang=20=28=E5=BC=A0=E5=BD=AA=29?=
 <52267892+zhangbiao-phy@users.noreply.github.com>
Date: Mon, 16 Dec 2024 19:40:46 +0100
Subject: [PATCH] [PWGHF] Add the missed rapidity for D0 (#9010)

---
 PWGHF/D2H/Tasks/taskDirectedFlowCharmHadrons.cxx | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/PWGHF/D2H/Tasks/taskDirectedFlowCharmHadrons.cxx b/PWGHF/D2H/Tasks/taskDirectedFlowCharmHadrons.cxx
index 763a1292a52..aa6ba1525eb 100644
--- a/PWGHF/D2H/Tasks/taskDirectedFlowCharmHadrons.cxx
+++ b/PWGHF/D2H/Tasks/taskDirectedFlowCharmHadrons.cxx
@@ -243,6 +243,7 @@ struct HfTaskDirectedFlowCharmHadrons {
         switch (channel) {
           case DecayChannel::D0ToPiK:
             massCand = hfHelper.invMassD0ToPiK(candidate);
+            rapCand = hfHelper.yD0(candidate);
             if constexpr (std::is_same_v<T1, CandD0DataWMl>) {
               for (unsigned int iclass = 0; iclass < classMl->size(); iclass++)
                 outputMl[iclass] = candidate.mlProbD0()[classMl->at(iclass)];
@@ -250,6 +251,7 @@ struct HfTaskDirectedFlowCharmHadrons {
             break;
           case DecayChannel::D0ToKPi:
             massCand = hfHelper.invMassD0barToKPi(candidate);
+            rapCand = hfHelper.yD0(candidate);
             if constexpr (std::is_same_v<T1, CandD0DataWMl>) {
               for (unsigned int iclass = 0; iclass < classMl->size(); iclass++)
                 outputMl[iclass] = candidate.mlProbD0bar()[classMl->at(iclass)];