mirror of https://github.com/Kozea/pygal.git
Python to generate nice looking SVG graph
http://pygal.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
518 lines
13 KiB
518 lines
13 KiB
#!/usr/bin/env python |
|
import SVG |
|
from itertools import izip, count, chain |
|
|
|
def get_pairs( i ): |
|
i = iter( i ) |
|
while True: yield i.next(), i.next() |
|
|
|
def float_range( start = 0, stop = None, step = 1 ): |
|
"Much like the built-in function range, but accepts floats" |
|
while start < stop: |
|
yield float( start ) |
|
start += step |
|
|
|
class Plot( SVG.Graph ): |
|
"""=== For creating SVG plots of scalar data |
|
|
|
= Synopsis |
|
|
|
require 'SVG/Graph/Plot' |
|
|
|
# Data sets are x,y pairs |
|
# Note that multiple data sets can differ in length, and that the |
|
# data in the datasets needn't be in order; they will be ordered |
|
# by the plot along the X-axis. |
|
projection = [ |
|
6, 11, 0, 5, 18, 7, 1, 11, 13, 9, 1, 2, 19, 0, 3, 13, |
|
7, 9 |
|
] |
|
actual = [ |
|
0, 18, 8, 15, 9, 4, 18, 14, 10, 2, 11, 6, 14, 12, |
|
15, 6, 4, 17, 2, 12 |
|
] |
|
|
|
graph = SVG::Graph::Plot.new({ |
|
:height => 500, |
|
:width => 300, |
|
:key => true, |
|
:scale_x_integers => true, |
|
:scale_y_integerrs => true, |
|
}) |
|
|
|
graph.add_data({ |
|
:data => projection |
|
:title => 'Projected', |
|
}) |
|
|
|
graph.add_data({ |
|
:data => actual, |
|
:title => 'Actual', |
|
}) |
|
|
|
print graph.burn() |
|
|
|
= Description |
|
|
|
Produces a graph of scalar data. |
|
|
|
This object aims to allow you to easily create high quality |
|
SVG[http://www.w3c.org/tr/svg] scalar plots. You can either use the |
|
default style sheet or supply your own. Either way there are many options |
|
which can be configured to give you control over how the graph is |
|
generated - with or without a key, data elements at each point, title, |
|
subtitle etc. |
|
|
|
= Examples |
|
|
|
http://www.germane-software/repositories/public/SVG/test/plot.rb |
|
|
|
= Notes |
|
|
|
The default stylesheet handles upto 10 data sets, if you |
|
use more you must create your own stylesheet and add the |
|
additional settings for the extra data sets. You will know |
|
if you go over 10 data sets as they will have no style and |
|
be in black. |
|
|
|
Unlike the other types of charts, data sets must contain x,y pairs: |
|
|
|
[ 1, 2 ] # A data set with 1 point: (1,2) |
|
[ 1,2, 5,6] # A data set with 2 points: (1,2) and (5,6) |
|
|
|
= See also |
|
|
|
* SVG::Graph::Graph |
|
* SVG::Graph::BarHorizontal |
|
* SVG::Graph::Bar |
|
* SVG::Graph::Line |
|
* SVG::Graph::Pie |
|
* SVG::Graph::TimeSeries |
|
|
|
== Author |
|
|
|
Sean E. Russell <serATgermaneHYPHENsoftwareDOTcom> |
|
|
|
Copyright 2004 Sean E. Russell |
|
This software is available under the Ruby license[LICENSE.txt]""" |
|
|
|
top_align = right_align = top_font = right_font = 1 |
|
|
|
|
|
"""Determines the scaling for the Y axis divisions. |
|
|
|
graph.scale_y_divisions = 0.5 |
|
|
|
would cause the graph to attempt to generate labels stepped by 0.5; EG: |
|
0, 0.5, 1, 1.5, 2, ...""" |
|
scale_y_divisions = None |
|
"Make the X axis labels integers" |
|
scale_x_integers = False |
|
"Make the Y axis labels integers" |
|
scale_y_integers = False |
|
"Fill the area under the line" |
|
area_fill = False |
|
"""Show a small circle on the graph where the line |
|
goes from one point to the next.""" |
|
show_data_points = True |
|
"Indicate whether the lines should be drawn between points" |
|
draw_lines_between_points = True |
|
"Set the minimum value of the X axis" |
|
min_x_value = None |
|
"Set the minimum value of the Y axis" |
|
min_y_value = None |
|
"Set the maximum value of the X axis" |
|
max_x_value = None |
|
"Set the maximum value of the Y axis" |
|
max_y_value = None |
|
|
|
stacked = False |
|
|
|
@apply |
|
def scale_x_divisions(): |
|
doc = """Determines the scaling for the X axis divisions. |
|
|
|
graph.scale_x_divisions = 2 |
|
|
|
would cause the graph to attempt to generate labels stepped by 2; EG: |
|
0,2,4,6,8...""" |
|
def fget( self ): |
|
return getattr( self, '_scale_x_divisions', None ) |
|
def fset( self, val ): |
|
self._scale_x_divisions = val |
|
return property(**locals()) |
|
|
|
def validate_data( self, data ): |
|
if len( data['data'] ) % 2 != 0: |
|
raise "Expecting x,y pairs for data points for %s." % self.__class__.__name__ |
|
|
|
def process_data( self, data ): |
|
pairs = list( get_pairs( data['data'] ) ) |
|
pairs.sort() |
|
data['data'] = zip( *pairs ) |
|
|
|
def calculate_left_margin( self ): |
|
super( Plot, self ).calculate_left_margin() |
|
label_left = len( str( self.get_x_labels()[0] ) ) / 2 * self.font_size * 0.6 |
|
self.border_left = max( label_left, self.border_left ) |
|
|
|
def calculate_right_margin( self ): |
|
super( Plot, self ).calculate_right_margin() |
|
label_right = len( str( self.get_x_labels()[-1] ) ) / 2 * self.font_size * 0.6 |
|
self.border_right = max( label_right, self.border_right ) |
|
|
|
def data_max( self, axis ): |
|
data_index = getattr( self, '%s_data_index' % axis ) |
|
max_value = max( chain( *map( lambda set: set['data'][data_index], self.data ) ) ) |
|
# above is same as |
|
#max_value = max( map( lambda set: max( set['data'][data_index] ), self.data ) ) |
|
spec_max = getattr( self, 'max_%s_value' % axis ) |
|
max_value = max( max_value, spec_max ) |
|
return max_value |
|
|
|
def data_min( self, axis ): |
|
data_index = getattr( self, '%s_data_index' % axis ) |
|
min_value = min( chain( *map( lambda set: set['data'][data_index], self.data ) ) ) |
|
spec_min = getattr( self, 'min_%s_value' % axis ) |
|
if spec_min is not None: |
|
min_value = min( min_value, spec_min ) |
|
return min_value |
|
|
|
x_data_index = 0 |
|
y_data_index = 1 |
|
def data_range( self, axis ): |
|
side = { 'x': 'right', 'y': 'top' }[axis] |
|
|
|
min_value = self.data_min( axis ) |
|
max_value = self.data_max( axis ) |
|
range = max_value - min_value |
|
|
|
side_pad = range / 20.0 or 10 |
|
scale_range = ( max_value + side_pad ) - min_value |
|
|
|
scale_division = getattr( self, 'scale_%s_divisions' % axis ) or ( scale_range / 10.0 ) |
|
|
|
if getattr( self, 'scale_%s_integers' % axis ): |
|
scale_division = scale_division.round() or 1 |
|
|
|
return min_value, max_value, scale_division |
|
|
|
def x_range( self ): return self.data_range( 'x' ) |
|
def y_range( self ): return self.data_range( 'y' ) |
|
|
|
def get_data_values( self, axis ): |
|
min_value, max_value, scale_division = self.data_range( axis ) |
|
return tuple( float_range( *self.data_range( axis ) ) ) |
|
|
|
def get_x_values( self ): return self.get_data_values( 'x' ) |
|
def get_y_values( self ): return self.get_data_values( 'y' ) |
|
|
|
def get_x_labels( self ): |
|
return map( str, self.get_x_values() ) |
|
def get_y_labels( self ): |
|
return map( str, self.get_y_values() ) |
|
|
|
def field_size( self, axis ): |
|
size = { 'x': 'width', 'y': 'height' }[axis] |
|
side = { 'x': 'right', 'y': 'top' }[axis] |
|
values = getattr( self, 'get_%s_values' % axis )() |
|
max_d = self.data_max( axis ) |
|
dx = float( max_d - values[-1] ) / ( values[-1] - values[-2] ) |
|
graph_size = getattr( self, 'graph_%s' % size ) |
|
side_font = getattr( self, '%s_font' % side ) |
|
side_align = getattr( self, '%s_align' % side ) |
|
result = ( float( graph_size ) - self.font_size*2*side_font ) / \ |
|
( len( values ) + dx - side_align ) |
|
return result |
|
|
|
def field_width( self ): return self.field_size( 'x' ) |
|
def field_height( self ): return self.field_size( 'y' ) |
|
|
|
def draw_data( self ): |
|
self.load_transform_parameters() |
|
for line, data in izip( count(1), self.data ): |
|
x_start, y_start = self.transform_output_coordinates( |
|
( data['data'][self.x_data_index][0], |
|
data['data'][self.y_data_index][0] ) |
|
) |
|
data_points = zip( *data['data'] ) |
|
graph_points = self.get_graph_points( data_points ) |
|
lpath = self.get_lpath( graph_points ) |
|
if self.area_fill: |
|
graph_height = self.graph_height |
|
path = self._create_element( 'path', { |
|
'd': 'M%(x_start)f %(graph_height)f %(lpath)s V%(graph_height)f Z' % vars(), |
|
'class': 'fill%(line)d' % vars() } ) |
|
self.graph.appendChild( path ) |
|
if self.draw_lines_between_points: |
|
path = self._create_element( 'path', { |
|
'd': 'M%(x_start)f %(y_start)f %(lpath)s' % vars(), |
|
'class': 'line%(line)d' % vars() } ) |
|
self.graph.appendChild( path ) |
|
self.draw_data_points( line, data_points, graph_points ) |
|
self._draw_constant_lines( ) |
|
del self.__transform_parameters |
|
|
|
def add_constant_line( self, value, label = None, style = None ): |
|
self.constant_lines = getattr( self, 'constant_lines', [] ) |
|
self.constant_lines.append( ( value, label, style ) ) |
|
|
|
def _draw_constant_lines( self ): |
|
if hasattr( self, 'constant_lines' ): |
|
map( self.__draw_constant_line, self.constant_lines ) |
|
|
|
def __draw_constant_line( self, ( value, label, style ) ): |
|
"Draw a constant line on the y-axis with the label" |
|
start = self.transform_output_coordinates( ( 0, value ) )[1] |
|
stop = self.graph_width |
|
path = self._create_element( 'path', { |
|
'd': 'M 0 %(start)s h%(stop)s' % vars(), |
|
'class': 'constantLine' } ) |
|
if style: |
|
path['style'] = style |
|
self.graph.appendChild( path ) |
|
text = self._create_element( 'text', { |
|
'x': str( 2 ), |
|
'y': str( start - 2 ), |
|
'class': 'constantLine' } ) |
|
text.appendChild( self._doc.createTextNode( label ) ) |
|
self.graph.appendChild( text ) |
|
|
|
def load_transform_parameters( self ): |
|
"Cache the parameters necessary to transform x & y coordinates" |
|
x_min, x_max, x_div = self.x_range() |
|
y_min, y_max, y_div = self.y_range() |
|
x_step = ( float( self.graph_width ) - self.font_size*2 ) / \ |
|
( x_max - x_min ) |
|
y_step = ( float( self.graph_height ) - self.font_size*2 ) / \ |
|
( y_max - y_min ) |
|
self.__transform_parameters = dict( vars() ) |
|
del self.__transform_parameters['self'] |
|
|
|
def get_graph_points( self, data_points ): |
|
return map( self.transform_output_coordinates, data_points ) |
|
|
|
def get_lpath( self, points ): |
|
points = map( lambda p: "%f %f" % p, points ) |
|
return 'L' + ' '.join( points ) |
|
|
|
def transform_output_coordinates( self, (x,y) ): |
|
x_min = self.__transform_parameters['x_min'] |
|
x_step = self.__transform_parameters['x_step'] |
|
y_min = self.__transform_parameters['y_min'] |
|
y_step = self.__transform_parameters['y_step'] |
|
#locals().update( self.__transform_parameters ) |
|
#vars().update( self.__transform_parameters ) |
|
x = ( x - x_min ) * x_step |
|
y = self.graph_height - ( y - y_min ) * y_step |
|
return x,y |
|
|
|
def draw_data_points( self, line, data_points, graph_points ): |
|
if not self.show_data_points \ |
|
and not self.show_data_values: return |
|
for ((dx,dy),(gx,gy)) in izip( data_points, graph_points ): |
|
if self.show_data_points: |
|
circle = self._create_element( 'circle', { |
|
'cx': str( gx ), |
|
'cy': str( gy ), |
|
'r': '2.5', |
|
'class': 'dataPoint%(line)s' % vars() } ) |
|
self.graph.appendChild( circle ) |
|
if self.show_data_values: |
|
self.add_popup( gx, gy, self.format( dx, dy ) ) |
|
self.make_datapoint_text( gx, gy-6, dy ) |
|
|
|
def format( self, x, y ): |
|
return '(%0.2f, %0.2f)' % (x,y) |
|
|
|
def get_css( self ): |
|
return """/* default line styles */ |
|
.line1{ |
|
fill: none; |
|
stroke: #ff0000; |
|
stroke-width: 1px; |
|
} |
|
.line2{ |
|
fill: none; |
|
stroke: #0000ff; |
|
stroke-width: 1px; |
|
} |
|
.line3{ |
|
fill: none; |
|
stroke: #00ff00; |
|
stroke-width: 1px; |
|
} |
|
.line4{ |
|
fill: none; |
|
stroke: #ffcc00; |
|
stroke-width: 1px; |
|
} |
|
.line5{ |
|
fill: none; |
|
stroke: #00ccff; |
|
stroke-width: 1px; |
|
} |
|
.line6{ |
|
fill: none; |
|
stroke: #ff00ff; |
|
stroke-width: 1px; |
|
} |
|
.line7{ |
|
fill: none; |
|
stroke: #00ffff; |
|
stroke-width: 1px; |
|
} |
|
.line8{ |
|
fill: none; |
|
stroke: #ffff00; |
|
stroke-width: 1px; |
|
} |
|
.line9{ |
|
fill: none; |
|
stroke: #ccc6666; |
|
stroke-width: 1px; |
|
} |
|
.line10{ |
|
fill: none; |
|
stroke: #663399; |
|
stroke-width: 1px; |
|
} |
|
.line11{ |
|
fill: none; |
|
stroke: #339900; |
|
stroke-width: 1px; |
|
} |
|
.line12{ |
|
fill: none; |
|
stroke: #9966FF; |
|
stroke-width: 1px; |
|
} |
|
/* default fill styles */ |
|
.fill1{ |
|
fill: #cc0000; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill2{ |
|
fill: #0000cc; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill3{ |
|
fill: #00cc00; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill4{ |
|
fill: #ffcc00; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill5{ |
|
fill: #00ccff; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill6{ |
|
fill: #ff00ff; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill7{ |
|
fill: #00ffff; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill8{ |
|
fill: #ffff00; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill9{ |
|
fill: #cc6666; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill10{ |
|
fill: #663399; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill11{ |
|
fill: #339900; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
.fill12{ |
|
fill: #9966FF; |
|
fill-opacity: 0.2; |
|
stroke: none; |
|
} |
|
/* default line styles */ |
|
.key1,.dataPoint1{ |
|
fill: #ff0000; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key2,.dataPoint2{ |
|
fill: #0000ff; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key3,.dataPoint3{ |
|
fill: #00ff00; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key4,.dataPoint4{ |
|
fill: #ffcc00; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key5,.dataPoint5{ |
|
fill: #00ccff; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key6,.dataPoint6{ |
|
fill: #ff00ff; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key7,.dataPoint7{ |
|
fill: #00ffff; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key8,.dataPoint8{ |
|
fill: #ffff00; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key9,.dataPoint9{ |
|
fill: #cc6666; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key10,.dataPoint10{ |
|
fill: #663399; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key11,.dataPoint11{ |
|
fill: #339900; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.key12,.dataPoint12{ |
|
fill: #9966FF; |
|
stroke: none; |
|
stroke-width: 1px; |
|
} |
|
.constantLine{ |
|
color: navy; |
|
stroke: navy; |
|
stroke-width: 1px; |
|
stroke-dasharray: 9 1 1; |
|
} |
|
""" |