/usr/lib/python3/dist-packages/leather/grid.py is in python3-leather 0.3.3-1.
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 | #!/usr/bin/env python
import math
import os
import xml.etree.ElementTree as ET
import six
import leather.svg as svg
from leather import theme
from leather.utils import IPythonSVG
class Grid(object):
"""
A container for a set of :class:`.Chart` instances that are rendered in a
grid layout.
"""
def __init__(self):
self._charts = []
def add_one(self, chart):
"""
Add a :class:`.Chart` to the grid.
"""
self._charts.append(chart)
def add_many(self, charts):
"""
Add a sequence of charts to this grid.
"""
self._charts.extend(charts)
def to_svg(self, path=None, width=None, height=None):
"""
Render the grid to an SVG.
The :code:`width` and :code:`height` arguments refer to the size of the
entire grid. The size of individual charts will be inferred
automatically.
See :meth:`.Chart.to_svg` for arguments.
"""
if not width or not height:
count = len(self._charts)
columns = math.ceil(math.sqrt(count))
rows = math.ceil(count / columns)
width = columns * theme.default_chart_width
height = rows * theme.default_chart_height
root = ET.Element('svg',
width=six.text_type(width),
height=six.text_type(height),
version='1.1',
xmlns='http://www.w3.org/2000/svg'
)
# Root / background
root_group = ET.Element('g')
root_group.append(ET.Element('rect',
x=six.text_type(0),
y=six.text_type(0),
width=six.text_type(width),
height=six.text_type(height),
fill=theme.background_color
))
root.append(root_group)
# Charts
grid_group = ET.Element('g')
chart_count = len(self._charts)
grid_width = math.ceil(math.sqrt(chart_count))
grid_height = math.ceil(chart_count / grid_width)
chart_width = width / grid_width
chart_height = height / grid_height
for i, chart in enumerate(self._charts):
x = (i % grid_width) * chart_width
y = math.floor(i / grid_width) * chart_height
group = ET.Element('g')
group.set('transform', svg.translate(x, y))
chart = chart.to_svg_group(chart_width, chart_height)
group.append(chart)
grid_group.append(group)
root_group.append(grid_group)
svg_text = svg.stringify(root)
close = True
if path:
f = None
try:
if hasattr(path, 'write'):
f = path
close = False
else:
dirpath = os.path.dirname(path)
if dirpath and not os.path.exists(dirpath):
os.makedirs(dirpath)
f = open(path, 'w')
f.write(svg.HEADER)
f.write(svg_text)
finally:
if close and f is not None:
f.close()
else:
return IPythonSVG(svg_text)
|