コード例 #1
0
ファイル: graph-grid.py プロジェクト: kratsg/Optimization
              'bkgd':     [r['bkgd'] for r in summary],
              'mgluino':  [r['m_gluino'] for r in summary],
              'mlsp':     [r['m_lsp'] for r in summary],
              'ratio':    [r['ratio'] for r in summary]}

  plotting.init_palette()

  c = plotting.init_canvas(args.x_dim, args.y_dim)
  labels = ['sig','signal','bkgd', 'ratio']
  zlabels = ['Significance in optimal cut','Exp. num. signal in optimal cut','Exp. num. bkgd in optimal cut', 'Signal/Background']
  nSigs = [2, 3, 3, 2]
  for label,zlabel,nSig in zip(labels,zlabels,nSigs):
    h = plotting.init_hist(zlabel, args.g_min, args.g_max, args.l_min, args.l_max, args.x_bin_size, args.y_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")
コード例 #2
0
    plotting.init_palette()

    c = plotting.init_canvas(args.x_dim, args.y_dim)
    labels = ['sig', 'signal', 'bkgd', 'ratio']
    zlabels = [
        'Significance in optimal cut', 'Exp. num. signal in optimal cut',
        'Exp. num. bkgd in optimal cut', 'Signal/Background'
    ]
    nSigs = [2, 3, 3, 2]
    for label, zlabel, nSig in zip(labels, zlabels, nSigs):
        h = plotting.init_hist(zlabel, args.g_min, args.g_max, args.l_min,
                               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)
コード例 #3
0
                'mgluino':  [r['m_c1n2'] for r in summary],
                'mlsp':     [r['m_n1'] for r in summary],
                'ratio':    [r['ratio'] for r in summary]}

  plotting.init_palette()

  c = plotting.init_canvas(args.x_dim, args.y_dim)
  labels = ['sig','signal','bkgd', 'ratio']
  #zlabels = ['Significance in optimal cut','Exp. num. signal in optimal cut','Exp. num. bkgd in optimal cut', 'Signal/Background']
  zlabels = ['Z_{{n}} in {}'.format(args.region),'Exp. num. signal in {}'.format(args.region),'Exp. num. bkgd in {}'.format(args.region), 'Signal/Background']
  nSigs = [2, 2, 2, 2]
  for label,zlabel,nSig in zip(labels,zlabels,nSigs):
    h = plotting.init_hist(zlabel, args.g_min, args.g_max, args.l_min, args.l_max, args.x_bin_size, args.y_bin_size, "grid", args.x_label, args.y_label)
    plotting.fill_hist(h,plot_array,label, label=='sig')
    plotting.draw_hist(h, nSig)
    plotting.draw_labels(args.lumi, "#tilde{#chi}^{#pm}_{1}#tilde{#chi}^{0}_{2} #rightarrow #tilde{#chi}^{0}_{1}#tilde{#chi}^{0}_{1} Wh #rightarrow b#bar{b}q#bar{q} + E_{T}^{miss}")
    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)