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

WeatherGod · tacaswell · commit 081b3e11ddc9 · 2020-08-10T16:39:20.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
@@ -398,6 +398,18 @@ 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):
+        super().set_facecolor(c)
+        self._facecolor3d = self.get_facecolor()
+
+    set_facecolors = set_facecolor
+
+    def set_edgecolor(self, c):
+        super().set_edgecolor(c)
+        self._edgecolor3d = self.get_edgecolor()
+
+    set_edgecolors = set_edgecolor
+
     def set_sort_zpos(self, val):
         """Set the position to use for z-sorting."""
         self._sort_zpos = val
@@ -485,6 +497,18 @@ def set_3d_properties(self, zs, zdir):
         self._sizes3d = self.get_sizes()
         self.stale = True
 
+    def set_facecolor(self, c):
+        super().set_facecolor(c)
+        self._facecolor3d = self.get_facecolor()
+
+    set_facecolors = set_facecolor
+
+    def set_edgecolor(self, c):
+        super().set_edgecolor(c)
+        self._edgecolor3d = self.get_edgecolor()
+
+    set_edgecolors = set_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)
