Merge pull request #4045 from pelson/quadmesh_big

tacaswell · tacaswell · commit cad373fe4453 · 2015-01-31T13:13:25.000-05:00
BUG : Fix run-away memory usage with pcolormesh + bbox_inches='tight'
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -1705,6 +1705,9 @@ def set_paths(self):
         self._paths = self.convert_mesh_to_paths(
             self._meshWidth, self._meshHeight, self._coordinates)
 
+    def get_datalim(self, transData):
+        return (self.get_transform() - transData).transform_bbox(self._bbox)
+
     @staticmethod
     def convert_mesh_to_paths(meshWidth, meshHeight, coordinates):
         """
diff --git a/lib/matplotlib/transforms.py b/lib/matplotlib/transforms.py
@@ -1344,6 +1344,15 @@ def transform_non_affine(self, values):
         """
         return values
 
+    def transform_bbox(self, bbox):
+        """
+        Transform the given bounding box.
+
+        Note, for smarter transforms including caching (a common
+        requirement for matplotlib figures), see :class:`TransformedBbox`.
+        """
+        return Bbox(self.transform(bbox.get_points()))
+
     def get_affine(self):
         """
         Get the affine part of this transform.
