Skip to content

Commit e92e3ed

Browse files
[PWGHF] Add the missed rapidity for D0 (AliceO2Group#9010)
1 parent a669a49 commit e92e3ed

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

PWGHF/D2H/Tasks/taskDirectedFlowCharmHadrons.cxx

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -243,13 +243,15 @@ struct HfTaskDirectedFlowCharmHadrons {
243243
switch (channel) {
244244
case DecayChannel::D0ToPiK:
245245
massCand = hfHelper.invMassD0ToPiK(candidate);
246+
rapCand = hfHelper.yD0(candidate);
246247
if constexpr (std::is_same_v<T1, CandD0DataWMl>) {
247248
for (unsigned int iclass = 0; iclass < classMl->size(); iclass++)
248249
outputMl[iclass] = candidate.mlProbD0()[classMl->at(iclass)];
249250
}
250251
break;
251252
case DecayChannel::D0ToKPi:
252253
massCand = hfHelper.invMassD0barToKPi(candidate);
254+
rapCand = hfHelper.yD0(candidate);
253255
if constexpr (std::is_same_v<T1, CandD0DataWMl>) {
254256
for (unsigned int iclass = 0; iclass < classMl->size(); iclass++)
255257
outputMl[iclass] = candidate.mlProbD0bar()[classMl->at(iclass)];

0 commit comments

Comments
 (0)