Skip to content
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

Jump to bottom

Fix lepton selection fields #46

Merged
merged 2 commits into from
Nov 15, 2024
Merged

Fix lepton selection fields #46

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
bfa38a5
only allow specific data samples to pass the different channel select…
nprouvost Nov 13, 2024
2dcec63
choose prettier names, remove changes for double counting
nprouvost Nov 14, 2024
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: 42 additions & 25 deletions hbt/selection/lepton.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,27 @@ def trigger_object_matching(
return any_match


def update_channel_ids(
events: ak.Array,
previous_channel_ids: ak.Array,
correct_channel_id: int,
channel_mask: ak.Array,
) -> ak.Array:
"""
Check if the events in the is_mask can be inside the given channel
or have already been sorted in another channel before.
"""
events_not_in_channel = (previous_channel_ids != 0) & (previous_channel_ids != correct_channel_id)
channel_id_overwrite = events_not_in_channel & channel_mask
if ak.any(channel_id_overwrite):
raise ValueError(
"The channel_ids of some events are being set to two different values. "
"The first event of this chunk concerned has index",
ak.where(channel_id_overwrite)[0],
)
return ak.where(channel_mask, correct_channel_id, previous_channel_ids)


@selector(
uses={
"Electron.pt", "Electron.eta", "Electron.phi", "Electron.dxy", "Electron.dz",
Expand Down Expand Up @@ -439,6 +460,7 @@ def lepton_selection(
)

# lepton pair selecton per trigger via lepton counting

if trigger.has_tag({"single_e", "cross_e_tau"}):
# expect 1 electron, 1 veto electron (the same one), 0 veto muons, and at least one tau
is_etau = (
Expand All @@ -454,14 +476,13 @@ def lepton_selection(
tau_charge = ak.firsts(events.Tau[tau_indices].charge, axis=1)
is_os = e_charge == -tau_charge
# store global variables
where = (channel_id == 0) & is_etau
channel_id = ak.where(where, ch_etau.id, channel_id)
tau2_isolated = ak.where(where, is_iso, tau2_isolated)
leptons_os = ak.where(where, is_os, leptons_os)
single_triggered = ak.where(where & is_single, True, single_triggered)
cross_triggered = ak.where(where & is_cross, True, cross_triggered)
sel_electron_indices = ak.where(where, electron_indices, sel_electron_indices)
sel_tau_indices = ak.where(where, tau_indices, sel_tau_indices)
channel_id = update_channel_ids(events, channel_id, ch_etau.id, is_etau)
tau2_isolated = ak.where(is_etau, is_iso, tau2_isolated)
leptons_os = ak.where(is_etau, is_os, leptons_os)
single_triggered = ak.where(is_etau & is_single, True, single_triggered)
cross_triggered = ak.where(is_etau & is_cross, True, cross_triggered)
sel_electron_indices = ak.where(is_etau, electron_indices, sel_electron_indices)
sel_tau_indices = ak.where(is_etau, tau_indices, sel_tau_indices)

elif trigger.has_tag({"single_mu", "cross_mu_tau"}):
# expect 1 muon, 1 veto muon (the same one), 0 veto electrons, and at least one tau
Expand All @@ -478,14 +499,13 @@ def lepton_selection(
tau_charge = ak.firsts(events.Tau[tau_indices].charge, axis=1)
is_os = mu_charge == -tau_charge
# store global variables
where = (channel_id == 0) & is_mutau
channel_id = ak.where(where, ch_mutau.id, channel_id)
tau2_isolated = ak.where(where, is_iso, tau2_isolated)
leptons_os = ak.where(where, is_os, leptons_os)
single_triggered = ak.where(where & is_single, True, single_triggered)
cross_triggered = ak.where(where & is_cross, True, cross_triggered)
sel_muon_indices = ak.where(where, muon_indices, sel_muon_indices)
sel_tau_indices = ak.where(where, tau_indices, sel_tau_indices)
channel_id = update_channel_ids(events, channel_id, ch_mutau.id, is_mutau)
tau2_isolated = ak.where(is_mutau, is_iso, tau2_isolated)
leptons_os = ak.where(is_mutau, is_os, leptons_os)
single_triggered = ak.where(is_mutau & is_single, True, single_triggered)
cross_triggered = ak.where(is_mutau & is_cross, True, cross_triggered)
sel_muon_indices = ak.where(is_mutau, muon_indices, sel_muon_indices)
sel_tau_indices = ak.where(is_mutau, tau_indices, sel_tau_indices)

elif trigger.has_tag({"cross_tau_tau", "cross_tau_tau_vbf", "cross_tau_tau_jet"}):
# expect 0 veto electrons, 0 veto muons and at least two taus of which one is isolated
Expand All @@ -496,12 +516,10 @@ def lepton_selection(
(ak.num(tau_indices, axis=1) >= 2) &
(ak.sum(tau_iso_mask, axis=1) >= 1)
)

# special case for cross tau vbf trigger:
# to avoid overlap, with non-vbf triggers, only one tau is allowed to have pt > 40
if trigger.has_tag("cross_tau_tau_vbf"):
is_tautau = is_tautau & (ak.sum(events.Tau[tau_indices].pt > 40, axis=1) <= 1)

is_iso = ak.sum(tau_iso_mask, axis=1) >= 2
# tau_indices are sorted by highest isolation as cond. 1 and highest pt as cond. 2, so
# the first two indices are exactly those selected by the full-blown pairing algorithm
Expand All @@ -511,13 +529,12 @@ def lepton_selection(
tau2_charge = ak.firsts(events.Tau[tau_indices].charge[..., 1:], axis=1)
is_os = tau1_charge == -tau2_charge
# store global variables
where = (channel_id == 0) & is_tautau
channel_id = ak.where(where, ch_tautau.id, channel_id)
tau2_isolated = ak.where(where, is_iso, tau2_isolated)
leptons_os = ak.where(where, is_os, leptons_os)
single_triggered = ak.where(where & is_single, True, single_triggered)
cross_triggered = ak.where(where & is_cross, True, cross_triggered)
sel_tau_indices = ak.where(where, tau_indices, sel_tau_indices)
channel_id = update_channel_ids(events, channel_id, ch_tautau.id, is_tautau)
tau2_isolated = ak.where(is_tautau, is_iso, tau2_isolated)
leptons_os = ak.where(is_tautau, is_os, leptons_os)
single_triggered = ak.where(is_tautau & is_single, True, single_triggered)
cross_triggered = ak.where(is_tautau & is_cross, True, cross_triggered)
sel_tau_indices = ak.where(is_tautau, tau_indices, sel_tau_indices)

# some final type conversions
channel_id = ak.values_astype(channel_id, np.uint8)
Expand Down
Loading