Skip to content

Adding GenJets to LST with indexing #224

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
kk428 wants to merge 5 commits into
base: master
Choose a base branch
from
+345 −195
Open

Adding GenJets to LST with indexing #224

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
4e37108
Add jet branches to TrackingNtuple.cc
kk428 Jan 17, 2026
b5ba5a1
Updated how jet branches are read and indexed
kk428 Jan 19, 2026
a72526e
Fixed minor errors
kk428 Jan 19, 2026
f56bb1b
Adjusted indexing and plotting
kk428 Jan 20, 2026
9e5437a
Reverted ph2_clustSize to <uint16_t>
kk428 Jan 20, 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
59 changes: 40 additions & 19 deletions RecoTracker/LSTCore/standalone/code/core/write_lst_ntuple.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,9 @@ void fillOutputBranches(LSTEvent* event) {

unsigned int n_accepted_simtrk = setSimTrackContainerBranches(event);

if (ana.jet_branches)
setGenJetBranches(event);

if (ana.occ_branches)
setOccupancyBranches(event);

Expand Down Expand Up @@ -192,12 +195,13 @@ void createT4DNNBranches() {

//________________________________________________________________________________________________________________________________
void createJetBranches() {
ana.tx->createBranch<std::vector<float>>("sim_deltaEta");
ana.tx->createBranch<std::vector<float>>("sim_deltaPhi");
ana.tx->createBranch<std::vector<float>>("sim_deltaR");
ana.tx->createBranch<std::vector<float>>("sim_jet_eta");
ana.tx->createBranch<std::vector<float>>("sim_jet_phi");
ana.tx->createBranch<std::vector<float>>("sim_jet_pt");
ana.tx->createBranch<std::vector<float>>("sim_genjet_deltaEta");
ana.tx->createBranch<std::vector<float>>("sim_genjet_deltaPhi");
ana.tx->createBranch<std::vector<float>>("sim_genjet_deltaR");
ana.tx->createBranch<std::vector<int>>("sim_genjet_idx");
ana.tx->createBranch<std::vector<float>>("genjet_eta");
ana.tx->createBranch<std::vector<float>>("genjet_phi");
ana.tx->createBranch<std::vector<float>>("genjet_pt");
}

//________________________________________________________________________________________________________________________________
Expand Down Expand Up @@ -616,6 +620,27 @@ void createOccupancyBranches() {
ana.tx->createBranch<int>("pT5_occupancies");
}

//________________________________________________________________________________________________________________________________
void setGenJetBranches(LSTEvent* event) {
//--------------------------------------------
//
//
// Gen Jets
//
//
//--------------------------------------------

auto const& trk_genjet_eta = trk.getVF("genjet_eta");
auto const& trk_genjet_phi = trk.getVF("genjet_phi");
auto const& trk_genjet_pt = trk.getVF("genjet_pt");

for (unsigned int ijet = 0; ijet < trk_genjet_pt.size(); ++ijet) {
ana.tx->pushbackToBranch<float>("genjet_eta", trk_genjet_eta[ijet]);
ana.tx->pushbackToBranch<float>("genjet_phi", trk_genjet_phi[ijet]);
ana.tx->pushbackToBranch<float>("genjet_pt", trk_genjet_pt[ijet]);
}
}

//________________________________________________________________________________________________________________________________
unsigned int setSimTrackContainerBranches(LSTEvent* event) {
//--------------------------------------------
Expand Down Expand Up @@ -669,19 +694,15 @@ unsigned int setSimTrackContainerBranches(LSTEvent* event) {
// Now we have a list of "accepted" tracks (no condition on vtx_z/perp, nor pt, eta etc are applied yet)

if (ana.jet_branches) {
auto const& trk_sim_deltaEta = trk.getVF("sim_deltaEta");
auto const& trk_sim_deltaPhi = trk.getVF("sim_deltaPhi");
auto const& trk_sim_deltaR = trk.getVF("sim_deltaR");
auto const& trk_sim_jet_eta = trk.getVF("sim_jet_eta");
auto const& trk_sim_jet_phi = trk.getVF("sim_jet_phi");
auto const& trk_sim_jet_pt = trk.getVF("sim_jet_pt");

ana.tx->pushbackToBranch<float>("sim_deltaEta", trk_sim_deltaEta[isimtrk]);
ana.tx->pushbackToBranch<float>("sim_deltaPhi", trk_sim_deltaPhi[isimtrk]);
ana.tx->pushbackToBranch<float>("sim_deltaR", trk_sim_deltaR[isimtrk]);
ana.tx->pushbackToBranch<float>("sim_jet_eta", trk_sim_jet_eta[isimtrk]);
ana.tx->pushbackToBranch<float>("sim_jet_phi", trk_sim_jet_phi[isimtrk]);
ana.tx->pushbackToBranch<float>("sim_jet_pt", trk_sim_jet_pt[isimtrk]);
auto const& trk_sim_genjet_deltaEta = trk.getVF("sim_genjet_deltaEta");
auto const& trk_sim_genjet_deltaPhi = trk.getVF("sim_genjet_deltaPhi");
auto const& trk_sim_genjet_deltaR = trk.getVF("sim_genjet_deltaR");
auto const& trk_sim_genjet_idx = trk.getVI("sim_genjet_idx");

ana.tx->pushbackToBranch<float>("sim_genjet_deltaEta", trk_sim_genjet_deltaEta[isimtrk]);
ana.tx->pushbackToBranch<float>("sim_genjet_deltaPhi", trk_sim_genjet_deltaPhi[isimtrk]);
ana.tx->pushbackToBranch<float>("sim_genjet_deltaR", trk_sim_genjet_deltaR[isimtrk]);
ana.tx->pushbackToBranch<int>("sim_genjet_idx", trk_sim_genjet_idx[isimtrk]);
}

// Fill the branch with simulated tracks.
Expand Down
1 change: 1 addition & 0 deletions RecoTracker/LSTCore/standalone/code/core/write_lst_ntuple.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ void createOccupancyBranches();

void fillOutputBranches(LSTEvent* event);
void setOccupancyBranches(LSTEvent* event);
void setGenJetBranches(LSTEvent* event);
unsigned int setSimTrackContainerBranches(LSTEvent* event);
void setTrackCandidateBranches(LSTEvent* event,
unsigned int n_accepted_tracks,
Expand Down
20 changes: 12 additions & 8 deletions RecoTracker/LSTCore/standalone/efficiency/python/lst_plot_performance.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@

sel_choices = ["base", "loweta", "xtr", "vtr", "none"]
metric_choices = ["eff", "fakerate", "duplrate", "fakeorduplrate", "avgOTlen"]
variable_choices = ["pt", "ptmtv", "ptlow", "eta", "phi", "dxy", "dz", "vxy", "deltaEta", "deltaPhi", "deltaR", "jet_eta", "jet_phi", "jet_pt"]
variable_choices = ["pt", "ptmtv", "ptlow", "eta", "phi", "dxy", "dz", "vxy", "deltaEta", "deltaPhi", "deltaR", "genJetEta", "genJetPhi", "genJetPt"]
objecttype_choices = ["TC", "pT5", "T5", "pT3", "pLS", "T4", "pT5_lower", "pT3_lower", "T5_lower", "pLS_lower"]
#lowerObjectType = ["pT5_lower", "pT3_lower", "T5_lower"]

Expand Down Expand Up @@ -478,11 +478,11 @@ def set_label(eff, output_name, raw_number):
title = "#phi diffs"
elif "_deltaR" in output_name:
title = "#Delta R"
elif "_jet_eta" in output_name:
elif "_genJetEta" in output_name:
title = "jet #eta"
elif "_jet_phi" in output_name:
elif "_genJetPhi" in output_name:
title = "jet #phi"
elif "_jet_pt" in output_name:
elif "_genJetPt" in output_name:
title = "jet pT"
elif "_dz" in output_name:
title = "z [cm]"
Expand Down Expand Up @@ -735,7 +735,11 @@ def plot_standard_performance_plots(args):
"fakeorduplrate": ["pt", "ptlow", "ptmtv", "eta", "phi"],
"avgOTlen": ["eta"],
}
if (args.jet_branches): variables["eff"] = ["pt", "ptlow", "ptmtv", "eta", "phi", "dxy", "dz", "vxy", "deltaEta", "deltaPhi", "deltaR", "jet_eta", "jet_phi", "jet_pt"]
if (args.jet_branches):
variables["eff"] = ["pt", "ptlow", "ptmtv", "eta", "phi", "dxy", "dz", "vxy", "deltaEta", "deltaPhi", "deltaR", "genJetEta", "genJetPhi", "genJetPt"]
variables["duplrate"] = ["pt", "ptlow", "ptmtv", "eta", "phi", "deltaR"]
variables["fakerate"] = ["pt", "ptlow", "ptmtv", "eta", "phi", "deltaR"]
variables["fakeorduplrate"] = ["pt", "ptlow", "ptmtv", "eta", "phi", "deltaR"]
sels = {
"eff": ["base", "loweta"],
"fakerate": ["none"],
Expand All @@ -757,9 +761,9 @@ def plot_standard_performance_plots(args):
xcoarses["deltaEta"] = [False, True]
xcoarses["deltaPhi"] = [False, True]
xcoarses["deltaR"] = [False, True]
xcoarses["jet_eta"] = [False, True]
xcoarses["jet_phi"] = [False, True]
xcoarses["jet_pt"] = [False, True]
xcoarses["genJetEta"] = [False, True]
xcoarses["genJetPhi"] = [False, True]
xcoarses["genJetPt"] = [False, True]

types = objecttype_choices
breakdowns = {
Expand Down
Loading