chart.py revision 2006:3ca085495c69
1# Copyright (c) 2005 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 30class ChartOptions(object): 31 defaults = { 'chart_size' : (8, 4), 32 'figure_size' : [0.1, 0.1, 0.6, 0.85], 33 'title' : None, 34 'legend' : None, 35 'legend_loc' : 'upper right', 36 'legend_size' : 6, 37 'colormap' : 'jet', 38 'xlabel' : None, 39 'ylabel' : None, 40 'xticks' : None, 41 'yticks' : None, 42 'ylim' : None, 43 } 44 45 def __init__(self, options=None, **kwargs): 46 self.init(options, **kwargs) 47 48 def clear(self): 49 self.options = {} 50 51 def init(self, options=None, **kwargs): 52 self.clear() 53 self.update(options, **kwargs) 54 55 def update(self, options=None, **kwargs): 56 if options is not None: 57 if not isinstance(options, ChartOptions): 58 raise AttributeError, \ 59 'attribute options of type %s should be %s' % \ 60 (type(options), ChartOptions) 61 self.options.update(options.options) 62 63 for key,value in kwargs.iteritems(): 64 if key not in ChartOptions.defaults: 65 raise AttributeError, \ 66 "%s instance has no attribute '%s'" % (type(self), key) 67 self.options[key] = value 68 69 def __getattr__(self, attr): 70 if attr in self.options: 71 return self.options[attr] 72 73 if attr in ChartOptions.defaults: 74 return ChartOptions.defaults[attr] 75 76 raise AttributeError, \ 77 "%s instance has no attribute '%s'" % (type(self), attr) 78 79 def __setattr__(self, attr, value): 80 if attr in ChartOptions.defaults: 81 self.options[attr] = value 82 else: 83 super(ChartOptions, self).__setattr__(attr, value) 84 85