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