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 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()
print "Read %d events from %s %s, found %s events with interesting jets" % (i, role, filename, len(ret)) return ret events_ref = readEvents(args[0], "reference") events_new = readEvents(args[1], "new") common = sorted(set(events_ref.keys()).intersection(events_new.keys())) print "%d common events. now doing jet-by-jet comparison" % len(common) from PhysicsTools.HeppyCore.utils.deltar import matchObjectCollection3, deltaR from PhysicsTools.HeppyCore.statistics.tree import Tree fout = ROOT.TFile.Open(args[2], "RECREATE") tree = Tree("t","t","F"); for X in "run", "lumi", "event": tree.var(X,int) 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)