Fix custom scales in pcolormesh (thanks, Matthew Turk)

mdboom · mdboom · commit 7ff5e8105088 · 2008-07-07T17:48:36.000Z
svn path=/trunk/matplotlib/; revision=5716
diff --git a/examples/pylab_examples/quadmesh_demo.py b/examples/pylab_examples/quadmesh_demo.py
@@ -29,6 +29,8 @@
 ax = fig.add_subplot(121)
 ax.set_axis_bgcolor("#bdb76b")
 ax.pcolormesh(Qx,Qz,Z)
+ax.set_xscale('log')
+ax.set_yscale('log')
 ax.set_title('Without masked values')
 
 ax = fig.add_subplot(122)
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -512,14 +512,24 @@ def draw(self, renderer):
         if clippath_trans is not None:
             clippath_trans = clippath_trans.frozen()
 
-        assert transform.is_affine
+        if not transform.is_affine:
+            coordinates = self._coordinates.reshape(
+                (self._coordinates.shape[0] *
+                 self._coordinates.shape[1],
+                 2))
+            coordinates = transform.transform(coordinates)
+            coordinates = coordinates.reshape(self._coordinates.shape)
+            transform = transforms.IdentityTransform()
+        else:
+            coordinates = self._coordinates
+
         if not transOffset.is_affine:
             offsets = transOffset.transform_non_affine(offsets)
             transOffset = transOffset.get_affine()
 
         renderer.draw_quad_mesh(
             transform.frozen(), self.clipbox, clippath, clippath_trans,
-            self._meshWidth, self._meshHeight, self._coordinates,
+            self._meshWidth, self._meshHeight, coordinates,
             offsets, transOffset, self._facecolors, self._antialiased,
             self._showedges)
         renderer.close_group(self.__class__.__name__)
