Functions
	topy (o)
	alleq (l)
	isint (o)
	form_list (l)
	squash_list (l)
	tostr (o)
	common_length (s1, s2)
	merge_names (blist)
	dumptree (tt, by_branch)
	dumphist (h)
	dumpdir (f, by_branch, pref='')
	dumpit (file, by_branch)

Variables
str	file = 'test1.root'
bool	by_branch = False

Function Documentation

◆ alleq()

alleq ( l )

Definition at line 55 of file dumptuple_any.py.

def alleq (l):
    if len(l) == 0: return True
    for i in range(1, len(l)):
        if l[i] != l[0]: return False
    return True
 
 

◆ common_length()

common_length	(		s1,
			s2 )

Definition at line 120 of file dumptuple_any.py.

def common_length (s1, s2):
    i = 0
    while i < len(s1) and i < len(s2) and s1[i] == s2[i]:
        i += 1
    return i
 

◆ dumpdir()

dumpdir	(	f,
		by_branch,
		pref = '' )

Definition at line 203 of file dumptuple_any.py.

def dumpdir (f, by_branch, pref=''):
    kk = [k.GetName() for k in f.GetListOfKeys()]
    kk.sort()
    for k in kk:
        o = f.Get (k)
        if k == 'Schema':
            pass
        elif isinstance (o, ROOT.TTree):
            print ('\n\n\nTree', pref+k)
            dumptree (o, by_branch)
        elif isinstance (o, ROOT.TH1):
            print ('\n\n\nTH1', pref+k)
            dumphist (o)
        elif isinstance (o, ROOT.TDirectory):
            dumpdir (o, by_branch, pref + k + '/')
        elif isinstance (o, ROOT.TObjString):
            print ('\n\n\nString', pref+k)
            if k == '_pickle':
                print ('[pickle data skipped]')
            else:
                print (o, end='')
            print ('__END_OF_STRING__')
        else:
            print ('\n\n\nKey', pref+k)
            print (o)
    return
              
 
 

◆ dumphist()

dumphist ( h )

Definition at line 194 of file dumptuple_any.py.

def dumphist (h):
    print ('bins: ', [h.GetBinContent(i) for i in range(h.GetSize())])
    if h.GetSumw2N():
        print ('errs: ', [h.GetBinError(i) for i in range(h.GetSize())])
    return

◆ dumpit()

dumpit	(		file,
			by_branch )

Definition at line 232 of file dumptuple_any.py.

def dumpit (file, by_branch):
    f=ROOT.TFile(file)
    dumpdir (f, by_branch=by_branch)
    return
 
 

◆ dumptree()

dumptree	(		tt,
			by_branch )

Definition at line 157 of file dumptuple_any.py.

def dumptree (tt, by_branch):
    n = tt.GetEntries()
    if tt.GetName().startswith ('dum_') and n > 10000000:
        print ('[Dummy tree skipped]')
        return
    bb = [b.GetName() for b in tt.GetListOfBranches()]
    bb.sort()
    if by_branch:
        bb = merge_names (bb)
        print ('\nBranches')
        print ('-----------')
        lasttag = ''
        for b in bb:
            if b[0] == ':':
                print (b)
                lasttag = b[1:]
                continue
            if b == '$':
                bname = lasttag
            else:
                bname = lasttag + b
            br = tt.GetBranch(bname)
            data = []
            for i in range(n):
                br.GetEntry(i)
                data.append (topy (getattr(tt, bname)))
            print (b, tostr (data))
    else:
        for i in range(n):
            tt.GetEntry(i)
            print ('\nEvent', i)
            print ('-----------')
            for b in bb:
                print (b, tostr (topy (getattr (tt, b))))
    return
 
 

◆ form_list()

form_list ( l )

Definition at line 70 of file dumptuple_any.py.

def form_list(l):
    if not isinstance(l, list):
        return l
    return '[' + ','.join(l) + ']'

◆ isint()

isint ( o )

Definition at line 62 of file dumptuple_any.py.

def isint(o):
    try:
        _ = int(o)
    except ValueError:
        return False
    return True
 
 

◆ merge_names()

merge_names ( blist )

Definition at line 126 of file dumptuple_any.py.

def merge_names (blist):
    out = []
    lasttag = ''
    for (i,b) in enumerate(blist):
        if i < len(blist)-1:
            b2 = blist[i+1]
        else:
            b2 = ''
        comm1 = common_length (lasttag, b)
        comm2 = common_length (b, b2)
        if comm1 >= comm2:
            if comm1 != len(lasttag):
                while comm1 > 0 and b[comm1-1] != '_':
                    comm1 -= 1
                if b[:comm1] != lasttag:
                    lasttag = b[:comm1]
                    out.append (':' + lasttag) 
        else:
            while comm2 > 0 and b[comm2-1] != '_':
                comm2 -= 1
            if b[:comm2] != lasttag:
                lasttag = b[:comm2]
                out.append (':' + lasttag)
 
        nm = b[len(lasttag):]
        if len(nm) == 0: nm = '$'
        out.append (nm)
    return out
                
 
 

◆ squash_list()

squash_list ( l )

Definition at line 74 of file dumptuple_any.py.

def squash_list(l):
    if not l: return l
    frags=[]
    nextfrag = []
    i = 0
    while i < len (l):
        if i < len(l)-1 and l[i] == l[i+1]:
            if nextfrag: frags.append(nextfrag)
            nextfrag = []
            j = i+2
            while j < len(l) and l[i] == l[j]:
                j += 1
            frags.append ('%d*[%s]' % (j-i, l[i]))
            i = j
        #elif (i < len(l)-2 and isint(l[i]) and
        #      isint(l[i+1]) and isint(l[i+2]) and
        #      int(l[i+1]) == int(l[i])+1 and int(l[i+2]) == int(l[i])+2):
        #    nextfrag = []
        #    j = i+3
        #    while j < len(l) and isint(l[j]) and int(l[j]) == int(l[i]) + (j-i):
        #        j += 1
        #    frags.append ('range(%s,%d)' % (l[i], int(l[i]) + (j-i)))
        #    i = j
        else:
            nextfrag.append (l[i])
            i += 1
    if nextfrag: frags.append (nextfrag)
    frags = [form_list(s) for s in frags]
    return '+'.join (frags)
                          
            
    

◆ topy()

topy ( o )

Definition at line 33 of file dumptuple_any.py.

def topy (o):
    if type(o).__name__ in ['map<string,string>',
                            'map<string,float>',
                            'map<string,int>']:
        keys = ROOT.D3PDTest.MapDumper.keys (o)
        values = ROOT.D3PDTest.MapDumper.values (o)
        return OrderedDict (sorted (zip (keys, values), key=lambda t: t[0]))
    if (type(o).__name__.startswith ('vector<char') or
        type(o).__name__.startswith ('vector<unsigned char')):
        ll = list(o)
        lens = [len(x) for x in ll]
        if lens and min(lens) == 1 and max(lens) == 1:
            ll = [ord(x) for x in ll]
        return ll
    if type(o).__name__.startswith ('vector<'):
        return [topy(x) for x in o]
    if type(o).__name__ in ['PyFloatBuffer',
                            'PyIntBuffer']:
        return list(o)
    return o
 
 

◆ tostr()

tostr ( o )

Definition at line 106 of file dumptuple_any.py.

def tostr (o):
    if isinstance(o, list):
        l = [tostr(x) for x in o]
        if len(l) > 1:
            return squash_list(l)
        return '[' + ', '.join (l) + ']'
    if isinstance(o, str):
        s = repr(o)
    else:
        s = str(o)
    if s.endswith ('.0'): s = s[:-2]
    return s
 
 

Variable Documentation

◆ by_branch

bool by_branch = False

Definition at line 241 of file dumptuple_any.py.

◆ file

python.dumptuple_any.file = 'test1.root'

Definition at line 240 of file dumptuple_any.py.

Functions

Variables