Avoid creating a Tick in Axis.get_tick_space.

QuLogic · QuLogic · commit 5404ee728737 · 2020-05-07T01:03:11.000-04:00
Doing so is expensive, and triggers the figure to be marked stale. We
don't need a whole tick, just the font size that it will eventually use,
so calculate that directly.
diff --git a/lib/matplotlib/axis.py b/lib/matplotlib/axis.py
@@ -1315,6 +1315,21 @@ def _get_tick(self, major):
         """Return the default tick instance."""
         raise NotImplementedError('derived must override')
 
+    def _get_tick_label_size(self, axis_name, major):
+        """
+        Return the text size of tick labels for this Axis.
+
+        This is a convenience function to avoid having to create a `Tick` in
+        `.get_tick_space`, since it is expensive.
+        """
+        if major:
+            tick_kw = self._major_tick_kw
+        else:
+            tick_kw = self._minor_tick_kw
+        size = tick_kw.get('labelsize',
+                           mpl.rcParams[f'{axis_name}tick.labelsize'])
+        return mtext.FontProperties(size).get_size_in_points()
+
     def _copy_tick_props(self, src, dest):
         """Copy the properties from *src* tick to *dest* tick."""
         if src is None or dest is None:
@@ -2186,10 +2201,9 @@ def set_default_intervals(self):
     def get_tick_space(self):
         ends = self.axes.transAxes.transform([[0, 0], [1, 0]])
         length = ((ends[1][0] - ends[0][0]) / self.axes.figure.dpi) * 72
-        tick = self._get_tick(True)
         # There is a heuristic here that the aspect ratio of tick text
         # is no more than 3:1
-        size = tick.label1.get_size() * 3
+        size = self._get_tick_label_size('x', True) * 3
         if size > 0:
             return int(np.floor(length / size))
         else:
@@ -2472,9 +2486,8 @@ def set_default_intervals(self):
     def get_tick_space(self):
         ends = self.axes.transAxes.transform([[0, 0], [0, 1]])
         length = ((ends[1][1] - ends[0][1]) / self.axes.figure.dpi) * 72
-        tick = self._get_tick(True)
         # Having a spacing of at least 2 just looks good.
-        size = tick.label1.get_size() * 2.0
+        size = self._get_tick_label_size('y', True) * 2
         if size > 0:
             return int(np.floor(length / size))
         else:
diff --git a/lib/matplotlib/tests/test_figure.py b/lib/matplotlib/tests/test_figure.py
@@ -1,6 +1,7 @@
 from datetime import datetime
 from pathlib import Path
 import platform
+from types import SimpleNamespace
 import warnings
 try:
     from contextlib import nullcontext
@@ -533,3 +534,17 @@ def test_removed_axis():
     fig, axs = plt.subplots(2, sharex=True)
     axs[0].remove()
     fig.canvas.draw()
+
+
+@pytest.mark.style('mpl20')
+def test_picking_staleness():
+    fig, ax = plt.subplots()
+    col = ax.scatter([0], [0], [1000], picker=True)
+    fig.canvas.draw()
+    assert not fig.stale
+
+    mouse_event = SimpleNamespace(x=ax.bbox.x0 + ax.bbox.width / 2,
+                                  y=ax.bbox.y0 + ax.bbox.height / 2,
+                                  inaxes=ax, guiEvent=None)
+    fig.pick(mouse_event)
+    assert not fig.stale
