Merge pull request matplotlib#27343 from anntzer/sfad

ksunden · web-flow · commit 461f0b069b69 · 2023-12-05T13:20:35.000-06:00
Fix draggable annotations on subfigures.
diff --git a/lib/matplotlib/artist.py b/lib/matplotlib/artist.py
@@ -537,7 +537,8 @@ def pick(self, mouseevent):
         for a in self.get_children():
             # make sure the event happened in the same Axes
             ax = getattr(a, 'axes', None)
-            if (mouseevent.inaxes is None or ax is None
+            if (isinstance(a, mpl.figure.SubFigure)
+                    or mouseevent.inaxes is None or ax is None
                     or mouseevent.inaxes == ax):
                 # we need to check if mouseevent.inaxes is None
                 # because some objects associated with an Axes (e.g., a
diff --git a/lib/matplotlib/offsetbox.py b/lib/matplotlib/offsetbox.py
@@ -1458,7 +1458,7 @@ def __init__(self, ref_artist, use_blit=False):
             ref_artist.set_picker(True)
         self.got_artist = False
         self._use_blit = use_blit and self.canvas.supports_blit
-        callbacks = ref_artist.figure._canvas_callbacks
+        callbacks = self.canvas.callbacks
         self._disconnectors = [
             functools.partial(
                 callbacks.disconnect, callbacks._connect_picklable(name, func))
@@ -1471,7 +1471,6 @@ def __init__(self, ref_artist, use_blit=False):
 
     # A property, not an attribute, to maintain picklability.
     canvas = property(lambda self: self.ref_artist.figure.canvas)
-
     cids = property(lambda self: [
         disconnect.args[0] for disconnect in self._disconnectors[:2]])
 
diff --git a/lib/matplotlib/tests/test_offsetbox.py b/lib/matplotlib/tests/test_offsetbox.py
@@ -450,3 +450,13 @@ def test_remove_draggable():
     an.draggable(True)
     an.remove()
     MouseEvent("button_release_event", fig.canvas, 1, 1)._process()
+
+
+def test_draggable_in_subfigure():
+    fig = plt.figure()
+    # Put annotation at lower left corner to make it easily pickable below.
+    ann = fig.subfigures().add_axes([0, 0, 1, 1]).annotate("foo", (0, 0))
+    ann.draggable(True)
+    fig.canvas.draw()  # Texts are non-pickable until the first draw.
+    MouseEvent("button_press_event", fig.canvas, 1, 1)._process()
+    assert ann._draggable.got_artist
