121 lines
3.3 KiB
Python
Raw Normal View History

2019-02-12 11:53:27 -08:00
def _prefer_non_zero(*args):
for arg in args:
if arg != 0:
return arg
return 0.
2019-02-08 14:57:10 -08:00
def _ntos(n):
# %f likes to add unnecessary 0's, %g isn't consistent about # decimals
2019-02-12 11:53:27 -08:00
return ('%.3f' % n).rstrip('0').rstrip('.')
2019-02-07 19:00:26 -08:00
class PathBuilder(object):
2019-02-12 11:53:27 -08:00
def __init__(self):
self.pathes = []
def _start_path(self, initial_path=''):
self.pathes.append(initial_path)
def _end_path(self):
self._add('z')
def _add(self, path_snippet):
path = self.pathes[-1]
if path:
path += ' ' + path_snippet
else:
path = path_snippet
self.pathes[-1] = path
def _move(self, c, x, y):
self._add('%s%s,%s' % (c, _ntos(x), _ntos(y)))
def M(self, x, y):
self._move('M', x, y)
def m(self, x, y):
self._move('m', x, y)
def _arc(self, c, rx, ry, x, y, large_arc):
self._add('%s%s,%s 0 %d 1 %s,%s' % (c, _ntos(rx), _ntos(ry), large_arc,
_ntos(x), _ntos(y)))
def A(self, rx, ry, x, y, large_arc = 0):
self._arc('A', rx, ry, x, y, large_arc)
def a(self, rx, ry, x, y, large_arc = 0):
self._arc('a', rx, ry, x, y, large_arc)
def _vhline(self, c, x):
self._add('%s%s' % (c, _ntos(x)))
def H(self, x):
self._vhline('H', x)
def h(self, x):
self._vhline('h', x)
def V(self, y):
self._vhline('V', y)
def v(self, x):
self._vhline('v', y)
def _parse_rect(self, rect):
x = float(rect.attrib.get('x', 0))
y = float(rect.attrib.get('y', 0))
w = float(rect.attrib.get('width'))
h = float(rect.attrib.get('height'))
rx = float(rect.attrib.get('rx', 0))
ry = float(rect.attrib.get('ry', 0))
rx = _prefer_non_zero(rx, ry)
ry = _prefer_non_zero(ry, rx)
# TODO there are more rules for adjusting rx, ry
self._start_path()
self.M(x + rx, y)
self.H(x + w -rx)
if rx > 0:
self.A(rx, ry, x + w, y + ry)
self.V(y + h -ry)
if rx > 0:
self.A(rx, ry, x + w - rx, y + h)
self.H(x + rx)
if rx > 0:
self.A(rx, ry, x, y + h - ry)
self.V(y + ry)
if rx > 0:
self.A(rx, ry, x + rx, y)
self._end_path()
def _parse_path(self, path):
if 'd' in path.attrib:
self._start_path(initial_path=path.attrib['d'])
def _parse_polygon(self, poly):
if 'points' in poly.attrib:
self._start_path('M' + poly.attrib['points'])
self._end_path()
def _parse_circle(self, circle):
cx = float(circle.attrib.get('cx', 0))
cy = float(circle.attrib.get('cy', 0))
r = float(circle.attrib.get('r'))
# arc doesn't seem to like being a complete shape, draw two halves
self._start_path()
self.M(cx - r, cy)
self.A(r, r, cx + r, cy, large_arc=1)
self.A(r, r, cx - r, cy, large_arc=1)
def add_path_from_element(self, el):
tag = el.tag
if '}' in el.tag:
tag = el.tag.split('}', 1)[1] # from https://bugs.python.org/issue18304
parse_fn = getattr(self, '_parse_%s' % tag.lower(), None)
if not callable(parse_fn):
return False
parse_fn(el)
return True
2019-02-08 14:57:10 -08:00