def __init__(self, weight_str, name=None):
if not name:
self.name = escape(weight_str)
else:
self.name = name
#Weight needs to be bracketed to avoid crazy results when mulitplying
self.weight_str = "(%s)" % weight_str
def plot_sherpa_vs_madgraph(var, cut_name, cut, samples, out_dir, recreate=False, **kwargs):
hname = var["varname"]
out_dir = out_dir + "/" + cut_name
if recreate and os.path.exists(out_dir):
logger.info("Output directory %s exists, removing" % out_dir)
shutil.rmtree(out_dir)
mkdir_p(out_dir)
logger.info("Using output directory %s" % out_dir)
logger.info("Using output directory %s" % out_dir)
coll = data_mc(var["var"], cut_name, cut, Weights.total()*Weights.mu, samples, out_dir, recreate, LUMI_TOTAL, reweight_madgraph=True, flavour_split=True, plot_range=var["range"], **kwargs)
logging.debug(str(coll.hists))
for hn, hist in coll.hists.items():
sample_name = coll.metadata[hn].sample_name
process_name = coll.metadata[hn].process_name
match = re.match(".*/cut__flavour__(W_[Hl][Hl])/.*", hn)
if match:
flavour_scenario = match.group(1)
else:
flavour_scenario = None
try:
if sample_types.is_mc(sample_name):
Styling.mc_style(hist, process_name)
else:
Styling.data_style(hist)
except KeyError as e:
logger.warning("Couldn't style histogram %s" % hn)
if flavour_scenario:
logger.debug("Matched flavour split histogram %s, %s" % (hn, flavour_scenario))
#Styling.mc_style(hist, process_name)
if re.match("W_H[lH]", flavour_scenario):
logger.debug("Changing colour of %s" % (hn))
hist.SetFillColor(hist.GetFillColor()+1)
hist.SetLineColor(hist.GetLineColor()+1)
logger.debug("pre merge: %s" % str([ (hn, coll.hists[hn].GetLineColor()) for hn in coll.hists.keys() if "sherpa" in hn]))
merges = dict()
merge_cmds = get_merge_cmds()
merge_cmds.pop("WJets")
merges["madgraph/unweighted"] = merge_cmds.copy()
merges["madgraph/weighted"] = merge_cmds.copy()
merges["sherpa/unweighted"] = merge_cmds.copy()
merges["sherpa/weighted"] = merge_cmds.copy()
merges["sherpa/unweighted"]["WJets_hf"] = ["weight__nominal/cut__flavour__W_heavy/WJets_sherpa_nominal"]
merges["sherpa/unweighted"]["WJets_lf"] = ["weight__nominal/cut__flavour__W_light/WJets_sherpa_nominal"]
merges["sherpa/weighted"]["WJets_hf"] = ["weight__sherpa_flavour/cut__flavour__W_heavy/WJets_sherpa_nominal"]
merges["sherpa/weighted"]["WJets_lf"] = ["weight__sherpa_flavour/cut__flavour__W_light/WJets_sherpa_nominal"]
merges["madgraph/unweighted"]["WJets_hf"] = ["weight__nominal/cut__flavour__W_heavy/W[1-4]Jets_exclusive"]
merges["madgraph/unweighted"]["WJets_lf"] = ["weight__nominal/cut__flavour__W_light/W[1-4]Jets_exclusive"]
merges["madgraph/weighted"]["WJets_hf"] = ["weight__reweight_madgraph/cut__flavour__W_heavy/W[1-4]Jets_exclusive"]
merges["madgraph/weighted"]["WJets_lf"] = ["weight__reweight_madgraph/cut__flavour__W_light/W[1-4]Jets_exclusive"]
hmerged = dict()
for k in merges.keys():
hmerged[k] = merge_hists(copy.deepcopy(coll.hists), merges[k])
logger.debug("post merge: %s" % str([ (hn, hmerged["sherpa/weighted"][hn].GetLineColor()) for hn in hmerged["sherpa/weighted"].keys()]))
#w_mg_sh = 1.0416259307303726 #sherpa to madgraph ratio
w_mg_sh = 1.0821535639376414
hmerged["sherpa/weighted"]["WJets_hf"].Scale(w_mg_sh)
hmerged["sherpa/weighted"]["WJets_lf"].Scale(w_mg_sh)
logger.info("Drawing madgraph unweighted plot")
canv = ROOT.TCanvas("c2", "c2")
suffix = "__%s__%s" % (var["var"], cut_name)
suffix = escape(suffix)
plot(canv, "madgraph_unw"+suffix, hmerged["madgraph/unweighted"], out_dir, **kwargs)
kwargs = dict({"x_label": var["varname"]}, **kwargs)
for k, v in hmerged.items():
logger.debug("Group %s" % k)
for hn, h in v.items():
logger.debug("Sample %s = %.2f" % (hn, h.Integral()))
logger.info("%s data=%.2f" % (k, v["data"].Integral()))
logger.info("%s MC=%.2f" % (k, sum([h.Integral() for k, h in v.items() if k!="data"])))
hists_flavours_merged = dict()
hists_flavours_merged["madgraph/weighted"] = merge_hists(hmerged["madgraph/weighted"], {"WJets": ["WJets_hf", "WJets_lf"]})
hists_flavours_merged["madgraph/unweighted"] = merge_hists(hmerged["madgraph/unweighted"], {"WJets": ["WJets_hf", "WJets_lf"]})
hists_flavours_merged["sherpa/unweighted"] = merge_hists(hmerged["sherpa/unweighted"], {"WJets": ["WJets_hf", "WJets_lf"]})
hists_flavours_merged["sherpa/weighted"] = merge_hists(hmerged["sherpa/weighted"], {"WJets": ["WJets_hf", "WJets_lf"]})
logger.info("Drawing sherpa weighted plot")
canv = ROOT.TCanvas("c1", "c1")
plot(canv, "sherpa_rew"+suffix, hmerged["sherpa/weighted"], out_dir, **kwargs)
logger.info("Drawing sherpa unweighted plot")
canv = ROOT.TCanvas("c1", "c1")
plot(canv, "sherpa_unw"+suffix, hmerged["sherpa/unweighted"], out_dir, **kwargs)
logger.info("Drawing madgraph plot")
canv = ROOT.TCanvas("c2", "c2")
plot(canv, "madgraph_rew"+suffix, hmerged["madgraph/weighted"], out_dir, **kwargs)
total_madgraph = copy.deepcopy(hmerged["madgraph/unweighted"])
merged_colls = dict()
for k, v in hmerged.items():
merged_colls[k] = HistCollection(copy.deepcopy(v), name=k)
logger.info("Drawing sherpa vs. madgraph shape comparison plots")
hists = [
("sherpa unw hf", hmerged["sherpa/unweighted"]["WJets_hf"]),
("sherpa rew hf", hmerged["sherpa/weighted"]["WJets_hf"]),
("madgraph unw hf", hmerged["madgraph/unweighted"]["WJets_hf"]),
("madgraph rew hf", hmerged["madgraph/weighted"]["WJets_hf"]),
]
hists = copy.deepcopy(hists)
for hn, h in hists:
h.SetTitle(hn + " %.2f" % h.Integral())
h.Scale(1.0/h.Integral())
hists = [h[1] for h in hists]
ColorStyleGen.style_hists(hists)
canv = plot_hists(hists, x_label=var["varname"], do_chi2=True)
leg = legend(hists, styles=["f", "f"], **kwargs)
canv.SaveAs(out_dir + "/weighted_flavour_hf_%s.png" % hname)
canv.Close()
hists = [
("data", hmerged["madgraph/unweighted"]["data"]),
("sherpa", hists_flavours_merged["sherpa/unweighted"]["WJets"]),
("madgraph", hists_flavours_merged["madgraph/unweighted"]["WJets"]),
]
hists = copy.deepcopy(hists)
for hn, h in hists:
h.SetTitle(hn + " %.2f" % h.Integral())
h.Scale(1.0/h.Integral())
hists = [h[1] for h in hists]
ColorStyleGen.style_hists(hists)
canv = plot_hists(hists, x_label=var["varname"], do_chi2=True)
leg = legend(hists, styles=["f", "f"], **kwargs)
canv.SaveAs(out_dir + "/unweighted_sherpa_mg_%s.png" % hname)
canv.Close()
hists = [
("data", hmerged["madgraph/unweighted"]["data"]),
("sherpa", hists_flavours_merged["sherpa/weighted"]["WJets"]),
("madgraph", hists_flavours_merged["madgraph/weighted"]["WJets"]),
]
hists = copy.deepcopy(hists)
for hn, h in hists:
h.SetTitle(hn + " %.2f" % h.Integral())
h.Scale(1.0/h.Integral())
hists = [h[1] for h in hists]
ColorStyleGen.style_hists(hists)
canv = plot_hists(hists, x_label=var["varname"], do_chi2=True)
leg = legend(hists, styles=["f", "f"], **kwargs)
canv.SaveAs(out_dir + "/weighted_sherpa_mg_%s.png" % hname)
canv.Close()
hists = [
("sherpa unw lf", hmerged["sherpa/unweighted"]["WJets_lf"]),
("sherpa rew lf", hmerged["sherpa/weighted"]["WJets_lf"]),
("madgraph unw lf", hmerged["madgraph/unweighted"]["WJets_lf"]),
("madgraph rew lf", hmerged["madgraph/weighted"]["WJets_lf"]),
]
hists = copy.deepcopy(hists)
for hn, h in hists:
h.SetTitle(hn + " %.2f" % h.Integral())
h.Scale(1.0/h.Integral())
hists = [h[1] for h in hists]
ColorStyleGen.style_hists(hists)
canv = plot_hists(hists, x_label=var["varname"], do_chi2=True)
leg = legend(hists, styles=["f", "f"], **kwargs)
canv.SaveAs(out_dir + "/weighted_flavour_lf_%s.png" % hname)
canv.Close()
hists = [
("data", hmerged["madgraph/unweighted"]["data"]),
("madgraph unw", hists_flavours_merged["madgraph/unweighted"]["WJets"]),
("madgraph rew", hists_flavours_merged["madgraph/weighted"]["WJets"]),
("sherpa unw", hists_flavours_merged["sherpa/unweighted"]["WJets"]),
("sherpa rew", hists_flavours_merged["sherpa/weighted"]["WJets"]),
]
hists = copy.deepcopy(hists)
for hn, h in hists:
h.SetTitle(hn + " %.2f" % h.Integral())
h.Scale(1.0/h.Integral())
hists = [h[1] for h in hists]
ColorStyleGen.style_hists(hists)
canv = plot_hists(hists, x_label=var["varname"], do_chi2=True)
leg = legend(hists, styles=["f", "f"], **kwargs)
hists[0].SetTitle("")
canv.Update()
canv.SaveAs(out_dir + "/shapes_%s.png" % hname)
canv.Close()
# hists = [
# ("sherpa hf", hmerged["sherpa"]["WJets_hf"]),
# ("madgraph unw hf", hmerged["madgraph/unweighted"]["WJets_hf"]),
# ("madgraph rew hf", hmerged["madgraph/weighted"]["WJets_hf"]),
# ]
# hists = copy.deepcopy(hists)
# for hn, h in hists:
# h.SetTitle(hn + " %.2f" % h.Integral())
# h.Scale(1.0/h.Integral())
# hists = [h[1] for h in hists]
# ColorStyleGen.style_hists(hists)
# canv = plot_hists(hists, x_label=var["varname"], do_chi2=True)
# leg = legend(hists, styles=["f", "f"], **kwargs)
# hists[0].SetTitle("madgraph sherpa rew hf")
# canv.SaveAs(out_dir + "/shapes_hf_%s.png" % hname)
# canv.Close()
return coll, merged_colls
