/usr/share/pyshared/pychart/rose_plot.py is in python-pychart 1.39-7.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 | #
# 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" : (NumberType, 90, ""), # top of chart (north)
"center" : (CoordType, None, ""),
"base_radius" : (NumberType, 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
chart_object.T.check_integrity(self)
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)
|