barchart.py revision 1881
11881Sbinkertn@umich.edu# Copyright (c) 2005 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 311881Sbinkertn@umich.edufrom matplotlib.numerix import array, arange, reshape, shape, transpose, zeros 321881Sbinkertn@umich.edufrom matplotlib.numerix import Float 331881Sbinkertn@umich.edu 341881Sbinkertn@umich.edumatplotlib.interactive(False) 351881Sbinkertn@umich.edu 361881Sbinkertn@umich.educlass BarChart(object): 371881Sbinkertn@umich.edu def __init__(self, **kwargs): 381881Sbinkertn@umich.edu self.init(**kwargs) 391881Sbinkertn@umich.edu 401881Sbinkertn@umich.edu def init(self, **kwargs): 411881Sbinkertn@umich.edu self.colormap = 'jet' 421881Sbinkertn@umich.edu self.inputdata = None 431881Sbinkertn@umich.edu self.chartdata = None 441881Sbinkertn@umich.edu self.xlabel = None 451881Sbinkertn@umich.edu self.ylabel = None 461881Sbinkertn@umich.edu self.legend = None 471881Sbinkertn@umich.edu self.xticks = None 481881Sbinkertn@umich.edu self.yticks = None 491881Sbinkertn@umich.edu self.title = None 501881Sbinkertn@umich.edu 511881Sbinkertn@umich.edu for key,value in kwargs.iteritems(): 521881Sbinkertn@umich.edu self.__setattr__(key, value) 531881Sbinkertn@umich.edu 541881Sbinkertn@umich.edu def gen_colors(self, count): 551881Sbinkertn@umich.edu cmap = matplotlib.cm.get_cmap(self.colormap) 561881Sbinkertn@umich.edu if count == 1: 571881Sbinkertn@umich.edu return cmap([ 0.5 ]) 581881Sbinkertn@umich.edu else: 591881Sbinkertn@umich.edu return cmap(arange(count) / float(count - 1)) 601881Sbinkertn@umich.edu 611881Sbinkertn@umich.edu # The input data format does not match the data format that the 621881Sbinkertn@umich.edu # graph function takes because it is intuitive. The conversion 631881Sbinkertn@umich.edu # from input data format to chart data format depends on the 641881Sbinkertn@umich.edu # dimensionality of the input data. Check here for the 651881Sbinkertn@umich.edu # dimensionality and correctness of the input data 661881Sbinkertn@umich.edu def set_data(self, data): 671881Sbinkertn@umich.edu if data is None: 681881Sbinkertn@umich.edu self.inputdata = None 691881Sbinkertn@umich.edu self.chartdata = None 701881Sbinkertn@umich.edu return 711881Sbinkertn@umich.edu 721881Sbinkertn@umich.edu data = array(data) 731881Sbinkertn@umich.edu dim = len(shape(data)) 741881Sbinkertn@umich.edu if dim not in (1, 2, 3): 751881Sbinkertn@umich.edu raise AttributeError, "Input data must be a 1, 2, or 3d matrix" 761881Sbinkertn@umich.edu self.inputdata = data 771881Sbinkertn@umich.edu 781881Sbinkertn@umich.edu # If the input data is a 1d matrix, then it describes a 791881Sbinkertn@umich.edu # standard bar chart. 801881Sbinkertn@umich.edu if dim == 1: 811881Sbinkertn@umich.edu self.chartdata = array([[data]]) 821881Sbinkertn@umich.edu 831881Sbinkertn@umich.edu # If the input data is a 2d matrix, then it describes a bar 841881Sbinkertn@umich.edu # chart with groups. The matrix being an array of groups of 851881Sbinkertn@umich.edu # bars. 861881Sbinkertn@umich.edu if dim == 2: 871881Sbinkertn@umich.edu self.chartdata = transpose([data], axes=(2,0,1)) 881881Sbinkertn@umich.edu 891881Sbinkertn@umich.edu # If the input data is a 3d matrix, then it describes an array 901881Sbinkertn@umich.edu # of groups of bars with each bar being an array of stacked 911881Sbinkertn@umich.edu # values. 921881Sbinkertn@umich.edu if dim == 3: 931881Sbinkertn@umich.edu self.chartdata = transpose(data, axes=(1,2,0)) 941881Sbinkertn@umich.edu 951881Sbinkertn@umich.edu def get_data(self): 961881Sbinkertn@umich.edu return self.inputdata 971881Sbinkertn@umich.edu 981881Sbinkertn@umich.edu data = property(get_data, set_data) 991881Sbinkertn@umich.edu 1001881Sbinkertn@umich.edu # Graph the chart data. 1011881Sbinkertn@umich.edu # Input is a 3d matrix that describes a plot that has multiple 1021881Sbinkertn@umich.edu # groups, multiple bars in each group, and multiple values stacked 1031881Sbinkertn@umich.edu # in each bar. The underlying bar() function expects a sequence of 1041881Sbinkertn@umich.edu # bars in the same stack location and same group location, so the 1051881Sbinkertn@umich.edu # organization of the matrix is that the inner most sequence 1061881Sbinkertn@umich.edu # represents one of these bar groups, then those are grouped 1071881Sbinkertn@umich.edu # together to make one full stack of bars in each group, and then 1081881Sbinkertn@umich.edu # the outer most layer describes the groups. Here is an example 1091881Sbinkertn@umich.edu # data set and how it gets plotted as a result. 1101881Sbinkertn@umich.edu # 1111881Sbinkertn@umich.edu # e.g. data = [[[10,11,12], [13,14,15], [16,17,18], [19,20,21]], 1121881Sbinkertn@umich.edu # [[22,23,24], [25,26,27], [28,29,30], [31,32,33]]] 1131881Sbinkertn@umich.edu # 1141881Sbinkertn@umich.edu # will plot like this: 1151881Sbinkertn@umich.edu # 1161881Sbinkertn@umich.edu # 19 31 20 32 21 33 1171881Sbinkertn@umich.edu # 16 28 17 29 18 30 1181881Sbinkertn@umich.edu # 13 25 14 26 15 27 1191881Sbinkertn@umich.edu # 10 22 11 23 12 24 1201881Sbinkertn@umich.edu # 1211881Sbinkertn@umich.edu # Because this arrangement is rather conterintuitive, the rearrange 1221881Sbinkertn@umich.edu # function takes various matricies and arranges them to fit this 1231881Sbinkertn@umich.edu # profile. 1241881Sbinkertn@umich.edu # 1251881Sbinkertn@umich.edu # This code deals with one of the dimensions in the matrix being 1261881Sbinkertn@umich.edu # one wide. 1271881Sbinkertn@umich.edu # 1281881Sbinkertn@umich.edu def graph(self): 1291881Sbinkertn@umich.edu if self.chartdata is None: 1301881Sbinkertn@umich.edu raise AttributeError, "Data not set for bar chart!" 1311881Sbinkertn@umich.edu 1321881Sbinkertn@umich.edu self.figure = pylab.figure() 1331881Sbinkertn@umich.edu self.axes = self.figure.add_subplot(111) 1341881Sbinkertn@umich.edu 1351881Sbinkertn@umich.edu dim = len(shape(self.inputdata)) 1361881Sbinkertn@umich.edu cshape = shape(self.chartdata) 1371881Sbinkertn@umich.edu if dim == 1: 1381881Sbinkertn@umich.edu colors = self.gen_colors(cshape[2]) 1391881Sbinkertn@umich.edu colors = [ [ colors ] * cshape[1] ] * cshape[0] 1401881Sbinkertn@umich.edu 1411881Sbinkertn@umich.edu if dim == 2: 1421881Sbinkertn@umich.edu colors = self.gen_colors(cshape[0]) 1431881Sbinkertn@umich.edu colors = [ [ [ c ] * cshape[2] ] * cshape[1] for c in colors ] 1441881Sbinkertn@umich.edu 1451881Sbinkertn@umich.edu if dim == 3: 1461881Sbinkertn@umich.edu colors = self.gen_colors(cshape[1]) 1471881Sbinkertn@umich.edu colors = [ [ [ c ] * cshape[2] for c in colors ] ] * cshape[0] 1481881Sbinkertn@umich.edu 1491881Sbinkertn@umich.edu colors = array(colors) 1501881Sbinkertn@umich.edu 1511881Sbinkertn@umich.edu bars_in_group = len(self.chartdata) 1521881Sbinkertn@umich.edu if bars_in_group < 5: 1531881Sbinkertn@umich.edu width = 1.0 / ( bars_in_group + 1) 1541881Sbinkertn@umich.edu center = width / 2 1551881Sbinkertn@umich.edu else: 1561881Sbinkertn@umich.edu width = .8 / bars_in_group 1571881Sbinkertn@umich.edu center = .1 1581881Sbinkertn@umich.edu 1591881Sbinkertn@umich.edu bars = [] 1601881Sbinkertn@umich.edu for i,stackdata in enumerate(self.chartdata): 1611881Sbinkertn@umich.edu bottom = array([0] * len(stackdata[0])) 1621881Sbinkertn@umich.edu stack = [] 1631881Sbinkertn@umich.edu for j,bardata in enumerate(stackdata): 1641881Sbinkertn@umich.edu bardata = array(bardata) 1651881Sbinkertn@umich.edu ind = arange(len(bardata)) + i * width + center 1661881Sbinkertn@umich.edu bar = self.axes.bar(ind, bardata, width, bottom=bottom, 1671881Sbinkertn@umich.edu color=colors[i][j]) 1681881Sbinkertn@umich.edu stack.append(bar) 1691881Sbinkertn@umich.edu bottom += bardata 1701881Sbinkertn@umich.edu bars.append(stack) 1711881Sbinkertn@umich.edu 1721881Sbinkertn@umich.edu if self.xlabel is not None: 1731881Sbinkertn@umich.edu self.axes.set_xlabel(self.xlabel) 1741881Sbinkertn@umich.edu 1751881Sbinkertn@umich.edu if self.ylabel is not None: 1761881Sbinkertn@umich.edu self.axes.set_ylabel(self.ylabel) 1771881Sbinkertn@umich.edu 1781881Sbinkertn@umich.edu if self.yticks is not None: 1791881Sbinkertn@umich.edu ymin, ymax = self.axes.get_ylim() 1801881Sbinkertn@umich.edu nticks = float(len(self.yticks)) 1811881Sbinkertn@umich.edu ticks = arange(nticks) / (nticks - 1) * (ymax - ymin) + ymin 1821881Sbinkertn@umich.edu self.axes.set_yticks(ticks) 1831881Sbinkertn@umich.edu self.axes.set_yticklabels(self.yticks) 1841881Sbinkertn@umich.edu 1851881Sbinkertn@umich.edu if self.xticks is not None: 1861881Sbinkertn@umich.edu self.axes.set_xticks(arange(cshape[2]) + .5) 1871881Sbinkertn@umich.edu self.axes.set_xticklabels(self.xticks) 1881881Sbinkertn@umich.edu 1891881Sbinkertn@umich.edu if self.legend is not None: 1901881Sbinkertn@umich.edu if dim == 1: 1911881Sbinkertn@umich.edu lbars = bars[0][0] 1921881Sbinkertn@umich.edu if dim == 2: 1931881Sbinkertn@umich.edu lbars = [ bars[i][0][0] for i in xrange(len(bars))] 1941881Sbinkertn@umich.edu if dim == 3: 1951881Sbinkertn@umich.edu number = len(bars[0]) 1961881Sbinkertn@umich.edu lbars = [ bars[0][number - j - 1][0] for j in xrange(number)] 1971881Sbinkertn@umich.edu 1981881Sbinkertn@umich.edu self.axes.legend(lbars, self.legend, loc='best') 1991881Sbinkertn@umich.edu 2001881Sbinkertn@umich.edu if self.title is not None: 2011881Sbinkertn@umich.edu self.axes.set_title(self.title) 2021881Sbinkertn@umich.edu 2031881Sbinkertn@umich.edu def savefig(self, name): 2041881Sbinkertn@umich.edu self.figure.savefig(name) 2051881Sbinkertn@umich.edu 2061881Sbinkertn@umich.eduif __name__ == '__main__': 2071881Sbinkertn@umich.edu import random, sys 2081881Sbinkertn@umich.edu 2091881Sbinkertn@umich.edu dim = 3 2101881Sbinkertn@umich.edu number = 5 2111881Sbinkertn@umich.edu 2121881Sbinkertn@umich.edu args = sys.argv[1:] 2131881Sbinkertn@umich.edu if len(args) > 3: 2141881Sbinkertn@umich.edu sys.exit("invalid number of arguments") 2151881Sbinkertn@umich.edu elif len(args) > 0: 2161881Sbinkertn@umich.edu myshape = [ int(x) for x in args ] 2171881Sbinkertn@umich.edu else: 2181881Sbinkertn@umich.edu myshape = [ 3, 4, 8 ] 2191881Sbinkertn@umich.edu 2201881Sbinkertn@umich.edu # generate a data matrix of the given shape 2211881Sbinkertn@umich.edu size = reduce(lambda x,y: x*y, myshape) 2221881Sbinkertn@umich.edu #data = [ random.randrange(size - i) + 10 for i in xrange(size) ] 2231881Sbinkertn@umich.edu data = [ float(i)/100.0 for i in xrange(size) ] 2241881Sbinkertn@umich.edu data = reshape(data, myshape) 2251881Sbinkertn@umich.edu 2261881Sbinkertn@umich.edu # setup some test bar charts 2271881Sbinkertn@umich.edu if True: 2281881Sbinkertn@umich.edu chart1 = BarChart() 2291881Sbinkertn@umich.edu chart1.data = data 2301881Sbinkertn@umich.edu 2311881Sbinkertn@umich.edu chart1.xlabel = 'Benchmark' 2321881Sbinkertn@umich.edu chart1.ylabel = 'Bandwidth (GBps)' 2331881Sbinkertn@umich.edu chart1.legend = [ 'x%d' % x for x in xrange(myshape[-1]) ] 2341881Sbinkertn@umich.edu chart1.xticks = [ 'xtick%d' % x for x in xrange(myshape[0]) ] 2351881Sbinkertn@umich.edu chart1.title = 'this is the title' 2361881Sbinkertn@umich.edu chart1.graph() 2371881Sbinkertn@umich.edu #chart1.savefig('/tmp/test1.png') 2381881Sbinkertn@umich.edu 2391881Sbinkertn@umich.edu if False: 2401881Sbinkertn@umich.edu chart2 = BarChart() 2411881Sbinkertn@umich.edu chart2.data = data 2421881Sbinkertn@umich.edu chart2.colormap = 'gray' 2431881Sbinkertn@umich.edu chart2.graph() 2441881Sbinkertn@umich.edu #chart2.savefig('/tmp/test2.png') 2451881Sbinkertn@umich.edu 2461881Sbinkertn@umich.edu pylab.show() 247