[svg] fix rotation in svg
This commit is contained in:
parent
9ede16df5d
commit
06f8611a90
2 changed files with 86 additions and 1 deletions
70
masque/test/test_svg.py
Normal file
70
masque/test/test_svg.py
Normal file
|
|
@ -0,0 +1,70 @@
|
|||
from pathlib import Path
|
||||
import xml.etree.ElementTree as ET
|
||||
|
||||
import numpy
|
||||
import pytest
|
||||
from numpy.testing import assert_allclose
|
||||
|
||||
pytest.importorskip("svgwrite")
|
||||
|
||||
from ..library import Library
|
||||
from ..pattern import Pattern
|
||||
from ..file import svg
|
||||
|
||||
|
||||
SVG_NS = "{http://www.w3.org/2000/svg}"
|
||||
XLINK_HREF = "{http://www.w3.org/1999/xlink}href"
|
||||
|
||||
|
||||
def _child_transform(svg_path: Path) -> tuple[float, ...]:
|
||||
root = ET.fromstring(svg_path.read_text())
|
||||
for use in root.iter(f"{SVG_NS}use"):
|
||||
if use.attrib.get(XLINK_HREF) == "#child":
|
||||
raw = use.attrib["transform"]
|
||||
assert raw.startswith("matrix(") and raw.endswith(")")
|
||||
return tuple(float(value) for value in raw[7:-1].split())
|
||||
raise AssertionError("No child reference found in SVG output")
|
||||
|
||||
|
||||
def test_svg_ref_rotation_uses_correct_affine_transform(tmp_path: Path) -> None:
|
||||
lib = Library()
|
||||
child = Pattern()
|
||||
child.polygon("1", vertices=[[0, 0], [1, 0], [0, 1]])
|
||||
lib["child"] = child
|
||||
|
||||
top = Pattern()
|
||||
top.ref("child", offset=(3, 4), rotation=numpy.pi / 2, scale=2)
|
||||
lib["top"] = top
|
||||
|
||||
svg_path = tmp_path / "rotation.svg"
|
||||
svg.writefile(lib, "top", str(svg_path))
|
||||
|
||||
assert_allclose(_child_transform(svg_path), (0, 2, -2, 0, 3, 4), atol=1e-10)
|
||||
|
||||
|
||||
def test_svg_ref_mirroring_changes_affine_transform(tmp_path: Path) -> None:
|
||||
base = Library()
|
||||
child = Pattern()
|
||||
child.polygon("1", vertices=[[0, 0], [1, 0], [0, 1]])
|
||||
base["child"] = child
|
||||
|
||||
top_plain = Pattern()
|
||||
top_plain.ref("child", offset=(3, 4), rotation=numpy.pi / 2, scale=2, mirrored=False)
|
||||
base["plain"] = top_plain
|
||||
|
||||
plain_path = tmp_path / "plain.svg"
|
||||
svg.writefile(base, "plain", str(plain_path))
|
||||
plain_transform = _child_transform(plain_path)
|
||||
|
||||
mirrored = Library()
|
||||
mirrored["child"] = child.deepcopy()
|
||||
top_mirrored = Pattern()
|
||||
top_mirrored.ref("child", offset=(3, 4), rotation=numpy.pi / 2, scale=2, mirrored=True)
|
||||
mirrored["mirrored"] = top_mirrored
|
||||
|
||||
mirrored_path = tmp_path / "mirrored.svg"
|
||||
svg.writefile(mirrored, "mirrored", str(mirrored_path))
|
||||
mirrored_transform = _child_transform(mirrored_path)
|
||||
|
||||
assert_allclose(plain_transform, (0, 2, -2, 0, 3, 4), atol=1e-10)
|
||||
assert_allclose(mirrored_transform, (0, 2, 2, 0, 3, 4), atol=1e-10)
|
||||
Loading…
Add table
Add a link
Reference in a new issue