util/stats/output.py - public/gem5 - Git at Google

 # Copyright (c) 2005-2006 The Regents of The University of Michigan
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions are
 # met: redistributions of source code must retain the above copyright
 # notice, this list of conditions and the following disclaimer;
 # redistributions in binary form must reproduce the above copyright
 # notice, this list of conditions and the following disclaimer in the
 # documentation and/or other materials provided with the distribution;
 # neither the name of the copyright holders nor the names of its
 # contributors may be used to endorse or promote products derived from
 # this software without specific prior written permission.
 #
 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 from chart import ChartOptions

 class StatOutput(ChartOptions):
     def __init__(self, jobfile, info, stat=None):
         super(StatOutput, self).__init__()
         self.jobfile = jobfile
         self.stat = stat
         self.invert = False
         self.info = info

     def display(self, name, printmode = 'G'):
         import info

         if printmode == 'G':
             valformat = '%g'
         elif printmode != 'F' and value > 1e6:
             valformat = '%0.5e'
         else:
             valformat = '%f'

         for job in self.jobfile.jobs():
             value = self.info.get(job, self.stat)
             if value is None:
                 return

             if not isinstance(value, list):
                 value = [ value ]

             if self.invert:
                 for i,val in enumerate(value):
                     if val != 0.0:
                         value[i] = 1 / val

             valstring = ', '.join([ valformat % val for val in value ])
             print '%-50s    %s' % (job.name + ':', valstring)

     def graph(self, name, graphdir, proxy=None):
         from os.path import expanduser, isdir, join as joinpath
         from barchart import BarChart
         from matplotlib.numerix import Float, array, zeros
         import os, re, urllib
         from jobfile import crossproduct

         confgroups = self.jobfile.groups()
         ngroups = len(confgroups)
         skiplist = [ False ] * ngroups
         groupopts = []
         baropts = []
         groups = []
         for i,group in enumerate(confgroups):
             if group.flags.graph_group:
                 groupopts.append(group.subopts())
                 skiplist[i] = True
             elif group.flags.graph_bars:
                 baropts.append(group.subopts())
                 skiplist[i] = True
             else:
                 groups.append(group)

         has_group = bool(groupopts)
         if has_group:
             groupopts = [ group for group in crossproduct(groupopts) ]
         else:
             groupopts = [ None ]

         if baropts:
             baropts = [ bar for bar in crossproduct(baropts) ]
         else:
             raise AttributeError, 'No group selected for graph bars'

         directory = expanduser(graphdir)
         if not isdir(directory):
             os.mkdir(directory)
         html = file(joinpath(directory, '%s.html' % name), 'w')
         print >>html, '<html>'
         print >>html, '<title>Graphs for %s</title>' % name
         print >>html, '<body>'
         html.flush()

         for options in self.jobfile.options(groups):
             chart = BarChart(self)

             data = [ [ None ] * len(baropts) for i in xrange(len(groupopts)) ]
             enabled = False
             stacked = 0
             for g,gopt in enumerate(groupopts):
                 for b,bopt in enumerate(baropts):
                     if gopt is None:
                         gopt = []
                     job = self.jobfile.job(options + gopt + bopt)
                     if not job:
                         continue

                     if proxy:
                         import db
                         proxy.dict['system'] = self.info[job.system]
                     val = self.info.get(job, self.stat)
                     if val is None:
                         print 'stat "%s" for job "%s" not found' % \
                               (self.stat, job)

                     if isinstance(val, (list, tuple)):
                         if len(val) == 1:
                             val = val[0]
                         else:
                             stacked = len(val)

                     data[g][b] = val

             if stacked == 0:
                 for i in xrange(len(groupopts)):
                     for j in xrange(len(baropts)):
                         if data[i][j] is None:
                             data[i][j] = 0.0
             else:
                 for i in xrange(len(groupopts)):
                     for j in xrange(len(baropts)):
                         val = data[i][j]
                         if val is None:
                             data[i][j] = [ 0.0 ] * stacked
                         elif len(val) != stacked:
                             raise ValueError, "some stats stacked, some not"

             data = array(data)
             if data.sum() == 0:
                 continue

             dim = len(data.shape)
             x = data.shape[0]
             xkeep = [ i for i in xrange(x) if data[i].sum() != 0 ]
             y = data.shape[1]
             ykeep = [ i for i in xrange(y) if data[:,i].sum() != 0 ]
             data = data.take(xkeep, axis=0)
             data = data.take(ykeep, axis=1)
             if not has_group:
                 data = data.take([ 0 ], axis=0)
             chart.data = data


             bopts = [ baropts[i] for i in ykeep ]
             bdescs = [ ' '.join([o.desc for o in opt]) for opt in bopts]

             if has_group:
                 gopts = [ groupopts[i] for i in xkeep ]
                 gdescs = [ ' '.join([o.desc for o in opt]) for opt in gopts]

             if chart.legend is None:
                 if stacked:
                     try:
                         chart.legend = self.info.rcategories
                     except:
                         chart.legend = [ str(i) for i in xrange(stacked) ]
                 else:
                     chart.legend = bdescs

             if chart.xticks is None:
                 if has_group:
                     chart.xticks = gdescs
                 else:
                     chart.xticks = []
             chart.graph()

             names = [ opt.name for opt in options ]
             descs = [ opt.desc for opt in options ]

             if names[0] == 'run':
                 names = names[1:]
                 descs = descs[1:]

             basename = '%s-%s' % (name, ':'.join(names))
             desc = ' '.join(descs)

             pngname = '%s.png' % basename
             psname = '%s.eps' % re.sub(':', '-', basename)
             epsname = '%s.ps' % re.sub(':', '-', basename)
             chart.savefig(joinpath(directory, pngname))
             chart.savefig(joinpath(directory, epsname))
             chart.savefig(joinpath(directory, psname))
             html_name = urllib.quote(pngname)
             print >>html, '''%s<br><img src="%s"><br>''' % (desc, html_name)
             html.flush()

         print >>html, '</body>'
         print >>html, '</html>'
         html.close()
	# Copyright (c) 2005-2006 The Regents of The University of Michigan
	# All rights reserved.
	#
	# Redistribution and use in source and binary forms, with or without
	# modification, are permitted provided that the following conditions are
	# met: redistributions of source code must retain the above copyright
	# notice, this list of conditions and the following disclaimer;
	# redistributions in binary form must reproduce the above copyright
	# notice, this list of conditions and the following disclaimer in the
	# documentation and/or other materials provided with the distribution;
	# neither the name of the copyright holders nor the names of its
	# contributors may be used to endorse or promote products derived from
	# this software without specific prior written permission.
	#
	# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
	# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
	# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
	# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
	# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
	# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
	# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
	# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
	# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
	# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
	# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

	from chart import ChartOptions

	class StatOutput(ChartOptions):
	def __init__(self, jobfile, info, stat=None):
	super(StatOutput, self).__init__()
	self.jobfile = jobfile
	self.stat = stat
	self.invert = False
	self.info = info

	def display(self, name, printmode = 'G'):
	import info

	if printmode == 'G':
	valformat = '%g'
	elif printmode != 'F' and value > 1e6:
	valformat = '%0.5e'
	else:
	valformat = '%f'

	for job in self.jobfile.jobs():
	value = self.info.get(job, self.stat)
	if value is None:
	return

	if not isinstance(value, list):
	value = [ value ]

	if self.invert:
	for i,val in enumerate(value):
	if val != 0.0:
	value[i] = 1 / val

	valstring = ', '.join([ valformat % val for val in value ])
	print '%-50s %s' % (job.name + ':', valstring)

	def graph(self, name, graphdir, proxy=None):
	from os.path import expanduser, isdir, join as joinpath
	from barchart import BarChart
	from matplotlib.numerix import Float, array, zeros
	import os, re, urllib
	from jobfile import crossproduct

	confgroups = self.jobfile.groups()
	ngroups = len(confgroups)
	skiplist = [ False ] * ngroups
	groupopts = []
	baropts = []
	groups = []
	for i,group in enumerate(confgroups):
	if group.flags.graph_group:
	groupopts.append(group.subopts())
	skiplist[i] = True
	elif group.flags.graph_bars:
	baropts.append(group.subopts())
	skiplist[i] = True
	else:
	groups.append(group)

	has_group = bool(groupopts)
	if has_group:
	groupopts = [ group for group in crossproduct(groupopts) ]
	else:
	groupopts = [ None ]

	if baropts:
	baropts = [ bar for bar in crossproduct(baropts) ]
	else:
	raise AttributeError, 'No group selected for graph bars'

	directory = expanduser(graphdir)
	if not isdir(directory):
	os.mkdir(directory)
	html = file(joinpath(directory, '%s.html' % name), 'w')
	print >>html, '<html>'
	print >>html, '<title>Graphs for %s</title>' % name
	print >>html, '<body>'
	html.flush()

	for options in self.jobfile.options(groups):
	chart = BarChart(self)

	data = [ [ None ] * len(baropts) for i in xrange(len(groupopts)) ]
	enabled = False
	stacked = 0
	for g,gopt in enumerate(groupopts):
	for b,bopt in enumerate(baropts):
	if gopt is None:
	gopt = []
	job = self.jobfile.job(options + gopt + bopt)
	if not job:
	continue

	if proxy:
	import db
	proxy.dict['system'] = self.info[job.system]
	val = self.info.get(job, self.stat)
	if val is None:
	print 'stat "%s" for job "%s" not found' % \
	(self.stat, job)

	if isinstance(val, (list, tuple)):
	if len(val) == 1:
	val = val[0]
	else:
	stacked = len(val)

	data[g][b] = val

	if stacked == 0:
	for i in xrange(len(groupopts)):
	for j in xrange(len(baropts)):
	if data[i][j] is None:
	data[i][j] = 0.0
	else:
	for i in xrange(len(groupopts)):
	for j in xrange(len(baropts)):
	val = data[i][j]
	if val is None:
	data[i][j] = [ 0.0 ] * stacked
	elif len(val) != stacked:
	raise ValueError, "some stats stacked, some not"

	data = array(data)
	if data.sum() == 0:
	continue

	dim = len(data.shape)
	x = data.shape[0]
	xkeep = [ i for i in xrange(x) if data[i].sum() != 0 ]
	y = data.shape[1]
	ykeep = [ i for i in xrange(y) if data[:,i].sum() != 0 ]
	data = data.take(xkeep, axis=0)
	data = data.take(ykeep, axis=1)
	if not has_group:
	data = data.take([ 0 ], axis=0)
	chart.data = data


	bopts = [ baropts[i] for i in ykeep ]
	bdescs = [ ' '.join([o.desc for o in opt]) for opt in bopts]

	if has_group:
	gopts = [ groupopts[i] for i in xkeep ]
	gdescs = [ ' '.join([o.desc for o in opt]) for opt in gopts]

	if chart.legend is None:
	if stacked:
	try:
	chart.legend = self.info.rcategories
	except:
	chart.legend = [ str(i) for i in xrange(stacked) ]
	else:
	chart.legend = bdescs

	if chart.xticks is None:
	if has_group:
	chart.xticks = gdescs
	else:
	chart.xticks = []
	chart.graph()

	names = [ opt.name for opt in options ]
	descs = [ opt.desc for opt in options ]

	if names[0] == 'run':
	names = names[1:]
	descs = descs[1:]

	basename = '%s-%s' % (name, ':'.join(names))
	desc = ' '.join(descs)

	pngname = '%s.png' % basename
	psname = '%s.eps' % re.sub(':', '-', basename)
	epsname = '%s.ps' % re.sub(':', '-', basename)
	chart.savefig(joinpath(directory, pngname))
	chart.savefig(joinpath(directory, epsname))
	chart.savefig(joinpath(directory, psname))
	html_name = urllib.quote(pngname)
	print >>html, '''%s<br><img src="%s"><br>''' % (desc, html_name)
	html.flush()

	print >>html, '</body>'
	print >>html, '</html>'
	html.close()