rootstat Namespace Reference

Functions
def	help
def	analyze
def	main

Variables
	myargv = sys.argv

Function Documentation

def rootstat.analyze	(		root,
			level,
			gtrees,
			gbranches,
			doprint,
			sorttype
	)

Definition at line 72 of file rootstat.py.

def analyze(root, level, gtrees, gbranches, doprint, sorttype):

    trees = {}
    events = None
    keys = root.GetListOfKeys()
    for key in keys:
        objname = key.GetName()
        if objname not in trees:
            obj = root.Get(objname)
            if obj and obj.InheritsFrom('TTree'):
                trees[objname] = obj
                if objname == 'Events':
                    events = obj

    # Print summary of trees.

    if doprint:
        print('\nTrees:\n')
    for key in sorted(trees.keys()):
        tree = trees[key]
        nentry = tree.GetEntriesFast()
        if doprint:
            print('%s has %d entries.' % (key, nentry))

        # Remember information about trees.

        if key in gtrees:
            gtrees[key] = gtrees[key] + nentry
        else:
            gtrees[key] = nentry

    # Print summary of branches in Events tree.

    if doprint:
        print('\nBranches of Events tree:\n')

    # If level is zero, we are done (don't analyze branches).

    if level == 0:
        return

    if events:

        branch_tuples = {}

        if doprint:
            print('   Total bytes  Zipped bytes   Comp.  Branch name')
            print('   -----------  ------------   -----  -----------')
            
        branches = events.GetListOfBranches()
        ntotall = 0
        nzipall = 0

        # Loop over branche of Events tree.

        for branch in branches:
            branch_class = branch.GetClass().GetName()

            # Only look at data products (class art::Wrapper<T>).
            
            if branch_class[0: 13] == 'art::Wrapper<':

                # Loop over subbranches.
                
                subbranches = branch.GetListOfBranches()
                for subbranch in subbranches:
                    name = subbranch.GetName()

                    # Only look at '.obj' subbranch (wrapped object).
                    
                    if name[-4:] == '.obj':
                        ntot = subbranch.GetTotBytes("*")
                        nzip = subbranch.GetZipBytes("*")
                        ntotall = ntotall + ntot
                        nzipall = nzipall + nzip
                        if doprint:
                            if nzip != 0:
                                comp = float(ntot) / float(nzip)
                            else:
                                comp = 0.
                            branch_key = None
                            if sorttype == 1:
                                branch_key = ntot
                            elif sorttype == 2:
                                branch_key = nzip
                            else:
                                branch_key = name
                            branch_tuples[branch_key] = (ntot, nzip, comp, name)
                            #print('%14d%14d%8.2f  %s' % (ntot, nzip, comp, name))

                        # Remember information about branches.
                        
                        if name in gbranches:
                            gbranches[name][0] = gbranches[name][0] + ntot
                            gbranches[name][1] = gbranches[name][1] + nzip
                        else:
                            gbranches[name] = [ntot, nzip]

                        # Loop over subsubbranches (attributes of wrapped object).
                        
                        if level > 1:
                            subsubbranches = subbranch.GetListOfBranches()
                            for subsubbranch in subsubbranches:
                                name = subsubbranch.GetName()
                                ntot = subsubbranch.GetTotBytes("*")
                                nzip = subsubbranch.GetZipBytes("*")
                                if doprint:
                                    if nzip != 0:
                                        comp = float(ntot) / float(nzip)
                                    else:
                                        comp = 0.
                                branch_key = None
                                if sorttype == 1:
                                    branch_key = ntot
                                elif sorttype == 2:
                                    branch_key = nzip
                                else:
                                    branch_key = name
                                    branch_tuples[branch_key] = (ntot, nzip, comp, name)
                                    #print('%14d%14d%8.2f  %s' % (ntot, nzip, comp,
                                    #                             subsubbranch.GetName()))

                                # Remember information about branches.
                        
                                if name in gbranches:
                                    gbranches[name][0] = gbranches[name][0] + ntot
                                    gbranches[name][1] = gbranches[name][1] + nzip
                                else:
                                    gbranches[name] = [ntot, nzip]

        # Print sorted information about branches.

        if doprint:
            for branch_key in sorted(branch_tuples.keys()):
                branch_tuple = branch_tuples[branch_key]
                ntot = branch_tuple[0]
                nzip = branch_tuple[1]
                comp = branch_tuple[2]
                name = branch_tuple[3]
                print('%14d%14d%8.2f  %s' % (ntot, nzip, comp, name))

        # Do summary of all branches.

        name = 'All branches'
        if doprint:
            if nzipall != 0:
                comp = float(ntotall) / float(nzipall)
            else:
                comp = 0.
            print('%14d%14d%8.2f  %s' % (ntotall, nzipall, comp, name))

            # Print average event size.

            nev = events.GetEntriesFast()
            if nev != 0:
                nevtot = 1.e-6 * float(ntotall) / float(nev)
                nevzip = 1.e-6 * float(nzipall) / float(nev)
            else:
                nevtot = 0.
                nevzip = 0.
            print()
            print('%10d events.' % nev)
            print('%7.2f Mb average size per event.' % nevtot)
            print('%7.2f Mb average zipped size per event.' % nevzip)

        if name in gbranches:
            gbranches[name][0] = gbranches[name][0] + ntotall
            gbranches[name][1] = gbranches[name][1] + nzipall
        else:
            gbranches[name] = [ntotall, nzipall]


    # Done.                     
    
    return
                
# Main program.