Skip to content

[PWGJE] add hasColl pass/fail diagnostics for event selections #18199

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
Oct 7, 2026
Merged

[PWGJE] add hasColl pass/fail diagnostics for event selections #18199

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
58 changes: 53 additions & 5 deletions PWGJE/Tasks/jetCrossSectionEfficiency.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@
#include <THnSparse.h>

#include <algorithm>
#include <array>
#include <cinttypes>
#include <cmath>
#include <cstddef>
Expand Down Expand Up @@ -151,6 +152,7 @@ struct JetCrossSectionEfficiency {
static constexpr float ConfigSwitchHigh = 9998.0f;
static constexpr float BrokenPtHardSentinel = 1.0f;
static constexpr int MinITSClustersForOccupancy = 5;
static constexpr int NumberOfHasCollSelectionConditionDiagnostics = 5;
static constexpr int HasCollQaMaxCount = 500;
static constexpr int HasCollQaMaxDeltaGlobalBC = 100000;
static constexpr int64_t NoOtherTruthCollision = -1;
Expand Down Expand Up @@ -370,6 +372,8 @@ struct JetCrossSectionEfficiency {

if (isSel8FullPbPb) {
AxisSpec axisPbPbTruthSelectionOnly = {PbPbTruthSelectionOnlyNBins, 0.5, static_cast<double>(PbPbTruthSelectionOnlyNBins) + 0.5, "event selection (PbPb truth-only)"};
AxisSpec selectionConditionAxis = {NumberOfHasCollSelectionConditionDiagnostics, 0.5, static_cast<double>(NumberOfHasCollSelectionConditionDiagnostics) + 0.5, "selection condition"};
AxisSpec hasCollSelectionStateAxis = {4, 0.5, 4.5, "hasColl and selection-condition state"};
std::vector<std::string> timeRangeOnlyLabels = {"INEL", "+RCT_pass", "+kTVX(truth)", "+kNoTFB(truth)", "+kNoITSROFB(truth)", "+kNoCollInTimeRangeStandard(truth)", "+hasColl", "+|zReco|<10", "+noSplit"};
std::vector<std::string> rofOnlyLabels = {"INEL", "+RCT_pass", "+kTVX(truth)", "+kNoTFB(truth)", "+kNoITSROFB(truth)", "+kNoCollInRofStandard(truth)", "+hasColl", "+|zReco|<10", "+noSplit"};

Expand All @@ -390,6 +394,19 @@ struct JetCrossSectionEfficiency {
{HistType::kTH1F, {axisPbPbTruthSelectionOnly}});
setAxisLabels(registry.get<TH2>(HIST("h2_jet_pt_part_eventselection_bcBitsFirst_rofOnly_truth"))->GetYaxis(), rofOnlyLabels);
setAxisLabels(registry.get<TH1>(HIST("h_mccollisions_eventselection_bcBitsFirst_rofOnly_truth"))->GetXaxis(), rofOnlyLabels);

registry.add("h2_mccollisions_selection_bit_vs_has_coll_state",
"weighted MC collisions after common RCT+TVX baseline with valid PbPb truth-proxy evaluation;selection condition;hasColl and selection-condition state",
{HistType::kTH2F, {selectionConditionAxis, hasCollSelectionStateAxis}});
registry.add("h2_mcpjets_selection_bit_vs_has_coll_state",
"weighted selected MCP jets after common RCT+TVX baseline with valid PbPb truth-proxy evaluation;selection condition;hasColl and selection-condition state",
{HistType::kTH2F, {selectionConditionAxis, hasCollSelectionStateAxis}});
std::vector<std::string> selectionConditionLabels = {"kNoTimeFrameBorder", "kNoITSROFrameBorder", "kNoSameBunchPileup (truth proxy)", "kNoCollInTimeRangeStandard (truth proxy)", "kNoCollInRofStandard (truth proxy)"};
std::vector<std::string> hasCollSelectionStateLabels = {"noColl, fail", "noColl, pass", "hasColl, fail", "hasColl, pass"};
setAxisLabels(registry.get<TH2>(HIST("h2_mccollisions_selection_bit_vs_has_coll_state"))->GetXaxis(), selectionConditionLabels);
setAxisLabels(registry.get<TH2>(HIST("h2_mccollisions_selection_bit_vs_has_coll_state"))->GetYaxis(), hasCollSelectionStateLabels);
setAxisLabels(registry.get<TH2>(HIST("h2_mcpjets_selection_bit_vs_has_coll_state"))->GetXaxis(), selectionConditionLabels);
setAxisLabels(registry.get<TH2>(HIST("h2_mcpjets_selection_bit_vs_has_coll_state"))->GetYaxis(), hasCollSelectionStateLabels);
}
}

Expand Down Expand Up @@ -1369,11 +1386,12 @@ struct JetCrossSectionEfficiency {
bool passesNoTFBTruth = truthBC.selection_bit(aod::evsel::kNoTimeFrameBorder);
bool passesNoITSROFBTruth = truthBC.selection_bit(aod::evsel::kNoITSROFrameBorder);

bool truthNoSBP = true;
if (applySBP) {
bool truthNoSBPDiagnostic = true;
if (applySBP || isSel8FullPbPb) {
auto sameBC = allMcCollisions.sliceBy(mcCollsPerBC, mccollision.bcId());
truthNoSBP = (sameBC.size() == 1);
truthNoSBPDiagnostic = (sameBC.size() == 1);
}
bool truthNoSBP = applySBP ? truthNoSBPDiagnostic : true;

bool hasRecoColl = (collisions.size() >= 1);
bool passesZvtxCutReco = false;
Expand All @@ -1398,6 +1416,38 @@ struct JetCrossSectionEfficiency {
}

bool passesRct = applyRCT ? (truthBC.rct_raw() & rctMask) == 0 : true;
float weight = mccollision.weight();

if (isSel8FullPbPb && passesRct && passesTVXTruth && truthPbPbSelections.valid) {
// The five selection conditions are evaluated independently on the same RCT+TVX
// baseline with valid PbPb truth-proxy evaluation.
const std::array<bool, NumberOfHasCollSelectionConditionDiagnostics> selectionConditionPasses = {
passesNoTFBTruth,
passesNoITSROFBTruth,
truthNoSBPDiagnostic,
truthPbPbSelections.noCollInTimeRangeStandard,
truthPbPbSelections.noCollInRofStandard};
for (size_t conditionIndex = 0; conditionIndex < selectionConditionPasses.size(); ++conditionIndex) {
const int hasCollSelectionState = (hasRecoColl ? 3 : 1) + (selectionConditionPasses[conditionIndex] ? 1 : 0);
registry.fill(HIST("h2_mccollisions_selection_bit_vs_has_coll_state"), static_cast<double>(conditionIndex + 1), hasCollSelectionState, weight);
}

float pTHatDiagnostic = computePtHat(mccollision);
if (pTHatDiagnostic >= pTHatAbsoluteMin) {
for (const auto& jet : jets) {
if (!jetfindingutilities::isInEtaAcceptance(jet, jetEtaMin, jetEtaMax, trackEtaMin, trackEtaMax) ||
jet.pt() < jetPtMin || jet.pt() > pTHatMaxMCP * pTHatDiagnostic ||
!isAcceptedJet<aod::JetParticles>(jet)) {
continue;
}
for (size_t conditionIndex = 0; conditionIndex < selectionConditionPasses.size(); ++conditionIndex) {
const int hasCollSelectionState = (hasRecoColl ? 3 : 1) + (selectionConditionPasses[conditionIndex] ? 1 : 0);
registry.fill(HIST("h2_mcpjets_selection_bit_vs_has_coll_state"), static_cast<double>(conditionIndex + 1), hasCollSelectionState, weight);
}
}
}
}

std::vector<bool> pass = {true, passesRct, passesTVXTruth,
applyTFB ? passesNoTFBTruth : true,
applyROFB ? passesNoITSROFBTruth : true};
Expand All @@ -1414,8 +1464,6 @@ struct JetCrossSectionEfficiency {
pass.push_back(passesZvtxCutReco);
pass.push_back(hasRecoColl && noSplitPass);

float weight = mccollision.weight();

int sMaxTimeRangeOnly = 0;
int sMaxRofOnly = 0;
if (isSel8FullPbPb) {
Expand Down
Loading