Skip to content

[PWGJE] Add MC efficiency information to Ds jet task #18009

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

Merged
nzardosh merged 1 commit into from
Sep 22, 2026
+47 −20
Merged

[PWGJE] Add MC efficiency information to Ds jet task #18009

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
67 changes: 47 additions & 20 deletions PWGJE/Tasks/jetDsSpecSubs.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -52,10 +52,11 @@ consteval float getValFromBin(int bin)
enum BinExpColCntr { AllCollisions = 1,
Sel8ZCut = 2 };

enum BinMCColCntr { AllMCCollisions = 1,
SelectedMCCollisions = 2,
AssociatedRecoCollisions = 3,
SelectedAssociatedRecoCollisions = 4
enum BinMCColCntr {
AllMCCollisions = 1,
EventSelection = 2,
VertexZCut = 3,
HasMCPDsJet = 4
};

enum BinMCJetCntr {
Expand Down Expand Up @@ -318,13 +319,20 @@ struct JetDsSpecSubs {
// Detector-level sparse histograms
registry.add(
"hSparse_ds_mcd1",
";m_{D_{S}}^{rec};#it{p}_{T,D_{S}}^{det};#it{p}_{T,jet}^{det};z^{D_{S},jet}_{||,det};Origin(D_{S});Matching status",
";m_{D_{S}}^{rec};"
"#it{p}_{T,D_{S}}^{det};"
"#it{p}_{T,jet}^{det};"
"z^{D_{S},jet}_{||,det};"
"Origin(D_{S});"
"flagMcMatchRec;"
"Jet matching status",
{HistType::kTHnSparseF,
{{350, 1.6, 2.3},
{100, 0., 50.},
{100, 0., 100.},
{200, 0., 1.},
{3, -0.5, 2.5},
{21, -10.5, 10.5},
{2, -0.5, 1.5}}});

registry.add(
Expand Down Expand Up @@ -356,13 +364,20 @@ struct JetDsSpecSubs {
// Particle-level sparse with origin and matching status
registry.add(
"hSparse_ds_mcp",
";#it{p}_{T,D_{S}}^{part};#it{p}_{T,jet}^{part};z^{D_{S},jet}_{||,part};#DeltaR_{D_{S},jet}^{part};Origin(D_{S});Matching status",
";#it{p}_{T,D_{S}}^{part};"
"#it{p}_{T,jet}^{part};"
"z^{D_{S},jet}_{||,part};"
"#DeltaR_{D_{S},jet}^{part};"
"Origin(D_{S});"
"flagMcMatchGen;"
"Jet matching status",
{HistType::kTHnSparseF,
{{100, 0., 50.},
{100, 0., 100.},
{200, 0., 1.},
{200, 0., 1.},
{3, -0.5, 2.5},
{21, -10.5, 10.5},
{2, -0.5, 1.5}}});

// Matched detector-level Ds-tagged jets
Expand Down Expand Up @@ -418,10 +433,10 @@ struct JetDsSpecSubs {

// Counter labels
auto mcCollisionCounter = registry.get<TH1>(HIST("hMCColCounter"));
mcCollisionCounter->GetXaxis()->SetBinLabel(BinMCColCntr::AllMCCollisions, "All MC coll.");
mcCollisionCounter->GetXaxis()->SetBinLabel(BinMCColCntr::SelectedMCCollisions, "Selected MC coll.");
mcCollisionCounter->GetXaxis()->SetBinLabel(BinMCColCntr::AssociatedRecoCollisions, "MC-associated reco coll.");
mcCollisionCounter->GetXaxis()->SetBinLabel(BinMCColCntr::SelectedAssociatedRecoCollisions, "Selected MC-associated reco coll.");
mcCollisionCounter->GetXaxis()->SetBinLabel(BinMCColCntr::AllMCCollisions, "All MC collisions");
mcCollisionCounter->GetXaxis()->SetBinLabel(BinMCColCntr::EventSelection, "Event selection");
mcCollisionCounter->GetXaxis()->SetBinLabel(BinMCColCntr::VertexZCut, "|z_{vtx}| < cut");
mcCollisionCounter->GetXaxis()->SetBinLabel(BinMCColCntr::HasMCPDsJet, "Has MCP D_{s}-jet");

auto jetCounter = registry.get<TH1>(HIST("hMCJetCounter"));
jetCounter->GetXaxis()->SetBinLabel(BinMCJetCntr::MCPJets, "MCP Ds-jets");
Expand All @@ -432,12 +447,12 @@ struct JetDsSpecSubs {

// Matching-status labels
auto hSparseMCD = registry.get<THnSparse>(HIST("hSparse_ds_mcd1"));
hSparseMCD->GetAxis(5)->SetBinLabel(1, "Unmatched");
hSparseMCD->GetAxis(5)->SetBinLabel(2, "Matched");
hSparseMCD->GetAxis(6)->SetBinLabel(1, "Unmatched");
hSparseMCD->GetAxis(6)->SetBinLabel(2, "Matched");

auto hSparseMCP = registry.get<THnSparse>(HIST("hSparse_ds_mcp"));
hSparseMCP->GetAxis(5)->SetBinLabel(1, "Unmatched");
hSparseMCP->GetAxis(5)->SetBinLabel(2, "Matched");
hSparseMCP->GetAxis(6)->SetBinLabel(1, "Unmatched");
hSparseMCP->GetAxis(6)->SetBinLabel(2, "Matched");
}

void addMCPOnTheFlyHistograms()
Expand Down Expand Up @@ -755,17 +770,27 @@ struct JetDsSpecSubs {
// All MC collisions
registry.fill(HIST("hMCColCounter"), getValFromBin(BinMCColCntr::AllMCCollisions));

// MC collision selection
if (!jetderiveddatautilities::selectCollision(mccollision, eventSelectionBits) ||
!(std::abs(mccollision.posZ()) < vertexZCut)) {
// Event selection
if (!jetderiveddatautilities::selectCollision(mccollision, eventSelectionBits)) {
continue;
}
// Selected MC collisions
registry.fill(HIST("hMCColCounter"), getValFromBin(BinMCColCntr::SelectedMCCollisions));

registry.fill(HIST("hMCColCounter"), getValFromBin(BinMCColCntr::EventSelection));

// Vertex-z selection
if (std::abs(mccollision.posZ()) >= vertexZCut) {
continue;
}

registry.fill(HIST("hMCColCounter"), getValFromBin(BinMCColCntr::VertexZCut));

// Particle-level jets belonging to the current MC collision
const auto mcpJetsPerMCCollision = mcpjets.sliceBy(MCPJetsPerMCCollisionPreslice, mccollision.globalIndex());

if (mcpJetsPerMCCollision.size() > 0) {
registry.fill(HIST("hMCColCounter"), getValFromBin(BinMCColCntr::HasMCPDsJet));
}

// Particle-level Ds-tagged jets
for (const auto& mcpjet : mcpJetsPerMCCollision) {

Expand Down Expand Up @@ -806,6 +831,7 @@ struct JetDsSpecSubs {
zParallelMCP,
deltaRMCP,
originMCP,
flagMcMatchGen,
static_cast<float>(isMatchedMCP));

// Store all particle-level Ds-tagged jets
Expand Down Expand Up @@ -984,14 +1010,15 @@ struct JetDsSpecSubs {
registry.fill(HIST("h_ds_phi_mcd"), mcdcand.phi());
registry.fill(HIST("h_ds_mass_mcd"), mcdcand.m());

// Detector-level sparse: mass, pT Ds, pT jet, z_parallel, origin, matching status
// Detector-level sparse: mass, pT Ds, pT jet, z_parallel, origin, flagMcMatchRec, matching status
registry.fill(
HIST("hSparse_ds_mcd1"),
mcdcand.m(),
mcdcand.pt(),
mcdjet.pt(),
zParallelMCD,
originMCD,
flagMcMatchRec,
static_cast<float>(isMatchedMCD));

// Detector-level sparse: pT Ds, pT jet, DeltaR
Expand Down
Loading