Exemplo n.º 1
0
                               args.l_max, args.bin_size)
        plotting.fill_hist(h, plot_array, label, label == 'sig')
        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()
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))