import ROOT as R
import logging
logger = logging.getLogger("")
logger.setLevel(logging.DEBUG)
formatter = logging.Formatter("%(name)s - %(levelname)s - %(message)s")
handler = logging.StreamHandler()
handler.setFormatter(formatter)
logger.addHandler(handler)
# Open file
era = 2016
channel = "tt"
category = "{}_m_sv".format(channel)
rootfile = rootfile_parser.Rootfile_parser("2016_shapes.root", "smhtt", "Run2016", "m_sv", 125)
# Make plot
plot = dd.Plot([], "ModTDR", r=0.04, l=0.15)
# Add backgrounds
bkg_processes = ["ZTT", "W", "ZL", "ZJ", "TTT", "TTL", "TTJ", "VVT", "VVL", "VVJ"]
for process in bkg_processes:
plot.add_hist(rootfile.get(channel, category, process), process, "bkg")
plot.setGraphStyle(process, "hist", fillcolor=styles.color_dict[process])
plot.create_stack(bkg_processes, "stack")
# Add data
plot.add_hist(rootfile.get(channel, category, "data_obs"), "data_obs", "data_obs")
plot.setGraphStyle(
"data_obs",
def main(info):
args = info["args"]
variable = info["variable"]
channel = info["channel"]
channel_dict = {
"ee": "#font[42]{#scale[0.85]{ee}}",
"em": "#scale[0.85]{e}#mu",
"et": "#font[42]{#scale[0.85]{e}}#tau_{#font[42]{h}}",
"mm": "#mu#mu",
"mt": "#mu#tau_{#font[42]{h}}",
"tt": "#tau_{#font[42]{h}}#tau_{#font[42]{h}}"
}
if args.linear == True:
split_value = 0.1
else:
if args.normalize_by_bin_width:
split_value = 10001
else:
split_value = 101
split_dict = {c: split_value for c in ["et", "mt", "tt", "em", "mm"]}
bkg_processes = [
"VVL", "TTL", "ZL", "jetFakesEMB", "EMB"
]
if not args.fake_factor and args.embedding:
bkg_processes = [
"QCDEMB", "VVL", "VVJ", "W", "TTL", "TTJ", "ZJ", "ZL", "EMB"
]
if not args.embedding and args.fake_factor:
bkg_processes = [
"VVT", "VVL", "TTT", "TTL", "ZL", "jetFakes", "ZTT"
]
if not args.embedding and not args.fake_factor:
bkg_processes = [
"QCD", "VVT", "VVL", "VVJ", "W", "TTT", "TTL", "TTJ", "ZJ", "ZL", "ZTT"
]
if args.draw_jet_fake_variation is not None:
bkg_processes = [
"VVL", "TTL", "ZL", "EMB"
]
if not args.fake_factor and args.embedding:
bkg_processes = [
"VVL", "VVJ", "W", "TTL", "TTJ", "ZJ", "ZL", "EMB"
]
if not args.embedding and args.fake_factor:
bkg_processes = [
"VVT", "VVL", "TTT", "TTL", "ZL", "ZTT"
]
if not args.embedding and not args.fake_factor:
bkg_processes = [
"VVT", "VVL", "VVJ", "W", "TTT", "TTL", "TTJ", "ZJ", "ZL", "ZTT"
]
all_bkg_processes = [b for b in bkg_processes]
legend_bkg_processes = copy.deepcopy(bkg_processes)
legend_bkg_processes.reverse()
if "2016" in args.era:
era = "Run2016"
elif "2017" in args.era:
era = "Run2017"
elif "2018" in args.era:
era = "Run2018"
else:
logger.critical("Era {} is not implemented.".format(args.era))
raise Exception
# category = "_".join([channel, variable])
# if args.category_postfix is not None:
# category += "_%s"%args.category_postfix
rootfile = rootfile_parser.Rootfile_parser(args.input, variable)
bkg_processes = [b for b in all_bkg_processes]
if "em" in channel:
if not args.embedding:
bkg_processes = [
"QCD", "VVT", "VVL", "W", "TTT", "TTL", "ZL", "ZTT"
]
if args.embedding:
bkg_processes = [
"QCDEMB", "VVL", "W", "TTL", "ZL", "EMB"
]
if args.draw_jet_fake_variation is not None:
if not args.embedding:
bkg_processes = [
"VVT", "VVL", "W", "TTT", "TTL", "ZL", "ZTT"
]
if args.embedding:
bkg_processes = [
"VVL", "W", "TTL", "ZL", "EMB"
]
if "mm" in channel:
bkg_processes = [
"QCD", "VVT", "VVL", "W", "TTT", "TTL", "ZTT", "ZL"
]
legend_bkg_processes = copy.deepcopy(bkg_processes)
legend_bkg_processes.reverse()
# create plot
width = 600
if args.linear == True:
plot = dd.Plot(
[0.3, [0.3, 0.28]], "ModTDR", r=0.04, l=0.14, width=width)
else:
plot = dd.Plot(
[0.5, [0.3, 0.28]], "ModTDR", r=0.04, l=0.14, width=width)
# get background histograms
total_bkg = None
if args.draw_jet_fake_variation is None:
stype = "Nominal"
else:
stype = args.draw_jet_fake_variation
for index,process in enumerate(bkg_processes):
if index == 0:
total_bkg = rootfile.get(channel, process, shape_type=stype).Clone()
else:
total_bkg.Add(rootfile.get(channel, process, shape_type=stype))
if process in ["jetFakesEMB", "jetFakes"] and channel == "tt":
total_bkg.Add(rootfile.get(channel, "wFakes", shape_type=stype))
jetfakes_hist = rootfile.get(channel, process, shape_type=stype)
jetfakes_hist.Add(
rootfile.get(channel, "wFakes", shape_type=stype))
plot.add_hist(
jetfakes_hist, process, "bkg")
else:
plot.add_hist(
rootfile.get(channel, process, shape_type=stype), process, "bkg")
plot.setGraphStyle(
process, "hist", fillcolor=styles.color_dict[process])
# if "mm" not in channel:
# # add VH, ttH & HWW to total bkg histogram
# total_bkg.Add(rootfile.get(channel, "VH125"))
# total_bkg.Add(rootfile.get(channel, "ttH125"))
# total_bkg.Add(rootfile.get(channel, "HWW"))
plot.add_hist(total_bkg, "total_bkg")
plot.setGraphStyle(
"total_bkg",
"e2",
markersize=0,
fillcolor=styles.color_dict["unc"],
linecolor=0)
plot.add_hist(rootfile.get(channel, "data", shape_type=stype), "data_obs")
data_norm = plot.subplot(0).get_hist("data_obs").Integral()
plot.subplot(0).get_hist("data_obs").GetXaxis().SetMaxDigits(4)
plot.subplot(0).setGraphStyle("data_obs", "e0")
plot.subplot(0).setGraphStyle("data_obs", "e0")
if args.linear:
pass
else:
plot.subplot(1).setGraphStyle("data_obs", "e0")
if "mm" not in channel:
# get signal histograms
plot_idx_to_add_signal = [0,2] if args.linear else [1,2]
for i in plot_idx_to_add_signal:
ggH = rootfile.get(channel, "ggH125").Clone()
qqH = rootfile.get(channel, "qqH125").Clone()
# VH = rootfile.get(channel, "VH125").Clone()
# ttH = rootfile.get(channel, "ttH125").Clone()
# HWW = rootfile.get(channel, "HWW").Clone()
if ggH.Integral() > 0:
ggH_scale = 10
else:
ggH_scale = 0.0
if qqH.Integral() > 0:
qqH_scale = 10
else:
qqH_scale = 0.0
# if VH.Integral() > 0:
# VH_scale = 10
# else:
# VH_scale = 0.0
# if ttH.Integral() > 0:
# ttH_scale = 10
# else:
# ttH_scale = 0.0
# if HWW.Integral() > 0:
# HWW_scale = 10
# else:
# HWW_scale = 0
if i in [0,1]:
ggH.Scale(ggH_scale)
qqH.Scale(qqH_scale)
# VH.Scale(VH_scale)
# ttH.Scale(ttH_scale)
# HWW.Scale(HWW_scale)
plot.subplot(i).add_hist(ggH, "ggH")
plot.subplot(i).add_hist(ggH, "ggH_top")
plot.subplot(i).add_hist(qqH, "qqH")
plot.subplot(i).add_hist(qqH, "qqH_top")
# plot.subplot(i).add_hist(VH, "VH")
# plot.subplot(i).add_hist(VH, "VH_top")
# plot.subplot(i).add_hist(ttH, "ttH")
# plot.subplot(i).add_hist(ttH, "ttH_top")
# plot.subplot(i).add_hist(HWW, "HWW")
# plot.subplot(i).add_hist(HWW, "HWW_top")
if "mm" not in channel:
plot.subplot(0 if args.linear else 1).setGraphStyle(
"ggH", "hist", linecolor=styles.color_dict["ggH"], linewidth=3)
plot.subplot(0 if args.linear else 1).setGraphStyle("ggH_top", "hist", linecolor=0)
plot.subplot(0 if args.linear else 1).setGraphStyle(
"qqH", "hist", linecolor=styles.color_dict["qqH"], linewidth=3)
plot.subplot(0 if args.linear else 1).setGraphStyle("qqH_top", "hist", linecolor=0)
# plot.subplot(0 if args.linear else 1).setGraphStyle(
# "VH", "hist", linecolor=styles.color_dict["VH"], linewidth=3)
# plot.subplot(0 if args.linear else 1).setGraphStyle("VH_top", "hist", linecolor=0)
# plot.subplot(0 if args.linear else 1).setGraphStyle(
# "ttH", "hist", linecolor=styles.color_dict["ttH"], linewidth=3)
# plot.subplot(0 if args.linear else 1).setGraphStyle("ttH_top", "hist", linecolor=0)
# plot.subplot(0 if args.linear else 1).setGraphStyle(
# "HWW", "hist", linecolor=styles.color_dict["HWW"], linewidth=3)
# plot.subplot(0 if args.linear else 1).setGraphStyle("HWW_top", "hist", linecolor=0)
# assemble ratio
if "mm" not in channel:
bkg_ggH = plot.subplot(2).get_hist("ggH")
bkg_qqH = plot.subplot(2).get_hist("qqH")
bkg_ggH.Add(plot.subplot(2).get_hist("total_bkg"))
bkg_qqH.Add(plot.subplot(2).get_hist("total_bkg"))
plot.subplot(2).add_hist(bkg_ggH, "bkg_ggH")
plot.subplot(2).add_hist(bkg_ggH, "bkg_ggH_top")
plot.subplot(2).add_hist(bkg_qqH, "bkg_qqH")
plot.subplot(2).add_hist(bkg_qqH, "bkg_qqH_top")
plot.subplot(2).setGraphStyle(
"bkg_ggH",
"hist",
linecolor=styles.color_dict["ggH"],
linewidth=3)
plot.subplot(2).setGraphStyle("bkg_ggH_top", "hist", linecolor=0)
plot.subplot(2).setGraphStyle(
"bkg_qqH",
"hist",
linecolor=styles.color_dict["qqH"],
linewidth=3)
plot.subplot(2).setGraphStyle("bkg_qqH_top", "hist", linecolor=0)
plot.subplot(2).normalize([
"total_bkg", "bkg_ggH", "bkg_ggH_top", "bkg_qqH",
"bkg_qqH_top", "data_obs"
], "total_bkg")
else:
plot.subplot(2).normalize(["total_bkg", "data_obs"], "total_bkg")
# stack background processes
plot.create_stack(bkg_processes, "stack")
# normalize stacks by bin-width
if args.normalize_by_bin_width:
plot.subplot(0).normalizeByBinWidth()
plot.subplot(1).normalizeByBinWidth()
# set axes limits and labels
plot.subplot(0).setYlims(
split_dict[channel],
max(1.6 * plot.subplot(0).get_hist("data_obs").GetMaximum(),
split_dict[channel] * 2))
log_quantities = ["ME_ggh", "ME_vbf", "ME_z2j_1", "ME_z2j_2", "ME_q2v1", "ME_q2v2", "ME_vbf_vs_ggh", "ME_ggh_vs_Z"]
if variable in log_quantities:
plot.subplot(0).setLogY()
plot.subplot(0).setYlims(
1.0,
1000 * plot.subplot(0).get_hist("data_obs").GetMaximum())
plot.subplot(2).setYlims(0.75, 1.45)
if channel == "mm":
plot.subplot(0).setLogY()
plot.subplot(0).setYlims(1, 10**10)
if args.linear != True:
plot.subplot(1).setYlims(0.1, split_dict[channel])
plot.subplot(1).setYlabel(
"") # otherwise number labels are not drawn on axis
plot.subplot(1).setLogY()
# Check if variables should be plotted with log x axis
log_x_variables = ["puppimet"]
if variable in log_x_variables:
plot.subplot(0).setLogX()
plot.subplot(1).setLogX()
plot.subplot(2).setLogX()
if variable != None:
if variable in styles.x_label_dict[channel]:
x_label = styles.x_label_dict[channel][
variable]
else:
x_label = variable
plot.subplot(2).setXlabel(x_label)
else:
plot.subplot(2).setXlabel("NN output")
if args.normalize_by_bin_width:
plot.subplot(0).setYlabel("dN/d(NN output)")
else:
plot.subplot(0).setYlabel("N_{events}")
plot.subplot(2).setYlabel("")
plot.subplot(2).setGrid()
plot.scaleYLabelSize(0.8)
plot.scaleYTitleOffset(1.1)
category = ""
if not channel == "tt" and category in ["11", "12", "13", "14", "15", "16"]:
plot.subplot(2).changeXLabels(["0.2", "0.4", "0.6", "0.8", "1.0"])
# draw subplots. Argument contains names of objects to be drawn in corresponding order.
if "mm" not in channel:
# procs_to_draw = ["stack", "total_bkg", "ggH", "ggH_top", "qqH", "qqH_top", "VH", "VH_top", "ttH", "ttH_top", "data_obs"] if args.linear else ["stack", "total_bkg", "data_obs"]
procs_to_draw = ["stack", "total_bkg", "ggH", "ggH_top", "qqH", "qqH_top", "data_obs"] if args.linear else ["stack", "total_bkg", "data_obs"]
if args.draw_jet_fake_variation is not None:
procs_to_draw = ["stack", "total_bkg", "data_obs"]
plot.subplot(0).Draw(procs_to_draw)
if args.linear != True:
# plot.subplot(1).Draw([
# "stack", "total_bkg", "ggH", "ggH_top", "qqH", "qqH_top",
# "VH", "VH_top", "ttH", "ttH_top", "HWW", "HWW_top", "data_obs"
# ])
plot.subplot(1).Draw([
"stack", "total_bkg", "ggH", "ggH_top", "qqH", "qqH_top",
"data_obs"
])
if args.draw_jet_fake_variation is None:
plot.subplot(2).Draw([
"total_bkg", "bkg_ggH", "bkg_ggH_top", "bkg_qqH",
"bkg_qqH_top", "data_obs"
])
else:
plot.subplot(2).Draw([
"total_bkg", "data_obs"
])
else:
procs_to_draw = ["stack", "total_bkg", "data_obs"] if args.linear else ["stack", "total_bkg", "data_obs"]
plot.subplot(0).Draw(procs_to_draw)
if args.linear != True:
plot.subplot(1).Draw([
"stack", "total_bkg", "data_obs"
])
plot.subplot(2).Draw([
"total_bkg", "data_obs"
])
# create legends
suffix = ["", "_top"]
for i in range(2):
plot.add_legend(width=0.6, height=0.15)
for process in legend_bkg_processes:
plot.legend(i).add_entry(
0, process, styles.legend_label_dict[process.replace("TTL", "TT").replace("VVL", "VV").replace("NLO","")], 'f')
plot.legend(i).add_entry(0, "total_bkg", "Bkg. stat. unc.", 'f')
if "mm" not in channel and args.draw_jet_fake_variation is None:
plot.legend(i).add_entry(0 if args.linear else 1, "ggH%s" % suffix[i], "%s #times gg#rightarrowH"%str(int(ggH_scale)), 'l')
plot.legend(i).add_entry(0 if args.linear else 1, "qqH%s" % suffix[i], "%s #times qq#rightarrowH"%str(int(qqH_scale)), 'l')
# plot.legend(i).add_entry(0 if args.linear else 1, "VH%s" % suffix[i], "%s #times V(lep)H"%str(int(VH_scale)), 'l')
# plot.legend(i).add_entry(0 if args.linear else 1, "ttH%s" % suffix[i], "%s #times ttH"%str(int(ttH_scale)), 'l')
# # plot.legend(i).add_entry(0 if args.linear else 1, "HWW%s" % suffix[i], "%s #times H#rightarrowWW"%str(int(HWW_scale)), 'l')
plot.legend(i).add_entry(0, "data_obs", "Observed", 'PE2L')
plot.legend(i).setNColumns(3)
plot.legend(0).Draw()
plot.legend(1).setAlpha(0.0)
plot.legend(1).Draw()
for i in range(2):
plot.add_legend(
reference_subplot=2, pos=1, width=0.6, height=0.03)
plot.legend(i + 2).add_entry(0, "data_obs", "Observed", 'PE2L')
if "mm" not in channel and args.draw_jet_fake_variation is None:
plot.legend(i + 2).add_entry(0 if args.linear else 1, "ggH%s" % suffix[i],
"ggH+bkg.", 'l')
plot.legend(i + 2).add_entry(0 if args.linear else 1, "qqH%s" % suffix[i],
"qqH+bkg.", 'l')
plot.legend(i + 2).add_entry(0, "total_bkg", "Bkg. stat. unc.", 'f')
plot.legend(i + 2).setNColumns(4)
plot.legend(2).Draw()
plot.legend(3).setAlpha(0.0)
plot.legend(3).Draw()
# draw additional labels
plot.DrawCMS()
if "2016" in args.era:
plot.DrawLumi("35.9 fb^{-1} (2016, 13 TeV)")
elif "2017" in args.era:
plot.DrawLumi("41.5 fb^{-1} (2017, 13 TeV)")
elif "2018" in args.era:
plot.DrawLumi("59.7 fb^{-1} (2018, 13 TeV)")
else:
logger.critical("Era {} is not implemented.".format(args.era))
raise Exception
posChannelCategoryLabelLeft = None
plot.DrawChannelCategoryLabel(
"%s, %s" % (channel_dict[channel], "inclusive"),
begin_left=posChannelCategoryLabelLeft)
# save plot
if not args.embedding and not args.fake_factor:
postfix = "fully_classic"
if args.embedding and not args.fake_factor:
postfix = "emb_classic"
if not args.embedding and args.fake_factor:
postfix = "classic_ff"
if args.embedding and args.fake_factor:
postfix = "emb_ff"
if args.draw_jet_fake_variation is not None:
postfix = postfix + "_" + args.draw_jet_fake_variation
if not os.path.exists("%s_plots_%s"%(args.era,postfix)):
os.mkdir("%s_plots_%s"%(args.era,postfix))
if not os.path.exists("%s_plots_%s/%s"%(args.era,postfix,channel)):
os.mkdir("%s_plots_%s/%s"%(args.era,postfix,channel))
print "Trying to save the created plot"
plot.save("%s_plots_%s/%s/%s_%s_%s.%s" % (args.era, postfix, channel, args.era, channel, variable, "pdf"))
plot.save("%s_plots_%s/%s/%s_%s_%s.%s" % (args.era, postfix, channel, args.era, channel, variable, "png"))
def main(args):
hack = []
roc = {}
shapes = {}
for channel in args.channels:
roc[channel] = {}
shapes[channel] = {}
for category in args.categories:
roc[channel][category] = {}
shapes[channel][category] = {}
for variable in args.variables:
for channel in args.channels:
for category in args.categories:
rootfile = rootfile_parser.Rootfile_parser(
"2016_shapes.root", "smhtt", "Run2016", variable, 125)
#print rootfile.list_contents()
name = "_".join([channel, category])
out_name = "_".join([channel, category, variable])
print name
# create canvas:
# First argument defines subplot structure: List of splits from top to bottom (max. 1.0 to min. 0.0). A split can be a single position or a pair resulting in gap.
# Further arguments set general style.
plot = dd.Plot([0.05], "ModTDR", r=0.04, l=0.14)
#bkg_processes = ["EWK", "QCD", "VV", "W", "TTT", "TTJ", "ZJ", "ZL", "ZTT"]
bkg_processes = [
"EWK", "QCD", "VV", "W", "TTT", "TTJ", "ZL", "ZJ", "ZTT"
]
if channel == 'tt':
bkg_processes = [
"QCD", "VVT", "VVJ", "W", "TTT", "TTJ", "ZL", "ZJ",
"ZTT"
]
# register histograms in the subplots (can be done globally or for specific subplots). regustered histograms are not necessarily plotted later.
for process in bkg_processes:
plot.add_hist(
rootfile.get(channel, name, process), process, "bkg"
) # get(channel, category, process) and assign specific name and group name to histogram. The group name is optional.
plot.setGraphStyle(process,
"hist",
fillcolor=styles.color_dict[process])
# for i in range(1):
# plot.add_hist(
# rootfile.get(channel, name, "ggh"), "ggh"
# ) # signal histograms are used twice in order to realize a two color line style
# plot.add_hist(
# rootfile.get(channel, name, "ggh"), "ggh_top")
# plot.add_hist(rootfile.get(channel, name, "qqH"), "qqH")
# plot.add_hist(
# rootfile.get(channel, name, "qqH"), "qqH_top")
plot.add_hist(rootfile.get(channel, name, "data_obs"),
"data_obs", "data_obs")
# set some graph styles
# plot.setGraphStyle(
# "ggh", "hist", linecolor=styles.color_dict["ggh"], linewidth=3)
# plot.setGraphStyle("ggh_top", "hist", linecolor=0)
# plot.setGraphStyle(
# "qqH", "hist", linecolor=styles.color_dict["qqH"], linewidth=3)
# plot.setGraphStyle("qqH_top", "hist", linecolor=0)
plot.setGraphStyle("data_obs",
"e0",
markersize=1,
fillcolor=styles.color_dict["unc"],
linecolor=1)
plot.create_stack(bkg_processes, "stack")
# plot.subplot(1).normalize(["data_obs"], bkg_processes) # would also work but add up the single bkg histograms in the background
if channel == 'tt':
plot.subplot(0).setYlims(1, 1e5)
plot.DrawChannelCategoryLabel("#tau_{h}#tau_{h}")
elif channel == 'mt':
plot.subplot(0).setYlims(1, 1e7)
plot.DrawChannelCategoryLabel("#mu#tau_{h}")
elif channel == 'et':
plot.subplot(0).setYlims(0.1, 1e7)
plot.DrawChannelCategoryLabel("e#tau_{h}")
# plot.subplot(0).setXlims(-200, 200)
# plot.subplot(1).setXlims(-200, )
plot.subplot(1).setYlims(0, 2)
plot.subplot(0).setLogY()
plot.subplot(0).setXlabel(variable)
plot.subplot(0).setYlabel("N_{events}")
plot.subplot(1).setYlabel("ratio to bkg")
plot.scaleXTitleSize(0.8)
plot.scaleXLabelSize(0.8)
plot.scaleYTitleSize(0.8)
plot.scaleYLabelSize(0.8)
plot.scaleXLabelOffset(2.0)
plot.scaleYTitleOffset(1.1)
plot.subplot(0).Draw([
'stack',
"data_obs",
"ggh",
"qqH",
"ggh_top",
"qqH_top",
])
# plot.subplot(1).add_hist(R.TF1("line", "1", 0, 1000), "line")
# plot.subplot(1).Draw(["data_obs", "line"])
# create legends
bkg_processes.reverse()
suffix = ["", "_top"]
for i in range(2):
plot.add_legend(width=0.5, height=0.08)
for process in bkg_processes:
plot.legend(i).add_entry(
0, process, styles.legend_label_dict[process], 'f')
#plot.legend(i).add_entry(1, "ggh%s" % suffix[i], "ggh", 'l')
#plot.legend(i).add_entry(1, "qqH%s" % suffix[i], "qqH", 'l')
# plot.legend(i).add_entry(0, "data_obs", "Data", 'PE')
plot.legend(i).setNColumns(3)
plot.legend(0).Draw()
plot.subplot(1)._pad.SetGrid()
# draw additional labels
plot.DrawCMS()
plot.DrawLumi("35.9 fb^{-1} (13 TeV)")
# save plot
plot.save(out_name + ".png")
plot.save(out_name + ".pdf")
hack.append(plot)