Skip to content

Btag v2 #61

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
abolshov wants to merge 8 commits into
base: main
Choose a base branch
from
Open

Btag v2 #61

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
98298d0
start new branch
Jan 29, 2026
10bbf99
update DeepHME
Jan 29, 2026
9d88184
add everything in HH_bbWW
Jan 30, 2026
dcc0c6f
fix merge conflicts in HH_bbWW
Jan 30, 2026
8a443ba
improve lepton category descriptions in the config and add flag contr…
Jan 30, 2026
44016e5
add FLAF
Jan 30, 2026
85eec71
implement correction of btag shape weights
Jan 30, 2026
90683d1
update FLAF
Feb 1, 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
7 changes: 5 additions & 2 deletions Analysis/hh_bbww.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -77,14 +77,17 @@ def GetBTagWeight(global_cfg_dict, cat, applyBtag=False):
btagshape_weight = "weight_bTagShape_Central"
return f"{btag_weight}*{btagshape_weight}"


def GetWeight(channel, cat, boosted_categories): # do you need all these args?
def GetWeight(
channel, cat, boosted_categories, apply_btag_shape_weights=False
): # do you need all these args?
# weights_to_apply = ["weight_base", "ExtraDYWeight"]
weights_to_apply = ["weight_base"]
total_weight = "*".join(weights_to_apply)
for lep_index in [1, 2]:
total_weight = f"{total_weight} * {GetLepWeight(lep_index)}"
total_weight = f"{total_weight} * {GetTriggerWeight()}"
if apply_btag_shape_weights:
total_weight = f"{total_weight} * weight_bTagShape_Central"
return total_weight


Expand Down
87 changes: 86 additions & 1 deletion Analysis/histTupleDef.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,79 @@ def GetDfw(df, setup, dataset_name):


central_df_weights_computed = False
btag_shape_weight_corrected = False

cat_to_channelId = {"e": 1, "mu": 2, "eE": 11, "eMu": 12, "muMu": 22}


class BtagShapeWeightCorrector:
def __init__(self, btag_integral_ratios):
self.exisiting_srcScale_combs = [key for key in btag_integral_ratios.keys()]
# if the btag_integral_ratios dictionary is not empty, do stuff
if self.exisiting_srcScale_combs:
ROOT.gInterpreter.Declare("#include <map>")

for key in btag_integral_ratios.keys():
# key in btag_integral_ratios has form f"{source}_{scale}", so function expects that
# and creates a map and function to rescale btag weights for each f"{source}_{scale}" value
self._declare_cpp_map_and_resc_func(btag_integral_ratios, key)

def _declare_cpp_map_and_resc_func(self, btag_integral_ratios, unc_src_scale):
correction_factors = btag_integral_ratios[unc_src_scale]

# init c++ map
cpp_map_entries = []
for cat, multipl_dict in correction_factors.items():
channelId = cat_to_channelId[cat]
for key, ratio in multipl_dict.items():
# key has structure f"ratio_ncetnralJet_{number}""
num_jet = int(key.split("_")[-1])
cpp_map_entries.append(f"{{{{{channelId}, {num_jet}}}, {ratio}}}")
cpp_init = ", ".join(cpp_map_entries)

ROOT.gInterpreter.Declare(
f"""
static const std::map<std::pair<int, int>, float> ratios_{unc_src_scale} = {{
{cpp_init}
}};

float integral_correction_ratio_{unc_src_scale}(int ncentralJet, int channelId) {{
std::pair<int, int> key{{channelId, ncentralJet}};
try
{{
float ratio = ratios_{unc_src_scale}.at(key);
return ratio;
}}
catch (...)
{{
return 1.0f;
}}
}}"""
)

def UpdateBtagWeight(self, dfw, unc_src="Central", unc_scale=None):
# return original dfw if empty dict was passed to constructor
if not self.exisiting_srcScale_combs:
return dfw

if unc_scale is None:
unc_src_scale = unc_src
else:
unc_src_scale = f"{unc_src}_{unc_scale}"

if unc_src_scale not in self.exisiting_srcScale_combs:
raise RuntimeError(
f"`BtagShapeWeightCorrection.json` does not contain key `{unc_src_scale}`."
)

dfw.df = dfw.df.Redefine(
"weight_bTagShape_Central",
f"""if (ncentralJet >= 2 && ncentralJet <= 8)
return integral_correction_ratio_{unc_src_scale}(ncentralJet, channelId)*weight_bTagShape_Central;
return weight_bTagShape_Central;""",
)

return dfw

def DefineWeightForHistograms(
*,
Expand All @@ -55,6 +127,7 @@ def DefineWeightForHistograms(
global_params,
final_weight_name,
df_is_central,
btag_integral_ratios,
):
global central_df_weights_computed
is_central = uncName == central
Expand Down Expand Up @@ -87,12 +160,24 @@ def DefineWeightForHistograms(
if df_is_central:
central_df_weights_computed = True

# btag shape weight column appears here
correct_btagShape_weights = global_params.get("correct_btagShape_weights", False)
global btag_shape_weight_corrected
if correct_btagShape_weights and not btag_shape_weight_corrected and btag_integral_ratios:
isMC = not isData
if is_central and isMC:
weight_corrector = BtagShapeWeightCorrector(btag_integral_ratios)
print(f"Calling weight_corrector.UpdateBtagWeight for unc_source={uncName} unc_scale={uncScale}")
weight_corrector.UpdateBtagWeight(dfw, unc_src=uncName)
btag_shape_weight_corrected = True

categories = global_params["categories"]
boosted_categories = global_params.get("boosted_categories", [])
process_group = global_params["process_group"]
total_weight_expression = (
# channel, cat, boosted_categories --> these are not needed in the GetWeight function therefore I just put some placeholders
analysis.GetWeight("", "", boosted_categories)
# if btag shape weight was corrected => must be applied, else no
analysis.GetWeight("", "", boosted_categories, apply_btag_shape_weights=btag_shape_weight_corrected)
if process_group != "data"
else "1"
) # are we sure?
Expand Down
2 changes: 1 addition & 1 deletion Corrections
Open in desktop
Submodule Corrections updated 1 files
+1 −2 btagShape.h
2 changes: 1 addition & 1 deletion FLAF
Open in desktop
Submodule FLAF updated 10 files
+28 −52 AnaProd/tasks.py
+107 −26 Analysis/AnalysisCacheProducer.py
+74 −0 Analysis/BtagShapeProducer.py
+91 −0 Analysis/ComputeBtagShapeWeightCorrection.py
+15 −0 Analysis/HistTupleProducer.py
+264 −4 Analysis/tasks.py
+7 −0 Common/Setup.py
+1 −1 RunKit
+21 −9 include/GenLepton.h
+36 −0 run_tools/law_customizations.py
27 changes: 27 additions & 0 deletions config/global.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,8 @@ muIsoWP: "Loose"

treeName: "Events"

correct_btagShape_weights: True

nPbPerFile: 2_000 # 2fb-1 per split data file
nEventsPerFile: 100_000 # 100k events per MC file

Expand Down Expand Up @@ -256,6 +258,31 @@ payload_producers:
awkward_based: True
uproot_stepsize: '50MB'
dependencies:
BtagShape:
producers_module_name: BtagShapeProducer
producer_name: BtagShapeProducer
save_as: json
columns : []
lepton_categories:
e: 1
mu: 2
eE: 11
eMu: 12
muMu: 22
jet_multiplicities:
- 2
- 3
- 4
- 5
- 6
- 7
- 8
n_cpus: 4
max_runtime: 8.0
cmssw_env: False
awkward_based: True
uproot_stepsize: '50MB'
dependencies:

region: All
region_default: SR
Expand Down
Loading