Add transforms needed for Axes3D

trananso · trananso · commit d85f75d84983 · 2024-05-08T11:05:09.000-04:00
diff --git a/lib/mpl_toolkits/mplot3d/meson.build b/lib/mpl_toolkits/mplot3d/meson.build
@@ -4,6 +4,7 @@ python_sources = [
   'axes3d.py',
   'axis3d.py',
   'proj3d.py',
+  'transform3d.py',
 ]
 
 py3.install_sources(python_sources, subdir: 'mpl_toolkits/mplot3d')
diff --git a/lib/mpl_toolkits/mplot3d/transform3d.py b/lib/mpl_toolkits/mplot3d/transform3d.py
@@ -0,0 +1,80 @@
+import numpy as np
+import matplotlib.transforms as mtransforms
+
+
+# These transforms break the assumption that the last row is [0, 0, 0, 1], and is
+# therefore not affine. However, this is required to preserve the order that
+# transforms are performed
+class NonAffine3D(mtransforms.Affine3D):
+    pass
+
+
+class WorldTransform(mtransforms.Affine3D):
+    def __init__(self, xmin, xmax, ymin, ymax, zmin, zmax, pb_aspect=None):
+        dx = xmax - xmin
+        dy = ymax - ymin
+        dz = zmax - zmin
+        if pb_aspect is not None:
+            ax, ay, az = pb_aspect
+            dx /= ax
+            dy /= ay
+            dz /= az
+        mtx = np.array([
+            [1/dx, 0,    0,    -xmin/dx],
+            [0,    1/dy, 0,    -ymin/dy],
+            [0,    0,    1/dz, -zmin/dz],
+            [0,    0,    0,    1]
+        ])
+        super().__init__(matrix=mtx)
+
+
+class PerspectiveTransform(NonAffine3D):
+    def __init__(self, zfront, zback, focal_length):
+        e = focal_length
+        a = 1
+        b = (zfront + zback) / (zfront - zback)
+        c = -2 * (zfront * zback) / (zfront - zback)
+        mtx = np.array([[e,   0,  0, 0],
+                         [0, e/a,  0, 0],
+                         [0,   0,  b, c],
+                         [0,   0, -1, 0]])
+        super().__init__(matrix=mtx)
+
+
+class OrthographicTransform(NonAffine3D):
+    def __init__(self, zfront, zback):
+        a = -(zfront + zback)
+        b = -(zfront - zback)
+        mtx = np.array([[2, 0,  0, 0],
+                         [0, 2,  0, 0],
+                         [0, 0, -2, 0],
+                         [0, 0,  a, b]])
+        super().__init__(matrix=mtx)
+
+
+class ViewTransform(mtransforms.Affine3D):
+    def __init__(self, u, v, w, E):
+        """
+        Return the view transformation matrix.
+
+        Parameters
+        ----------
+        u : 3-element numpy array
+            Unit vector pointing towards the right of the screen.
+        v : 3-element numpy array
+            Unit vector pointing towards the top of the screen.
+        w : 3-element numpy array
+            Unit vector pointing out of the screen.
+        E : 3-element numpy array
+            The coordinates of the eye/camera.
+        """
+        self._u = u
+        self._v = v
+        self._w = w
+
+        Mr = np.eye(4)
+        Mt = np.eye(4)
+        Mr[:3, :3] = [u, v, w]
+        Mt[:3, -1] = -E
+        mtx = np.dot(Mr, Mt)
+        super().__init__(matrix=mtx)

Original file line number	Diff line number	Diff line change
`@@ -4,6 +4,7 @@ python_sources = [`
`4`	`4`	`'axes3d.py',`
`5`	`5`	`'axis3d.py',`
`6`	`6`	`'proj3d.py',`
	`7`	`+ 'transform3d.py',`
`7`	`8`	`]`
`8`	`9`
`9`	`10`	`py3.install_sources(python_sources, subdir: 'mpl_toolkits/mplot3d')`