|
import re |
|
|
|
|
|
def _prefer_non_zero(*args): |
|
for arg in args: |
|
if arg != 0: |
|
return arg |
|
return 0.0 |
|
|
|
|
|
def _ntos(n): |
|
|
|
return ("%.3f" % n).rstrip("0").rstrip(".") |
|
|
|
|
|
def _strip_xml_ns(tag): |
|
|
|
|
|
return tag.split("}", 1)[1] if "}" in tag else tag |
|
|
|
|
|
def _transform(raw_value): |
|
|
|
|
|
|
|
|
|
match = re.match(r"matrix\((.*)\)", raw_value) |
|
if not match: |
|
raise NotImplementedError |
|
matrix = tuple(float(p) for p in re.split(r"\s+|,", match.group(1))) |
|
if len(matrix) != 6: |
|
raise ValueError("wrong # of terms in %s" % raw_value) |
|
return matrix |
|
|
|
|
|
class PathBuilder(object): |
|
def __init__(self): |
|
self.paths = [] |
|
self.transforms = [] |
|
|
|
def _start_path(self, initial_path=""): |
|
self.paths.append(initial_path) |
|
self.transforms.append(None) |
|
|
|
def _end_path(self): |
|
self._add("z") |
|
|
|
def _add(self, path_snippet): |
|
path = self.paths[-1] |
|
if path: |
|
path += " " + path_snippet |
|
else: |
|
path = path_snippet |
|
self.paths[-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, y): |
|
self._vhline("v", y) |
|
|
|
def _line(self, c, x, y): |
|
self._add("%s%s,%s" % (c, _ntos(x), _ntos(y))) |
|
|
|
def L(self, x, y): |
|
self._line("L", x, y) |
|
|
|
def l(self, x, y): |
|
self._line("l", x, y) |
|
|
|
def _parse_line(self, line): |
|
x1 = float(line.attrib.get("x1", 0)) |
|
y1 = float(line.attrib.get("y1", 0)) |
|
x2 = float(line.attrib.get("x2", 0)) |
|
y2 = float(line.attrib.get("y2", 0)) |
|
|
|
self._start_path() |
|
self.M(x1, y1) |
|
self.L(x2, y2) |
|
|
|
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) |
|
|
|
|
|
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_polyline(self, poly): |
|
if "points" in poly.attrib: |
|
self._start_path("M" + poly.attrib["points"]) |
|
|
|
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")) |
|
|
|
|
|
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 _parse_ellipse(self, ellipse): |
|
cx = float(ellipse.attrib.get("cx", 0)) |
|
cy = float(ellipse.attrib.get("cy", 0)) |
|
rx = float(ellipse.attrib.get("rx")) |
|
ry = float(ellipse.attrib.get("ry")) |
|
|
|
|
|
self._start_path() |
|
self.M(cx - rx, cy) |
|
self.A(rx, ry, cx + rx, cy, large_arc=1) |
|
self.A(rx, ry, cx - rx, cy, large_arc=1) |
|
|
|
def add_path_from_element(self, el): |
|
tag = _strip_xml_ns(el.tag) |
|
parse_fn = getattr(self, "_parse_%s" % tag.lower(), None) |
|
if not callable(parse_fn): |
|
return False |
|
parse_fn(el) |
|
if "transform" in el.attrib: |
|
self.transforms[-1] = _transform(el.attrib["transform"]) |
|
return True |
|
|