Nullary offset dynamic computers.

anntzer · anntzer · commit 3fd822cf389a · 2022-02-13T16:56:26.000+01:00
diff --git a/lib/matplotlib/legend.py b/lib/matplotlib/legend.py
@@ -585,8 +585,11 @@ def _get_loc(self):
 
     _loc = property(_get_loc, _set_loc)
 
-    def _findoffset(self, width, height, xdescent, ydescent, renderer):
+    def _findoffset(self):
         """Helper function to locate the legend."""
+        renderer = self.figure._cachedRenderer or self._cached_renderer
+        width, height, xdescent, ydescent = self._legend_box.get_extent(
+            renderer)
 
         if self._loc == 0:  # "best".
             x, y = self._find_best_position(width, height, renderer)
@@ -882,6 +885,8 @@ def get_window_extent(self, renderer=None):
         # docstring inherited
         if renderer is None:
             renderer = self.figure._cachedRenderer
+        # May not be cached on the figure, so cache it ourselves.
+        self._cached_renderer = renderer
         return self._legend_box.get_window_extent(renderer=renderer)
 
     def get_tightbbox(self, renderer):
diff --git a/lib/matplotlib/offsetbox.py b/lib/matplotlib/offsetbox.py
@@ -263,11 +263,9 @@ def set_offset(self, xy):
         xy : (float, float) or callable
             The (x, y) coordinates of the offset in display units. These can
             either be given explicitly as a tuple (x, y), or by providing a
-            function that converts the extent into the offset. This function
-            must have the signature::
-
-                def offset(width, height, xdescent, ydescent, renderer) \
--> (float, float)
+            function dynamically computes an offset (taking the arguments
+            passed to `.OffsetBox.get_offset`).  It is recommended, but not
+            compulsory, to make such functions take no arguments.
         """
         self._offset = xy
         self.stale = True
@@ -340,7 +338,7 @@ def get_extent(self, renderer):
     def get_window_extent(self, renderer):
         # docstring inherited
         w, h, xd, yd = self.get_extent(renderer)
-        px, py = self.get_offset(w, h, xd, yd, renderer)
+        px, py = self.get_offset()
         return mtransforms.Bbox.from_bounds(px - xd, py - yd, w, h)
 
     def draw(self, renderer):
@@ -349,7 +347,7 @@ def draw(self, renderer):
         to the given *renderer*.
         """
         w, h, xdescent, ydescent, offsets = self.get_extent_offsets(renderer)
-        px, py = self.get_offset(w, h, xdescent, ydescent, renderer)
+        px, py = self.get_offset()
         for c, (ox, oy) in zip(self.get_visible_children(), offsets):
             c.set_offset((px + ox, py + oy))
             c.draw(renderer)
@@ -530,7 +528,7 @@ def get_extent_offsets(self, renderer):
     def draw(self, renderer):
         # docstring inherited
         w, h, xdescent, ydescent, offsets = self.get_extent_offsets(renderer)
-        px, py = self.get_offset(w, h, xdescent, ydescent, renderer)
+        px, py = self.get_offset()
         for c, (ox, oy) in zip(self.get_visible_children(), offsets):
             c.set_offset((px + ox, py + oy))
 
@@ -1037,9 +1035,10 @@ def get_window_extent(self, renderer):
         # docstring inherited
         # Update the offset func, which depends on the dpi of the renderer
         # (because of the padding).
+        w, h, xd, yd = self.get_extent(renderer)
         fontsize = renderer.points_to_pixels(self.prop.get_size_in_points())
 
-        def _offset(w, h, xd, yd, renderer):
+        def _offset(*args, **kwargs):  # args are ignored; left for backcompat.
             bbox = Bbox.from_bounds(0, 0, w, h)
             pad = self.borderpad * fontsize
             bbox_to_anchor = self.get_bbox_to_anchor()
@@ -1065,9 +1064,7 @@ def draw(self, renderer):
         self.update_frame(bbox, fontsize)
         self.patch.draw(renderer)
 
-        width, height, xdescent, ydescent = self.get_extent(renderer)
-
-        px, py = self.get_offset(width, height, xdescent, ydescent, renderer)
+        px, py = self.get_offset()
 
         self.get_child().set_offset((px, py))
         self.get_child().draw(renderer)
@@ -1566,8 +1563,7 @@ def __init__(self, ref_artist, offsetbox, use_blit=False):
     def save_offset(self):
         offsetbox = self.offsetbox
         renderer = offsetbox.figure._cachedRenderer
-        w, h, xd, yd = offsetbox.get_extent(renderer)
-        offset = offsetbox.get_offset(w, h, xd, yd, renderer)
+        offset = offsetbox.get_offset()
         self.offsetbox_x, self.offsetbox_y = offset
         self.offsetbox.set_offset(offset)
 
diff --git a/lib/mpl_toolkits/axes_grid1/inset_locator.py b/lib/mpl_toolkits/axes_grid1/inset_locator.py
@@ -71,7 +71,7 @@ def draw(self, renderer):
     def __call__(self, ax, renderer):
         self.axes = ax
         bbox = self.get_window_extent(renderer)
-        px, py = self.get_offset(bbox.width, bbox.height, 0, 0, renderer)
+        px, py = self.get_offset()
         bbox_canvas = Bbox.from_bounds(px, py, bbox.width, bbox.height)
         tr = ax.figure.transFigure.inverted()
         return TransformedBbox(bbox_canvas, tr)
