BUG : Add validation for masked arrays in _revalidate

tacaswell · tacaswell · commit c9244df82ec7 · 2015-01-18T13:05:00.000-05:00
Ensure that transforms that return masked arrays have the arrays
filled with nan

Improved the code readability.
diff --git a/lib/matplotlib/transforms.py b/lib/matplotlib/transforms.py
@@ -2615,14 +2615,25 @@ def __init__(self, path, transform):
         self._transformed_points = None
 
     def _revalidate(self):
-        # only recompute if the invalidation includes the non_affine part of the transform
-        if ((self._invalid & self.INVALID_NON_AFFINE == self.INVALID_NON_AFFINE)
-            or self._transformed_path is None):
-            self._transformed_path = \
-                self._transform.transform_path_non_affine(self._path)
-            self._transformed_points = \
-                Path._fast_from_codes_and_verts(self._transform.transform_non_affine(self._path.vertices),
-                        None, {'interpolation_steps': self._path._interpolation_steps})
+        # only recompute if the invalidation includes
+        # the non_affine part of the transform
+        if ((self._invalid &
+             self.INVALID_NON_AFFINE == self.INVALID_NON_AFFINE)
+           or self._transformed_path is None):
+
+            tpath = self._transform.transform_path_non_affine(self._path)
+            self._transformed_path = tpath
+
+            x = self._transform.transform_non_affine(self._path.vertices)
+            if ma.isMaskedArray(x):
+                x = x.astype(np.float_).filled(np.nan)
+
+            tpts = Path._fast_from_codes_and_verts(verts=x, codes=None,
+                    internals={'interpolation_steps':
+                               self._path._interpolation_steps})
+
+            self._transformed_points = tpts
+
         self._invalid = 0
 
     def get_transformed_points_and_affine(self):
