def compare_flavour_fraction_hists_vs_pt_from_contribs( contribs, flav, output_filename, title="", xtitle="p_{T}^{jet} [GeV]", **plot_kwargs): """Plot a specified flavour fraction vs pT for several sources. TODO: use this one more often - compare_flavour_fractions_vs_pt() is almost identical but has to deal with npartons """ flav_str = FLAV_STR_DICT[flav] ytitle = "Fraction of %s %ss" % (flav_str.lower(), get_jet_str('')) p = Plot(contribs, what='graph', xtitle=xtitle, ytitle=ytitle, title=title, xlim=(50, 2000), ylim=(0, 1), has_data=False, **plot_kwargs) p.default_canvas_size = (600, 600) try: p.plot("AP") p.main_pad.SetBottomMargin(0.16) p.get_modifier().GetXaxis().SetTitleOffset(1.4) p.get_modifier().GetXaxis().SetTitleSize(.045) p.legend.SetX1(0.56) p.legend.SetY1(0.65) p.legend.SetY2(0.87) if len(contribs) >= 4: p.legend.SetY1(0.7) p.legend.SetX1(0.5) p.legend.SetNColumns(2) p.set_logx(do_more_labels=True, do_exponent=False) p.save(output_filename) except ZeroContributions: warnings.warn("No contributions for %s" % output_filename)
def do_flavour_fraction_vs_eta(input_file, dirname, eta_bins, output_filename, title="", var_prepend="", which_jet="both", append=""): """Plot all flavour fractions vs eta for one input file & dirname in the ROOT file""" info = get_flavour_efficiencies( input_file, bins=eta_bins, hist_name=get_flavour_hist_name( dirname=dirname, var_prepend=var_prepend, which_jet="both", # since no jet/jet1/jet2 in hist name metric='eta' + append)) leg_draw_opt = "LP" plot_u = Contribution(info['u'].CreateGraph(), label="Up", line_color=ROOT.kRed, marker_color=ROOT.kRed, marker_style=20, leg_draw_opt=leg_draw_opt) plot_d = Contribution(info['d'].CreateGraph(), label="Down", line_color=ROOT.kBlue, marker_color=ROOT.kBlue, marker_style=21, leg_draw_opt=leg_draw_opt) plot_s = Contribution(info['s'].CreateGraph(), label="Strange", line_color=ROOT.kBlack, marker_color=ROOT.kBlack, marker_style=22, leg_draw_opt=leg_draw_opt) plot_c = Contribution(info['c'].CreateGraph(), label="Charm", line_color=ROOT.kGreen + 2, marker_color=ROOT.kGreen + 2, marker_style=23, leg_draw_opt=leg_draw_opt) plot_b = Contribution(info['b'].CreateGraph(), label="Bottom", line_color=ROOT.kOrange - 3, marker_color=ROOT.kOrange - 3, marker_style=33, leg_draw_opt=leg_draw_opt) plot_g = Contribution(info['g'].CreateGraph(), label="Gluon", line_color=ROOT.kViolet, marker_color=ROOT.kViolet, marker_style=29, leg_draw_opt=leg_draw_opt) plot_unknown = Contribution(info['unknown'].CreateGraph(), label="Unknown", line_color=ROOT.kGray + 1, marker_color=ROOT.kGray + 1, marker_style=26, leg_draw_opt=leg_draw_opt) p_flav = Plot( [plot_d, plot_u, plot_s, plot_c, plot_b, plot_g, plot_unknown], what='graph', xtitle="y^{%s}" % get_jet_str(''), ytitle="Fraction", title=title, xlim=(eta_bins[0][0], eta_bins[-1][1]), ylim=[0, 1], has_data=False) p_flav.default_canvas_size = (600, 600) p_flav.plot("AP") p_flav.main_pad.SetBottomMargin(0.16) p_flav.get_modifier().GetXaxis().SetTitleOffset(1.4) p_flav.get_modifier().GetXaxis().SetTitleSize(.045) p_flav.legend.SetX1(0.55) p_flav.legend.SetY1(0.72) p_flav.legend.SetY2(0.85) p_flav.legend.SetNColumns(2) p_flav.save(output_filename)
def do_flavour_fraction_vs_pt(input_file, hist_name, pt_bins, output_filename, title=""): """Plot all flavour fractions vs PT for one input file & hist_name in the ROOT file""" info = get_flavour_efficiencies(input_file, bins=pt_bins, hist_name=hist_name) leg_draw_opt = "LP" plot_u = Contribution(info['u'].CreateGraph(), label="Up", line_color=ROOT.kRed, marker_color=ROOT.kRed, marker_style=20, leg_draw_opt=leg_draw_opt) plot_d = Contribution(info['d'].CreateGraph(), label="Down", line_color=ROOT.kBlue, marker_color=ROOT.kBlue, marker_style=21, leg_draw_opt=leg_draw_opt) plot_s = Contribution(info['s'].CreateGraph(), label="Strange", line_color=ROOT.kBlack, marker_color=ROOT.kBlack, marker_style=22, leg_draw_opt=leg_draw_opt) plot_c = Contribution(info['c'].CreateGraph(), label="Charm", line_color=ROOT.kGreen + 2, marker_color=ROOT.kGreen + 2, marker_style=23, leg_draw_opt=leg_draw_opt) plot_b = Contribution(info['b'].CreateGraph(), label="Bottom", line_color=ROOT.kOrange - 3, marker_color=ROOT.kOrange - 3, marker_style=33, leg_draw_opt=leg_draw_opt) plot_g = Contribution(info['g'].CreateGraph(), label="Gluon", line_color=ROOT.kViolet, marker_color=ROOT.kViolet, marker_style=29, leg_draw_opt=leg_draw_opt) plot_unknown = Contribution(info['unknown'].CreateGraph(), label="Unknown", line_color=ROOT.kGray + 1, marker_color=ROOT.kGray + 1, marker_style=26, leg_draw_opt=leg_draw_opt) p_flav = Plot( [plot_d, plot_u, plot_s, plot_c, plot_b, plot_g, plot_unknown], what='graph', xtitle="p_{T}^{%s} [GeV]" % get_jet_str(''), ytitle="Fraction", title=title, xlim=(pt_bins[0][0], pt_bins[-1][1]), ylim=[0, 1], has_data=False) p_flav.default_canvas_size = (600, 600) p_flav.plot("AP") p_flav.main_pad.SetBottomMargin(0.16) p_flav.get_modifier().GetXaxis().SetTitleOffset(1.4) p_flav.get_modifier().GetXaxis().SetTitleSize(.045) p_flav.set_logx(do_more_labels=True, do_exponent=False) p_flav.legend.SetX1(0.55) p_flav.legend.SetY1(0.72) p_flav.legend.SetY2(0.85) p_flav.legend.SetNColumns(2) p_flav.save(output_filename)
def compare_flavour_fraction_hists_vs_pt(input_files, hist_names, pt_bins, labels, flav, output_filename, title="", xtitle="p_{T}^{jet} [GeV]", is_preliminary=True): """Plot a specified flavour fraction vs pT for several sources. Each entry in input_files, dirnames, and labels corresponds to one line n_partons can be a str, 'all', '1', etc, or a list of str to include TODO: use this one more often - compare_flavour_fractions_vs_pt() is almost identical but has to deal with npartons """ bin_centers = [0.5 * (x[0] + x[1]) for x in pt_bins] bin_widths = [0.5 * (x[1] - x[0]) for x in pt_bins] contribs = [] info = [ get_flavour_efficiencies(ifile, bins=pt_bins, hist_name=hname) for ifile, hname in zip(input_files, hist_names) ] N = len(bin_centers) colours = [ROOT.kBlack, ROOT.kBlue, ROOT.kRed, ROOT.kGreen + 2] for i, fdict in enumerate(info): if flav in ['u', 'd', 's', 'c', 'b', 't', 'g']: obj = fdict[flav].CreateGraph() else: raise RuntimeError("Robin broke 1-X functionality") obj = ROOT.TGraphErrors( N, np.array(bin_centers), 1. - np.array(fdict[flav.replace("1-", '')]), np.array(bin_widths), np.zeros(N)) if obj.GetN() == 0: continue c = Contribution(obj, label=labels[i], line_color=colours[i], line_width=1, line_style=1, marker_style=20 + i, marker_color=colours[i], marker_size=1, leg_draw_opt="LP") contribs.append(c) flav_str = FLAV_STR_DICT[flav] ytitle = "Fraction of %s %ss" % (flav_str.lower(), get_jet_str('')) p = Plot(contribs, what='graph', xtitle=xtitle, ytitle=ytitle, title=title, xlim=(50, 2000), ylim=(0, 1), has_data=False, is_preliminary=is_preliminary) p.default_canvas_size = (600, 600) try: p.plot("AP") p.main_pad.SetBottomMargin(0.16) p.get_modifier().GetXaxis().SetTitleOffset(1.4) p.get_modifier().GetXaxis().SetTitleSize(.045) p.legend.SetX1(0.56) p.legend.SetY1(0.65) p.legend.SetY2(0.87) p.set_logx(do_more_labels=True, do_exponent=False) p.save(output_filename) except ZeroContributions: pass
def compare_flavour_fractions_vs_pt(input_files, dirnames, pt_bins, labels, flav, output_filename, title="", var_prepend="", which_jet="both", xtitle="p_{T}^{jet} [GeV]", n_partons='all', is_preliminary=True): """Plot a specified flavour fraction vs pT for several sources. Each entry in input_files, dirnames, and labels corresponds to one line n_partons can be a str, 'all', '1', etc, or a list of str to include TODO: fix this - bit stupid input format """ bin_centers = [0.5 * (x[0] + x[1]) for x in pt_bins] bin_widths = [0.5 * (x[1] - x[0]) for x in pt_bins] if isinstance(n_partons, str): n_partons = [n_partons] contribs = [] for n_parton_ind, n_parton in enumerate(n_partons): metric = 'pt' if n_parton.lower() != 'all': metric = 'pt_npartons_%s' % n_parton info = [ get_flavour_efficiencies( ifile, bins=pt_bins, hist_name=get_flavour_hist_name( dirname=dname, var_prepend=var_prepend, which_jet=(which_jet if "Dijet" in dname else "both"), metric=metric)) for ifile, dname in zip(input_files, dirnames) ] N = len(bin_centers) colours = [ROOT.kBlack, ROOT.kBlue, ROOT.kRed, ROOT.kGreen + 2] for i, fdict in enumerate(info): if flav in ['u', 'd', 's', 'c', 'b', 't', 'g']: obj = fdict[flav].CreateGraph() else: raise RuntimeError("Robin broke 1-X functionality") obj = ROOT.TGraphErrors( N, np.array(bin_centers), 1. - np.array(fdict[flav.replace("1-", '')]), np.array(bin_widths), np.zeros(N)) if obj.GetN() == 0: continue n_parton_str = "" if n_parton == "all" else " (%s-parton)" % n_parton c = Contribution(obj, label="%s%s" % (labels[i], n_parton_str), line_color=colours[i] + n_parton_ind, line_width=1, line_style=n_parton_ind + 1, marker_style=20 + i, marker_color=colours[i] + n_parton_ind, marker_size=1, leg_draw_opt="LP") contribs.append(c) flav_str = FLAV_STR_DICT[flav] ytitle = "Fraction of %s %ss" % (flav_str.lower(), get_jet_str('')) p = Plot(contribs, what='graph', xtitle=xtitle, ytitle=ytitle, title=title, xlim=(pt_bins[0][0], pt_bins[-1][1]), ylim=(0, 1), has_data=False, is_preliminary=is_preliminary) p.default_canvas_size = (600, 600) try: p.plot("AP") p.main_pad.SetBottomMargin(0.16) p.get_modifier().GetXaxis().SetTitleOffset(1.4) p.get_modifier().GetXaxis().SetTitleSize(.045) p.legend.SetX1(0.56) p.legend.SetY1(0.65) p.legend.SetY2(0.87) p.set_logx(do_more_labels=True, do_exponent=False) p.save(output_filename) except ZeroContributions: pass