diff options
Diffstat (limited to 'bin/pychart/rose_plot.py')
| -rw-r--r-- | bin/pychart/rose_plot.py | 143 |
1 files changed, 143 insertions, 0 deletions
diff --git a/bin/pychart/rose_plot.py b/bin/pychart/rose_plot.py new file mode 100644 index 00000000000..dcfd06074d4 --- /dev/null +++ b/bin/pychart/rose_plot.py @@ -0,0 +1,143 @@ +# +# Copyright (C) 2000-2005 by Yasushi Saito (yasushi.saito@gmail.com) +# +# Jockey is free software; you can redistribute it and/or modify it +# under the terms of the GNU General Public License as published by the +# Free Software Foundation; either version 2, or (at your option) any +# later version. +# +# Jockey is distributed in the hope that it will be useful, but WITHOUT +# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or +# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License +# for more details. +# +import canvas +import fill_style +import line_style +import pychart_util +import chart_object +import legend +import font +import color +from pychart_types import * + +class T(chart_object.T): + """Plots sector diagram which can be superimposed on one another. + Sector diagrams are also known as wind roses""" + keys = { + "start_angle" : (NumType, 90, ""), # top of chart (north) + "center" : (CoordType, None, ""), + "base_radius" : (NumType, None, ""), + "line_style" : (line_style.T, line_style.T(color=color.black, width=0.3), ""), + "fill_styles" : (list, fill_style.standards.list()[:], + """The fill style of each item. The length of the + list should be equal to the length of the data. + """), + "sector_centred":(int, 1, + """Bool indicating whether the sectors should be centred on each sector_width(e.g. on 0)"""), + "dir_offset": (UnitType, None, + """The distance between the directions and the outermost circle. Defaults fine for most cases"""), + "data" : (AnyType, None, pychart_util.data_desc), + "label_col" : (int, 0, + """The column, within "data", from which the labels of items are retrieved."""), + "data_col": (int, 1, + """ The column, within "data", from which the data values are retrieved."""), + "dir_line_style": (line_style.T, None, ""), + "dir_fill_style": (fill_style.T, fill_style.default, ""), + "shadow": (ShadowType, None, pychart_util.shadow_desc), + "sector_width": (int, None, ""), # automatically generated + } + + def __init__(self, colour=True, **args): + chart_object.T.init(self, args) + if colour: + # the theme.color flag does not seem to affect the fill_style.standards, + #besides, I want the first two colors to resemble those of gnuplot's postscript terminal + self.fill_styles = [fill_style.Plain(bgcolor=color.red), + fill_style.Plain(bgcolor=color.green), + fill_style.Plain(bgcolor=color.blue), + fill_style.Plain(bgcolor=color.magenta)] + + def check_integrity(self): + nSectors = len(self.data[0][self.data_col]) + if (360%nSectors != 0): + raise Exception('Length of dataset ' + str(nSectors) + ' not a divisor of 360 degrees!') + for dataset in self.data: + length = len(dataset[self.data_col]) + if length != nSectors: + raise Exception('Lengths of datasets given is different!') + for val in dataset[self.data_col]: + if (val < 0) | (val > 1): + raise Exception('Data value ' + str(val) + ' not between 0 and 1!') + self.sector_width = 360/nSectors + self.type_check() + + def get_data_range(self, which): + return (0, 1) + + def get_legend_entry(self): + legends = [] + i = 0 + for dataset in self.data: + fill = self.fill_styles[i] + i = (i + 1) % len(self.fill_styles) + legends.append(legend.Entry(line_style=self.line_style, + fill_style=fill, + label=dataset[self.label_col])) + return legends + + def draw(self, ar, can): + center = self.center + if not center: + center = (ar.loc[0] + ar.size[0]/2.0, + ar.loc[1] + ar.size[1]/2.0) + base_radius = self.base_radius # the maximum radius of a wedge + if not base_radius: + base_radius = min(ar.size[0]/2.0, ar.size[1]/2.0) #* 0.8 + + sector_decrement = 1./(len(self.data)*2) * self.sector_width # each following sector diagram will have its sector width decremented by half this amount (in degrees) + i = 0 + for dataset in self.data: + cur_angle = self.start_angle + if self.sector_centred: + cur_angle -= self.sector_width/2. + fill = self.fill_styles[i] + x_center = center[0] + y_center = center[1] + + if not i: # draw directions around sector diagram once off + dir_offset = base_radius + (self.dir_offset or base_radius * 0.04) + directions = ['N', 'E', 'S', 'W'] + angle = self.start_angle + + can.ellipsis(line_style.T(color=color.black, width=0.3, dash=line_style.dash1), None, + x_center, y_center, base_radius, 1, + 0, 360) # + + for d in directions: + x_label, y_label = pychart_util.rotate(dir_offset, 0, angle) # coords for bottom left corner of box + tw = font.text_width(d) + half = 1/3. # normal arithmetic does not seem to apply to these text_box objects... + if (angle == 0): # east + y_label -= font.text_height(d)[0]*half # move down half + elif (angle == -180): # west + y_label -= font.text_height(d)[0]*half # move down half + x_label -= font.text_width(d) # move left full + elif (angle == 90): # north + x_label -= font.text_height(d)[0]*half # move left half + elif (angle == -90): # south + y_label -= font.text_height(d)[0]*.8 # move down (couldn't figure out how to set this dynamically so I fudged...) + x_label -= font.text_height(d)[0]*half # move left half + canvas.show(x_label + x_center, y_label + y_center, d) + angle -= 360/len(directions) + + for val in dataset[self.data_col]: # now draw the sectors + radius = base_radius*val # scale the radius + start = cur_angle-self.sector_width+i*sector_decrement + stop = cur_angle-i*sector_decrement # these may seem confusing, but remember that we need to go counterclockwise + + can.ellipsis(self.line_style, fill, + x_center, y_center, radius, 1, start, stop, self.shadow) + cur_angle = (cur_angle - self.sector_width) % 360 # we want to go in anticlockwise direction (North, West, South, etc. as in meteorology) + i = (i + 1) % len(self.fill_styles) + |