barchart.py revision 2160
12115Shsul@eecs.umich.edu# Copyright (c) 2005-2006 The Regents of The University of Michigan 21881Sbinkertn@umich.edu# All rights reserved. 31881Sbinkertn@umich.edu# 41881Sbinkertn@umich.edu# Redistribution and use in source and binary forms, with or without 51881Sbinkertn@umich.edu# modification, are permitted provided that the following conditions are 61881Sbinkertn@umich.edu# met: redistributions of source code must retain the above copyright 71881Sbinkertn@umich.edu# notice, this list of conditions and the following disclaimer; 81881Sbinkertn@umich.edu# redistributions in binary form must reproduce the above copyright 91881Sbinkertn@umich.edu# notice, this list of conditions and the following disclaimer in the 101881Sbinkertn@umich.edu# documentation and/or other materials provided with the distribution; 111881Sbinkertn@umich.edu# neither the name of the copyright holders nor the names of its 121881Sbinkertn@umich.edu# contributors may be used to endorse or promote products derived from 131881Sbinkertn@umich.edu# this software without specific prior written permission. 141881Sbinkertn@umich.edu# 151881Sbinkertn@umich.edu# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 161881Sbinkertn@umich.edu# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 171881Sbinkertn@umich.edu# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 181881Sbinkertn@umich.edu# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 191881Sbinkertn@umich.edu# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 201881Sbinkertn@umich.edu# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 211881Sbinkertn@umich.edu# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 221881Sbinkertn@umich.edu# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 231881Sbinkertn@umich.edu# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 241881Sbinkertn@umich.edu# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 251881Sbinkertn@umich.edu# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 261881Sbinkertn@umich.edu# 271881Sbinkertn@umich.edu# Authors: Nathan Binkert 281881Sbinkertn@umich.edu# Lisa Hsu 291881Sbinkertn@umich.edu 301881Sbinkertn@umich.eduimport matplotlib, pylab 312006Sbinkertn@umich.edufrom matplotlib.font_manager import FontProperties 321881Sbinkertn@umich.edufrom matplotlib.numerix import array, arange, reshape, shape, transpose, zeros 331881Sbinkertn@umich.edufrom matplotlib.numerix import Float 342119Shsul@eecs.umich.edufrom matplotlib.ticker import NullLocator 351881Sbinkertn@umich.edu 361881Sbinkertn@umich.edumatplotlib.interactive(False) 371881Sbinkertn@umich.edu 382006Sbinkertn@umich.edufrom chart import ChartOptions 391881Sbinkertn@umich.edu 402006Sbinkertn@umich.educlass BarChart(ChartOptions): 412006Sbinkertn@umich.edu def __init__(self, default=None, **kwargs): 422006Sbinkertn@umich.edu super(BarChart, self).__init__(default, **kwargs) 431881Sbinkertn@umich.edu self.inputdata = None 441881Sbinkertn@umich.edu self.chartdata = None 451881Sbinkertn@umich.edu 461881Sbinkertn@umich.edu def gen_colors(self, count): 471881Sbinkertn@umich.edu cmap = matplotlib.cm.get_cmap(self.colormap) 481881Sbinkertn@umich.edu if count == 1: 491881Sbinkertn@umich.edu return cmap([ 0.5 ]) 501881Sbinkertn@umich.edu else: 511881Sbinkertn@umich.edu return cmap(arange(count) / float(count - 1)) 521881Sbinkertn@umich.edu 531881Sbinkertn@umich.edu # The input data format does not match the data format that the 541881Sbinkertn@umich.edu # graph function takes because it is intuitive. The conversion 551881Sbinkertn@umich.edu # from input data format to chart data format depends on the 561881Sbinkertn@umich.edu # dimensionality of the input data. Check here for the 571881Sbinkertn@umich.edu # dimensionality and correctness of the input data 581881Sbinkertn@umich.edu def set_data(self, data): 591881Sbinkertn@umich.edu if data is None: 601881Sbinkertn@umich.edu self.inputdata = None 611881Sbinkertn@umich.edu self.chartdata = None 621881Sbinkertn@umich.edu return 631881Sbinkertn@umich.edu 641881Sbinkertn@umich.edu data = array(data) 651881Sbinkertn@umich.edu dim = len(shape(data)) 661881Sbinkertn@umich.edu if dim not in (1, 2, 3): 671881Sbinkertn@umich.edu raise AttributeError, "Input data must be a 1, 2, or 3d matrix" 681881Sbinkertn@umich.edu self.inputdata = data 691881Sbinkertn@umich.edu 701881Sbinkertn@umich.edu # If the input data is a 1d matrix, then it describes a 711881Sbinkertn@umich.edu # standard bar chart. 721881Sbinkertn@umich.edu if dim == 1: 731881Sbinkertn@umich.edu self.chartdata = array([[data]]) 741881Sbinkertn@umich.edu 751881Sbinkertn@umich.edu # If the input data is a 2d matrix, then it describes a bar 761881Sbinkertn@umich.edu # chart with groups. The matrix being an array of groups of 771881Sbinkertn@umich.edu # bars. 781881Sbinkertn@umich.edu if dim == 2: 791881Sbinkertn@umich.edu self.chartdata = transpose([data], axes=(2,0,1)) 801881Sbinkertn@umich.edu 811881Sbinkertn@umich.edu # If the input data is a 3d matrix, then it describes an array 821881Sbinkertn@umich.edu # of groups of bars with each bar being an array of stacked 831881Sbinkertn@umich.edu # values. 841881Sbinkertn@umich.edu if dim == 3: 851881Sbinkertn@umich.edu self.chartdata = transpose(data, axes=(1,2,0)) 861881Sbinkertn@umich.edu 871881Sbinkertn@umich.edu def get_data(self): 881881Sbinkertn@umich.edu return self.inputdata 891881Sbinkertn@umich.edu 901881Sbinkertn@umich.edu data = property(get_data, set_data) 911881Sbinkertn@umich.edu 921881Sbinkertn@umich.edu # Graph the chart data. 931881Sbinkertn@umich.edu # Input is a 3d matrix that describes a plot that has multiple 941881Sbinkertn@umich.edu # groups, multiple bars in each group, and multiple values stacked 951881Sbinkertn@umich.edu # in each bar. The underlying bar() function expects a sequence of 961881Sbinkertn@umich.edu # bars in the same stack location and same group location, so the 971881Sbinkertn@umich.edu # organization of the matrix is that the inner most sequence 981881Sbinkertn@umich.edu # represents one of these bar groups, then those are grouped 991881Sbinkertn@umich.edu # together to make one full stack of bars in each group, and then 1001881Sbinkertn@umich.edu # the outer most layer describes the groups. Here is an example 1011881Sbinkertn@umich.edu # data set and how it gets plotted as a result. 1021881Sbinkertn@umich.edu # 1031881Sbinkertn@umich.edu # e.g. data = [[[10,11,12], [13,14,15], [16,17,18], [19,20,21]], 1041881Sbinkertn@umich.edu # [[22,23,24], [25,26,27], [28,29,30], [31,32,33]]] 1051881Sbinkertn@umich.edu # 1061881Sbinkertn@umich.edu # will plot like this: 1071881Sbinkertn@umich.edu # 1081881Sbinkertn@umich.edu # 19 31 20 32 21 33 1091881Sbinkertn@umich.edu # 16 28 17 29 18 30 1101881Sbinkertn@umich.edu # 13 25 14 26 15 27 1111881Sbinkertn@umich.edu # 10 22 11 23 12 24 1121881Sbinkertn@umich.edu # 1131881Sbinkertn@umich.edu # Because this arrangement is rather conterintuitive, the rearrange 1141881Sbinkertn@umich.edu # function takes various matricies and arranges them to fit this 1151881Sbinkertn@umich.edu # profile. 1161881Sbinkertn@umich.edu # 1171881Sbinkertn@umich.edu # This code deals with one of the dimensions in the matrix being 1181881Sbinkertn@umich.edu # one wide. 1191881Sbinkertn@umich.edu # 1201881Sbinkertn@umich.edu def graph(self): 1211881Sbinkertn@umich.edu if self.chartdata is None: 1221881Sbinkertn@umich.edu raise AttributeError, "Data not set for bar chart!" 1231881Sbinkertn@umich.edu 1241881Sbinkertn@umich.edu dim = len(shape(self.inputdata)) 1251881Sbinkertn@umich.edu cshape = shape(self.chartdata) 1261881Sbinkertn@umich.edu if dim == 1: 1271881Sbinkertn@umich.edu colors = self.gen_colors(cshape[2]) 1281881Sbinkertn@umich.edu colors = [ [ colors ] * cshape[1] ] * cshape[0] 1291881Sbinkertn@umich.edu 1301881Sbinkertn@umich.edu if dim == 2: 1311881Sbinkertn@umich.edu colors = self.gen_colors(cshape[0]) 1321881Sbinkertn@umich.edu colors = [ [ [ c ] * cshape[2] ] * cshape[1] for c in colors ] 1331881Sbinkertn@umich.edu 1341881Sbinkertn@umich.edu if dim == 3: 1351881Sbinkertn@umich.edu colors = self.gen_colors(cshape[1]) 1361881Sbinkertn@umich.edu colors = [ [ [ c ] * cshape[2] for c in colors ] ] * cshape[0] 1371881Sbinkertn@umich.edu 1381881Sbinkertn@umich.edu colors = array(colors) 1391881Sbinkertn@umich.edu 1402115Shsul@eecs.umich.edu self.figure = pylab.figure(figsize=self.chart_size) 1412115Shsul@eecs.umich.edu 1422115Shsul@eecs.umich.edu outer_axes = None 1432115Shsul@eecs.umich.edu inner_axes = None 1442119Shsul@eecs.umich.edu if self.xsubticks is not None: 1452119Shsul@eecs.umich.edu color = self.figure.get_facecolor() 1462119Shsul@eecs.umich.edu self.metaaxes = self.figure.add_axes(self.figure_size, axisbg=color, frameon=False) 1472119Shsul@eecs.umich.edu for tick in self.metaaxes.xaxis.majorTicks: 1482119Shsul@eecs.umich.edu tick.tick1On = False 1492119Shsul@eecs.umich.edu tick.tick2On = False 1502115Shsul@eecs.umich.edu self.metaaxes.set_yticklabels([]) 1512115Shsul@eecs.umich.edu self.metaaxes.set_yticks([]) 1522115Shsul@eecs.umich.edu size = [0] * 4 1532115Shsul@eecs.umich.edu size[0] = self.figure_size[0] 1542160Shsul@eecs.umich.edu size[1] = self.figure_size[1] + .12 1552115Shsul@eecs.umich.edu size[2] = self.figure_size[2] 1562160Shsul@eecs.umich.edu size[3] = self.figure_size[3] - .12 1572115Shsul@eecs.umich.edu self.axes = self.figure.add_axes(size) 1582115Shsul@eecs.umich.edu outer_axes = self.metaaxes 1592115Shsul@eecs.umich.edu inner_axes = self.axes 1602115Shsul@eecs.umich.edu else: 1612115Shsul@eecs.umich.edu self.axes = self.figure.add_axes(self.figure_size) 1622115Shsul@eecs.umich.edu outer_axes = self.axes 1632115Shsul@eecs.umich.edu inner_axes = self.axes 1642115Shsul@eecs.umich.edu 1651881Sbinkertn@umich.edu bars_in_group = len(self.chartdata) 1662160Shsul@eecs.umich.edu 1672160Shsul@eecs.umich.edu width = 1.0 / ( bars_in_group + 1) 1682160Shsul@eecs.umich.edu center = width / 2 1691881Sbinkertn@umich.edu 1701881Sbinkertn@umich.edu bars = [] 1711881Sbinkertn@umich.edu for i,stackdata in enumerate(self.chartdata): 1722006Sbinkertn@umich.edu bottom = array([0.0] * len(stackdata[0]), Float) 1731881Sbinkertn@umich.edu stack = [] 1741881Sbinkertn@umich.edu for j,bardata in enumerate(stackdata): 1751881Sbinkertn@umich.edu bardata = array(bardata) 1761881Sbinkertn@umich.edu ind = arange(len(bardata)) + i * width + center 1771881Sbinkertn@umich.edu bar = self.axes.bar(ind, bardata, width, bottom=bottom, 1781881Sbinkertn@umich.edu color=colors[i][j]) 1792160Shsul@eecs.umich.edu if self.xsubticks is not None: 1802115Shsul@eecs.umich.edu self.metaaxes.bar(ind, [0] * len(bardata), width) 1811881Sbinkertn@umich.edu stack.append(bar) 1821881Sbinkertn@umich.edu bottom += bardata 1831881Sbinkertn@umich.edu bars.append(stack) 1841881Sbinkertn@umich.edu 1851881Sbinkertn@umich.edu if self.xlabel is not None: 1862115Shsul@eecs.umich.edu outer_axes.set_xlabel(self.xlabel) 1871881Sbinkertn@umich.edu 1881881Sbinkertn@umich.edu if self.ylabel is not None: 1892115Shsul@eecs.umich.edu inner_axes.set_ylabel(self.ylabel) 1901881Sbinkertn@umich.edu 1911881Sbinkertn@umich.edu if self.yticks is not None: 1921881Sbinkertn@umich.edu ymin, ymax = self.axes.get_ylim() 1931881Sbinkertn@umich.edu nticks = float(len(self.yticks)) 1941881Sbinkertn@umich.edu ticks = arange(nticks) / (nticks - 1) * (ymax - ymin) + ymin 1952115Shsul@eecs.umich.edu inner_axes.set_yticks(ticks) 1962115Shsul@eecs.umich.edu inner_axes.set_yticklabels(self.yticks) 1972006Sbinkertn@umich.edu elif self.ylim is not None: 1982115Shsul@eecs.umich.edu self.inner_axes.set_ylim(self.ylim) 1991881Sbinkertn@umich.edu 2001881Sbinkertn@umich.edu if self.xticks is not None: 2012115Shsul@eecs.umich.edu outer_axes.set_xticks(arange(cshape[2]) + .5) 2022115Shsul@eecs.umich.edu outer_axes.set_xticklabels(self.xticks) 2032119Shsul@eecs.umich.edu 2042115Shsul@eecs.umich.edu if self.xsubticks is not None: 2052115Shsul@eecs.umich.edu inner_axes.set_xticks(arange((cshape[0] + 1)*cshape[2])*width + 2*center) 2062115Shsul@eecs.umich.edu self.xsubticks.append('') 2072160Shsul@eecs.umich.edu inner_axes.set_xticklabels(self.xsubticks * cshape[2], fontsize=7, rotation=90) 2082119Shsul@eecs.umich.edu 2091881Sbinkertn@umich.edu if self.legend is not None: 2101881Sbinkertn@umich.edu if dim == 1: 2111881Sbinkertn@umich.edu lbars = bars[0][0] 2121881Sbinkertn@umich.edu if dim == 2: 2131881Sbinkertn@umich.edu lbars = [ bars[i][0][0] for i in xrange(len(bars))] 2141881Sbinkertn@umich.edu if dim == 3: 2151881Sbinkertn@umich.edu number = len(bars[0]) 2161881Sbinkertn@umich.edu lbars = [ bars[0][number - j - 1][0] for j in xrange(number)] 2171881Sbinkertn@umich.edu 2182006Sbinkertn@umich.edu self.figure.legend(lbars, self.legend, self.legend_loc, 2192006Sbinkertn@umich.edu prop=FontProperties(size=self.legend_size)) 2201881Sbinkertn@umich.edu 2211881Sbinkertn@umich.edu if self.title is not None: 2221881Sbinkertn@umich.edu self.axes.set_title(self.title) 2231881Sbinkertn@umich.edu 2241881Sbinkertn@umich.edu def savefig(self, name): 2251881Sbinkertn@umich.edu self.figure.savefig(name) 2261881Sbinkertn@umich.edu 2272006Sbinkertn@umich.edu def savecsv(self, name): 2282006Sbinkertn@umich.edu f = file(name, 'w') 2292006Sbinkertn@umich.edu data = array(self.inputdata) 2302006Sbinkertn@umich.edu dim = len(data.shape) 2312006Sbinkertn@umich.edu 2322006Sbinkertn@umich.edu if dim == 1: 2332006Sbinkertn@umich.edu #if self.xlabel: 2342006Sbinkertn@umich.edu # f.write(', '.join(list(self.xlabel)) + '\n') 2352006Sbinkertn@umich.edu f.write(', '.join([ '%f' % val for val in data]) + '\n') 2362006Sbinkertn@umich.edu if dim == 2: 2372006Sbinkertn@umich.edu #if self.xlabel: 2382006Sbinkertn@umich.edu # f.write(', '.join([''] + list(self.xlabel)) + '\n') 2392006Sbinkertn@umich.edu for i,row in enumerate(data): 2402006Sbinkertn@umich.edu ylabel = [] 2412006Sbinkertn@umich.edu #if self.ylabel: 2422006Sbinkertn@umich.edu # ylabel = [ self.ylabel[i] ] 2432006Sbinkertn@umich.edu f.write(', '.join(ylabel + [ '%f' % val for val in row]) + '\n') 2442006Sbinkertn@umich.edu if dim == 3: 2452006Sbinkertn@umich.edu f.write("don't do 3D csv files\n") 2462006Sbinkertn@umich.edu pass 2472006Sbinkertn@umich.edu 2482006Sbinkertn@umich.edu f.close() 2492006Sbinkertn@umich.edu 2501881Sbinkertn@umich.eduif __name__ == '__main__': 2512006Sbinkertn@umich.edu from random import randrange 2521881Sbinkertn@umich.edu import random, sys 2531881Sbinkertn@umich.edu 2541881Sbinkertn@umich.edu dim = 3 2551881Sbinkertn@umich.edu number = 5 2561881Sbinkertn@umich.edu 2571881Sbinkertn@umich.edu args = sys.argv[1:] 2581881Sbinkertn@umich.edu if len(args) > 3: 2591881Sbinkertn@umich.edu sys.exit("invalid number of arguments") 2601881Sbinkertn@umich.edu elif len(args) > 0: 2611881Sbinkertn@umich.edu myshape = [ int(x) for x in args ] 2621881Sbinkertn@umich.edu else: 2631881Sbinkertn@umich.edu myshape = [ 3, 4, 8 ] 2641881Sbinkertn@umich.edu 2651881Sbinkertn@umich.edu # generate a data matrix of the given shape 2661881Sbinkertn@umich.edu size = reduce(lambda x,y: x*y, myshape) 2671881Sbinkertn@umich.edu #data = [ random.randrange(size - i) + 10 for i in xrange(size) ] 2681881Sbinkertn@umich.edu data = [ float(i)/100.0 for i in xrange(size) ] 2691881Sbinkertn@umich.edu data = reshape(data, myshape) 2701881Sbinkertn@umich.edu 2711881Sbinkertn@umich.edu # setup some test bar charts 2721881Sbinkertn@umich.edu if True: 2731881Sbinkertn@umich.edu chart1 = BarChart() 2741881Sbinkertn@umich.edu chart1.data = data 2751881Sbinkertn@umich.edu 2761881Sbinkertn@umich.edu chart1.xlabel = 'Benchmark' 2771881Sbinkertn@umich.edu chart1.ylabel = 'Bandwidth (GBps)' 2781881Sbinkertn@umich.edu chart1.legend = [ 'x%d' % x for x in xrange(myshape[-1]) ] 2791881Sbinkertn@umich.edu chart1.xticks = [ 'xtick%d' % x for x in xrange(myshape[0]) ] 2801881Sbinkertn@umich.edu chart1.title = 'this is the title' 2812119Shsul@eecs.umich.edu if len(myshape) > 2: 2822117Shsul@eecs.umich.edu chart1.xsubticks = [ '%d' % x for x in xrange(myshape[1]) ] 2831881Sbinkertn@umich.edu chart1.graph() 2842006Sbinkertn@umich.edu chart1.savefig('/tmp/test1.png') 2852006Sbinkertn@umich.edu chart1.savefig('/tmp/test1.ps') 2862006Sbinkertn@umich.edu chart1.savefig('/tmp/test1.eps') 2872006Sbinkertn@umich.edu chart1.savecsv('/tmp/test1.csv') 2881881Sbinkertn@umich.edu 2891881Sbinkertn@umich.edu if False: 2901881Sbinkertn@umich.edu chart2 = BarChart() 2911881Sbinkertn@umich.edu chart2.data = data 2921881Sbinkertn@umich.edu chart2.colormap = 'gray' 2931881Sbinkertn@umich.edu chart2.graph() 2942006Sbinkertn@umich.edu chart2.savefig('/tmp/test2.png') 2952006Sbinkertn@umich.edu chart2.savefig('/tmp/test2.ps') 2961881Sbinkertn@umich.edu 2972160Shsul@eecs.umich.edu# pylab.show() 298