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

WeatherGod · QuLogic · commit e95a99095a9a · 2020-09-16T04:15:22.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,14 @@ 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()
+
+    def set_edgecolor(self, c):
+        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 +499,14 @@ def set_3d_properties(self, zs, zdir):
         self._linewidth3d = self.get_linewidth()
         self.stale = True
 
+    def set_facecolor(self, c):
+        super().set_facecolor(c)
+        self._facecolor3d = self.get_facecolor()
+
+    def set_edgecolor(self, c):
+        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)
