Skip to content

Commit 8b53467

Browse files
kgwizdzikgwizdzi
authored
[PWGCF] FU: efficiency and D0 tasks update (#16293)
1 parent ebeef5c commit 8b53467

2 files changed

Lines changed: 36 additions & 18 deletions

File tree

‎PWGCF/FemtoUniverse/Tasks/femtoUniverseEfficiencyBase.cxx‎

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -660,6 +660,13 @@ struct FemtoUniverseEfficiencyBase {
660660
}
661661
registryCuts.fill(HIST("part1/cutsVspT"), part.pt(), 4);
662662

663+
// TOF Matching efficiency
664+
if (part.mLambda() == 1) {
665+
registryTOFMatch.fill(HIST("part1/hTofMatchPtAfterPID"), part.pt());
666+
} else {
667+
registryTOFMatch.fill(HIST("part1/hTpcPt"), part.pt());
668+
}
669+
663670
// Get the coresponding MC particle
664671
const auto mcParticle = part.fdMCParticle();
665672

@@ -696,12 +703,6 @@ struct FemtoUniverseEfficiencyBase {
696703
registryPDG.fill(HIST("part1/PDGvspT"), part.pt(), mcParticle.pdgMCTruth());
697704
registryMCOrigin.fill(HIST("part1/hRecoPt"), part.pt());
698705
registryMCOrigin.fill(HIST("part1/hTruthPt"), mcParticle.pt());
699-
// TOF Matching efficiency
700-
if (part.mLambda() == 1) {
701-
registryTOFMatch.fill(HIST("part1/hTofMatchPtAfterPID"), part.pt());
702-
} else {
703-
registryTOFMatch.fill(HIST("part1/hTpcPt"), part.pt());
704-
}
705706
}
706707

707708
if (!confIsSame) {
@@ -738,6 +739,13 @@ struct FemtoUniverseEfficiencyBase {
738739
}
739740
registryCuts.fill(HIST("part2/cutsVspT"), part.pt(), 4);
740741

742+
// TOF Matching efficiency
743+
if (part.mLambda() == 1) {
744+
registryTOFMatch.fill(HIST("part2/hTofMatchPtAfterPID"), part.pt());
745+
} else {
746+
registryTOFMatch.fill(HIST("part2/hTpcPt"), part.pt());
747+
}
748+
741749
// Get the coresponding MC particle
742750
const auto mcParticle = part.fdMCParticle();
743751

@@ -774,12 +782,6 @@ struct FemtoUniverseEfficiencyBase {
774782
registryPDG.fill(HIST("part2/PDGvspT"), part.pt(), mcParticle.pdgMCTruth());
775783
registryMCOrigin.fill(HIST("part2/hRecoPt"), part.pt());
776784
registryMCOrigin.fill(HIST("part2/hTruthPt"), mcParticle.pt());
777-
// TOF Matching efficiency
778-
if (part.mLambda() == 1) {
779-
registryTOFMatch.fill(HIST("part2/hTofMatchPtAfterPID"), part.pt());
780-
} else {
781-
registryTOFMatch.fill(HIST("part2/hTpcPt"), mcParticle.pt());
782-
}
783785
}
784786
}
785787
}

‎PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx‎

Lines changed: 22 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -531,9 +531,15 @@ struct FemtoUniversePairTaskTrackD0 {
531531
registry.add("DebugBdt/hBdtScore2", ";BDT score;Entries", {HistType::kTH1F, {axisBdtScore}});
532532
registry.add("DebugBdt/hBdtScore3", ";BDT score;Entries", {HistType::kTH1F, {axisBdtScore}});
533533
if (fillBDTvsPt) {
534-
registry.add("DebugBdtMcReco/hBdtScore1", ";BDT score;Entries", {HistType::kTH1F, {axisBdtScore}});
535-
registry.add("DebugBdtMcReco/hBdtScore2", ";BDT score;Entries", {HistType::kTH1F, {axisBdtScore}});
536-
registry.add("DebugBdtMcReco/hBdtScore3", ";BDT score;Entries", {HistType::kTH1F, {axisBdtScore}});
534+
registry.add("DebugBdtMcReco/hBdtScore1VsPt", ";BDT score;#it{p}_{T} (GeV/#it{c})", {HistType::kTH2F, {axisBdtScore, {vbins}}});
535+
registry.add("DebugBdtMcReco/hBdtScore2VsPt", ";BDT score;#it{p}_{T} (GeV/#it{c})", {HistType::kTH2F, {axisBdtScore, {vbins}}});
536+
registry.add("DebugBdtMcReco/hBdtScore3VsPt", ";BDT score;#it{p}_{T} (GeV/#it{c})", {HistType::kTH2F, {axisBdtScore, {vbins}}});
537+
registry.add("DebugBdtMcReco/hBdtScore1VsPtD0", ";BDT score;#it{p}_{T} (GeV/#it{c})", {HistType::kTH2F, {axisBdtScore, {vbins}}});
538+
registry.add("DebugBdtMcReco/hBdtScore2VsPtD0", ";BDT score;#it{p}_{T} (GeV/#it{c})", {HistType::kTH2F, {axisBdtScore, {vbins}}});
539+
registry.add("DebugBdtMcReco/hBdtScore3VsPtD0", ";BDT score;#it{p}_{T} (GeV/#it{c})", {HistType::kTH2F, {axisBdtScore, {vbins}}});
540+
registry.add("DebugBdtMcReco/hBdtScore1VsPtD0bar", ";BDT score;#it{p}_{T} (GeV/#it{c})", {HistType::kTH2F, {axisBdtScore, {vbins}}});
541+
registry.add("DebugBdtMcReco/hBdtScore2VsPtD0bar", ";BDT score;#it{p}_{T} (GeV/#it{c})", {HistType::kTH2F, {axisBdtScore, {vbins}}});
542+
registry.add("DebugBdtMcReco/hBdtScore3VsPtD0bar", ";BDT score;#it{p}_{T} (GeV/#it{c})", {HistType::kTH2F, {axisBdtScore, {vbins}}});
537543
}
538544
if (applyMLOpt) {
539545
registry.add("D0D0bar_MLSel/hMassVsPt1", "2-prong candidates;inv. mass (#pi K) (GeV/#it{c}^{2});entries", {HistType::kTH2F, {confInvMassBins, {vbins, "#it{p}_{T} (GeV/#it{c})"}}});
@@ -1466,14 +1472,19 @@ struct FemtoUniversePairTaskTrackD0 {
14661472
}
14671473
} else if ((part.partType() == aod::femtouniverseparticle::ParticleType::kD0) && (part.pt() > ConfDmesons.confMinPtD0D0barReco) && (part.pt() < ConfDmesons.confMaxPtD0D0barReco)) {
14681474
if (fillBDTvsPt && std::abs(mcpart.pdgMCTruth()) == o2::constants::physics::Pdg::kD0) {
1469-
registry.fill(HIST("DebugBdtMcReco/hBdtScore1"), part.decayVtxX());
1470-
registry.fill(HIST("DebugBdtMcReco/hBdtScore2"), part.decayVtxY());
1471-
registry.fill(HIST("DebugBdtMcReco/hBdtScore3"), part.decayVtxZ());
1475+
registry.fill(HIST("DebugBdtMcReco/hBdtScore1VsPt"), part.decayVtxX(), part.pt());
1476+
registry.fill(HIST("DebugBdtMcReco/hBdtScore2VsPt"), part.decayVtxY(), part.pt());
1477+
registry.fill(HIST("DebugBdtMcReco/hBdtScore3VsPt"), part.decayVtxZ(), part.pt());
14721478
}
14731479
if (mcpart.pdgMCTruth() == ConfDmesons.confPDGCodeD0) {
14741480
mcRecoRegistry.fill(HIST("hMcRecD0"), part.pt(), part.eta());
14751481
mcRecoRegistry.fill(HIST("hMcRecD0Pt"), part.pt());
14761482
mcRecoRegistry.fill(HIST("hMcRecD0Phi"), part.phi());
1483+
if (fillBDTvsPt) {
1484+
registry.fill(HIST("DebugBdtMcReco/hBdtScore1VsPtD0"), part.decayVtxX(), part.pt());
1485+
registry.fill(HIST("DebugBdtMcReco/hBdtScore2VsPtD0"), part.decayVtxY(), part.pt());
1486+
registry.fill(HIST("DebugBdtMcReco/hBdtScore3VsPtD0"), part.decayVtxZ(), part.pt());
1487+
}
14771488
if (part.tpcNClsFound() == 0) { // prompt candidates
14781489
mcRecoRegistry.fill(HIST("hMcRecD0Prompt"), part.pt(), part.eta());
14791490
mcRecoRegistry.fill(HIST("hMcRecD0PromptPt"), part.pt());
@@ -1487,6 +1498,11 @@ struct FemtoUniversePairTaskTrackD0 {
14871498
mcRecoRegistry.fill(HIST("hMcRecD0bar"), part.pt(), part.eta());
14881499
mcRecoRegistry.fill(HIST("hMcRecD0barPt"), part.pt());
14891500
mcRecoRegistry.fill(HIST("hMcRecD0barPhi"), part.phi());
1501+
if (fillBDTvsPt) {
1502+
registry.fill(HIST("DebugBdtMcReco/hBdtScore1VsPtD0bar"), part.decayVtxX(), part.pt());
1503+
registry.fill(HIST("DebugBdtMcReco/hBdtScore2VsPtD0bar"), part.decayVtxY(), part.pt());
1504+
registry.fill(HIST("DebugBdtMcReco/hBdtScore3VsPtD0bar"), part.decayVtxZ(), part.pt());
1505+
}
14901506
if (part.tpcNClsFound() == 0) { // prompt candidates
14911507
mcRecoRegistry.fill(HIST("hMcRecD0barPrompt"), part.pt(), part.eta());
14921508
mcRecoRegistry.fill(HIST("hMcRecD0barPromptPt"), part.pt());

0 commit comments

Comments
 (0)