def main():
parser = argparse.ArgumentParser(description='get cutflow')
parser.add_argument('-r', dest='regions', help='Region')
parser.add_argument('-s', dest='samples', help='Samples separated with ,')
parser.add_argument('-i', dest='files', help='Files separated with ,')
parser.add_argument('-l', dest='luminosity', help='Luminosity [fb-1]')
parser.add_argument('-p', dest='percentage', action='store_true', help='Show percentage')
parser.add_argument('--latex', action='store_true', default=False, help='use LatexTable instead PrettyTable')
parser.add_argument('--sel', dest='selection', help='Selection')
parser.add_argument('--pre', dest='preselection', action='store_true', help='add preselection cutflow')
if len(sys.argv) < 2:
parser.print_usage()
sys.exit(1)
args = parser.parse_args()
## samples
if args.samples is None and args.files is None:
samples = [
'data',
'photonjet',
'wgamma',
'zllgamma',
'znunugamma',
'ttbar',
'ttbarg',
'wjets',
'zjets',
'diboson',
'efake',
'jfake'
]
if args.regions is None:
args.regions = 'Sel'
do_scale = True
if args.luminosity == "0":
do_scale = False
for region in args.regions.split(','):
#if args.samples is not None:
# files = [ os.path.join(MiniDir, '%s.t146_mini.root' % sample) for sample in args.samples.split(',') ]
#if args.files is not None:
# files = args.files.split(',')
try:
selection = getattr(regions_, region)
except:
selection = args.selection
flows = OrderedDict()
if args.samples is not None:
for sample in args.samples.split(','):
cutflow = get_cutflow(sample, selection=selection, lumi=args.luminosity, preselection=args.preselection, scale=do_scale)
cuts = [ cutflow.GetXaxis().GetBinLabel(b+1) for b in xrange(cutflow.GetNbinsX()) ]
flows[sample] = [ cutflow.GetBinContent(b+1) for b in xrange(cutflow.GetNbinsX()) ]
if args.files is not None:
for fname in args.files.split(','):
cutflow = get_cutflow(fname, selection=selection, preselection=args.preselection, scale=do_scale)
cuts = [ cutflow.GetXaxis().GetBinLabel(b+1) for b in xrange(cutflow.GetNbinsX()) ]
flows[os.path.basename(fname)] = [ cutflow.GetBinContent(b+1) for b in xrange(cutflow.GetNbinsX()) ]
if args.latex:
table = LatexTable()
else:
table = PrettyTable()
table.add_column(region, cuts)
for sample, flow in flows.iteritems():
total = float(flow[0])
if args.percentage:
table.add_column(sample, ['%.2f (%d%%)' % (n, 100*n/total) for n in flow])
else:
table.add_column(sample, ['%.2f' % n for n in flow])
print table
def main():
parser = argparse.ArgumentParser(description='get yields')
parser.add_argument('-r',
dest='regions',
default='',
help='Regions separated with ,')
parser.add_argument('-s', dest='samples', help='Samples separated with ,')
parser.add_argument('-i', dest='files', help='Files separated with ,')
parser.add_argument('-l', dest='lumi', help='Luminosity to scale')
parser.add_argument('-v', dest='version', help='force mini version')
parser.add_argument('--sel', dest='selection')
parser.add_argument('--prw', action='store_true', help='apply prw weights')
parser.add_argument('--data', help='Include data: data|data15|data16')
parser.add_argument('--unblind', action='store_true', help='Unblind data')
# backgrounds
parser.add_argument('--mc', action='store_true', help='Use MC backgrounds')
parser.add_argument('--muq', help='Normalization factor for gam+jet')
parser.add_argument('--muw', help='Normalization factor for W gamma')
parser.add_argument('--mut', help='Normalization factor for ttbar gamma')
# signal
parser.add_argument('--signal', action='store_true', help='Include signal')
parser.add_argument('--m3', default='1400', help='M3')
# others
parser.add_argument('--latex',
action='store_true',
help='use LatexTable instead PrettyTable')
parser.add_argument('--nw', action='store_true')
if len(sys.argv) < 2:
parser.print_usage()
sys.exit(1)
args = parser.parse_args()
## samples
bkgs = analysis.backgrounds_mc
dd_scale = 1.
if args.data is None and args.samples is None and args.lumi is not None:
bkgs.append('jfake')
bkgs.append('efake')
dd_scale = float(args.lumi) * 1000. / analysis.lumi_data
elif args.data == 'data15':
args.lumi = 'data15'
bkgs.append('jfake15')
bkgs.append('efake15')
elif args.data == 'data16':
args.lumi = 'data16'
bkgs.append('jfake16')
bkgs.append('efake16')
elif args.data == 'data':
args.lumi = 'data'
bkgs.append('jfake')
bkgs.append('efake')
if args.mc:
bkgs = [
'photonjet',
'multijet',
'wgamma',
'zgamma',
'ttbar',
'ttbarg',
'vjets',
]
signal = []
for (m3, mu) in sorted(grid_m3_mu.iterkeys()):
if int(args.m3) == m3:
signal.append('GGM_M3_mu_%d_%d' % (m3, mu))
if args.prw:
get_events = partial(miniutils.get_events,
lumi=args.lumi,
version=args.version,
prw=True)
elif args.nw:
get_events = partial(miniutils.get_events,
lumi=args.lumi,
version=args.version,
scale=False)
else:
get_events = partial(miniutils.get_events,
lumi=args.lumi,
version=args.version)
if args.regions:
regions = args.regions.split(',')
samples = args.samples.split(',') if args.samples is not None else []
if args.selection is not None:
regions = []
regions.append(args.selection)
if args.latex:
table = LatexTable()
else:
table = PrettyTable()
if samples:
table.add_column('', [s for s in samples])
elif args.data and args.signal:
table.add_column('', [
'Data',
] + bkgs + ['Total bkg'] + signal)
elif args.data:
table.add_column('', [
'Data',
] + bkgs + ['Total bkg'])
elif args.signal:
table.add_column('', bkgs + ['Total bkg'] + signal)
else:
table.add_column('', bkgs + ['Total bkg'])
for region in regions:
if not region:
continue
try:
selection = getattr(regions_, region)
except:
selection = region
cols = OrderedDict()
if samples:
for sample in samples:
evts = get_events(sample, region=region, selection=selection)
cols[sample] = evts
else:
# Data
if args.data is None:
pass
elif 'SR' in region and not args.unblind:
cols['data'] = '-1'
else:
cols['data'] = get_events(args.data,
region=region,
selection=selection)
# Bkgs
total_bkg = Value(0)
for sample in bkgs:
evts = get_events(sample, region=region, selection=selection)
if args.data is None and 'fake' in sample:
evts *= dd_scale
if not region.startswith('CR'):
if args.muq is not None and sample == 'photonjet':
evts *= float(args.muq)
if args.muw is not None and sample == 'wgamma':
evts *= float(args.muw)
if args.mut is not None and sample == 'ttbarg':
evts *= float(args.mut)
cols[sample] = evts
total_bkg += evts
cols['Total bkg'] = total_bkg
if region.startswith('CR') and args.data:
if 'CRQ' in region or 'CRM' in region:
mu = (cols['data'] -
(total_bkg - cols['photonjet'])) / cols['photonjet']
purity = cols['photonjet'] / total_bkg
cols['photonjet'] = '%s (%.2f, mu=%.2f)' % (
cols['photonjet'], purity.mean, mu.mean)
elif 'CRT' in region:
mu = (cols['data'] -
(total_bkg - cols['ttbarg'])) / cols['ttbarg']
purity = cols['ttbarg'] / total_bkg
cols['ttbarg'] = '%s (%.2f%%, mu=%.2f)' % (
cols['ttbarg'], purity.mean, mu.mean)
elif 'CRW' in region:
mu = (cols['data'] -
(total_bkg - cols['wgamma'])) / cols['wgamma']
purity = cols['wgamma'] / total_bkg
cols['wgamma'] = '%s (%.2f%%, mu=%.2f)' % (
cols['wgamma'], purity.mean, mu.mean)
# Signals
if args.signal:
for sig in signal:
n_s = get_events(sig, region=region, selection=selection)
cols[sig] = '%s (%s)' % (n_s,
get_significance(n_s, total_bkg))
table.add_column(region[:10], cols.values())
print table
def main():
parser = argparse.ArgumentParser(description='get cutflow')
parser.add_argument('-r', dest='regions', help='Region')
parser.add_argument('-s', dest='samples', help='Samples separated with ,')
parser.add_argument('-i', dest='files', help='Files separated with ,')
parser.add_argument('-l', dest='luminosity', help='Luminosity [fb-1]')
parser.add_argument('-p',
dest='percentage',
action='store_true',
help='Show percentage')
parser.add_argument('-v', dest='version', help='Mini version')
parser.add_argument('--latex',
action='store_true',
default=False,
help='use LatexTable instead PrettyTable')
parser.add_argument('--sel', dest='selection', help='Selection')
parser.add_argument('--pre',
dest='preselection',
action='store_true',
help='add preselection cutflow')
if len(sys.argv) < 2:
parser.print_usage()
sys.exit(1)
args = parser.parse_args()
## samples
if args.samples is None and args.files is None:
sys.exit(1)
if args.regions is None:
args.regions = 'Sel'
do_scale = True
if args.luminosity == "0":
do_scale = False
for region in args.regions.split(','):
try:
selection = getattr(regions_, region)
except:
selection = args.selection
flows = OrderedDict()
if args.samples is not None:
for sample in args.samples.split(','):
cutflow = get_cutflow(sample,
selection=selection,
lumi=args.luminosity,
preselection=args.preselection,
scale=do_scale,
version=args.version)
cuts = [
cutflow.GetXaxis().GetBinLabel(b + 1)
for b in xrange(cutflow.GetNbinsX())
]
flows[sample] = [
cutflow.GetBinContent(b + 1)
for b in xrange(cutflow.GetNbinsX())
]
if args.files is not None:
for fname in args.files.split(','):
cutflow = get_cutflow(fname,
selection=selection,
preselection=args.preselection,
scale=do_scale,
version=args.version)
cuts = [
cutflow.GetXaxis().GetBinLabel(b + 1)
for b in xrange(cutflow.GetNbinsX())
]
flows[os.path.basename(fname)] = [
cutflow.GetBinContent(b + 1)
for b in xrange(cutflow.GetNbinsX())
]
if args.latex:
table = LatexTable()
else:
table = PrettyTable()
table.add_column(region, cuts)
for sample, flow in flows.iteritems():
total = float(flow[0])
if args.percentage:
table.add_column(sample, [
'%.2f (%d%%)' % (n, int(round(100 * (n / total))))
for n in flow
])
else:
table.add_column(sample, ['%.2f' % n for n in flow])
print table
def main():
parser = argparse.ArgumentParser(description='get yields')
parser.add_argument('-r', dest='regions', default='', help='Regions separated with ,')
parser.add_argument('-s', dest='samples', help='Samples separated with ,')
parser.add_argument('-i', dest='files', help='Files separated with ,')
parser.add_argument('-l', dest='lumi', help='Luminosity to scale')
parser.add_argument('-v', dest='version', help='force mini version')
parser.add_argument('--sel', dest='selection')
parser.add_argument('--prw', action='store_true', help='apply prw weights')
parser.add_argument('--data', help='Include data: data|data15|data16')
parser.add_argument('--unblind', action='store_true', help='Unblind data')
# backgrounds
parser.add_argument('--mc', action='store_true', help='Use MC backgrounds')
parser.add_argument('--muq', help='Normalization factor for gam+jet')
parser.add_argument('--muw', help='Normalization factor for W gamma')
parser.add_argument('--mut', help='Normalization factor for ttbar gamma')
# signal
parser.add_argument('--signal', action='store_true', help='Include signal')
parser.add_argument('--m3', default='1400', help='M3')
# others
parser.add_argument('--latex', action='store_true', help='use LatexTable instead PrettyTable')
parser.add_argument('--nw', action='store_true')
if len(sys.argv) < 2:
parser.print_usage()
sys.exit(1)
args = parser.parse_args()
## samples
bkgs = analysis.backgrounds_mc
dd_scale = 1.
if args.data is None and args.samples is None and args.lumi is not None:
bkgs.append('jfake')
bkgs.append('efake')
dd_scale = float(args.lumi) * 1000. / analysis.lumi_data
elif args.data == 'data15':
args.lumi = 'data15'
bkgs.append('jfake15')
bkgs.append('efake15')
elif args.data == 'data16':
args.lumi = 'data16'
bkgs.append('jfake16')
bkgs.append('efake16')
elif args.data == 'data':
args.lumi = 'data'
bkgs.append('jfake')
bkgs.append('efake')
if args.mc:
bkgs = [
'photonjet',
'multijet',
'wgamma',
'zgamma',
'ttbar',
'ttbarg',
'vjets',
]
signal = []
for (m3, mu) in sorted(grid_m3_mu.iterkeys()):
if int(args.m3) == m3:
signal.append('GGM_M3_mu_%d_%d' % (m3, mu))
if args.prw:
get_events = partial(miniutils.get_events, lumi=args.lumi, version=args.version, prw=True)
elif args.nw:
get_events = partial(miniutils.get_events, lumi=args.lumi, version=args.version, scale=False)
else:
get_events = partial(miniutils.get_events, lumi=args.lumi, version=args.version)
if args.regions:
regions = args.regions.split(',')
samples = args.samples.split(',') if args.samples is not None else []
if args.selection is not None:
regions = []
regions.append(args.selection)
if args.latex:
table = LatexTable()
else:
table = PrettyTable()
if samples:
table.add_column('', [s for s in samples])
elif args.data and args.signal:
table.add_column('', ['Data',]+bkgs+['Total bkg']+signal)
elif args.data:
table.add_column('', ['Data',]+bkgs+['Total bkg'])
elif args.signal:
table.add_column('', bkgs+['Total bkg']+signal)
else:
table.add_column('', bkgs+['Total bkg'])
for region in regions:
if not region:
continue
try:
selection = getattr(regions_, region)
except:
selection = region
cols = OrderedDict()
if samples:
for sample in samples:
evts = get_events(sample, region=region, selection=selection)
cols[sample] = evts
else:
# Data
if args.data is None:
pass
elif 'SR' in region and not args.unblind:
cols['data'] = '-1'
else:
cols['data'] = get_events(args.data, region=region, selection=selection)
# Bkgs
total_bkg = Value(0)
for sample in bkgs:
evts = get_events(sample, region=region, selection=selection)
if args.data is None and 'fake' in sample:
evts *= dd_scale
if not region.startswith('CR'):
if args.muq is not None and sample == 'photonjet':
evts *= float(args.muq)
if args.muw is not None and sample == 'wgamma':
evts *= float(args.muw)
if args.mut is not None and sample == 'ttbarg':
evts *= float(args.mut)
cols[sample] = evts
total_bkg += evts
cols['Total bkg'] = total_bkg
if region.startswith('CR') and args.data:
if 'CRQ' in region or 'CRM' in region:
mu = (cols['data']-(total_bkg-cols['photonjet']))/cols['photonjet']
purity = cols['photonjet'] / total_bkg
cols['photonjet'] = '%s (%.2f, mu=%.2f)' % (cols['photonjet'], purity.mean, mu.mean)
elif 'CRT' in region:
mu = (cols['data']-(total_bkg-cols['ttbarg']))/cols['ttbarg']
purity = cols['ttbarg'] / total_bkg
cols['ttbarg'] = '%s (%.2f%%, mu=%.2f)' % (cols['ttbarg'], purity.mean, mu.mean)
elif 'CRW' in region:
mu = (cols['data']-(total_bkg-cols['wgamma']))/cols['wgamma']
purity = cols['wgamma'] / total_bkg
cols['wgamma'] = '%s (%.2f%%, mu=%.2f)' % (cols['wgamma'], purity.mean, mu.mean)
# Signals
if args.signal:
for sig in signal:
n_s = get_events(sig, region=region, selection=selection)
cols[sig] = '%s (%s)' % (n_s, get_significance(n_s, total_bkg))
table.add_column(region[:10], cols.values())
print table
