barchart.py revision 2179
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 ]) 502179Sbinkertn@umich.edu 512179Sbinkertn@umich.edu if count < 5: 522179Sbinkertn@umich.edu return cmap(arange(5) / float(4))[:count] 532179Sbinkertn@umich.edu 542179Sbinkertn@umich.edu return cmap(arange(count) / float(count - 1)) 551881Sbinkertn@umich.edu 561881Sbinkertn@umich.edu # The input data format does not match the data format that the 571881Sbinkertn@umich.edu # graph function takes because it is intuitive. The conversion 581881Sbinkertn@umich.edu # from input data format to chart data format depends on the 591881Sbinkertn@umich.edu # dimensionality of the input data. Check here for the 601881Sbinkertn@umich.edu # dimensionality and correctness of the input data 611881Sbinkertn@umich.edu def set_data(self, data): 621881Sbinkertn@umich.edu if data is None: 631881Sbinkertn@umich.edu self.inputdata = None 641881Sbinkertn@umich.edu self.chartdata = None 651881Sbinkertn@umich.edu return 661881Sbinkertn@umich.edu 671881Sbinkertn@umich.edu data = array(data) 681881Sbinkertn@umich.edu dim = len(shape(data)) 691881Sbinkertn@umich.edu if dim not in (1, 2, 3): 701881Sbinkertn@umich.edu raise AttributeError, "Input data must be a 1, 2, or 3d matrix" 711881Sbinkertn@umich.edu self.inputdata = data 721881Sbinkertn@umich.edu 731881Sbinkertn@umich.edu # If the input data is a 1d matrix, then it describes a 741881Sbinkertn@umich.edu # standard bar chart. 751881Sbinkertn@umich.edu if dim == 1: 761881Sbinkertn@umich.edu self.chartdata = array([[data]]) 771881Sbinkertn@umich.edu 781881Sbinkertn@umich.edu # If the input data is a 2d matrix, then it describes a bar 791881Sbinkertn@umich.edu # chart with groups. The matrix being an array of groups of 801881Sbinkertn@umich.edu # bars. 811881Sbinkertn@umich.edu if dim == 2: 821881Sbinkertn@umich.edu self.chartdata = transpose([data], axes=(2,0,1)) 831881Sbinkertn@umich.edu 841881Sbinkertn@umich.edu # If the input data is a 3d matrix, then it describes an array 851881Sbinkertn@umich.edu # of groups of bars with each bar being an array of stacked 861881Sbinkertn@umich.edu # values. 871881Sbinkertn@umich.edu if dim == 3: 881881Sbinkertn@umich.edu self.chartdata = transpose(data, axes=(1,2,0)) 891881Sbinkertn@umich.edu 901881Sbinkertn@umich.edu def get_data(self): 911881Sbinkertn@umich.edu return self.inputdata 921881Sbinkertn@umich.edu 931881Sbinkertn@umich.edu data = property(get_data, set_data) 941881Sbinkertn@umich.edu 951881Sbinkertn@umich.edu # Graph the chart data. 961881Sbinkertn@umich.edu # Input is a 3d matrix that describes a plot that has multiple 971881Sbinkertn@umich.edu # groups, multiple bars in each group, and multiple values stacked 981881Sbinkertn@umich.edu # in each bar. The underlying bar() function expects a sequence of 991881Sbinkertn@umich.edu # bars in the same stack location and same group location, so the 1001881Sbinkertn@umich.edu # organization of the matrix is that the inner most sequence 1011881Sbinkertn@umich.edu # represents one of these bar groups, then those are grouped 1021881Sbinkertn@umich.edu # together to make one full stack of bars in each group, and then 1031881Sbinkertn@umich.edu # the outer most layer describes the groups. Here is an example 1041881Sbinkertn@umich.edu # data set and how it gets plotted as a result. 1051881Sbinkertn@umich.edu # 1061881Sbinkertn@umich.edu # e.g. data = [[[10,11,12], [13,14,15], [16,17,18], [19,20,21]], 1071881Sbinkertn@umich.edu # [[22,23,24], [25,26,27], [28,29,30], [31,32,33]]] 1081881Sbinkertn@umich.edu # 1091881Sbinkertn@umich.edu # will plot like this: 1101881Sbinkertn@umich.edu # 1111881Sbinkertn@umich.edu # 19 31 20 32 21 33 1121881Sbinkertn@umich.edu # 16 28 17 29 18 30 1131881Sbinkertn@umich.edu # 13 25 14 26 15 27 1141881Sbinkertn@umich.edu # 10 22 11 23 12 24 1151881Sbinkertn@umich.edu # 1161881Sbinkertn@umich.edu # Because this arrangement is rather conterintuitive, the rearrange 1171881Sbinkertn@umich.edu # function takes various matricies and arranges them to fit this 1181881Sbinkertn@umich.edu # profile. 1191881Sbinkertn@umich.edu # 1201881Sbinkertn@umich.edu # This code deals with one of the dimensions in the matrix being 1211881Sbinkertn@umich.edu # one wide. 1221881Sbinkertn@umich.edu # 1231881Sbinkertn@umich.edu def graph(self): 1241881Sbinkertn@umich.edu if self.chartdata is None: 1251881Sbinkertn@umich.edu raise AttributeError, "Data not set for bar chart!" 1261881Sbinkertn@umich.edu 1271881Sbinkertn@umich.edu dim = len(shape(self.inputdata)) 1281881Sbinkertn@umich.edu cshape = shape(self.chartdata) 1291881Sbinkertn@umich.edu if dim == 1: 1301881Sbinkertn@umich.edu colors = self.gen_colors(cshape[2]) 1311881Sbinkertn@umich.edu colors = [ [ colors ] * cshape[1] ] * cshape[0] 1321881Sbinkertn@umich.edu 1331881Sbinkertn@umich.edu if dim == 2: 1341881Sbinkertn@umich.edu colors = self.gen_colors(cshape[0]) 1351881Sbinkertn@umich.edu colors = [ [ [ c ] * cshape[2] ] * cshape[1] for c in colors ] 1361881Sbinkertn@umich.edu 1371881Sbinkertn@umich.edu if dim == 3: 1381881Sbinkertn@umich.edu colors = self.gen_colors(cshape[1]) 1391881Sbinkertn@umich.edu colors = [ [ [ c ] * cshape[2] for c in colors ] ] * cshape[0] 1401881Sbinkertn@umich.edu 1411881Sbinkertn@umich.edu colors = array(colors) 1421881Sbinkertn@umich.edu 1432115Shsul@eecs.umich.edu self.figure = pylab.figure(figsize=self.chart_size) 1442115Shsul@eecs.umich.edu 1452115Shsul@eecs.umich.edu outer_axes = None 1462115Shsul@eecs.umich.edu inner_axes = None 1472119Shsul@eecs.umich.edu if self.xsubticks is not None: 1482119Shsul@eecs.umich.edu color = self.figure.get_facecolor() 1492119Shsul@eecs.umich.edu self.metaaxes = self.figure.add_axes(self.figure_size, axisbg=color, frameon=False) 1502119Shsul@eecs.umich.edu for tick in self.metaaxes.xaxis.majorTicks: 1512119Shsul@eecs.umich.edu tick.tick1On = False 1522119Shsul@eecs.umich.edu tick.tick2On = False 1532115Shsul@eecs.umich.edu self.metaaxes.set_yticklabels([]) 1542115Shsul@eecs.umich.edu self.metaaxes.set_yticks([]) 1552115Shsul@eecs.umich.edu size = [0] * 4 1562115Shsul@eecs.umich.edu size[0] = self.figure_size[0] 1572160Shsul@eecs.umich.edu size[1] = self.figure_size[1] + .12 1582115Shsul@eecs.umich.edu size[2] = self.figure_size[2] 1592160Shsul@eecs.umich.edu size[3] = self.figure_size[3] - .12 1602115Shsul@eecs.umich.edu self.axes = self.figure.add_axes(size) 1612115Shsul@eecs.umich.edu outer_axes = self.metaaxes 1622115Shsul@eecs.umich.edu inner_axes = self.axes 1632115Shsul@eecs.umich.edu else: 1642115Shsul@eecs.umich.edu self.axes = self.figure.add_axes(self.figure_size) 1652115Shsul@eecs.umich.edu outer_axes = self.axes 1662115Shsul@eecs.umich.edu inner_axes = self.axes 1672115Shsul@eecs.umich.edu 1681881Sbinkertn@umich.edu bars_in_group = len(self.chartdata) 1692160Shsul@eecs.umich.edu 1702160Shsul@eecs.umich.edu width = 1.0 / ( bars_in_group + 1) 1712160Shsul@eecs.umich.edu center = width / 2 1721881Sbinkertn@umich.edu 1731881Sbinkertn@umich.edu bars = [] 1741881Sbinkertn@umich.edu for i,stackdata in enumerate(self.chartdata): 1752006Sbinkertn@umich.edu bottom = array([0.0] * len(stackdata[0]), Float) 1761881Sbinkertn@umich.edu stack = [] 1771881Sbinkertn@umich.edu for j,bardata in enumerate(stackdata): 1781881Sbinkertn@umich.edu bardata = array(bardata) 1791881Sbinkertn@umich.edu ind = arange(len(bardata)) + i * width + center 1801881Sbinkertn@umich.edu bar = self.axes.bar(ind, bardata, width, bottom=bottom, 1811881Sbinkertn@umich.edu color=colors[i][j]) 1822160Shsul@eecs.umich.edu if self.xsubticks is not None: 1832115Shsul@eecs.umich.edu self.metaaxes.bar(ind, [0] * len(bardata), width) 1841881Sbinkertn@umich.edu stack.append(bar) 1851881Sbinkertn@umich.edu bottom += bardata 1861881Sbinkertn@umich.edu bars.append(stack) 1871881Sbinkertn@umich.edu 1881881Sbinkertn@umich.edu if self.xlabel is not None: 1892115Shsul@eecs.umich.edu outer_axes.set_xlabel(self.xlabel) 1901881Sbinkertn@umich.edu 1911881Sbinkertn@umich.edu if self.ylabel is not None: 1922115Shsul@eecs.umich.edu inner_axes.set_ylabel(self.ylabel) 1931881Sbinkertn@umich.edu 1941881Sbinkertn@umich.edu if self.yticks is not None: 1951881Sbinkertn@umich.edu ymin, ymax = self.axes.get_ylim() 1961881Sbinkertn@umich.edu nticks = float(len(self.yticks)) 1971881Sbinkertn@umich.edu ticks = arange(nticks) / (nticks - 1) * (ymax - ymin) + ymin 1982115Shsul@eecs.umich.edu inner_axes.set_yticks(ticks) 1992115Shsul@eecs.umich.edu inner_axes.set_yticklabels(self.yticks) 2002006Sbinkertn@umich.edu elif self.ylim is not None: 2012115Shsul@eecs.umich.edu self.inner_axes.set_ylim(self.ylim) 2021881Sbinkertn@umich.edu 2031881Sbinkertn@umich.edu if self.xticks is not None: 2042115Shsul@eecs.umich.edu outer_axes.set_xticks(arange(cshape[2]) + .5) 2052115Shsul@eecs.umich.edu outer_axes.set_xticklabels(self.xticks) 2062119Shsul@eecs.umich.edu 2072115Shsul@eecs.umich.edu if self.xsubticks is not None: 2082115Shsul@eecs.umich.edu inner_axes.set_xticks(arange((cshape[0] + 1)*cshape[2])*width + 2*center) 2092115Shsul@eecs.umich.edu self.xsubticks.append('') 2102160Shsul@eecs.umich.edu inner_axes.set_xticklabels(self.xsubticks * cshape[2], fontsize=7, rotation=90) 2112119Shsul@eecs.umich.edu 2121881Sbinkertn@umich.edu if self.legend is not None: 2131881Sbinkertn@umich.edu if dim == 1: 2141881Sbinkertn@umich.edu lbars = bars[0][0] 2151881Sbinkertn@umich.edu if dim == 2: 2161881Sbinkertn@umich.edu lbars = [ bars[i][0][0] for i in xrange(len(bars))] 2171881Sbinkertn@umich.edu if dim == 3: 2181881Sbinkertn@umich.edu number = len(bars[0]) 2191881Sbinkertn@umich.edu lbars = [ bars[0][number - j - 1][0] for j in xrange(number)] 2201881Sbinkertn@umich.edu 2212006Sbinkertn@umich.edu self.figure.legend(lbars, self.legend, self.legend_loc, 2222006Sbinkertn@umich.edu prop=FontProperties(size=self.legend_size)) 2231881Sbinkertn@umich.edu 2241881Sbinkertn@umich.edu if self.title is not None: 2251881Sbinkertn@umich.edu self.axes.set_title(self.title) 2261881Sbinkertn@umich.edu 2271881Sbinkertn@umich.edu def savefig(self, name): 2281881Sbinkertn@umich.edu self.figure.savefig(name) 2291881Sbinkertn@umich.edu 2302006Sbinkertn@umich.edu def savecsv(self, name): 2312006Sbinkertn@umich.edu f = file(name, 'w') 2322006Sbinkertn@umich.edu data = array(self.inputdata) 2332006Sbinkertn@umich.edu dim = len(data.shape) 2342006Sbinkertn@umich.edu 2352006Sbinkertn@umich.edu if dim == 1: 2362006Sbinkertn@umich.edu #if self.xlabel: 2372006Sbinkertn@umich.edu # f.write(', '.join(list(self.xlabel)) + '\n') 2382006Sbinkertn@umich.edu f.write(', '.join([ '%f' % val for val in data]) + '\n') 2392006Sbinkertn@umich.edu if dim == 2: 2402006Sbinkertn@umich.edu #if self.xlabel: 2412006Sbinkertn@umich.edu # f.write(', '.join([''] + list(self.xlabel)) + '\n') 2422006Sbinkertn@umich.edu for i,row in enumerate(data): 2432006Sbinkertn@umich.edu ylabel = [] 2442006Sbinkertn@umich.edu #if self.ylabel: 2452006Sbinkertn@umich.edu # ylabel = [ self.ylabel[i] ] 2462006Sbinkertn@umich.edu f.write(', '.join(ylabel + [ '%f' % val for val in row]) + '\n') 2472006Sbinkertn@umich.edu if dim == 3: 2482006Sbinkertn@umich.edu f.write("don't do 3D csv files\n") 2492006Sbinkertn@umich.edu pass 2502006Sbinkertn@umich.edu 2512006Sbinkertn@umich.edu f.close() 2522006Sbinkertn@umich.edu 2531881Sbinkertn@umich.eduif __name__ == '__main__': 2542006Sbinkertn@umich.edu from random import randrange 2551881Sbinkertn@umich.edu import random, sys 2561881Sbinkertn@umich.edu 2571881Sbinkertn@umich.edu dim = 3 2581881Sbinkertn@umich.edu number = 5 2591881Sbinkertn@umich.edu 2601881Sbinkertn@umich.edu args = sys.argv[1:] 2611881Sbinkertn@umich.edu if len(args) > 3: 2621881Sbinkertn@umich.edu sys.exit("invalid number of arguments") 2631881Sbinkertn@umich.edu elif len(args) > 0: 2641881Sbinkertn@umich.edu myshape = [ int(x) for x in args ] 2651881Sbinkertn@umich.edu else: 2661881Sbinkertn@umich.edu myshape = [ 3, 4, 8 ] 2671881Sbinkertn@umich.edu 2681881Sbinkertn@umich.edu # generate a data matrix of the given shape 2691881Sbinkertn@umich.edu size = reduce(lambda x,y: x*y, myshape) 2701881Sbinkertn@umich.edu #data = [ random.randrange(size - i) + 10 for i in xrange(size) ] 2711881Sbinkertn@umich.edu data = [ float(i)/100.0 for i in xrange(size) ] 2721881Sbinkertn@umich.edu data = reshape(data, myshape) 2731881Sbinkertn@umich.edu 2741881Sbinkertn@umich.edu # setup some test bar charts 2751881Sbinkertn@umich.edu if True: 2761881Sbinkertn@umich.edu chart1 = BarChart() 2771881Sbinkertn@umich.edu chart1.data = data 2781881Sbinkertn@umich.edu 2791881Sbinkertn@umich.edu chart1.xlabel = 'Benchmark' 2801881Sbinkertn@umich.edu chart1.ylabel = 'Bandwidth (GBps)' 2811881Sbinkertn@umich.edu chart1.legend = [ 'x%d' % x for x in xrange(myshape[-1]) ] 2821881Sbinkertn@umich.edu chart1.xticks = [ 'xtick%d' % x for x in xrange(myshape[0]) ] 2831881Sbinkertn@umich.edu chart1.title = 'this is the title' 2842119Shsul@eecs.umich.edu if len(myshape) > 2: 2852117Shsul@eecs.umich.edu chart1.xsubticks = [ '%d' % x for x in xrange(myshape[1]) ] 2861881Sbinkertn@umich.edu chart1.graph() 2872006Sbinkertn@umich.edu chart1.savefig('/tmp/test1.png') 2882006Sbinkertn@umich.edu chart1.savefig('/tmp/test1.ps') 2892006Sbinkertn@umich.edu chart1.savefig('/tmp/test1.eps') 2902006Sbinkertn@umich.edu chart1.savecsv('/tmp/test1.csv') 2911881Sbinkertn@umich.edu 2921881Sbinkertn@umich.edu if False: 2931881Sbinkertn@umich.edu chart2 = BarChart() 2941881Sbinkertn@umich.edu chart2.data = data 2951881Sbinkertn@umich.edu chart2.colormap = 'gray' 2961881Sbinkertn@umich.edu chart2.graph() 2972006Sbinkertn@umich.edu chart2.savefig('/tmp/test2.png') 2982006Sbinkertn@umich.edu chart2.savefig('/tmp/test2.ps') 2991881Sbinkertn@umich.edu 3002160Shsul@eecs.umich.edu# pylab.show() 301