Mplot3d: fix scatter3d markers bug

heeres · heeres · commit 1131d0525df6 · 2009-12-10T00:09:03.000Z
svn path=/branches/v0_99_maint/; revision=8016
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -240,6 +240,7 @@ class Patch3DCollection(PatchCollection):
 
     def __init__(self, *args, **kwargs):
         PatchCollection.__init__(self, *args, **kwargs)
+        self._old_draw = lambda x: PatchCollection.draw(self, x)
 
     def set_3d_properties(self, zs, zdir):
         xs, ys = zip(*self.get_offsets())
@@ -259,10 +260,15 @@ def do_3d_projection(self, renderer):
         return min(vzs)
 
     def draw(self, renderer):
-        PatchCollection.draw(self, renderer)
+        self._old_draw(renderer)
 
 def patch_collection_2d_to_3d(col, zs=0, zdir='z'):
     """Convert a PatchCollection to a Patch3DCollection object."""
+
+    # The tricky part here is that there are several classes that are
+    # derived from PatchCollection. We need to use the right draw method.
+    col._old_draw = col.draw
+
     col.__class__ = Patch3DCollection
     col.set_3d_properties(zs, zdir)
 
