jan/masque
1
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

[apply_transforms] fix handling of rotations while mirrored

Browse source
This commit is contained in:
jan 2026-03-08 21:38:47 -07:00
commit 5989e45906
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
masque/utils/transform.py
View file

@ -114,9 +114,13 @@ def apply_transforms(
tot = numpy.empty((outer.shape[0], inner.shape[0], 4))
tot[:, :, :2] = outer[:, None, :2] + xy
tot[:, :, 2:] = outer[:, None, 2:] + inner[None, :, 2:] # sum rotations and mirrored
tot[:, :, 2] %= 2 * pi # clamp rot
tot[:, :, 3] %= 2 # clamp mirrored
# If mirrored, flip inner rotation
mirrored_outer = outer[:, None, 3].astype(bool)
rotations = outer[:, None, 2] + numpy.where(mirrored_outer, -inner[None, :, 2], inner[None, :, 2])
tot[:, :, 2] = rotations % (2 * pi)
tot[:, :, 3] = (outer[:, None, 3] + inner[None, :, 3]) % 2 # net mirrored
if tensor:
return tot