[PWGHF] Add the missed rapidity for D0 (AliceO2Group#9010)

ddobrigk · Dec 16, 2024 · e92e3ed · e92e3ed
1 parent a669a49
commit e92e3ed
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/PWGHF/D2H/Tasks/taskDirectedFlowCharmHadrons.cxx b/PWGHF/D2H/Tasks/taskDirectedFlowCharmHadrons.cxx
@@ -243,13 +243,15 @@ 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)];
             }
             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)];