Skip to content

Commit 3b170e6

Browse files
adding a THnSpare to study the soft pion relative fraction
1 parent b27bfbd commit 3b170e6

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

PWGHF/D2H/Tasks/taskDstarToD0Pi.cxx

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ struct HfTaskDstarToD0Pi {
6363
Configurable<bool> studyD0ToPiKPi0{"studyD0ToPiKPi0", false, "Flag to study D*->D0(piKpi0)pi channel"};
6464
Configurable<bool> ptShapeStudy{"ptShapeStudy", false, "Flag to enable pT shape study"};
6565
Configurable<bool> useWeightOnline{"useWeightOnline", false, "Flag to enable use of weights for pT shape study online"};
66+
Configurable<bool> studySoftPiFraction{"studySoftPiFraction", false, "Flag to enable study of soft pion fraction, currently implemented for ML-based analysis only"};
6667

6768
// CCDB configuration
6869
Configurable<std::string> ccdbUrl{"ccdbUrl", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
@@ -397,6 +398,10 @@ struct HfTaskDstarToD0Pi {
397398
return;
398399
}
399400
}
401+
402+
if (studySoftPiFraction && doprocessDataWML) {
403+
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);
404+
}
400405
}
401406

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

480488
if (doprocessDataWoML) {

0 commit comments

Comments
 (0)