ENH: Scroll to zoom

Implements a minimal version of matplotlib#20317, in particular https://github
.com/matplotlib/pull/20317#issuecomment-2233156558:

When any of the axes manipulation tools is active (pan or zoom tool), a
mouse scroll results in a zoom towards the cursor, keeping aspect ratio.

I've decided to require an active manipulation tool, so that without any
active tool the plot cannot be changed (accidentally) - as before.
For convenience, scroll-to-zoom is allowed with both the zoom and pan
tools. Limiting further feels unnecessarily restrictive.

Zooming is also limited to not having a modifier key pressed. This is
because we might later want to add scroll+modifiers for other operations
. It's better for now not to react to these at all to not introduce
behaviors we later want to change.

Author: timhoffm

lib/matplotlib/backend_bases.py

@@ -2574,6 +2574,46 @@ def button_press_handler(event, canvas=None, toolbar=None):
             toolbar.forward()
 
 
+def scroll_handler(event, canvas=None, toolbar=None):
+    ax = event.inaxes
+    if ax is None:
+        return
+
+    if toolbar is None:
+        if canvas is None:
+            canvas = event.canvas
+        toolbar = canvas.toolbar
+
+    if toolbar is None or toolbar.mode == _Mode.NONE:
+        return
+
+    if event.key is None:  # zoom
+        toolbar.push_current()
+
+        # Get the current x and y limits
+        xmin, xmax = ax.get_xlim()
+        ymin, ymax = ax.get_ylim()
+
+        # Get the data coordinates of the mouse
+        x = event.xdata
+        y = event.ydata
+
+        # Calculate the scale factor based on the step
+        scale_factor = 1.0 - 0.05 * event.step
+
+        # Calculate the new limits while keeping the mouse position fixed
+        new_xmin = x - (x - xmin) * scale_factor
+        new_xmax = x + (xmax - x) * scale_factor
+        new_ymin = y - (y - ymin) * scale_factor
+        new_ymax = y + (ymax - y) * scale_factor
+
+        # Set the new limits
+        ax.set_xlim(new_xmin, new_xmax)
+        ax.set_ylim(new_ymin, new_ymax)
+
+        ax.figure.canvas.draw_idle()
+
+
 class NonGuiException(Exception):
     """Raised when trying show a figure in a non-GUI backend."""
     pass
@@ -2653,11 +2693,14 @@ def __init__(self, canvas, num):
 
         self.key_press_handler_id = None
         self.button_press_handler_id = None
+        self.scroll_handler_id = None
         if rcParams['toolbar'] != 'toolmanager':
             self.key_press_handler_id = self.canvas.mpl_connect(
                 'key_press_event', key_press_handler)
             self.button_press_handler_id = self.canvas.mpl_connect(
                 'button_press_event', button_press_handler)
+            self.scroll_handler_id = self.canvas.mpl_connect(
+                'scroll_event', scroll_handler)
 
         self.toolmanager = (ToolManager(canvas.figure)
                             if mpl.rcParams['toolbar'] == 'toolmanager'