Fix containment test with nonlinear transforms.

anntzer · anntzer · commit 050b20cff887 · 2017-01-16T11:48:59.000-08:00
diff --git a/lib/matplotlib/path.py b/lib/matplotlib/path.py
@@ -493,9 +493,15 @@ def contains_point(self, point, transform=None, radius=0.0):
         """
         if transform is not None:
             transform = transform.frozen()
-        result = _path.point_in_path(point[0], point[1], radius, self,
-                                     transform)
-        return result
+        from .transforms import Affine2D
+        # `point_in_path` does not handle nonlinear transforms, so we
+        # transform the path ourselves.  If `transform` is affine, letting
+        # `point_in_path` handle the transform avoids allocating an extra
+        # buffer.
+        if transform and not isinstance(transform, Affine2D):
+            self = transform.transform_path(self)
+            transform = None
+        return _path.point_in_path(point[0], point[1], radius, self, transform)
 
     def contains_points(self, points, transform=None, radius=0.0):
         """
