Collaboration diagram for D3PDSizeSummary.Filestats:

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

Public Attributes
	fname = fname
return	treename = treename
dict	branchgroups = {}
	file_size = os.stat(self.fname).st_size
	rfile = ROOT.TFile.Open (self.fname)
	tree = self.rfile.Get (self.treename)
	nev = self.tree.GetEntries()

Detailed Description

Definition at line 172 of file D3PDSizeSummary.py.

Constructor & Destructor Documentation

◆ init()

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()

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()

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()

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()

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()

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()

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 = os.stat(self.fname).st_size

Definition at line 217 of file D3PDSizeSummary.py.

◆ fname

D3PDSizeSummary.Filestats.fname = fname

Definition at line 174 of file D3PDSizeSummary.py.

◆ nev

D3PDSizeSummary.Filestats.nev = self.tree.GetEntries()

Definition at line 225 of file D3PDSizeSummary.py.

◆ rfile

D3PDSizeSummary.Filestats.rfile = ROOT.TFile.Open (self.fname)

Definition at line 218 of file D3PDSizeSummary.py.

◆ tree

D3PDSizeSummary.Filestats.tree = self.rfile.Get (self.treename)

Definition at line 222 of file D3PDSizeSummary.py.

◆ treename

return D3PDSizeSummary.Filestats.treename = treename

Definition at line 175 of file D3PDSizeSummary.py.

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

D3PDSizeSummary.py

Public Member Functions

Public Attributes