示例#1
0
#! /usr/bin/env python

import sys

from miniutils import get_sample_datasets

if len(sys.argv) < 2:
    print 'usage: show_samples.py [sample-name] [version]'
    sys.exit(1)

sample = sys.argv[1]

version = None
if len(sys.argv) > 2:
    version = sys.argv[2]


datasets = get_sample_datasets(sample, version)

for ds in datasets:
    print ds['path']

h_srl300.SetDirectory(0)
h_srh.SetDirectory(0)

ROOT.SetOwnership(h_srl200, False)
ROOT.SetOwnership(h_srl300, False)
ROOT.SetOwnership(h_srh, False)
ROOT.SetOwnership(h_sel_srl200, False)
ROOT.SetOwnership(h_sel_srl300, False)
ROOT.SetOwnership(h_sel_srh, False)

for (m3, mu), (mgl, mn1) in sorted(mg_gg_grid.iteritems()):

    name = 'GGM_GG_bhmix_%i_%i' % (m3, mu)

    # total events
    ds = get_sample_datasets(name)[0]
    total_events = get_sumw(ds)

    if total_events == 0:
        continue

    srl200_events_scaled = get_events(name, selection=SRL200, lumi='data').mean
    srl300_events_scaled = get_events(name, selection=SRL300, lumi='data').mean
    srh_events_scaled = get_events(name, selection=SRH, lumi='data').mean

    h_sel_srl200.Fill(mgl, mn1, round(srl200_events_scaled, 2))
    h_sel_srl300.Fill(mgl, mn1, round(srl300_events_scaled, 2))
    h_sel_srh.Fill(mgl, mn1, round(srh_events_scaled, 2))

    srl200_events = get_events(name, selection=SRL200, scale=False).mean
    srl300_events = get_events(name, selection=SRL300, scale=False).mean
h_srih.SetDirectory(0)

ROOT.SetOwnership(h_srl, False)
ROOT.SetOwnership(h_srh, False)
ROOT.SetOwnership(h_sril, False)
ROOT.SetOwnership(h_srih, False)


for (m3, mu) in sorted(mass_dict.iterkeys()):

    mgl, mn1 = mass_dict[(int(m3), int(mu))]

    name = 'GGM_M3_mu_%i_%i' % (m3, mu)

    # total events
    ds = get_sample_datasets(name)[0]
    total_events = get_sumw(ds)

    if total_events == 0:
        continue

    srl_events_scaled = get_events(name, selection=SR_L, lumi='data').mean
    srh_events_scaled = get_events(name, selection=SR_H, lumi='data').mean

    sril_events_scaled = get_events(name, selection=SRincl_L, lumi='data').mean
    srih_events_scaled = get_events(name, selection=SRincl_H, lumi='data').mean

    h_sel_srl.Fill(mgl, mn1, round(srl_events_scaled, 2))
    h_sel_srh.Fill(mgl, mn1, round(srh_events_scaled, 2))

    h_sel_sril.Fill(mgl, mn1, round(sril_events_scaled, 2))