barchart.py revision 2119:f0de10227ae5
1# Copyright (c) 2005-2006 The Regents of The University of Michigan 2# All rights reserved. 3# 4# Redistribution and use in source and binary forms, with or without 5# modification, are permitted provided that the following conditions are 6# met: redistributions of source code must retain the above copyright 7# notice, this list of conditions and the following disclaimer; 8# redistributions in binary form must reproduce the above copyright 9# notice, this list of conditions and the following disclaimer in the 10# documentation and/or other materials provided with the distribution; 11# neither the name of the copyright holders nor the names of its 12# contributors may be used to endorse or promote products derived from 13# this software without specific prior written permission. 14# 15# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 16# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 17# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 18# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 19# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 21# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 25# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26# 27# Authors: Nathan Binkert 28# Lisa Hsu 29 30import matplotlib, pylab 31from matplotlib.font_manager import FontProperties 32from matplotlib.numerix import array, arange, reshape, shape, transpose, zeros 33from matplotlib.numerix import Float 34from matplotlib.ticker import NullLocator 35 36matplotlib.interactive(False) 37 38from chart import ChartOptions 39 40class BarChart(ChartOptions): 41 def __init__(self, default=None, **kwargs): 42 super(BarChart, self).__init__(default, **kwargs) 43 self.inputdata = None 44 self.chartdata = None 45 46 def gen_colors(self, count): 47 cmap = matplotlib.cm.get_cmap(self.colormap) 48 if count == 1: 49 return cmap([ 0.5 ]) 50 else: 51 return cmap(arange(count) / float(count - 1)) 52 53 # The input data format does not match the data format that the 54 # graph function takes because it is intuitive. The conversion 55 # from input data format to chart data format depends on the 56 # dimensionality of the input data. Check here for the 57 # dimensionality and correctness of the input data 58 def set_data(self, data): 59 if data is None: 60 self.inputdata = None 61 self.chartdata = None 62 return 63 64 data = array(data) 65 dim = len(shape(data)) 66 if dim not in (1, 2, 3): 67 raise AttributeError, "Input data must be a 1, 2, or 3d matrix" 68 self.inputdata = data 69 70 # If the input data is a 1d matrix, then it describes a 71 # standard bar chart. 72 if dim == 1: 73 self.chartdata = array([[data]]) 74 75 # If the input data is a 2d matrix, then it describes a bar 76 # chart with groups. The matrix being an array of groups of 77 # bars. 78 if dim == 2: 79 self.chartdata = transpose([data], axes=(2,0,1)) 80 81 # If the input data is a 3d matrix, then it describes an array 82 # of groups of bars with each bar being an array of stacked 83 # values. 84 if dim == 3: 85 self.chartdata = transpose(data, axes=(1,2,0)) 86 87 def get_data(self): 88 return self.inputdata 89 90 data = property(get_data, set_data) 91 92 # Graph the chart data. 93 # Input is a 3d matrix that describes a plot that has multiple 94 # groups, multiple bars in each group, and multiple values stacked 95 # in each bar. The underlying bar() function expects a sequence of 96 # bars in the same stack location and same group location, so the 97 # organization of the matrix is that the inner most sequence 98 # represents one of these bar groups, then those are grouped 99 # together to make one full stack of bars in each group, and then 100 # the outer most layer describes the groups. Here is an example 101 # data set and how it gets plotted as a result. 102 # 103 # e.g. data = [[[10,11,12], [13,14,15], [16,17,18], [19,20,21]], 104 # [[22,23,24], [25,26,27], [28,29,30], [31,32,33]]] 105 # 106 # will plot like this: 107 # 108 # 19 31 20 32 21 33 109 # 16 28 17 29 18 30 110 # 13 25 14 26 15 27 111 # 10 22 11 23 12 24 112 # 113 # Because this arrangement is rather conterintuitive, the rearrange 114 # function takes various matricies and arranges them to fit this 115 # profile. 116 # 117 # This code deals with one of the dimensions in the matrix being 118 # one wide. 119 # 120 def graph(self): 121 if self.chartdata is None: 122 raise AttributeError, "Data not set for bar chart!" 123 124 dim = len(shape(self.inputdata)) 125 cshape = shape(self.chartdata) 126 if dim == 1: 127 colors = self.gen_colors(cshape[2]) 128 colors = [ [ colors ] * cshape[1] ] * cshape[0] 129 130 if dim == 2: 131 colors = self.gen_colors(cshape[0]) 132 colors = [ [ [ c ] * cshape[2] ] * cshape[1] for c in colors ] 133 134 if dim == 3: 135 colors = self.gen_colors(cshape[1]) 136 colors = [ [ [ c ] * cshape[2] for c in colors ] ] * cshape[0] 137 138 colors = array(colors) 139 140 self.figure = pylab.figure(figsize=self.chart_size) 141 142 outer_axes = None 143 inner_axes = None 144 if self.xsubticks is not None: 145 color = self.figure.get_facecolor() 146 self.metaaxes = self.figure.add_axes(self.figure_size, axisbg=color, frameon=False) 147 for tick in self.metaaxes.xaxis.majorTicks: 148 tick.tick1On = False 149 tick.tick2On = False 150 self.metaaxes.set_yticklabels([]) 151 self.metaaxes.set_yticks([]) 152 size = [0] * 4 153 size[0] = self.figure_size[0] 154 size[1] = self.figure_size[1] + .05 155 size[2] = self.figure_size[2] 156 size[3] = self.figure_size[3] - .05 157 self.axes = self.figure.add_axes(size) 158 outer_axes = self.metaaxes 159 inner_axes = self.axes 160 else: 161 self.axes = self.figure.add_axes(self.figure_size) 162 outer_axes = self.axes 163 inner_axes = self.axes 164 165 bars_in_group = len(self.chartdata) 166 if bars_in_group < 5: 167 width = 1.0 / ( bars_in_group + 1) 168 center = width / 2 169 else: 170 width = .8 / bars_in_group 171 center = .1 172 173 bars = [] 174 for i,stackdata in enumerate(self.chartdata): 175 bottom = array([0.0] * len(stackdata[0]), Float) 176 stack = [] 177 for j,bardata in enumerate(stackdata): 178 bardata = array(bardata) 179 ind = arange(len(bardata)) + i * width + center 180 bar = self.axes.bar(ind, bardata, width, bottom=bottom, 181 color=colors[i][j]) 182 if dim != 1: 183 self.metaaxes.bar(ind, [0] * len(bardata), width) 184 stack.append(bar) 185 bottom += bardata 186 bars.append(stack) 187 188 if self.xlabel is not None: 189 outer_axes.set_xlabel(self.xlabel) 190 191 if self.ylabel is not None: 192 inner_axes.set_ylabel(self.ylabel) 193 194 if self.yticks is not None: 195 ymin, ymax = self.axes.get_ylim() 196 nticks = float(len(self.yticks)) 197 ticks = arange(nticks) / (nticks - 1) * (ymax - ymin) + ymin 198 inner_axes.set_yticks(ticks) 199 inner_axes.set_yticklabels(self.yticks) 200 elif self.ylim is not None: 201 self.inner_axes.set_ylim(self.ylim) 202 203 if self.xticks is not None: 204 outer_axes.set_xticks(arange(cshape[2]) + .5) 205 outer_axes.set_xticklabels(self.xticks) 206 207 if self.xsubticks is not None: 208 inner_axes.set_xticks(arange((cshape[0] + 1)*cshape[2])*width + 2*center) 209 self.xsubticks.append('') 210 inner_axes.set_xticklabels(self.xsubticks * cshape[2], fontsize=8) 211 212 if self.legend is not None: 213 if dim == 1: 214 lbars = bars[0][0] 215 if dim == 2: 216 lbars = [ bars[i][0][0] for i in xrange(len(bars))] 217 if dim == 3: 218 number = len(bars[0]) 219 lbars = [ bars[0][number - j - 1][0] for j in xrange(number)] 220 221 self.figure.legend(lbars, self.legend, self.legend_loc, 222 prop=FontProperties(size=self.legend_size)) 223 224 if self.title is not None: 225 self.axes.set_title(self.title) 226 227 def savefig(self, name): 228 self.figure.savefig(name) 229 230 def savecsv(self, name): 231 f = file(name, 'w') 232 data = array(self.inputdata) 233 dim = len(data.shape) 234 235 if dim == 1: 236 #if self.xlabel: 237 # f.write(', '.join(list(self.xlabel)) + '\n') 238 f.write(', '.join([ '%f' % val for val in data]) + '\n') 239 if dim == 2: 240 #if self.xlabel: 241 # f.write(', '.join([''] + list(self.xlabel)) + '\n') 242 for i,row in enumerate(data): 243 ylabel = [] 244 #if self.ylabel: 245 # ylabel = [ self.ylabel[i] ] 246 f.write(', '.join(ylabel + [ '%f' % val for val in row]) + '\n') 247 if dim == 3: 248 f.write("don't do 3D csv files\n") 249 pass 250 251 f.close() 252 253if __name__ == '__main__': 254 from random import randrange 255 import random, sys 256 257 dim = 3 258 number = 5 259 260 args = sys.argv[1:] 261 if len(args) > 3: 262 sys.exit("invalid number of arguments") 263 elif len(args) > 0: 264 myshape = [ int(x) for x in args ] 265 else: 266 myshape = [ 3, 4, 8 ] 267 268 # generate a data matrix of the given shape 269 size = reduce(lambda x,y: x*y, myshape) 270 #data = [ random.randrange(size - i) + 10 for i in xrange(size) ] 271 data = [ float(i)/100.0 for i in xrange(size) ] 272 data = reshape(data, myshape) 273 274 # setup some test bar charts 275 if True: 276 chart1 = BarChart() 277 chart1.data = data 278 279 chart1.xlabel = 'Benchmark' 280 chart1.ylabel = 'Bandwidth (GBps)' 281 chart1.legend = [ 'x%d' % x for x in xrange(myshape[-1]) ] 282 chart1.xticks = [ 'xtick%d' % x for x in xrange(myshape[0]) ] 283 chart1.title = 'this is the title' 284 chart1.figure_size = [0.1, 0.2, 0.7, 0.85 ] 285 if len(myshape) > 2: 286 chart1.xsubticks = [ '%d' % x for x in xrange(myshape[1]) ] 287 chart1.graph() 288 chart1.savefig('/tmp/test1.png') 289 chart1.savefig('/tmp/test1.ps') 290 chart1.savefig('/tmp/test1.eps') 291 chart1.savecsv('/tmp/test1.csv') 292 293 if False: 294 chart2 = BarChart() 295 chart2.data = data 296 chart2.colormap = 'gray' 297 chart2.graph() 298 chart2.savefig('/tmp/test2.png') 299 chart2.savefig('/tmp/test2.ps') 300 301 pylab.myshow() 302