Add get_tightbbox to Legend

jklymak · jklymak · commit d6c5c7d41898 · 2018-03-05T09:38:15.000-08:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -4154,12 +4154,9 @@ def get_tightbbox(self, renderer, call_axes_locator=True,
             bbox_artists = self.get_default_bbox_extra_artists()
 
         for a in bbox_artists:
-            if isinstance(a, Legend) and a.get_visible():
-                bb.append(a._legend_box.get_window_extent(renderer))
-            else:
-                bbox = a.get_tightbbox(renderer)
-                if bbox is not None and (bbox.width != 0 or bbox.height != 0):
-                    bb.append(bbox)
+            bbox = a.get_tightbbox(renderer)
+            if bbox is not None and (bbox.width != 0 or bbox.height != 0):
+                bb.append(bbox)
 
         _bbox = mtransforms.Bbox.union(
             [b for b in bb if b.width != 0 or b.height != 0])
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -2241,13 +2241,9 @@ def get_tightbbox(self, renderer, bbox_extra_artists=None):
         else:
             artists = bbox_extra_artists
         for a in artists:
-            if isinstance(a, mlegend.Legend) and a.get_visible():
-                bb.append(a._legend_box.get_window_extent(renderer))
-            else:
-                bbox = a.get_tightbbox(renderer)
-                if bbox is not None and (bbox.width != 0 or
-                                         bbox.height != 0):
-                    bb.append(bbox)
+            bbox = a.get_tightbbox(renderer)
+            if bbox is not None and (bbox.width != 0 or bbox.height != 0):
+                bb.append(bbox)
 
         for ax in self.axes:
             if ax.get_visible():
diff --git a/lib/matplotlib/legend.py b/lib/matplotlib/legend.py
@@ -1125,6 +1125,10 @@ def get_window_extent(self, *args, **kwargs):
         'Return extent of the legend.'
         return self.legendPatch.get_window_extent(*args, **kwargs)
 
+    def get_tightbbox(self, renderer):
+        'Return extent of the legend.'
+        return self._legend_box.get_window_extent(renderer)
+
     def get_frame_on(self):
         """Get whether the legend box patch is drawn."""
         return self._drawFrame
