windowextent-for-annotationbbox

ImportanceOfBeingErnest · QuLogic · commit 01cb983fac87 · 2020-05-27T00:07:54.000-04:00
diff --git a/lib/matplotlib/offsetbox.py b/lib/matplotlib/offsetbox.py
@@ -1544,6 +1544,24 @@ def get_fontsize(self, s=None):
         """Return the fontsize in points."""
         return self.prop.get_size_in_points()
 
+    def get_window_extent(self, renderer):
+        """
+        get the bounding box in display space.
+        """
+        bboxes = [child.get_window_extent(renderer)
+                  for child in self.get_children()]
+
+        return Bbox.union(bboxes)
+
+    def get_tightbbox(self, renderer):
+        """
+        get tight bounding box in display space.
+        """
+        bboxes = [child.get_tightbbox(renderer)
+                  for child in self.get_children()]
+
+        return Bbox.union(bboxes)
+
     def update_positions(self, renderer):
         """
         Update the pixel positions of the annotated point and the text.
diff --git a/lib/matplotlib/tests/test_offsetbox.py b/lib/matplotlib/tests/test_offsetbox.py
@@ -1,4 +1,5 @@
 from collections import namedtuple
+import io
 
 import numpy as np
 from numpy.testing import assert_allclose
@@ -238,3 +239,70 @@ def test_picking(child_type, boxcoords):
     calls.clear()
     fig.canvas.button_press_event(x, y, MouseButton.LEFT)
     assert len(calls) == 0
+
+
+def test_annotationbbox_extents():
+    plt.rcParams.update(plt.rcParamsDefault)
+    fig, ax = plt.subplots(figsize=(4, 3), dpi=100)
+
+    ax.axis([0, 1, 0, 1])
+
+    an1 = ax.annotate("Annotation", xy=(.9, .9), xytext=(1.1, 1.1),
+                      arrowprops=dict(arrowstyle="->"), clip_on=False,
+                      va="baseline", ha="left")
+
+    da = DrawingArea(20, 20, 0, 0, clip=True)
+    p = mpatches.Circle((-10, 30), 32)
+    da.add_artist(p)
+
+    ab3 = AnnotationBbox(da, [.5, .5], xybox=(-0.2, 0.5), xycoords='data',
+                         boxcoords="axes fraction", box_alignment=(0., .5),
+                         arrowprops=dict(arrowstyle="->"))
+    ax.add_artist(ab3)
+
+    im = OffsetImage(np.random.rand(10, 10), zoom=3)
+    im.image.axes = ax
+    ab6 = AnnotationBbox(im, (0.5, -.3), xybox=(0, 75),
+                         xycoords='axes fraction',
+                         boxcoords="offset points", pad=0.3,
+                         arrowprops=dict(arrowstyle="->"))
+    ax.add_artist(ab6)
+
+    fig.canvas.draw()
+    renderer = fig.canvas.get_renderer()
+
+    # Test Annotation
+    bb1w = an1.get_window_extent(renderer)
+    bb1e = an1.get_tightbbox(renderer)
+
+    target1 = [332.9, 242.8, 467.0, 298.9]
+    assert_allclose(bb1w.extents, target1, atol=2)
+    assert_allclose(bb1e.extents, target1, atol=2)
+
+    # Test AnnotationBbox
+    bb3w = ab3.get_window_extent(renderer)
+    bb3e = ab3.get_tightbbox(renderer)
+
+    target3 = [-17.6, 129.0, 200.7, 167.9]
+    assert_allclose(bb3w.extents, target3, atol=2)
+    assert_allclose(bb3e.extents, target3, atol=2)
+
+    bb6w = ab6.get_window_extent(renderer)
+    bb6e = ab6.get_tightbbox(renderer)
+
+    target6 = [180.0, -32.0, 230.0, 92.9]
+    assert_allclose(bb6w.extents, target6, atol=2)
+    assert_allclose(bb6e.extents, target6, atol=2)
+
+    # Test bbox_inches='tight'
+    buf = io.BytesIO()
+    fig.savefig(buf, bbox_inches='tight')
+    buf.seek(0)
+    shape = plt.imread(buf).shape
+    targetshape = (350, 504, 4)
+    assert_allclose(shape, targetshape, atol=2)
+
+    # Simple smoke test for tight_layout, to make sure it does not error out.
+    fig.canvas.draw()
+    fig.tight_layout()
+    fig.canvas.draw()
