Fix 3D rotation precession

scottshambaugh · scottshambaugh · commit 29e4c75ad9fe · 2024-09-14T22:16:09.000-06:00
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -1568,10 +1568,11 @@ def _on_move(self, event):
             q = _Quaternion.from_cardan_angles(elev, azim, roll)
 
             # Update quaternion - a variation on Ken Shoemake's ARCBALL
-            current_vec = self._arcball(self._sx/w, self._sy/h)
-            new_vec = self._arcball(x/w, y/h)
+            scale = np.sqrt(2)/2  # slow down the rate of rotation
+            current_vec = self._arcball(self._sx*scale/w, self._sy*scale/h)
+            new_vec = self._arcball(x*scale/w, y*scale/h)
             dq = _Quaternion.rotate_from_to(current_vec, new_vec)
-            q = dq * q
+            q = dq * dq * q
 
             # Convert to elev, azim, roll
             elev, azim, roll = q.as_cardan_angles()
diff --git a/lib/mpl_toolkits/mplot3d/tests/test_axes3d.py b/lib/mpl_toolkits/mplot3d/tests/test_axes3d.py
@@ -1952,8 +1952,8 @@ def test_rotate():
             [30, 0.5, 0, 30, -90, 0],
             [0, 0, 0.5, -90, 0, 0],
             [30, 0, 0.5, -60, -90, 90],
-            [0, 0.5, 0.5, -45, -90, 45],
-            [30, 0.5, 0.5, -15, -90, 45]]:
+            [0, 0.5, 0.5, 0, -180, 90],
+            [30, 0.5, 0.5, 0, -180, 60]]:
         fig = plt.figure()
         ax = fig.add_subplot(1, 1, 1, projection='3d')
         ax.view_init(0, 0, roll)
@@ -1967,9 +1967,8 @@ def test_rotate():
                            xdata=dx*ax._pseudo_w, ydata=dy*ax._pseudo_h))
         fig.canvas.draw()
 
-        assert np.isclose(ax.elev, new_elev)
-        assert np.isclose(ax.azim, new_azim)
-        assert np.isclose(ax.roll, new_roll)
+        np.testing.assert_allclose((ax.elev, ax.azim, ax.roll),
+                                   (new_elev, new_azim, new_roll), atol=1e-5)
 
 
 def test_pan():
