playground
/
pygal
mirror of https://github.com/Kozea/pygal.git
1
0
Fork 0
Code Issues Releases Activity
Browse Source

Simplify since charts are recreated from scratch every time

pull/8/head
Florian Mounier 12 years ago
parent
1fb64acb97
commit
9782569f47
5 changed files with 47 additions and 57 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 13
      pygal/graph/horizontal.py
  2. 25
      pygal/graph/line.py
  3. 43
      pygal/graph/radar.py
  4. 10
      pygal/graph/stackedline.py
  5. 13
      pygal/graph/xy.py

13
pygal/graph/horizontal.py
Unescape View File

@ -26,21 +26,16 @@ from pygal.graph.graph import Graph
class HorizontalGraph(Graph):
"""Horizontal graph"""
def __init__(self, *args, **kwargs):
self.first_pass = True
self.horizontal = True
super(HorizontalGraph, self).__init__(*args, **kwargs)
def _compute(self):
self.first_pass = False
# Stupid pylint
# pylint: disable-msg=E0203,W0201
if self.first_pass and self.x_labels:
if self.x_labels:
self.x_labels = list(reversed(self.x_labels))
# pylint: enable-msg=W0201,E0203
super(HorizontalGraph, self)._compute()
self._x_labels, self._y_labels = self._y_labels, self._x_labels
self._box.swap()
# Y axis is inverted
if self.first_pass:
for serie in self.series:
serie.values = list(reversed(serie.values))
for serie in self.series:
serie.values = list(reversed(serie.values))

25
pygal/graph/line.py
Unescape View File

@ -38,18 +38,11 @@ class Line(Graph):
@cached_property
def _values(self):
if self.interpolate:
return [
val[1]
for serie in self.series
for val in serie.interpolated
if val[1] is not None and (not self.logarithmic or val[1] > 0)]
else:
return [
val[1]
for serie in self.series
for val in serie.points
if val[1] is not None and (not self.logarithmic or val[1] > 0)]
return [
val[1]
for serie in self.series
for val in serie.points
if val[1] is not None and (not self.logarithmic or val[1] > 0)]
def _fill(self, values):
"""Add extra values to fill the line"""
@ -88,8 +81,6 @@ class Line(Graph):
y + self.value_font_size)
if self.stroke:
if self.interpolate:
view_values = map(self.view, serie.interpolated)
if self.fill:
view_values = self._fill(view_values)
self.svg.line(
@ -102,12 +93,12 @@ class Line(Graph):
] if self._len != 1 else [.5] # Center if only one value
for serie in self.series:
if not hasattr(serie, 'points'):
if self.interpolate:
serie.points = self._interpolate(serie.values, x_pos)
else:
serie.points = [
(x_pos[i], v)
for i, v in enumerate(serie.values)]
if self.interpolate:
serie.interpolated = self._interpolate(serie.values, x_pos)
if self.include_x_axis:
self._box.ymin = min(self._min, 0)

43
pygal/graph/radar.py
Unescape View File

@ -23,7 +23,7 @@ Radar chart
from __future__ import division
from pygal.graph.line import Line
from pygal.adapters import positive
from pygal.adapters import positive, none_to_zero
from pygal.view import PolarView
from pygal.util import deg, cached_property, compute_scale
from math import cos, pi
@ -32,7 +32,7 @@ from math import cos, pi
class Radar(Line):
"""Kiviat graph"""
_adapters = [positive]
_adapters = [positive, none_to_zero]
def __init__(self, *args, **kwargs):
self.x_pos = None
@ -49,7 +49,7 @@ class Radar(Line):
def _values(self):
if self.interpolate:
return [val[0] for serie in self.series
for val in serie.interpolated]
for val in serie.points]
else:
return super(Line, self)._values
@ -70,15 +70,16 @@ class Radar(Line):
for label, theta in self._x_labels:
guides = self.svg.node(axis, class_='guides')
end = self.view((r, theta))
self.svg.node(guides, 'path',
d='M%s L%s' % (format_(center), format_(end)),
class_='line')
self.svg.node(
guides, 'path',
d='M%s L%s' % (format_(center), format_(end)),
class_='line')
r_txt = (1 - self._box.__class__.margin) * self._box.ymax
pos_text = self.view((r_txt, theta))
text = self.svg.node(guides, 'text',
x=pos_text[0],
y=pos_text[1]
)
text = self.svg.node(
guides, 'text',
x=pos_text[0],
y=pos_text[1])
text.text = label
angle = - theta + pi / 2
if cos(angle) < 0:
@ -99,19 +100,15 @@ class Radar(Line):
close=True,
class_='guide line')
x, y = self.view((r, self.x_pos[0]))
self.svg.node(guides, 'text',
x=x - 5,
y=y
).text = label
self.svg.node(
guides, 'text',
x=x - 5,
y=y).text = label
def _compute(self):
delta = 2 * pi / self._len
x_pos = [.5 * pi + i * delta for i in range(self._len + 1)]
for serie in self.series:
vals = [val if val != None else 0 for val in serie.values]
serie.points = [
(v, x_pos[i])
for i, v in enumerate(vals)]
if self.interpolate:
extend = 2
extended_x_pos = (
@ -119,9 +116,15 @@ class Radar(Line):
x_pos +
[.5 * pi + i * delta for i in range(
self._len + 1, self._len + 1 + extend)])
extended_vals = vals[-extend:] + vals + vals[:extend]
serie.interpolated = self._interpolate(
extended_vals = (serie.values[-extend:] +
serie.values +
serie.values[:extend])
serie.point = self._interpolate(
extended_vals, extended_x_pos, polar=True)
else:
serie.points = [
(v, x_pos[i])
for i, v in enumerate(serie.values)]
self._box.margin *= 2
self._box.xmin = self._box.ymin = - self._max

10
pygal/graph/stackedline.py
Unescape View File

@ -49,9 +49,11 @@ class StackedLine(Line):
accumulation = [0] * self._len
for serie in self.series:
accumulation = map(sum, zip(accumulation, serie.values))
serie.points = [
(x_pos[i], v)
for i, v in enumerate(accumulation)]
if self.interpolate:
serie.interpolated = self._interpolate(accumulation, x_pos)
serie.points = self._interpolate(accumulation, x_pos)
else:
serie.points = [
(x_pos[i], v)
for i, v in enumerate(accumulation)]
return super(StackedLine, self)._compute()

13
pygal/graph/xy.py
Unescape View File

@ -46,26 +46,25 @@ class XY(Line):
rng = (xmax - xmin)
for serie in self.series:
serie.points = serie.values
if self.interpolate:
vals = zip(*sorted(serie.points, key=lambda x: x[0]))
serie.interpolated = self._interpolate(
serie.points = self._interpolate(
vals[1], vals[0], xy=True, xy_xmin=xmin, xy_rng=rng)
if not serie.interpolated:
serie.interpolated = serie.values
else:
serie.points = serie.values
if self.interpolate:
xvals = [val[0]
for serie in self.series
for val in serie.interpolated]
for val in serie.points]
yvals = [val[1]
for serie in self.series
for val in serie.interpolated]
for val in serie.points]
self._box.xmin, self._box.xmax = min(xvals), max(xvals)
self._box.ymin, self._box.ymax = min(yvals), max(yvals)
x_pos = compute_scale(
self._box.xmin, self._box.xmax, self.logarithmic, self.order_min)
self._box.xmin, self._box.xmax, self.logarithmic, self.order_min)
y_pos = compute_scale(
self._box.ymin, self._box.ymax, self.logarithmic, self.order_min)

Write Preview
Loading…
Cancel
Save