Change cursor when hovering over draggable artists

daniilS · daniilS · commit 8ce6c89f9294 · 2023-07-07T15:11:04.000+01:00
diff --git a/lib/matplotlib/offsetbox.py b/lib/matplotlib/offsetbox.py
@@ -32,6 +32,7 @@
 import matplotlib.path as mpath
 import matplotlib.text as mtext
 import matplotlib.transforms as mtransforms
+from matplotlib.backend_tools import Cursors
 from matplotlib.font_manager import FontProperties
 from matplotlib.image import BboxImage
 from matplotlib.patches import (
@@ -1485,6 +1486,7 @@ def __init__(self, ref_artist, use_blit=False):
         if not ref_artist.pickable():
             ref_artist.set_picker(True)
         self.got_artist = False
+        self._hover = False
         self._use_blit = use_blit and self.canvas.supports_blit
         callbacks = ref_artist.figure._canvas_callbacks
         self._disconnectors = [
@@ -1504,7 +1506,31 @@ def __init__(self, ref_artist, use_blit=False):
         disconnect.args[0] for disconnect in self._disconnectors[:2]])
 
     def on_motion(self, evt):
-        if self._check_still_parented() and self.got_artist:
+        # Only check if the widget lock is available, setting it would prevent
+        # picking.
+        if not (
+            self._check_still_parented() and self.canvas.widgetlock.available(self)
+        ):
+            return
+
+        picker = self.ref_artist.get_picker()
+        if callable(picker):
+            inside, _ = picker(self, evt)
+        else:
+            inside, _ = self.ref_artist.contains(evt)
+
+        # If the mouse is moving quickly while dragging, it may leave the artist,
+        # but should still use the move cursor.
+        if inside or self.got_artist:
+            self._hover = True
+            self.canvas.set_cursor(Cursors.MOVE)
+        elif self._hover:
+            # Only change the cursor back if this is the widget that set it, to
+            # avoid multiple draggable widgets fighting over the cursor.
+            self._hover = False
+            self.canvas.set_cursor(Cursors.POINTER)
+
+        if self.got_artist:
             dx = evt.x - self.mouse_x
             dy = evt.y - self.mouse_y
             self.update_offset(dx, dy)
@@ -1550,6 +1576,9 @@ def disconnect(self):
         for disconnector in self._disconnectors:
             disconnector()
 
+        if self._hover:
+            self.canvas.set_cursor(Cursors.POINTER)
+
     def save_offset(self):
         pass
 
