Added pick methods specific to host and parasite axes.

productivememberofsociety666 · smheidrich · commit 8ccffbaac4a7 · 2018-05-17T11:31:52.000+02:00
diff --git a/lib/mpl_toolkits/axes_grid1/parasite_axes.py b/lib/mpl_toolkits/axes_grid1/parasite_axes.py
@@ -39,6 +39,36 @@ def cla(self):
             self.xaxis.set_zorder(2.5)
             self.yaxis.set_zorder(2.5)
 
+    # Overwritten so events that actually happened in the host axes still count
+    def pick(self, mouseevent):
+        """
+        call signature::
+
+          pick(mouseevent)
+
+        each child artist will fire a pick event if *mouseevent* is over
+        the artist and the artist has picker set
+        """
+        # Pick self
+        if self.pickable():
+            picker = self.get_picker()
+            if six.callable(picker):
+                inside, prop = picker(self, mouseevent)
+            else:
+                inside, prop = self.contains(mouseevent)
+            if inside:
+                self.figure.canvas.pick_event(mouseevent, self, **prop)
+
+        # Pick children
+        for a in self.get_children():
+            # make sure the event happened in the same axes (never happens but
+            # whatever) or in host axes
+            ax = getattr(a, 'axes', None)
+            if mouseevent.inaxes is None or ax is None or \
+                    mouseevent.inaxes == ax or \
+                    (hasattr(mouseevent.inaxes, "parasites") and \
+                    self in mouseevent.inaxes.parasites):
+                a.pick(mouseevent)
 
 @functools.lru_cache(None)
 def parasite_axes_class_factory(axes_class=None):
@@ -266,6 +296,45 @@ def cla(self):
             ax.cla()
         self._get_base_axes_attr("cla")(self)
 
+    # Need to overwrite this so children of parasite axes get picked as well
+    def pick(self, mouseevent):
+        """
+        call signature::
+
+          pick(mouseevent)
+
+        each child artist will fire a pick event if *mouseevent* is over
+        the artist and the artist has picker set
+        """
+        # Pick self
+        if self.pickable():
+            picker = self.get_picker()
+            if six.callable(picker):
+                inside, prop = picker(self, mouseevent)
+            else:
+                inside, prop = self.contains(mouseevent)
+            if inside:
+                self.figure.canvas.pick_event(mouseevent, self, **prop)
+
+        # Pick children
+        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 or \
+                    mouseevent.inaxes == ax:
+                # we need to check if mouseevent.inaxes is None
+                # because some objects associated with an axes (e.g., a
+                # tick label) can be outside the bounding box of the
+                # axes and inaxes will be None
+                # also check that ax is None so that it traverse objects
+                # which do no have an axes property but children might
+                a.pick(mouseevent)
+
+        # Pick parasite axes
+        for a in self.parasites:
+            a.pick(mouseevent)
+
+
     def twinx(self, axes_class=None):
         """
         create a twin of Axes for generating a plot with a sharex
