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