@@ -298,7 +298,7 @@ struct HfCorrelatorD0Hadrons {
298298 registry.add (" hMass1D" , " D0, D0bar candidates mass" , {HistType::kTH1F , {axisMassD}});
299299 registry.add (" hMassD01D" , " D0 candidates mass" , {HistType::kTH1F , {axisMassD}});
300300 registry.add (" hMassD0bar1D" , " D0bar candidates mass" , {HistType::kTH1F , {axisMassD}});
301- registry.add (" hMLScoresVsMassVsPt " , " D0, D0bar candidates massVsPt" , {HistType::kTHnSparseD , {{axisBdtScoreBkg}, {axisBdtScorePrompt}, {axisMassD}, {axisPtD}, {axisOrigin}}});
301+ registry.add (" hMLScoresVsMassVsPtVsOrigin " , " D0, D0bar candidates massVsPt" , {HistType::kTHnSparseD , {{axisBdtScoreBkg}, {axisBdtScorePrompt}, {axisMassD}, {axisPtD}, {axisOrigin}}});
302302 // Histograms for MC Reco
303303 registry.add (" hPtCandRec" , " D0, D0bar candidates - MC reco" , {HistType::kTH1F , {axisPtD}});
304304 registry.add (" hPtProng0Rec" , " D0, D0bar candidates prong 0 - MC reco" , {HistType::kTH1F , {axisPtD}});
@@ -407,7 +407,7 @@ struct HfCorrelatorD0Hadrons {
407407 for (unsigned int iclass = 0 ; iclass < classMl->size (); iclass++) {
408408 outputMlD0[iclass] = candidate.mlProbD0 ()[classMl->at (iclass)];
409409 }
410- registry.fill (HIST (" hMLScoresVsMassVsPt " ), outputMlD0[0 ], outputMlD0[2 ], hfHelper.invMassD0ToPiK (candidate), candidate.pt (), candidate.isSelD0bar () ? o2::aod::hf_correlation_d0_hadron::D0D0barBoth : o2::aod::hf_correlation_d0_hadron::D0Only);
410+ registry.fill (HIST (" hMLScoresVsMassVsPtVsOrigin " ), outputMlD0[0 ], outputMlD0[2 ], hfHelper.invMassD0ToPiK (candidate), candidate.pt (), candidate.isSelD0bar () ? o2::aod::hf_correlation_d0_hadron::D0D0barBoth : o2::aod::hf_correlation_d0_hadron::D0Only);
411411 }
412412 if (candidate.isSelD0bar () >= selectionFlagD0bar) {
413413 registry.fill (HIST (" hMass" ), hfHelper.invMassD0barToKPi (candidate), candidate.pt (), efficiencyWeight);
@@ -416,7 +416,7 @@ struct HfCorrelatorD0Hadrons {
416416 for (unsigned int iclass = 0 ; iclass < classMl->size (); iclass++) {
417417 outputMlD0bar[iclass] = candidate.mlProbD0bar ()[classMl->at (iclass)];
418418 }
419- registry.fill (HIST (" hMLScoresVsMassVsPt " ), outputMlD0bar[0 ], outputMlD0bar[2 ], hfHelper.invMassD0barToKPi (candidate), candidate.pt (), candidate.isSelD0 () ? o2::aod::hf_correlation_d0_hadron::D0D0barBoth : o2::aod::hf_correlation_d0_hadron::D0barOnly);
419+ registry.fill (HIST (" hMLScoresVsMassVsPtVsOrigin " ), outputMlD0bar[0 ], outputMlD0bar[2 ], hfHelper.invMassD0barToKPi (candidate), candidate.pt (), candidate.isSelD0 () ? o2::aod::hf_correlation_d0_hadron::D0D0barBoth : o2::aod::hf_correlation_d0_hadron::D0barOnly);
420420 }
421421 entryD0CandRecoInfo (hfHelper.invMassD0ToPiK (candidate), hfHelper.invMassD0barToKPi (candidate), candidate.pt (), outputMlD0[0 ], outputMlD0[2 ], outputMlD0bar[0 ], outputMlD0bar[2 ]);
422422
0 commit comments