def getData(version='v1', nentries=-1): chains = getChains(version) data = {} for name, chain in chains.items(): mass = RooRealVar('mass', 'mass', 60, 120) weight = RooRealVar('weight', 'weight', 0, 999) if nentries > 0: cuts = ['Entry$ < %d' % nentries] else: cuts = [] data[name] = dataset.get(tree=chain, variable=mass, weight=weight, cuts=cuts) return data
def getData(version='v1', nentries=-1): chains = getChains(version) data = {} for name, chain in chains.items(): mass = RooRealVar('mass', 'mass', 60, 120) weight = RooRealVar('weight', 'weight', 0, 999) if nentries > 0: cuts = ['Entry$ < %d' % nentries] else: cuts = [] data[name] = dataset.get(tree=chain, variable=mass, weight=weight, cuts=cuts) return data
#getFitPlot(ws, "mass", ).Draw() ## print "# x, xerr, m0(%), m0_err(%), sigma, sigma_err(%), cut, cut_err, power, power_err" ## for i in range(len(xvalues)): ## print "% 5.3g 0.0 " % (100*(xvalues[i]-1.),), ## factor = 100./91.19 ## print "% 8.3g %8.2g " % (params[i][0].getVal() * factor, params[i][0].getError() * factor), ## factor = factor * math.sqrt(2) ## print "% 8.3g %8.2g " % (params[i][1].getVal() * factor, params[i][1].getError() * factor), ## for j in range(2, 4): ## print "% 8.3g %8.2g " % (params[i][j].getVal(), params[i][j].getError(),), ## print nentries = 5000 chains = getChains('v1') mcTree = chains['mc'] test1Tree = chains['test1'] w = RooWorkspace('w') mass = w.factory('mass[60, 120]') trange = (log(mass.getMin()/91.2), log(mass.getMax()/91.2)) t = w.factory('t[%f,%f]' % trange) t.SetTitle('log(mass/91.2)') weight = w.factory('weight[0, 999]') cuts = ['Entry$ < %d' % nentries] mData = dataset.get(tree=mcTree, variable=mass, weight=weight, cuts=cuts, name='mData')