def read(self, variables, weights, cut, files=None):
if files is None:
files = self.files
return read_files(filenames=files,
branches=variables + weights,
cut=cut,
treename=self.tree_name)
def draw(order):
if order == 'nlo':
f = f_nlo
weightstr = tTIMES('normalizedWeight', nlo_corrections[args.process])
else:
f = f_lo
weightstr = 'normalizedWeight'
cutstr = tAND(selections[args.selection], proc_cuts[args.process])
h = hbase.Clone('h_' + order)
xarr = read_files([f], [ptstr, weightstr], cutstr, treename='Events')
draw_hist(h, xarr, [ptstr], weightstr)
print order, h.Integral(), cutstr
return h
branches.setdefault(cat + '_hbb1', []).extend(
[add_idx(x.replace('[]', '%i' % i), 1) for i in xrange(5)])
elif x.endswith('[0]'):
branches.setdefault(cat + '_hbb0', []).append(x)
elif x.endswith('[1]'):
branches.setdefault(cat + '_hbb1', []).append(x)
else:
branches.setdefault(cat + '_hbb0', []).append(add_idx(x, 0))
branches.setdefault(cat + '_hbb1', []).append(add_idx(x, 1))
cut = 'hbbm>0'
all_branches = [cut]
for _, v in branches.iteritems():
all_branches += v
xarr = read_files(filenames=[infile], branches=all_branches, cut=cut)
# flatten them
flattened = {}
for b in all_branches:
# flattened[b] = np.concatenate(xarr[b])
flattened[b] = xarr[b].astype(float)
# now merge some
data = {}
for cat in branches:
if cat.startswith('input'):
data[cat] = np.vstack([flattened[b] for b in branches[cat]]).T
data['shape' + cat.replace('inputs', '')] = data[cat].shape
else:
for b in branches[cat]:
def read(self, variables, weights, cut):
return read_files(filenames = self.__files,
branches = variables+weights,
cut = cut,
treename = self.tree_name)
