plotting.draw_hist(h, nSig) plotting.draw_labels(args.lumi) plotting.draw_text(args.text_file) plotting.draw_line(args.g_min, args.l_min, args.g_max, args.l_max, args.top_mass) savefilename = os.path.join(args.out_directory, '_'.join([args.output, label])) if args.do_run1: gr = plotting.get_run1(args.run1_excl, 1, 3, args.run1_color) gr.Draw("C") gr_1sigma = plotting.get_run1(args.run1_1sigma, 3, 1, args.run1_color) gr_1sigma.Draw("C") plotting.draw_run1_text(args.run1_color) savefilename += "_wrun1" if args.do_run2: gr = plotting.get_run2(args.run2_excl, 1, 3, args.run2_color) gr.Draw("C") gr_1sigma = plotting.get_run2(args.run2_1sigma, 3, 1, args.run2_color) gr_1sigma.Draw("C") plotting.draw_run2_text(args.run2_color) savefilename += "_wrun2" #p = exclusion() #p.Draw() c.SaveAs(savefilename + ".pdf") print("Saving file " + savefilename) c.Clear() exit(0)
save_canvas(c, '{0}_optimalSR_grid_lumi{1}'.format(os.path.join(args.output_dir, args.output), args.lumi)) # now make a plot of the actual significances c = init_canvas() h = plotting.init_hist("Significance of optimal SR", 200, 2500, 0, 2300, 100, 200) for did, vals in significances.iteritems(): winningSR = vals['winner'] mgluino, mstop, mlsp = mdict[did] if mstop != 5000: continue set_bin(h, mgluino, mlsp, vals[winningSR]) draw_hist(h, "1.1f") draw_text(args) draw_line() if args.do_run1: gr = plotting.get_run1(args.run1_excl,1,3,args.run1_color) gr.Draw("C") gr_1sigma = plotting.get_run1(args.run1_1sigma,3,1,args.run1_color) gr_1sigma.Draw("C") plotting.draw_run1_text(args.run1_color) if args.do_run2: gr = plotting.get_run2(args.run2_excl,1,3,args.run2_color) gr.Draw("C") gr_1sigma = plotting.get_run2(args.run2_1sigma,3,1,args.run2_color) gr_1sigma.Draw("C") plotting.draw_run2_text(args.run2_color) save_canvas(c, '{0}_optimalSR_sig_lumi{1}'.format(os.path.join(args.output_dir, args.output), args.lumi))