Add accessors for 3D collection's depth shading.

QuLogic · QuLogic · commit 264f5c91e5ec · 2020-09-17T03:05:04.000-04:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -403,6 +403,22 @@ def __init__(self, *args, zs=0, zdir='z', depthshade=True, **kwargs):
         super().__init__(*args, **kwargs)
         self.set_3d_properties(zs, zdir)
 
+    def get_depthshade(self):
+        return self._depthshade
+
+    def set_depthshade(self, depthshade):
+        """
+        Set whether depth shading is performed on collection members.
+
+        Parameters
+        ----------
+        depthshade : bool
+            Whether to shade the patches in order to give the appearance of
+            depth.
+        """
+        self._depthshade = depthshade
+        self.stale = True
+
     def set_facecolor(self, c):
         # docstring inherited
         super().set_facecolor(c)
@@ -501,6 +517,22 @@ def set_3d_properties(self, zs, zdir):
         self._linewidth3d = self.get_linewidth()
         self.stale = True
 
+    def get_depthshade(self):
+        return self._depthshade
+
+    def set_depthshade(self, depthshade):
+        """
+        Set whether depth shading is performed on collection members.
+
+        Parameters
+        ----------
+        depthshade : bool
+            Whether to shade the patches in order to give the appearance of
+            depth.
+        """
+        self._depthshade = depthshade
+        self.stale = True
+
     def set_facecolor(self, c):
         # docstring inherited
         super().set_facecolor(c)
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -251,12 +251,15 @@ def test_scatter3d_modification(fig_ref, fig_test):
                         marker='o')
     c.set_facecolor('C1')
     c.set_edgecolor('C2')
+    c.set_alpha([0.3, 0.7] * 5)
+    c.set_depthshade(False)
     c.set_sizes(np.full(10, 75))
     c.set_linewidths(3)
 
     ax_ref = fig_ref.add_subplot(projection='3d')
     ax_ref.scatter(np.arange(10), np.arange(10), np.arange(10), marker='o',
-                   facecolor='C1', edgecolor='C2', s=75, linewidths=3)
+                   facecolor='C1', edgecolor='C2', alpha=[0.3, 0.7] * 5,
+                   depthshade=False, s=75, linewidths=3)
 
 
 @pytest.mark.parametrize('depthshade', [True, False])
@@ -552,10 +555,13 @@ def test_patch_collection_modification(fig_test, fig_ref):
     ax_test.add_collection3d(c)
     c.set_edgecolor('C2')
     c.set_facecolor('C3')
+    c.set_alpha(0.7)
+    c.set_depthshade(False)
 
     patch = Circle((0, 0), 0.05)
     c = art3d.Patch3DCollection([patch], linewidths=3,
-                                edgecolor='C2', facecolor='C3')
+                                edgecolor='C2', facecolor='C3', alpha=0.7,
+                                depthshade=False)
 
     ax_ref = fig_ref.add_subplot(projection='3d')
     ax_ref.add_collection3d(c)
