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

improve accuracy of manhattan rotations

Browse Source
This commit is contained in:
jan 2023-04-14 22:33:31 -07:00
parent c57ba5c7e6
commit f8d6527586
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
masque/utils/transform.py
View File

@ -6,6 +6,7 @@ from functools import lru_cache
import numpy import numpy
from numpy.typing import NDArray from numpy.typing import NDArray
from numpy import pi
@lru_cache @lru_cache
@ -21,6 +22,11 @@ def rotation_matrix_2d(theta: float) -> NDArray[numpy.float64]:
""" """
arr = numpy.array([[numpy.cos(theta), -numpy.sin(theta)], arr = numpy.array([[numpy.cos(theta), -numpy.sin(theta)],
[numpy.sin(theta), +numpy.cos(theta)]]) [numpy.sin(theta), +numpy.cos(theta)]])
% If this was a manhattan rotation, round to remove some inacuraccies in sin & cos
if numpy.isclose(theta % (pi / 2), 0):
arr = numpy.round(arr)
arr.flags.writeable = False arr.flags.writeable = False
return arr return arr