|
|
|
|
@ -1,5 +1,5 @@
|
|
|
|
|
#!python |
|
|
|
|
# -*- coding: UTF-8 -*- |
|
|
|
|
# -*- coding: utf-8 -*- |
|
|
|
|
|
|
|
|
|
""" |
|
|
|
|
svg.charts.graph |
|
|
|
|
@ -14,6 +14,7 @@ import functools
|
|
|
|
|
|
|
|
|
|
import cssutils |
|
|
|
|
from lxml import etree |
|
|
|
|
from xml import xpath |
|
|
|
|
|
|
|
|
|
from svg.charts import css # causes the SVG profile to be loaded |
|
|
|
|
|
|
|
|
|
@ -22,12 +23,15 @@ try:
|
|
|
|
|
except ImportError: |
|
|
|
|
zlib = None |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def sort_multiple(arrays): |
|
|
|
|
"sort multiple lists (of equal size) using the first list for the sort keys" |
|
|
|
|
"sort multiple lists (of equal size) " |
|
|
|
|
"using the first list for the sort keys" |
|
|
|
|
tuples = zip(*arrays) |
|
|
|
|
tuples.sort() |
|
|
|
|
return zip(*tuples) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Graph(object): |
|
|
|
|
""" |
|
|
|
|
Base object for generating SVG Graphs |
|
|
|
|
@ -67,14 +71,16 @@ class Graph(object):
|
|
|
|
|
show_x_title = False |
|
|
|
|
x_title = 'X Field names' |
|
|
|
|
show_y_title = False |
|
|
|
|
y_title_text_direction= 'bt' # 'bt' for bottom to top; 'tb' for top to bottom |
|
|
|
|
# 'bt' for bottom to top; 'tb' for top to bottom |
|
|
|
|
y_title_text_direction = 'bt' |
|
|
|
|
y_title = 'Y Scale' |
|
|
|
|
show_graph_title = False |
|
|
|
|
graph_title = 'Graph Title' |
|
|
|
|
show_graph_subtitle = False |
|
|
|
|
graph_subtitle = 'Graph Subtitle' |
|
|
|
|
key = True |
|
|
|
|
key_position= 'right' # 'bottom' or 'right', |
|
|
|
|
# 'bottom' or 'right', |
|
|
|
|
key_position = 'right' |
|
|
|
|
|
|
|
|
|
font_size = 12 |
|
|
|
|
title_font_size = 16 |
|
|
|
|
@ -124,12 +130,15 @@ class Graph(object):
|
|
|
|
|
try: |
|
|
|
|
assert(isinstance(conf['data'], (tuple, list))) |
|
|
|
|
except TypeError, e: |
|
|
|
|
raise TypeError, "conf should be dictionary with 'data' and other items" |
|
|
|
|
raise TypeError( |
|
|
|
|
"conf should be dictionary with 'data' and other items") |
|
|
|
|
except AssertionError: |
|
|
|
|
if not hasattr(conf['data'], '__iter__'): |
|
|
|
|
raise TypeError, "conf['data'] should be tuple or list or iterable" |
|
|
|
|
raise TypeError( |
|
|
|
|
"conf['data'] should be tuple or list or iterable") |
|
|
|
|
|
|
|
|
|
def process_data(self, data): pass |
|
|
|
|
def process_data(self, data): |
|
|
|
|
pass |
|
|
|
|
|
|
|
|
|
def clear_data(self): |
|
|
|
|
""" |
|
|
|
|
@ -148,9 +157,11 @@ class Graph(object):
|
|
|
|
|
Raises ValueError when no data set has |
|
|
|
|
been added to the graph object. |
|
|
|
|
""" |
|
|
|
|
if not self.data: raise ValueError("No data available") |
|
|
|
|
if not self.data: |
|
|
|
|
raise ValueError("No data available") |
|
|
|
|
|
|
|
|
|
if hasattr(self, 'calculations'): self.calculations() |
|
|
|
|
if hasattr(self, 'calculations'): |
|
|
|
|
self.calculations() |
|
|
|
|
|
|
|
|
|
self.start_svg() |
|
|
|
|
self.calculate_graph_dimensions() |
|
|
|
|
@ -166,9 +177,12 @@ class Graph(object):
|
|
|
|
|
|
|
|
|
|
def _burn_compressed(self): |
|
|
|
|
if self.compress and not zlib: |
|
|
|
|
self.root.addprevious(etree.Comment('Python zlib not available for SVGZ')) |
|
|
|
|
self.root.addprevious( |
|
|
|
|
etree.Comment('Python zlib not available for SVGZ')) |
|
|
|
|
|
|
|
|
|
data = etree.tostring(self.root, pretty_print=True, xml_declaration=True, encoding='utf-8') |
|
|
|
|
data = etree.tostring( |
|
|
|
|
self.root, pretty_print=True, |
|
|
|
|
xml_declaration=True, encoding='utf-8') |
|
|
|
|
|
|
|
|
|
if self.compress and zlib: |
|
|
|
|
data = zlib.compress(data) |
|
|
|
|
@ -189,10 +203,14 @@ class Graph(object):
|
|
|
|
|
else: |
|
|
|
|
label_lengths = map(len, self.get_y_labels()) |
|
|
|
|
max_y_label_len = max(label_lengths) |
|
|
|
|
max_y_label_height_px = 0.6 * max_y_label_len * self.y_label_font_size |
|
|
|
|
if self.show_y_labels: bl += max_y_label_height_px |
|
|
|
|
if self.stagger_y_labels: bl += max_y_label_height_px + 10 |
|
|
|
|
if self.show_y_title: bl += self.y_title_font_size + 5 |
|
|
|
|
max_y_label_height_px = (0.6 * max_y_label_len * |
|
|
|
|
self.y_label_font_size) |
|
|
|
|
if self.show_y_labels: |
|
|
|
|
bl += max_y_label_height_px |
|
|
|
|
if self.stagger_y_labels: |
|
|
|
|
bl += max_y_label_height_px + 10 |
|
|
|
|
if self.show_y_title: |
|
|
|
|
bl += self.y_title_font_size + 5 |
|
|
|
|
self.border_left = bl |
|
|
|
|
|
|
|
|
|
def max_y_label_width_px(self): |
|
|
|
|
@ -222,9 +240,11 @@ class Graph(object):
|
|
|
|
|
border_top. |
|
|
|
|
""" |
|
|
|
|
self.border_top = 5 |
|
|
|
|
if self.show_graph_title: self.border_top += self.title_font_size |
|
|
|
|
if self.show_graph_title: |
|
|
|
|
self.border_top += self.title_font_size |
|
|
|
|
self.border_top += 5 |
|
|
|
|
if self.show_graph_subtitle: self.border_top += self.subtitle_font_size |
|
|
|
|
if self.show_graph_subtitle: |
|
|
|
|
self.border_top += self.subtitle_font_size |
|
|
|
|
|
|
|
|
|
def add_popup(self, x, y, label): |
|
|
|
|
""" |
|
|
|
|
@ -245,7 +265,8 @@ class Graph(object):
|
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
# add the circle element to the foreground |
|
|
|
|
visibility = "document.getElementById('%s').setAttribute('visibility', %%s)" % id |
|
|
|
|
visibility = ("document.getElementById('%s')." |
|
|
|
|
"setAttribute('visibility', %%s)" % id) |
|
|
|
|
t = etree.SubElement(self.foreground, 'circle', { |
|
|
|
|
'cx': str(x), |
|
|
|
|
'cy': str(y), |
|
|
|
|
@ -271,8 +292,10 @@ class Graph(object):
|
|
|
|
|
max_x_label_len = reduce(max, label_lengths) |
|
|
|
|
max_x_label_height_px *= 0.6 * max_x_label_len |
|
|
|
|
bb += max_x_label_height_px |
|
|
|
|
if self.stagger_x_labels: bb += max_x_label_height_px + 10 |
|
|
|
|
if self.show_x_title: bb += self.x_title_font_size + 5 |
|
|
|
|
if self.stagger_x_labels: |
|
|
|
|
bb += max_x_label_height_px + 10 |
|
|
|
|
if self.show_x_title: |
|
|
|
|
bb += self.x_title_font_size + 5 |
|
|
|
|
self.border_bottom = bb |
|
|
|
|
|
|
|
|
|
def draw_graph(self): |
|
|
|
|
@ -336,7 +359,8 @@ class Graph(object):
|
|
|
|
|
'y': str(y), |
|
|
|
|
'class': 'dataPointLabel'}) |
|
|
|
|
e.text = str(value) |
|
|
|
|
if style: e.set('style', style) |
|
|
|
|
if style: |
|
|
|
|
e.set('style', style) |
|
|
|
|
|
|
|
|
|
def draw_x_labels(self): |
|
|
|
|
"Draw the X axis labels" |
|
|
|
|
@ -388,13 +412,14 @@ class Graph(object):
|
|
|
|
|
return 0 |
|
|
|
|
|
|
|
|
|
def get_field_width(self): |
|
|
|
|
return float(self.graph_width - self.font_size*2*self.right_font) / \ |
|
|
|
|
(len(self.get_x_labels()) - self.right_align) |
|
|
|
|
return (float( |
|
|
|
|
self.graph_width - self.font_size * 2 * self.right_font) / |
|
|
|
|
(len(self.get_x_labels()) - self.right_align)) |
|
|
|
|
field_width = get_field_width |
|
|
|
|
|
|
|
|
|
def get_field_height(self): |
|
|
|
|
return float(self.graph_height - self.font_size*2*self.top_font) / \ |
|
|
|
|
(len(self.get_y_labels()) - self.top_align) |
|
|
|
|
return (float(self.graph_height - self.font_size * 2 * self.top_font) / |
|
|
|
|
(len(self.get_y_labels()) - self.top_align)) |
|
|
|
|
field_height = get_field_height |
|
|
|
|
|
|
|
|
|
def draw_y_labels(self): |
|
|
|
|
@ -414,7 +439,8 @@ class Graph(object):
|
|
|
|
|
|
|
|
|
|
def get_y_offset(self): |
|
|
|
|
result = self.graph_height + self.y_label_offset(self.field_height()) |
|
|
|
|
if not self.rotate_y_labels: result += self.font_size/1.2 |
|
|
|
|
if not self.rotate_y_labels: |
|
|
|
|
result += self.font_size / 1.2 |
|
|
|
|
return result |
|
|
|
|
y_offset = property(get_y_offset) |
|
|
|
|
|
|
|
|
|
@ -449,7 +475,8 @@ class Graph(object):
|
|
|
|
|
|
|
|
|
|
def draw_x_guidelines(self, label_height, count): |
|
|
|
|
"Draw the X-axis guidelines" |
|
|
|
|
if not self.show_x_guidelines: return |
|
|
|
|
if not self.show_x_guidelines: |
|
|
|
|
return |
|
|
|
|
# skip the first one |
|
|
|
|
for count in range(1, count): |
|
|
|
|
start = label_height * count |
|
|
|
|
@ -460,7 +487,8 @@ class Graph(object):
|
|
|
|
|
|
|
|
|
|
def draw_y_guidelines(self, label_height, count): |
|
|
|
|
"Draw the Y-axis guidelines" |
|
|
|
|
if not self.show_y_guidelines: return |
|
|
|
|
if not self.show_y_guidelines: |
|
|
|
|
return |
|
|
|
|
for count in range(1, count): |
|
|
|
|
start = self.graph_height - label_height * count |
|
|
|
|
stop = self.graph_width |
|
|
|
|
@ -470,10 +498,14 @@ class Graph(object):
|
|
|
|
|
|
|
|
|
|
def draw_titles(self): |
|
|
|
|
"Draws the graph title and subtitle" |
|
|
|
|
if self.show_graph_title: self.draw_graph_title() |
|
|
|
|
if self.show_graph_subtitle: self.draw_graph_subtitle() |
|
|
|
|
if self.show_x_title: self.draw_x_title() |
|
|
|
|
if self.show_y_title: self.draw_y_title() |
|
|
|
|
if self.show_graph_title: |
|
|
|
|
self.draw_graph_title() |
|
|
|
|
if self.show_graph_subtitle: |
|
|
|
|
self.draw_graph_subtitle() |
|
|
|
|
if self.show_x_title: |
|
|
|
|
self.draw_x_title() |
|
|
|
|
if self.show_y_title: |
|
|
|
|
self.draw_y_title() |
|
|
|
|
|
|
|
|
|
def draw_graph_title(self): |
|
|
|
|
text = etree.SubElement(self.root, 'text', { |
|
|
|
|
@ -483,7 +515,8 @@ class Graph(object):
|
|
|
|
|
text.text = self.graph_title |
|
|
|
|
|
|
|
|
|
def draw_graph_subtitle(self): |
|
|
|
|
y_subtitle_options = [subtitle_font_size, title_font_size+10] |
|
|
|
|
y_subtitle_options = [self.subtitle_font_size, |
|
|
|
|
self.title_font_size + 10] |
|
|
|
|
y_subtitle = y_subtitle_options[self.show_graph_title] |
|
|
|
|
text = etree.SubElement(self.root, 'text', { |
|
|
|
|
'x': str(self.width / 2), |
|
|
|
|
@ -496,7 +529,8 @@ class Graph(object):
|
|
|
|
|
y = self.graph_height + self.border_top + self.x_title_font_size |
|
|
|
|
if self.show_x_labels: |
|
|
|
|
y_size = self.x_label_font_size + 5 |
|
|
|
|
if self.stagger_x_labels: y_size*=2 |
|
|
|
|
if self.stagger_x_labels: |
|
|
|
|
y_size *= 2 |
|
|
|
|
y += y_size |
|
|
|
|
x = self.width / 2 |
|
|
|
|
|
|
|
|
|
@ -560,16 +594,16 @@ class Graph(object):
|
|
|
|
|
x_offset = self.border_left + 20 |
|
|
|
|
y_offset = self.border_top + self.graph_height + 5 |
|
|
|
|
if self.show_x_labels: |
|
|
|
|
max_x_label_height_px = x_label_font_size |
|
|
|
|
max_x_label_height_px = self.x_label_font_size |
|
|
|
|
if self.rotate_x_labels: |
|
|
|
|
longest_label_length = max(map(len, self.get_x_labels())) |
|
|
|
|
# note: I think 0.6 is the ratio of width to height of characters |
|
|
|
|
# I think 0.6 is the ratio of width to height of characters |
|
|
|
|
max_x_label_height_px *= longest_label_length * 0.6 |
|
|
|
|
y_offset += max_x_label_height_px |
|
|
|
|
if self.stagger_x_labels: |
|
|
|
|
y_offset += max_x_label_height_px + 5 |
|
|
|
|
if self.show_x_title: |
|
|
|
|
y_offset += x_title_font_size + 5 |
|
|
|
|
y_offset += self.x_title_font_size + 5 |
|
|
|
|
return x_offset, y_offset |
|
|
|
|
|
|
|
|
|
def render_inline_styles(self): |
|
|
|
|
@ -616,7 +650,8 @@ class Graph(object):
|
|
|
|
|
'width': str(self.width), |
|
|
|
|
'height': str(self.height), |
|
|
|
|
'viewBox': '0 0 %s %s' % (self.width, self.height), |
|
|
|
|
'{http://ns.adobe.com/AdobeSVGViewerExtensions/3.0/}scriptImplementation': 'Adobe', |
|
|
|
|
'{http://ns.adobe.com/AdobeSVGViewerExtensions/3.0/}' |
|
|
|
|
'scriptImplementation': 'Adobe', |
|
|
|
|
}, nsmap=NSMAP) |
|
|
|
|
if hasattr(self, 'style_sheet_href'): |
|
|
|
|
pi = etree.ProcessingInstruction( |
|
|
|
|
@ -638,10 +673,11 @@ class Graph(object):
|
|
|
|
|
self.add_defs(defs) |
|
|
|
|
|
|
|
|
|
if not hasattr(self, 'style_sheet_href') and not self.css_inline: |
|
|
|
|
self.root.append(etree.Comment(' include default stylesheet if none specified ')) |
|
|
|
|
self.root.append(etree.Comment( |
|
|
|
|
' include default stylesheet if none specified ')) |
|
|
|
|
style = etree.SubElement(defs, 'style', type='text/css') |
|
|
|
|
# TODO: the text was previously escaped in a CDATA declaration... how |
|
|
|
|
# to do that with etree? |
|
|
|
|
# TODO: the text was previously escaped in a CDATA declaration... |
|
|
|
|
# how to do that with etree? |
|
|
|
|
style.text = self.get_stylesheet().cssText |
|
|
|
|
|
|
|
|
|
self.root.append(etree.Comment('SVG Background')) |
|
|
|
|
@ -679,11 +715,13 @@ class Graph(object):
|
|
|
|
|
|
|
|
|
|
def get_stylesheet(self): |
|
|
|
|
cssutils.log.setLevel(30) # disable INFO log messages |
|
|
|
|
|
|
|
|
|
def merge_sheets(s1, s2): |
|
|
|
|
map(s1.add, s2) |
|
|
|
|
return s1 |
|
|
|
|
return reduce(merge_sheets, self.get_stylesheet_resources()) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class class_dict(object): |
|
|
|
|
"Emulates a dictionary, but retrieves class attributes" |
|
|
|
|
def __init__(self, obj): |
|
|
|
|
|
Florian Mounier
13 years ago