mirror of https://github.com/Kozea/pygal.git
Florian Mounier
10 years ago
15 changed files with 167 additions and 3332 deletions
|
Before Width: | Height: | Size: 151 KiB |
|
Before Width: | Height: | Size: 584 KiB |
|
Before Width: | Height: | Size: 370 KiB |
@ -0,0 +1,113 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2014 Kozea |
||||
# |
||||
# This library is free software: you can redistribute it and/or modify it under |
||||
# the terms of the GNU Lesser General Public License as published by the Free |
||||
# Software Foundation, either version 3 of the License, or (at your option) any |
||||
# later version. |
||||
# |
||||
# This library 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 Lesser General Public License for more |
||||
# details. |
||||
# |
||||
# You should have received a copy of the GNU Lesser General Public License |
||||
# along with pygal. If not, see <http://www.gnu.org/licenses/>. |
||||
|
||||
from __future__ import division |
||||
from pygal.graph.graph import Graph |
||||
from pygal.util import cut, cached_property, decorate |
||||
from pygal.etree import etree |
||||
from numbers import Number |
||||
|
||||
|
||||
class BaseMap(Graph): |
||||
"""Base map.""" |
||||
_dual = True |
||||
|
||||
@cached_property |
||||
def _values(self): |
||||
"""Getter for series values (flattened)""" |
||||
return [val[1] |
||||
for serie in self.series |
||||
for val in serie.values |
||||
if val[1] is not None] |
||||
|
||||
def get_values(self, serie): |
||||
return serie.values |
||||
|
||||
def _plot(self): |
||||
map = etree.fromstring(self.svg_map) |
||||
map.set('width', str(self.view.width)) |
||||
map.set('height', str(self.view.height)) |
||||
|
||||
for i, serie in enumerate(self.series): |
||||
safe_vals = list(filter( |
||||
lambda x: x is not None, cut(serie.values, 1))) |
||||
if not safe_vals: |
||||
continue |
||||
min_ = min(safe_vals) |
||||
max_ = max(safe_vals) |
||||
for j, (area_code, value) in enumerate(self.get_values(serie)): |
||||
# TODO: Generalize |
||||
if isinstance(area_code, Number): |
||||
area_code = '%2d' % area_code |
||||
|
||||
if value is None: |
||||
continue |
||||
if max_ == min_: |
||||
ratio = 1 |
||||
else: |
||||
ratio = .3 + .7 * (value - min_) / (max_ - min_) |
||||
try: |
||||
areae = map.findall( |
||||
".//*[@class='%s%s %s map-element']" % ( |
||||
self.area_prefix, area_code, |
||||
self.kind)) |
||||
except SyntaxError: |
||||
# Python 2.6 (you'd better install lxml) |
||||
areae = [] |
||||
for g in map: |
||||
for e in g: |
||||
if '%s%s' % ( |
||||
self.area_prefix, area_code |
||||
) in e.attrib.get('class', ''): |
||||
areae.append(e) |
||||
|
||||
if not areae: |
||||
continue |
||||
|
||||
for area in areae: |
||||
cls = area.get('class', '').split(' ') |
||||
cls.append('color-%d' % i) |
||||
area.set('class', ' '.join(cls)) |
||||
area.set('style', 'fill-opacity: %f' % (ratio)) |
||||
|
||||
metadata = serie.metadata.get(j) |
||||
if metadata: |
||||
node = decorate(self.svg, area, metadata) |
||||
if node != area: |
||||
area.remove(node) |
||||
for g in map: |
||||
if area not in g: |
||||
continue |
||||
index = list(g).index(area) |
||||
g.remove(area) |
||||
node.append(area) |
||||
g.insert(index, node) |
||||
|
||||
last_node = len(area) > 0 and area[-1] |
||||
if last_node is not None and last_node.tag == 'title': |
||||
title_node = last_node |
||||
text = title_node.text + '\n' |
||||
else: |
||||
title_node = self.svg.node(area, 'title') |
||||
text = '' |
||||
title_node.text = text + '[%s] %s: %s' % ( |
||||
serie.title, |
||||
self.area_names[area_code], self._format(value)) |
||||
|
||||
self.nodes['plot'].append(map) |
||||
@ -1,109 +0,0 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2014 Kozea |
||||
# |
||||
# This library is free software: you can redistribute it and/or modify it under |
||||
# the terms of the GNU Lesser General Public License as published by the Free |
||||
# Software Foundation, either version 3 of the License, or (at your option) any |
||||
# later version. |
||||
# |
||||
# This library 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 Lesser General Public License for more |
||||
# details. |
||||
# |
||||
# You should have received a copy of the GNU Lesser General Public License |
||||
# along with pygal. If not, see <http://www.gnu.org/licenses/>. |
||||
""" |
||||
Supranational Worldmap chart |
||||
|
||||
""" |
||||
|
||||
from __future__ import division |
||||
from pygal.graph.worldmap import Worldmap |
||||
from pygal.i18n import SUPRANATIONAL |
||||
from pygal.util import cut, decorate |
||||
from pygal.etree import etree |
||||
import os |
||||
|
||||
with open(os.path.join( |
||||
os.path.dirname(__file__), |
||||
'worldmap.svg')) as file: |
||||
MAP = file.read() |
||||
|
||||
|
||||
class SupranationalWorldmap(Worldmap): |
||||
"""SupranationalWorldmap graph""" |
||||
def _plot(self): |
||||
map = etree.fromstring(MAP) |
||||
map.set('width', str(self.view.width)) |
||||
map.set('height', str(self.view.height)) |
||||
|
||||
for i, serie in enumerate(self.series): |
||||
safe_vals = list(filter( |
||||
lambda x: x is not None, cut(serie.values, 1))) |
||||
if not safe_vals: |
||||
continue |
||||
min_ = min(safe_vals) |
||||
max_ = max(safe_vals) |
||||
serie.values = self.replace_supranationals(serie.values) |
||||
for j, (country_code, value) in enumerate(serie.values): |
||||
if value is None: |
||||
continue |
||||
if max_ == min_: |
||||
ratio = 1 |
||||
else: |
||||
ratio = .3 + .7 * (value - min_) / (max_ - min_) |
||||
|
||||
try: |
||||
country = map.find('.//*[@id="%s"]' % country_code) |
||||
except SyntaxError: |
||||
# Python 2.6 (you'd better install lxml) |
||||
country = None |
||||
for e in map: |
||||
if e.attrib.get('id', '') == country_code: |
||||
country = e |
||||
|
||||
if country is None: |
||||
continue |
||||
cls = country.get('class', '').split(' ') |
||||
cls.append('color-%d' % i) |
||||
country.set('class', ' '.join(cls)) |
||||
country.set( |
||||
'style', 'fill-opacity: %f' % ( |
||||
ratio)) |
||||
|
||||
metadata = serie.metadata.get(j) |
||||
if metadata: |
||||
node = decorate(self.svg, country, metadata) |
||||
if node != country: |
||||
country.remove(node) |
||||
index = list(map).index(country) |
||||
map.remove(country) |
||||
node.append(country) |
||||
map.insert(index, node) |
||||
|
||||
last_node = len(country) > 0 and country[-1] |
||||
if last_node is not None and last_node.tag == 'title': |
||||
title_node = last_node |
||||
text = title_node.text + '\n' |
||||
else: |
||||
title_node = self.svg.node(country, 'title') |
||||
text = '' |
||||
title_node.text = text + '[%s] %s: %d' % ( |
||||
serie.title, |
||||
self.country_names[country_code], value) |
||||
|
||||
self.nodes['plot'].append(map) |
||||
|
||||
def replace_supranationals(self, values): |
||||
"""Replaces the values if it contains a supranational code.""" |
||||
for i, (code, value) in enumerate(values[:]): |
||||
for suprakey in SUPRANATIONAL.keys(): |
||||
if suprakey == code: |
||||
values.extend( |
||||
[(country, value) for country in SUPRANATIONAL[code]]) |
||||
values.remove((code, value)) |
||||
return values |
||||
|
Before Width: | Height: | Size: 754 KiB |
Loading…
Reference in new issue