diff options
Diffstat (limited to 'bin/pychart/rose_plot.py')
| -rw-r--r-- | bin/pychart/rose_plot.py | 144 |
1 files changed, 0 insertions, 144 deletions
diff --git a/bin/pychart/rose_plot.py b/bin/pychart/rose_plot.py deleted file mode 100644 index 1f54a4e7cb6..00000000000 --- a/bin/pychart/rose_plot.py +++ /dev/null @@ -1,144 +0,0 @@ -# -*- coding: utf-8 -*- -# -# 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) - |