Fix bbox for axisline style

oscargus · leejjoon · oscargus · commit fe7c9ed3c455 · 2022-11-24T18:16:51.000+01:00
Co-authored-by: Jae-Joon Lee &lt;lee.j.joon@gmail.com&gt;
diff --git a/lib/mpl_toolkits/axisartist/axis_artist.py b/lib/mpl_toolkits/axisartist/axis_artist.py
@@ -1044,11 +1044,17 @@ def get_tightbbox(self, renderer=None):
         self._axis_artist_helper.update_lim(self.axes)
         self._update_ticks(renderer)
         self._update_label(renderer)
+
+        self.line.set_path(self._axis_artist_helper.get_line(self.axes))
+        if self.get_axisline_style() is not None:
+            self.line.set_line_mutation_scale(self.major_ticklabels.get_size())
+
         bb = [
             *self.major_ticklabels.get_window_extents(renderer),
             *self.minor_ticklabels.get_window_extents(renderer),
             self.label.get_window_extent(renderer),
             self.offsetText.get_window_extent(renderer),
+            self.line.get_window_extent(renderer),
         ]
         bb = [b for b in bb if b and (b.width != 0 or b.height != 0)]
         if bb:
diff --git a/lib/mpl_toolkits/axisartist/axisline_style.py b/lib/mpl_toolkits/axisartist/axisline_style.py
@@ -67,6 +67,15 @@ def draw(self, renderer):
             self._path_original = extended_path
             FancyArrowPatch.draw(self, renderer)
 
+        def get_window_extent(self, renderer=None):
+
+            path_in_disp = self._line_transform.transform_path(self._line_path)
+            mutation_size = self.get_mutation_scale()  # line_mutation_scale()
+            extended_path = self._extend_path(path_in_disp,
+                                              mutation_size=mutation_size)
+            self._path_original = extended_path
+            return FancyArrowPatch.get_window_extent(self, renderer)
+
     class FilledArrow(SimpleArrow):
         """The artist class that will be returned for FilledArrow style."""
         _ARROW_STYLE = "-|>"
