ENH: Set the default plotbox aspect to match how renders looked previously

eric-wieser · eric-wieser · commit 58e961de365b · 2019-06-23T22:06:27.000-07:00
This sets it to have a 4:3 aspect ratio, which matches what would result from a typical figure layout
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -1027,6 +1027,9 @@ def get_proj(self):
         point.
 
         """
+        # chosen for similarity with the initial view before gh-8896
+        pb_aspect = np.array([4, 4, 3]) / 3.5
+
         relev, razim = np.pi * self.elev/180, np.pi * self.azim/180
 
         xmin, xmax = self.get_xlim3d()
@@ -1036,10 +1039,10 @@ def get_proj(self):
         # transform to uniform world coordinates 0-1.0,0-1.0,0-1.0
         worldM = proj3d.world_transformation(xmin, xmax,
                                              ymin, ymax,
-                                             zmin, zmax)
+                                             zmin, zmax, pb_aspect=pb_aspect)
 
         # look into the middle of the new coordinates
-        R = np.array([0.5, 0.5, 0.5])
+        R = pb_aspect / 2
 
         xp = R[0] + np.cos(razim) * np.cos(relev) * self.dist
         yp = R[1] + np.sin(razim) * np.cos(relev) * self.dist
diff --git a/lib/mpl_toolkits/mplot3d/proj3d.py b/lib/mpl_toolkits/mplot3d/proj3d.py
@@ -86,8 +86,20 @@ def mod(v):
 
 def world_transformation(xmin, xmax,
                          ymin, ymax,
-                         zmin, zmax):
-    dx, dy, dz = (xmax-xmin), (ymax-ymin), (zmax-zmin)
+                         zmin, zmax, pb_aspect=None):
+    """
+    produce a matrix that scales homogenous coords in the specified ranges
+    to [0, 1], or [0, pb_aspect[i]] if the plotbox aspect ratio is specified
+    """
+    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
+
     return np.array([[1/dx, 0,    0,    -xmin/dx],
                      [0,    1/dy, 0,    -ymin/dy],
                      [0,    0,    1/dz, -zmin/dz],
