D3PDSizeSummary.Filestats Class Reference

Collaboration diagram for D3PDSizeSummary.Filestats:

Public Member Functions
def	__init__ (self, fname, treename=None)
def	get_treename (self)
def	get_branchgroup (self, b)
def	get_stats (self)
def	scan_branches (self)
def	print_stats_text (self)
def	draw_pie (self)

Public Attributes
	fname
	treename
	branchgroups
	file_size
	rfile
	tree
	nev

Detailed Description

Definition at line 172 of file D3PDSizeSummary.py.

Constructor & Destructor Documentation

__init__()

def D3PDSizeSummary.Filestats.__init__	(		self,
			fname,
			treename = None
	)

Definition at line 173 of file D3PDSizeSummary.py.

    def __init__ (self, fname, treename = None):
        self.fname = fname
        self.treename = treename
        self.branchgroups = {}
        self.get_stats()
        return
 
 

Member Function Documentation

draw_pie()

def D3PDSizeSummary.Filestats.draw_pie

(

self

)

Definition at line 251 of file D3PDSizeSummary.py.

    def draw_pie (self):
        
        if options.debug:
            print "Number of branch groups:", len(self.branchgroups)
            print "Values of branch groups:", list(self.branchgroups.keys()) 
        
        blist = self.branchgroups.values()
        blist.sort (lambda a, b: int(b.filebytes - a.filebytes))            
        pie = ROOT.TPie("BranchPie", "", len(self.branchgroups))
        for i, bgroup in enumerate( blist ):
            bginfo = bgroup.get_dict(self.nev) 
            
            if options.debug:
                print "Branch group #%d, named %s with size %f = %f%%" % (i ,  bginfo['name'], (float(bgroup.filebytes) / 1024 / self.nev ), (100. * float(bgroup.filebytes) / float(self.file_size)) )
            
            pie.SetEntryLabel(i, bginfo['name'])
            pie.SetEntryVal  (i, (100. * float(bgroup.filebytes) / float(self.file_size)) )
            pie.SetEntryFillColor(i, i+2)
          
        
        c1 = ROOT.TCanvas( 'c1' , 'c1' )
        
        pie.SetCircle(.5,.45,.2)    
        pie.SetLabelFormat("#splitline{%txt}{(%perc)}")
        pie.SetLabelFormat("%txt")
        pie.SetRadius(0.2)
        pie.SetTextSize(0.03)
        pie.SetCircle(0.4808696, 0.7940109, 0.2)
        pie.SetValueFormat("%4.2f")
        pie.SetLabelFormat("%txt")
        pie.SetPercentFormat("%3.1f")
        pie.SetLabelsOffset(0.005)
        pie.SetAngularOffset(265.2655)
        leg = ROOT.TLegend(0.05580866, 0.06554878, 0.785877, 0.4512195)
        for i, bgroup in enumerate( blist ):
          if ( (float(bgroup.filebytes) / float(self.file_size)) > 0.05):
              pie.SetEntryRadiusOffset(i,.07)
              bginfo = bgroup.get_dict(self.nev) 
              leg.AddEntry( pie.GetSlice(i), "%s : %1.1lf%%" % (bginfo['name'] , 100. * float(bgroup.filebytes) / float(self.file_size)) , "f" )
          else:
              pie.SetEntryLabel(i,"")
 
        leg.SetFillColor(0);
        leg.SetFillStyle(0);
        
        pie.Draw("3d")
        leg.Draw();
        
        c1.SaveAs(options.filename + "." + options.extension)
        
        
  
       
 

get_branchgroup()

def D3PDSizeSummary.Filestats.get_branchgroup	(		self,
			b
	)

Definition at line 190 of file D3PDSizeSummary.py.

    def get_branchgroup (self, b):
        bname = b.GetName()
        ll = bname.split ('_')
        if bname in ['RunNumber',
                     'EventNumber',
                     'timestamp',
                     'timestamp_ns',
                     'lbn',
                     'bcid',
                     'detmask0',
                     'detmask1',
                     'collcand']:
            gname = 'EventInfo'
        elif len(ll) > 1:
            gname = ll[0]
        elif (bname.endswith ('Error') or bname.endswith ('Flags')):
            gname = 'DetFlags'
        else:
            gname = bname
        group = self.branchgroups.get (gname)
        if not group:
            group = Branchgroup (gname)
            self.branchgroups[gname] = group
        return group
 
 

get_stats()

def D3PDSizeSummary.Filestats.get_stats

(

self

)

Definition at line 216 of file D3PDSizeSummary.py.

    def get_stats (self):
        self.file_size = os.stat(self.fname).st_size
        self.rfile = ROOT.TFile.Open (self.fname)
        self.get_treename()
        if self.treename == None:
            raise Exception ("Can't find tree name in file.")
        self.tree = self.rfile.Get (self.treename)
        if not self.tree:
            raise Exception ("Can't find tree " + self.treename + " in file.")
        self.nev = self.tree.GetEntries()
        self.scan_branches()
        return
 
 

get_treename()

def D3PDSizeSummary.Filestats.get_treename

(

self

)

Definition at line 181 of file D3PDSizeSummary.py.

    def get_treename (self):
        if self.treename: return
        for kk in self.rfile.GetListOfKeys():
            if (kk.GetClassName() == 'TTree' and
                kk.GetName() != 'CollectionTree'):
                self.treename = kk.GetName()
        return
 
 

print_stats_text()

def D3PDSizeSummary.Filestats.print_stats_text

(

self

)

Definition at line 237 of file D3PDSizeSummary.py.

    def print_stats_text (self):
        print "Number of events:", self.nev
        print "File size: %dk" % int(float(self.file_size)/1024 + 0.5)
        print "Overall size/event: %.1fk" % (float(self.file_size) / self.nev / 1024)
 
        bl = self.branchgroups.values()
        bl.sort (lambda a, b: int(b.filebytes - a.filebytes))
        Branchgroup.print_header()
        for bg in bl:
            bg.print_stats (self.nev)
        Branchgroup.print_trailer()
        return
 
 

scan_branches()

def D3PDSizeSummary.Filestats.scan_branches

(

self

)

Definition at line 230 of file D3PDSizeSummary.py.

    def scan_branches (self):
        for b in self.tree.GetListOfBranches():
            bg = self.get_branchgroup (b)
            bg.add (b)
        return
 
 

Member Data Documentation

branchgroups

D3PDSizeSummary.Filestats.branchgroups

Definition at line 176 of file D3PDSizeSummary.py.

file_size

D3PDSizeSummary.Filestats.file_size

Definition at line 217 of file D3PDSizeSummary.py.

fname

D3PDSizeSummary.Filestats.fname

Definition at line 174 of file D3PDSizeSummary.py.

nev

D3PDSizeSummary.Filestats.nev

Definition at line 225 of file D3PDSizeSummary.py.

rfile

D3PDSizeSummary.Filestats.rfile

Definition at line 218 of file D3PDSizeSummary.py.

tree

D3PDSizeSummary.Filestats.tree

Definition at line 222 of file D3PDSizeSummary.py.

treename

D3PDSizeSummary.Filestats.treename

Definition at line 175 of file D3PDSizeSummary.py.

---

The documentation for this class was generated from the following file:

• D3PDSizeSummary.py