if args.logx: axes[0].SetMoreLogLabels() mainPad.SetGrid() # ROOT.gPad.Update() # ROOT.gPad.RedrawAxis('G') legend.Draw() title = TitleAsLatex('[{}] {}'.format(chan.replace('mu', '#mu'), htitle.split(';')[0])) title.Draw() draw_labels('59.74 fb^{-1} (13 TeV)', cms_position='left', extra_text='work-in-progress') if args.vline is not None: vline = Line(args.vline, mainPad.GetUymin(), args.vline, axes[1].GetXmax()) vline.color = 'black' vline.linewidth = 2 vline.linestyle = 'dashed' vline.Draw() if args.dataset == 'all': # draw ratio on subpad subPad.cd() _ratio = hData.clone() _ratio.Divide(sumHistStack(hstack)) _ratio.SetMarkerSize(0.8) _ratio.yaxis.SetTitle('Data/MC') _ratio.yaxis.SetTitleOffset(0.35) _ratio.yaxis.CenterTitle() _ratio.yaxis.SetTitleSize(0.1)
h.get_correlation_factor()) h.Draw('colz') h.GetListOfFunctions().FindObject("palette").SetX2NDC(0.92) title = TitleAsLatex('[4#mu VR] ' + h.title) title.Draw() canvas.SaveAs('{}/ch4mu_vr_dphiisopre.pdf'.format(outdir)) canvas.Clear() h.Draw('colz') h.GetListOfFunctions().FindObject("palette").SetX2NDC(0.92) title.Draw() dphi_bounds = [0.7 * math.pi, 0.8 * math.pi, 0.9 * math.pi] vlines = [] for i, b in enumerate(dphi_bounds): vline = Line(b, h.yaxis.GetXmin(), b, h.yaxis.GetXmax()) vline.color = COLORS[i] vline.linewidth = 2 vlines.append(vline) for l in vlines: l.Draw() canvas.SaveAs('{}/ch4mu_vr_dphiisopre_vline.pdf'.format(outdir)) canvas.Clear() h.Draw('colz') h.GetListOfFunctions().FindObject("palette").SetX2NDC(0.92) title.Draw() iso_bounds = [0.2, 0.3, 0.4] hlines = [] for i, b in enumerate(iso_bounds): hline = Line(h.xaxis.GetXmin(), b, h.xaxis.GetXmax(), b) hline.color = COLORS[i]
leg = Legend( 3, pad=canvas, topmargin=0.05, margin=0.2, entryheight=0.02, entrysep=0.01, textsize=12, ) leg.AddEntry(d0inc, label='inclusive') leg.AddEntry(d0wbj, label='N_{bjet}#geq1') leg.AddEntry(d0wbj_c, label='N_{bjet}#geq1 (scaled)') leg.Draw() title = TitleAsLatex( '[4#mu VR] muon-type lepton-jet min |d_{0}| shape comparison') title.Draw() frame = canvas.FindObject('TFrame') lo, hi = frame.GetY1(), frame.GetY2() vline = Line(100, lo, 100, hi) vline.color = 'black' vline.linewidth = 2 vline.linestyle = 'dashed' vline.Draw() canvas.SaveAs('{}/ch4mu_muljd0Shape.pdf'.format(outdir)) canvas.clear() pf.close()