In This Package:

howBig.py

Go to the documentation of this file.

#!/usr/bin/env python
#
'''

 size of headers in mixed (or any) data in file.root
 using TTree::Print()

options can be obtained with
  nuwa.py -m"howBig --help"

usage: 
 nuwa.py -l 5 --no-history -n 1 -m "howBig " file.root



Notes:
Apparently must process one execution cycle "-n 1" to ensure file.root is read.
Use of "-l 5" and cd "--no-history" just avoids copious and uninteresting output.
'''

__all__ = ['howBig']
# from GaudiPython.GaudiAlgs import GaudiAlgo
from DybPython.DybPythonAlg import DybPythonAlg
from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl, loaddict

from DybPython.Util import irange
from GaudiKernel import SystemOfUnits as units
import PyCintex
import re
import ROOT
import math

printLevel = 0

class howBig(DybPythonAlg):

    def __init__(self,name):

        DybPythonAlg.__init__(self,name)

        return

    
    def initialize(self):
        print "Initializing the howBig: ", self.name()
        status = DybPythonAlg.initialize(self)
        if status.isFailure(): return status

        global printlevel

        return SUCCESS

    def execute(self):

        return SUCCESS
    
    def finalize(self):


# get TDir for input file
        self.infileObj = gbl.gDirectory

# get filename of input file. append ":/" to use as path
# Check on validity of filename
        self.filename = self.infileObj.GetName() 
        if self.filename.count( 'PyROOT' ) > 0 :
            print "\n ERROR  It looks like you did not read any events. You need to read at least one event for howBig to work ERROR  \n"
            return FAILURE 

        print "\n Contents of ",self.filename
        self.filename += ":/"
        self.TotBytes = 0
        self.ZipBytes = 0
        self.process_path("")
        nt = self.TotBytes
        nz = self.ZipBytes
        if printlevel == 0  or printlevel == 1 :
            print "{2:10.2f}MB(on disk)  {1:10.2f}MB(uncompressed)  {0:30}  \n".format( 'TOTALS', nt/1e6, nz/1e6 )


        status = DybPythonAlg.finalize(self)
        return status

    def process_path(self, itemPath) :
        ''' process path in root file. cribbed from Dan '''
        #print " process_path: ",itemPath

        filePath  = self.filename
        if len(itemPath)>0 :
            filePath += "/" + itemPath
        content = self.infileObj.Get(filePath)
        if not content and itemPath=="":
            filePath += ":/"
            content = self.infileObj # self.infileObj.Get(filePath)
        if not content :
            print "Failed to get item at ",filePath
            return
        if content.IsA().InheritsFrom("TTree") :
            name = content.GetName()
            nt   = content.GetTotBytes()
            self.TotBytes += nt
            nz   = content.GetZipBytes()
            self.ZipBytes += nz
            if printlevel > 1 : 
                content.Print()
            if printlevel == 1 : 
                print "{2:10.2f}MB(on disk)  {1:10.2f}MB(uncompressed)  {0:30}".format( name, nt/1e6, nz/1e6 )

        elif content.IsA().InheritsFrom("TDirectory") :
            keylist = content.GetListOfKeys()
            for key in keylist :
                dirItem = key.ReadObj()
                dirItemPath = None
                if len(itemPath)>0:
                    dirItemPath = itemPath + "/" + dirItem.GetName()
                else :
                    dirItemPath = dirItem.GetName()
                self.process_path( dirItemPath )
        else:
            print "Not processing",filePath,"of type",content.ClassName()

        
        return 

def configure(argv=[]) : #None) :
    # note: if argv=None is specified as the arguments to configure, then
    #      the parser expects to have at least one option explicitly specified.
    #      with argv=[], one does not have specify any arguments (the defaults are used)
#    from DetHelpers.DetHelpersConf import CoordSysSvc
#    from GaudiSvc.GaudiSvcConf import THistSvc
#    from GaudiSvc.GaudiSvcConf import DetectorDataSvc


    # deal with input arguments
    from optparse import OptionParser
    parser = OptionParser() #usage=self.__doc__)

    parser.add_option("-p","--print-level",
                      default=1, type="int",
                      help="Set print level. 0=minimal(Totals only), 1=more(per Header totals), 2=most (same as TTree->Print()).   [default: %default]")

    (options,args) = parser.parse_args(args=argv)

    global printlevel 
    printlevel = options.print_level
    print " Set output print level to ",printlevel
    

    from Gaudi.Configuration import ApplicationMgr
    theApp = ApplicationMgr()
#    theApp.ExtSvc.append(coorSvc)



    return

def run(app):
    '''Configure and add this algorithm to job'''
#    from Gaudi.Configuration import ApplicationMgr
#    app = ApplicationMgr()
    import sys
#    app.ExtSvc += ["StatisticsSvc", "GiveHeader", "StaticCableSvc", "PmtGeomInfoSvc"]
#    app.ExtSvc += ["StatisticsSvc",  "StaticCableSvc", "PmtGeomInfoSvc"]
    plotBasics = howBig("HowBig")
    app.addAlgorithm(plotBasics)

    pass

---