Skip to content

[PWGHF/D2H] adding a THnSpare to study the soft pion relative fraction #16049

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
deependra170598 wants to merge 2 commits into
base: master
Choose a base branch
from
+37 −29
Open

[PWGHF/D2H] adding a THnSpare to study the soft pion relative fraction #16049

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
3b170e6
adding a THnSpare to study the soft pion relative fraction
deependra170598 Apr 30, 2026
46daa17
O2 linter wants UpperCamelCase for the enum type name
deependra170598 Apr 30, 2026
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
66 changes: 37 additions & 29 deletions PWGHF/D2H/Tasks/taskDstarToD0Pi.cxx
View file
Open in desktop
Copy link
Copy Markdown
Collaborator

@vkucera vkucera Apr 30, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please fix the warning.

Copy link
Copy Markdown
Contributor Author

@deependra170598 deependra170598 Apr 30, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

could you please review it again?

Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@ struct HfTaskDstarToD0Pi {
Configurable<bool> studyD0ToPiKPi0{"studyD0ToPiKPi0", false, "Flag to study D*->D0(piKpi0)pi channel"};
Configurable<bool> ptShapeStudy{"ptShapeStudy", false, "Flag to enable pT shape study"};
Configurable<bool> useWeightOnline{"useWeightOnline", false, "Flag to enable use of weights for pT shape study online"};
Configurable<bool> studySoftPiFraction{"studySoftPiFraction", false, "Flag to enable study of soft pion fraction, currently implemented for ML-based analysis only"};

// CCDB configuration
Configurable<std::string> ccdbUrl{"ccdbUrl", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
Expand All @@ -82,7 +83,7 @@ struct HfTaskDstarToD0Pi {
std::vector<TH1D*> hWeights;
int const nWeights = 2; // prompt and non-prompt weights
std::vector<std::string> const weightHistNames = {"promptWeightVsPt", "nonPromptWeightVsPt"};
enum weightType {
enum WeightType {
Prompt = 0,
NonPrompt = 1
};
Expand Down Expand Up @@ -368,27 +369,27 @@ struct HfTaskDstarToD0Pi {
// Ensure hWeights is properly sized
hWeights.resize(nWeights); // prompt and non-prompt

hWeights[weightType::Prompt] = dynamic_cast<TH1D*>(weightFile->Get(weightHistNames[weightType::Prompt].c_str()));
hWeights[weightType::NonPrompt] = dynamic_cast<TH1D*>(weightFile->Get(weightHistNames[weightType::NonPrompt].c_str()));
if (hWeights[weightType::Prompt] == nullptr) {
LOGF(fatal, "Histogram %s not found in weight file!", weightHistNames[weightType::Prompt].c_str());
hWeights[WeightType::Prompt] = dynamic_cast<TH1D*>(weightFile->Get(weightHistNames[WeightType::Prompt].c_str()));
hWeights[WeightType::NonPrompt] = dynamic_cast<TH1D*>(weightFile->Get(weightHistNames[WeightType::NonPrompt].c_str()));
if (hWeights[WeightType::Prompt] == nullptr) {
LOGF(fatal, "Histogram %s not found in weight file!", weightHistNames[WeightType::Prompt].c_str());
return;
}
if (hWeights[weightType::NonPrompt] == nullptr) {
LOGF(fatal, "Histogram %s not found in weight file!", weightHistNames[weightType::NonPrompt].c_str());
if (hWeights[WeightType::NonPrompt] == nullptr) {
LOGF(fatal, "Histogram %s not found in weight file!", weightHistNames[WeightType::NonPrompt].c_str());
return;
}
// checking if bin wdith of weight histograms are not finner than pT axis of Dstar
if (hWeights[weightType::Prompt]->GetXaxis()->GetBinWidth(1) >= vecPtBins[1] - vecPtBins[0]) {
if (hWeights[WeightType::Prompt]->GetXaxis()->GetBinWidth(1) >= vecPtBins[1] - vecPtBins[0]) {
LOGF(fatal, "Bin width of weight histogram should be finer than pT axis of Dstar!");
return;
}
if (hWeights[weightType::NonPrompt]->GetXaxis()->GetBinWidth(1) >= vecPtBins[1] - vecPtBins[0]) {
if (hWeights[WeightType::NonPrompt]->GetXaxis()->GetBinWidth(1) >= vecPtBins[1] - vecPtBins[0]) {
LOGF(fatal, "Bin width of weight histogram should be finer than pT axis of Dstar!");
return;
}
hWeights[weightType::Prompt]->SetDirectory(nullptr);
hWeights[weightType::NonPrompt]->SetDirectory(nullptr);
hWeights[WeightType::Prompt]->SetDirectory(nullptr);
hWeights[WeightType::NonPrompt]->SetDirectory(nullptr);

weightFile->Close();
delete weightFile;
Expand All @@ -397,6 +398,10 @@ struct HfTaskDstarToD0Pi {
return;
}
}

if (studySoftPiFraction && doprocessDataWML) {
registry.add("SoftPiFraction/hPtSoftPiVsPtDtstarVsCentVsBDTScore", "Pt of Soft Pi vs Pt of D* vs Centrality vs BDT Score", {HistType::kTHnSparseF, {{100, 0.1, 1.0}, {vecPtBins, "#it{p}_{T} of Soft Pi (GeV/#it{c})"}, {axisCentrality}, {axisBDTScoreBackground}, {axisBDTScorePrompt}, {axisBDTScoreNonPrompt}}}, true);
}
}

// Comparator function to sort based on the second argument of a tuple
Expand Down Expand Up @@ -475,6 +480,9 @@ struct HfTaskDstarToD0Pi {
if constexpr (ApplyMl) {
auto mlBdtScore = candDstar.mlProbDstarToD0Pi();
registry.fill(HIST("Yield/hDeltaInvMassVsPtVsCentVsBDTScore"), deltaMDstar, candDstar.pt(), centrality, mlBdtScore[0], mlBdtScore[1], mlBdtScore[2], invD0);
if (studySoftPiFraction) {
registry.fill(HIST("SoftPiFraction/hPtSoftPiVsPtDtstarVsCentVsBDTScore"), candDstar.ptSoftPi(), candDstar.pt(), centrality, mlBdtScore[0], mlBdtScore[1], mlBdtScore[2]);
}
}

if (doprocessDataWoML) {
Expand Down Expand Up @@ -568,8 +576,8 @@ struct HfTaskDstarToD0Pi {
LOGF(fatal, "Weight histograms are not initialized or empty. Check CCDB path or weight file.");
return;
}
ptShapeWeightValues[weightType::Prompt] = hWeights[weightType::Prompt]->GetBinContent(hWeights[weightType::Prompt]->FindBin(ptDstarRecSig));
ptShapeWeightValues[weightType::NonPrompt] = hWeights[weightType::NonPrompt]->GetBinContent(hWeights[weightType::NonPrompt]->FindBin(ptMother));
ptShapeWeightValues[WeightType::Prompt] = hWeights[WeightType::Prompt]->GetBinContent(hWeights[WeightType::Prompt]->FindBin(ptDstarRecSig));
ptShapeWeightValues[WeightType::NonPrompt] = hWeights[WeightType::NonPrompt]->GetBinContent(hWeights[WeightType::NonPrompt]->FindBin(ptMother));
}

if (qaEnabled) {
Expand All @@ -581,9 +589,9 @@ struct HfTaskDstarToD0Pi {
auto bdtScore = candDstarMcRec.mlProbDstarToD0Pi();
registry.fill(HIST("Efficiency/hPtVsCentVsBDTScoreVsPvContribRecSig"), ptDstarRecSig, centrality, bdtScore[0], bdtScore[1], bdtScore[2], nPVContributors, weightValue);
if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::Prompt) {
registry.fill(HIST("Efficiency/hPtPromptVsCentVsBDTScorePvContribRecSig"), ptDstarRecSig, centrality, bdtScore[0], bdtScore[1], bdtScore[2], nPVContributors, ptShapeWeightValues[weightType::Prompt]);
registry.fill(HIST("Efficiency/hPtPromptVsCentVsBDTScorePvContribRecSig"), ptDstarRecSig, centrality, bdtScore[0], bdtScore[1], bdtScore[2], nPVContributors, ptShapeWeightValues[WeightType::Prompt]);
} else if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::NonPrompt) {
registry.fill(HIST("Efficiency/hPtNonPrompRectVsCentVsBDTScorePvContribRecSig"), ptDstarRecSig, centrality, bdtScore[0], bdtScore[1], bdtScore[2], nPVContributors, ptShapeWeightValues[weightType::NonPrompt]);
registry.fill(HIST("Efficiency/hPtNonPrompRectVsCentVsBDTScorePvContribRecSig"), ptDstarRecSig, centrality, bdtScore[0], bdtScore[1], bdtScore[2], nPVContributors, ptShapeWeightValues[WeightType::NonPrompt]);
}
if (ptShapeStudy && !useWeightOnline) {
if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::Prompt) {
Expand All @@ -596,9 +604,9 @@ struct HfTaskDstarToD0Pi {
auto bdtScore = candDstarMcRec.mlProbDstarToD0Pi();
registry.fill(HIST("Efficiency/hPtVsBDTScoreRecSig"), ptDstarRecSig, bdtScore[0], bdtScore[1], bdtScore[2], weightValue);
if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::Prompt) {
registry.fill(HIST("Efficiency/hPtPromptVsBDTScoreRecSig"), ptDstarRecSig, bdtScore[0], bdtScore[1], bdtScore[2], ptShapeWeightValues[weightType::Prompt]);
registry.fill(HIST("Efficiency/hPtPromptVsBDTScoreRecSig"), ptDstarRecSig, bdtScore[0], bdtScore[1], bdtScore[2], ptShapeWeightValues[WeightType::Prompt]);
} else if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::NonPrompt) {
registry.fill(HIST("Efficiency/hPtNonPromptVsBDTScoreRecSig"), ptDstarRecSig, bdtScore[0], bdtScore[1], bdtScore[2], ptShapeWeightValues[weightType::NonPrompt]);
registry.fill(HIST("Efficiency/hPtNonPromptVsBDTScoreRecSig"), ptDstarRecSig, bdtScore[0], bdtScore[1], bdtScore[2], ptShapeWeightValues[WeightType::NonPrompt]);
}
if (ptShapeStudy && !useWeightOnline) {
if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::Prompt) {
Expand All @@ -612,9 +620,9 @@ struct HfTaskDstarToD0Pi {
if (isCentStudy) {
registry.fill(HIST("Efficiency/hPtVsCentVsPvContribRecSig"), ptDstarRecSig, centrality, nPVContributors, weightValue);
if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::Prompt) {
registry.fill(HIST("Efficiency/hPtPromptVsCentVsPvContribRecSig"), ptDstarRecSig, centrality, nPVContributors, ptShapeWeightValues[weightType::Prompt]);
registry.fill(HIST("Efficiency/hPtPromptVsCentVsPvContribRecSig"), ptDstarRecSig, centrality, nPVContributors, ptShapeWeightValues[WeightType::Prompt]);
} else if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::NonPrompt) {
registry.fill(HIST("Efficiency/hPtNonPromptVsCentVsPvContribRecSig"), ptDstarRecSig, centrality, nPVContributors, ptShapeWeightValues[weightType::NonPrompt]);
registry.fill(HIST("Efficiency/hPtNonPromptVsCentVsPvContribRecSig"), ptDstarRecSig, centrality, nPVContributors, ptShapeWeightValues[WeightType::NonPrompt]);
}
if (ptShapeStudy && !useWeightOnline) {
if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::Prompt) {
Expand All @@ -626,9 +634,9 @@ struct HfTaskDstarToD0Pi {
} else {
registry.fill(HIST("Efficiency/hPtVsPvContribRecSig"), ptDstarRecSig, nPVContributors, weightValue);
if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::Prompt) {
registry.fill(HIST("Efficiency/hPtPromptVsPvContribRecSig"), ptDstarRecSig, nPVContributors, ptShapeWeightValues[weightType::Prompt]);
registry.fill(HIST("Efficiency/hPtPromptVsPvContribRecSig"), ptDstarRecSig, nPVContributors, ptShapeWeightValues[WeightType::Prompt]);
} else if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::NonPrompt) {
registry.fill(HIST("Efficiency/hPtNonPromptVsPvContribRecSig"), ptDstarRecSig, nPVContributors, ptShapeWeightValues[weightType::NonPrompt]);
registry.fill(HIST("Efficiency/hPtNonPromptVsPvContribRecSig"), ptDstarRecSig, nPVContributors, ptShapeWeightValues[WeightType::NonPrompt]);
}
if (ptShapeStudy && !useWeightOnline) {
if (candDstarMcRec.originMcRec() == RecoDecay::OriginType::Prompt) {
Expand Down Expand Up @@ -782,8 +790,8 @@ struct HfTaskDstarToD0Pi {
LOGF(fatal, "Weight histograms are not initialized or empty. Check CCDB path or weight file.");
return;
}
ptShapeWeightValues[weightType::Prompt] = hWeights[weightType::Prompt]->GetBinContent(hWeights[weightType::Prompt]->FindBin(ptGen));
ptShapeWeightValues[weightType::NonPrompt] = hWeights[weightType::NonPrompt]->GetBinContent(hWeights[weightType::NonPrompt]->FindBin(ptBMother));
ptShapeWeightValues[WeightType::Prompt] = hWeights[WeightType::Prompt]->GetBinContent(hWeights[WeightType::Prompt]->FindBin(ptGen));
ptShapeWeightValues[WeightType::NonPrompt] = hWeights[WeightType::NonPrompt]->GetBinContent(hWeights[WeightType::NonPrompt]->FindBin(ptBMother));
}

registry.fill(HIST("Efficiency/hPtVsYDstarGen"), ptGen, yGen, weightValue);
Expand All @@ -798,20 +806,20 @@ struct HfTaskDstarToD0Pi {

// Prompt
if (mcParticle.originMcGen() == RecoDecay::OriginType::Prompt) {
registry.fill(HIST("Efficiency/hPtVsYPromptDstarGen"), ptGen, yGen, ptShapeWeightValues[weightType::Prompt]);
registry.fill(HIST("Efficiency/hPtVsYPromptDstarGen"), ptGen, yGen, ptShapeWeightValues[WeightType::Prompt]);
if (isCentStudy) {
registry.fill(HIST("Efficiency/hPtPromptVsCentVsPvContribGen"), ptGen, centFT0MGen, pvContributors, ptShapeWeightValues[weightType::Prompt]);
registry.fill(HIST("Efficiency/hPtPromptVsCentVsPvContribGen"), ptGen, centFT0MGen, pvContributors, ptShapeWeightValues[WeightType::Prompt]);
} else {
registry.fill(HIST("Efficiency/hPtPromptVsGen"), ptGen, ptShapeWeightValues[weightType::Prompt]);
registry.fill(HIST("Efficiency/hPtPromptVsGen"), ptGen, ptShapeWeightValues[WeightType::Prompt]);
}
// Non-Prompt
} else if (mcParticle.originMcGen() == RecoDecay::OriginType::NonPrompt) {
registry.fill(HIST("Efficiency/hPtVsYNonPromptDstarGen"), ptGen, yGen, ptShapeWeightValues[weightType::NonPrompt]);
registry.fill(HIST("Efficiency/hPtVsYNonPromptDstarGen"), ptGen, yGen, ptShapeWeightValues[WeightType::NonPrompt]);
if (isCentStudy) {
registry.fill(HIST("Efficiency/hPtNonPromptVsCentVsPvContribGen"), ptGen, centFT0MGen, pvContributors, ptShapeWeightValues[weightType::NonPrompt]);
registry.fill(HIST("Efficiency/hPtNonPromptVsCentVsPvContribGen"), ptGen, centFT0MGen, pvContributors, ptShapeWeightValues[WeightType::NonPrompt]);

} else {
registry.fill(HIST("Efficiency/hPtNonPromptVsGen"), ptGen, ptShapeWeightValues[weightType::NonPrompt]);
registry.fill(HIST("Efficiency/hPtNonPromptVsGen"), ptGen, ptShapeWeightValues[WeightType::NonPrompt]);
}
}
if (ptShapeStudy && !useWeightOnline) {
Expand Down
Loading