pcolor range bug fix

bblay · bblay · commit 5b194a45a012 · 2012-10-15T15:29:38.000+01:00
diff --git a/lib/matplotlib/axes.py b/lib/matplotlib/axes.py
@@ -7364,6 +7364,20 @@ def pcolor(self, *args, **kwargs):
 
         x = X.compressed()
         y = Y.compressed()
+        
+        # Transform from native to data coordinates?
+        t = collection._transform
+        if (not isinstance(t, mtransforms.Transform)
+            and hasattr(t, '_as_mpl_transform')):
+            t = t._as_mpl_transform(self.axes)
+        
+        if t and any(t.contains_branch_seperately(self.transData)):
+            trans_to_data = t - self.transData
+            pts = np.vstack([x, y]).T.astype(np.float)
+            transformed_pts = trans_to_data.transform(pts)
+            x = transformed_pts[..., 0]
+            y = transformed_pts[..., 1]
+
         minx = np.amin(x)
         maxx = np.amax(x)
         miny = np.amin(y)
