chart.py revision 2117
12117Shsul@eecs.umich.edu# Copyright (c) 2005-2006 The Regents of The University of Michigan 22006Sbinkertn@umich.edu# All rights reserved. 32006Sbinkertn@umich.edu# 42006Sbinkertn@umich.edu# Redistribution and use in source and binary forms, with or without 52006Sbinkertn@umich.edu# modification, are permitted provided that the following conditions are 62006Sbinkertn@umich.edu# met: redistributions of source code must retain the above copyright 72006Sbinkertn@umich.edu# notice, this list of conditions and the following disclaimer; 82006Sbinkertn@umich.edu# redistributions in binary form must reproduce the above copyright 92006Sbinkertn@umich.edu# notice, this list of conditions and the following disclaimer in the 102006Sbinkertn@umich.edu# documentation and/or other materials provided with the distribution; 112006Sbinkertn@umich.edu# neither the name of the copyright holders nor the names of its 122006Sbinkertn@umich.edu# contributors may be used to endorse or promote products derived from 132006Sbinkertn@umich.edu# this software without specific prior written permission. 142006Sbinkertn@umich.edu# 152006Sbinkertn@umich.edu# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 162006Sbinkertn@umich.edu# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 172006Sbinkertn@umich.edu# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 182006Sbinkertn@umich.edu# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 192006Sbinkertn@umich.edu# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 202006Sbinkertn@umich.edu# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 212006Sbinkertn@umich.edu# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 222006Sbinkertn@umich.edu# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 232006Sbinkertn@umich.edu# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 242006Sbinkertn@umich.edu# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 252006Sbinkertn@umich.edu# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 262006Sbinkertn@umich.edu# 272006Sbinkertn@umich.edu# Authors: Nathan Binkert 282006Sbinkertn@umich.edu# Lisa Hsu 292006Sbinkertn@umich.edu 302006Sbinkertn@umich.educlass ChartOptions(object): 312006Sbinkertn@umich.edu defaults = { 'chart_size' : (8, 4), 322006Sbinkertn@umich.edu 'figure_size' : [0.1, 0.1, 0.6, 0.85], 332006Sbinkertn@umich.edu 'title' : None, 342006Sbinkertn@umich.edu 'legend' : None, 352006Sbinkertn@umich.edu 'legend_loc' : 'upper right', 362006Sbinkertn@umich.edu 'legend_size' : 6, 372006Sbinkertn@umich.edu 'colormap' : 'jet', 382006Sbinkertn@umich.edu 'xlabel' : None, 392006Sbinkertn@umich.edu 'ylabel' : None, 402006Sbinkertn@umich.edu 'xticks' : None, 412117Shsul@eecs.umich.edu 'xsubticks' : None, 422006Sbinkertn@umich.edu 'yticks' : None, 432006Sbinkertn@umich.edu 'ylim' : None, 442006Sbinkertn@umich.edu } 452006Sbinkertn@umich.edu 462006Sbinkertn@umich.edu def __init__(self, options=None, **kwargs): 472006Sbinkertn@umich.edu self.init(options, **kwargs) 482006Sbinkertn@umich.edu 492006Sbinkertn@umich.edu def clear(self): 502006Sbinkertn@umich.edu self.options = {} 512006Sbinkertn@umich.edu 522006Sbinkertn@umich.edu def init(self, options=None, **kwargs): 532006Sbinkertn@umich.edu self.clear() 542006Sbinkertn@umich.edu self.update(options, **kwargs) 552006Sbinkertn@umich.edu 562006Sbinkertn@umich.edu def update(self, options=None, **kwargs): 572006Sbinkertn@umich.edu if options is not None: 582006Sbinkertn@umich.edu if not isinstance(options, ChartOptions): 592006Sbinkertn@umich.edu raise AttributeError, \ 602006Sbinkertn@umich.edu 'attribute options of type %s should be %s' % \ 612006Sbinkertn@umich.edu (type(options), ChartOptions) 622006Sbinkertn@umich.edu self.options.update(options.options) 632006Sbinkertn@umich.edu 642006Sbinkertn@umich.edu for key,value in kwargs.iteritems(): 652006Sbinkertn@umich.edu if key not in ChartOptions.defaults: 662006Sbinkertn@umich.edu raise AttributeError, \ 672006Sbinkertn@umich.edu "%s instance has no attribute '%s'" % (type(self), key) 682006Sbinkertn@umich.edu self.options[key] = value 692006Sbinkertn@umich.edu 702006Sbinkertn@umich.edu def __getattr__(self, attr): 712006Sbinkertn@umich.edu if attr in self.options: 722006Sbinkertn@umich.edu return self.options[attr] 732006Sbinkertn@umich.edu 742006Sbinkertn@umich.edu if attr in ChartOptions.defaults: 752006Sbinkertn@umich.edu return ChartOptions.defaults[attr] 762006Sbinkertn@umich.edu 772006Sbinkertn@umich.edu raise AttributeError, \ 782006Sbinkertn@umich.edu "%s instance has no attribute '%s'" % (type(self), attr) 792006Sbinkertn@umich.edu 802006Sbinkertn@umich.edu def __setattr__(self, attr, value): 812006Sbinkertn@umich.edu if attr in ChartOptions.defaults: 822006Sbinkertn@umich.edu self.options[attr] = value 832006Sbinkertn@umich.edu else: 842006Sbinkertn@umich.edu super(ChartOptions, self).__setattr__(attr, value) 852006Sbinkertn@umich.edu 86