FIX: Add get/set color methods for some 3d collections

WeatherGod · QuLogic · commit 6a1a6e9c388a · 2020-09-17T02:53:18.000-04:00
closes #3370 We need to do an extra step to propagate the color information back to the 3D projections
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -403,6 +403,16 @@ def __init__(self, *args, zs=0, zdir='z', depthshade=True, **kwargs):
         super().__init__(*args, **kwargs)
         self.set_3d_properties(zs, zdir)
 
+    def set_facecolor(self, c):
+        # docstring inherited
+        super().set_facecolor(c)
+        self._facecolor3d = self.get_facecolor()
+
+    def set_edgecolor(self, c):
+        # docstring inherited
+        super().set_edgecolor(c)
+        self._edgecolor3d = self.get_edgecolor()
+
     def set_sort_zpos(self, val):
         """Set the position to use for z-sorting."""
         self._sort_zpos = val
@@ -491,6 +501,16 @@ def set_3d_properties(self, zs, zdir):
         self._linewidth3d = self.get_linewidth()
         self.stale = True
 
+    def set_facecolor(self, c):
+        # docstring inherited
+        super().set_facecolor(c)
+        self._facecolor3d = self.get_facecolor()
+
+    def set_edgecolor(self, c):
+        # docstring inherited
+        super().set_edgecolor(c)
+        self._edgecolor3d = self.get_edgecolor()
+
     def do_3d_projection(self, renderer):
         xs, ys, zs = self._offsets3d
         vxs, vys, vzs, vis = proj3d.proj_transform_clip(xs, ys, zs, renderer.M)
@@ -722,10 +742,12 @@ def do_3d_projection(self, renderer):
             return np.nan
 
     def set_facecolor(self, colors):
+        # docstring inherited
         super().set_facecolor(colors)
         self._facecolors3d = PolyCollection.get_facecolor(self)
 
     def set_edgecolor(self, colors):
+        # docstring inherited
         super().set_edgecolor(colors)
         self._edgecolors3d = PolyCollection.get_edgecolor(self)
 
