Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 17 additions & 0 deletions PWGHF/HFC/Tasks/taskCorrelationD0Hadrons.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -164,6 +164,12 @@ struct HfTaskCorrelationD0Hadrons {
AxisSpec axisCentFT0M = {binsCentFt0m, "Centrality percentile (FT0M)"};

// Histograms for data

registry.add("hBdtScorePrompt", "D0 BDT prompt score", {HistType::kTH1F, {axisBdtScore}});
registry.add("hBdtScoreBkg", "D0 BDT bkg score", {HistType::kTH1F, {axisBdtScore}});
registry.add("hMassD0VsPt", "D0 candidates massVsPt", {HistType::kTH2F, {{axisMassD}, {axisPtD}}});
registry.add("hMassD0VsPtWoEff", "D0 candidates massVsPt without efficiency", {HistType::kTH2F, {{axisMassD}, {axisPtD}}});

registry.add("hDeltaEtaPtIntSignalRegion", "D0-h deltaEta signal region", {HistType::kTH1F, {axisDeltaEta}});
registry.add("hDeltaPhiPtIntSignalRegion", "D0-h deltaPhi signal region", {HistType::kTH1F, {axisDeltaPhi}});
registry.add("hCorrel2DPtIntSignalRegion", "D0-h deltaPhi vs deltaEta signal region", {HistType::kTH2F, {{axisDeltaPhi}, {axisDeltaEta}}});
Expand Down Expand Up @@ -308,6 +314,7 @@ struct HfTaskCorrelationD0Hadrons {
aod::D0CandRecoInfo const& candidates)
{
for (const auto& candidate : candidates) {
float const massD = candidate.mD();
float const ptD = candidate.ptD();
float const bdtScorePromptD0 = candidate.mlScorePromptD0();
float const bdtScoreBkgD0 = candidate.mlScoreBkgD0();
Expand All @@ -324,6 +331,16 @@ struct HfTaskCorrelationD0Hadrons {
(bdtScorePromptD0bar < mlOutputPromptD0bar->at(effBinD) || bdtScoreBkgD0bar > mlOutputBkgD0bar->at(effBinD))) {
continue;
}

double efficiencyWeightD = 1.;
if (applyEfficiency != 0) {
efficiencyWeightD = 1. / efficiencyDmeson->at(o2::analysis::findBin(binsPtEfficiencyD, ptD));
}

registry.fill(HIST("hMassD0VsPt"), massD, ptD, efficiencyWeightD);
registry.fill(HIST("hMassD0VsPtWoEff"), massD, ptD);
registry.fill(HIST("hBdtScorePrompt"), bdtScorePromptD0);
registry.fill(HIST("hBdtScoreBkg"), bdtScoreBkgD0);
}

for (const auto& pairEntry : pairEntries) {
Expand Down
Loading