def test_fill(self):
fi = TFile('tree.root','RECREATE')
tr = Tree('test_tree', 'A test tree')
tr.var('a')
tr.var('b')
tr.fill('a', 3)
tr.fill('a', 4)
tr.fill('b', 5)
tr.tree.Fill()
fi.Write()
fi.Close()
def create_tree(filename="test_tree.root"):
outfile = TFile(filename, 'recreate')
tree = Tree('test_tree', 'A test tree')
tree.var('var1')
for i in range(100):
tree.fill('var1', i)
tree.tree.Fill()
print('creating a tree', tree.tree.GetName(),\
tree.tree.GetEntries(), 'entries in',\
outfile.GetName())
outfile.Write()
def create_tree(filename="test_tree.root"):
outfile = TFile(filename, 'recreate')
tree = Tree('test_tree', 'A test tree')
tree.var('var1')
for i in range(100):
tree.fill('var1', i)
tree.tree.Fill()
print('creating a tree', tree.tree.GetName(),\
tree.tree.GetEntries(), 'entries in',\
outfile.GetName())
outfile.Write()
def test_fill(self):
fi = TFile('tree.root', 'RECREATE')
tr = Tree('test_tree', 'A test tree')
tr.var('a')
tr.var('b')
tr.fill('a', 3)
tr.fill('a', 4)
tr.fill('b', 5)
tr.tree.Fill()
fi.Write()
fi.Close()
class SimpleTreeProducer(Analyzer):
def beginLoop(self):
super(SimpleTreeProducer, self).beginLoop()
self.rootfile = TFile('/'.join([self.dirName, 'simple_tree.root']),
'recreate')
self.tree = Tree(self.cfg_ana.tree_name, self.cfg_ana.tree_title)
self.tree.var('test_variable')
def process(self, event):
self.tree.fill('test_variable', event.input.var1)
self.tree.tree.Fill()
def write(self):
self.rootfile.Write()
self.rootfile.Close()
class SimpleTreeProducer(Analyzer):
def beginLoop(self, setup):
super(SimpleTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName,
'simple_tree.root']),
'recreate')
self.tree = Tree( self.cfg_ana.tree_name,
self.cfg_ana.tree_title )
self.tree.var('test_variable')
def process(self, event):
self.tree.fill('test_variable', event.input.var1)
self.tree.tree.Fill()
def write(self, setup):
self.rootfile.Write()
self.rootfile.Close()
def write(self, dirName, fileName='RLTInfo.root'):
f = TFile('/'.join([dirName, fileName]), 'RECREATE')
t = Tree('RLTInfo', 'HLT/Run/Lumi information')
t.var('run', int)
t.var('lumi', int)
t.var('counts', int)
t.var('trigger', int)
for rlt, count in six.iteritems(self.dict):
t.fill('run', rlt[1])
t.fill('lumi', rlt[2])
t.fill('counts', count.integer)
t.tree.Fill()
f.Write()
f.Close()
def write(self, dirName, fileName='RLTInfo.root'):
f = TFile('/'.join( [dirName, fileName]), 'RECREATE')
t = Tree('RLTInfo','HLT/Run/Lumi information')
t.var('run', int )
t.var('lumi', int )
t.var('counts', int )
t.var('trigger', int )
for rlt, count in six.iteritems(self.dict):
t.fill('run', rlt[1])
t.fill('lumi', rlt[2])
t.fill( 'counts', count.integer)
t.tree.Fill()
f.Write()
f.Close()
from ROOT import TFile
from PhysicsTools.HeppyCore.statistics.tree import Tree
outfile = TFile('test_tree.root', 'recreate')
tree = Tree('test_tree', 'A test tree')
tree.var('var1')
for i in range(100):
tree.fill('var1', i)
tree.tree.Fill()
print 'creating a tree', tree.tree.GetName(),\
tree.tree.GetEntries(), 'entries in',\
outfile.GetName()
outfile.Write()
jetvars = events_ref[common[0]][0].keys()
for Y in jetvars:
tree.var("r_"+Y)
tree.var("" +Y)
tree.var("deltaR")
goodevents = set()
print "%-14s | %8s %8s %8s %7s | %7s %7s %7s %7s %7s %7s %7s %7s %7s %7s" % (
"event", "new pt", "ref pt", "gen pt", "gen eta", "c had","ref", "n had","ref", "c em","ref", "n em","ref", "mu","ref")
for key in common:
(run,lumi,event) = key
jets_ref = events_ref[key]
jets_new = events_new[key]
matching = matchObjectCollection3(jets_ref, jets_new, 0.3, filter = lambda r,n : abs(r.val("gen")-n.val("gen"))<0.01)
tree.fill("run", run)
tree.fill("lumi", lumi)
tree.fill("event", event)
for jr in jets_ref:
jm = matching[jr]
if not jm: continue
for v in jetvars:
tree.fill("r_"+v,jr.val(v))
tree.fill(v,jm.val(v))
tree.fill("deltaR", deltaR(jr,jm))
tree.tree.Fill()
goodevents.add(key)
newpt, refpt, genpt, gen_eta = jm.val("pt"), jr.val("pt"), jr.val("gen"), jr.val("gen_eta")
toPrint = options.allprint
if abs(newpt/refpt-1) > 0.2 and abs(refpt/genpt-1)<0.2 and abs(newpt/genpt-1)>2*abs(refpt/genpt-1): toPrint = True
if options.regression:
