Vectorize bounding box limits updates

scottshambaugh · scottshambaugh · commit dfe72ed19c2a · 2025-01-05T15:22:15.000-07:00
diff --git a/lib/matplotlib/transforms.py b/lib/matplotlib/transforms.py
@@ -46,8 +46,7 @@
 from numpy.linalg import inv
 
 from matplotlib import _api
-from matplotlib._path import (
-    affine_transform, count_bboxes_overlapping_bbox, update_path_extents)
+from matplotlib._path import affine_transform, count_bboxes_overlapping_bbox
 from .path import Path
 
 DEBUG = False
@@ -868,10 +867,35 @@ def update_from_path(self, path, ignore=None, updatex=True, updatey=True):
         if path.vertices.size == 0:
             return
 
-        points, minpos, changed = update_path_extents(
-            path, None, self._points, self._minpos, ignore)
+        if ignore:
+            points = np.array([[np.inf, np.inf], [-np.inf, -np.inf]])
+            minpos = np.array([np.inf, np.inf])
+        else:
+            points = self._points.copy()
+            minpos = self._minpos.copy()
+
+        if updatex and updatey:
+            where = (np.isfinite(path.vertices[..., 0])
+                     & np.isfinite(path.vertices[..., 1]))
+        elif updatex:
+            where = np.isfinite(path.vertices[..., 0])
+        elif updatey:
+            where = np.isfinite(path.vertices[..., 1])
+        else:
+            return
 
-        if changed:
+        if updatex:
+            x = path.vertices[..., 0][where]
+            points[0, 0] = min(points[0, 0], np.min(x, initial=np.inf))
+            points[1, 0] = max(points[1, 0], np.max(x, initial=-np.inf))
+            minpos[0] = min(minpos[0], np.min(x[x > 0], initial=np.inf))
+        if updatey:
+            y = path.vertices[..., 1][where]
+            points[0, 1] = min(points[0, 1], np.min(y, initial=np.inf))
+            points[1, 1] = max(points[1, 1], np.max(y, initial=-np.inf))
+            minpos[1] = min(minpos[1], np.min(y[y > 0], initial=np.inf))
+
+        if np.any(points != self._points) or np.any(minpos != self._minpos):
             self.invalidate()
             if updatex:
                 self._points[:, 0] = points[:, 0]
