heppy_report Namespace Reference

Functions
def	root2map

Variables
string	oname = "%s/%s"
tuple	parser
tuple	summary = root2map(a,options.jsonAnalyzer,options.treeName)

Function Documentation

def heppy_report.root2map	(		dir,
			ana,
			treename
	)

Definition at line 5 of file heppy_report.py.

References bitset_utilities.append(), and list().

def root2map(dir,ana,treename):
    import ROOT
    tfile = ROOT.TFile.Open("%s/%s/%s.root"%(dir,ana,treename))
    if not tfile:
        print "Error: dir %s does not contain %s/%s.root" % (dir,ana,treename)
        return None
    tree = tfile.Get(treename)
    if not tree:
        print "Error: rootfile %s/%s/%s.root does not contain a TTree %s" % (dir,ana,treename,treename)
        return None
    jsonind = {}
    for e in xrange(tree.GetEntries()):
        tree.GetEntry(e)
        run,lumi = tree.run, tree.lumi
        if run not in jsonind:
            jsonind[run] = [lumi]
        else:
            jsonind[run].append(lumi)
    # remove duplicates
    for run in jsonind:
        jsonind[run] =  list(set(jsonind[run]))

    nruns = len(jsonind)
    nlumis = sum(len(v) for v in jsonind.itervalues())
    jsonmap = {}
    for r,lumis in jsonind.iteritems():
        if len(lumis) == 0: continue # shouldn't happen
        lumis.sort()
        ranges = [ [ lumis[0], lumis[0] ] ]
        for lumi in lumis[1:]:
            if lumi == ranges[-1][1] + 1:
                ranges[-1][1] = lumi
            else:
                ranges.append([lumi,lumi])
        jsonmap[r] = ranges
    return (jsonmap, nruns, nlumis)