Merge pull request #16660 from anntzer/fancybox_demo

timhoffm · web-flow · commit 943444a1c2de · 2020-03-06T01:43:18.000+01:00
Cleanup fancybox_demo.
diff --git a/examples/shapes_and_collections/fancybox_demo.py b/examples/shapes_and_collections/fancybox_demo.py
@@ -27,169 +27,77 @@
              size=fontsize,
              bbox=dict(boxstyle=stylename, fc="w", ec="k"))
 
-plt.show()
 
 ###############################################################################
 # Next we'll show off multiple fancy boxes at once.
 
-# Bbox object around which the fancy box will be drawn.
-bb = mtransforms.Bbox([[0.3, 0.4], [0.7, 0.6]])
-
-
-def draw_bbox(ax, bb):
-    # boxstyle=square with pad=0, i.e. bbox itself.
-    p_bbox = FancyBboxPatch((bb.xmin, bb.ymin),
-                            abs(bb.width), abs(bb.height),
-                            boxstyle="square,pad=0.",
-                            ec="k", fc="none", zorder=10.,
-                            )
-    ax.add_patch(p_bbox)
-
-
-def test1(ax):
-
-    # a fancy box with round corners. pad=0.1
-    p_fancy = FancyBboxPatch((bb.xmin, bb.ymin),
-                             abs(bb.width), abs(bb.height),
-                             boxstyle="round,pad=0.1",
-                             fc=(1., .8, 1.),
-                             ec=(1., 0.5, 1.))
-
-    ax.add_patch(p_fancy)
-
-    ax.text(0.1, 0.8,
-            r' boxstyle="round,pad=0.1"',
-            size=10, transform=ax.transAxes)
-
-    # draws control points for the fancy box.
-    # l = p_fancy.get_path().vertices
-    # ax.plot(l[:,0], l[:,1], ".")
-
-    # draw the original bbox in black
-    draw_bbox(ax, bb)
-
-
-def test2(ax):
-
-    # bbox=round has two optional argument. pad and rounding_size.
-    # They can be set during the initialization.
-    p_fancy = FancyBboxPatch((bb.xmin, bb.ymin),
-                             abs(bb.width), abs(bb.height),
-                             boxstyle="round,pad=0.1",
-                             fc=(1., .8, 1.),
-                             ec=(1., 0.5, 1.))
-
-    ax.add_patch(p_fancy)
-
-    # boxstyle and its argument can be later modified with
-    # set_boxstyle method. Note that the old attributes are simply
-    # forgotten even if the boxstyle name is same.
-
-    p_fancy.set_boxstyle("round,pad=0.1, rounding_size=0.2")
-    # or
-    # p_fancy.set_boxstyle("round", pad=0.1, rounding_size=0.2)
-
-    ax.text(0.1, 0.8,
-            ' boxstyle="round,pad=0.1\n rounding_size=0.2"',
-            size=10, transform=ax.transAxes)
-
-    # draws control points for the fancy box.
-    # l = p_fancy.get_path().vertices
-    # ax.plot(l[:,0], l[:,1], ".")
 
-    draw_bbox(ax, bb)
+def add_fancy_patch_around(ax, bb, **kwargs):
+    fancy = FancyBboxPatch((bb.xmin, bb.ymin), bb.width, bb.height,
+                           fc=(1, 0.8, 1, 0.5), ec=(1, 0.5, 1, 0.5),
+                           **kwargs)
+    ax.add_patch(fancy)
+    return fancy
 
 
-def test3(ax):
+def draw_control_points_for_patches(ax):
+    for patch in ax.patches:
+        patch.axes.plot(*patch.get_path().vertices.T, ".",
+                        c=patch.get_edgecolor())
 
-    # mutation_scale determine overall scale of the mutation,
-    # i.e. both pad and rounding_size is scaled according to this
-    # value.
-    p_fancy = FancyBboxPatch((bb.xmin, bb.ymin),
-                             abs(bb.width), abs(bb.height),
-                             boxstyle="round,pad=0.1",
-                             mutation_scale=2.,
-                             fc=(1., .8, 1.),
-                             ec=(1., 0.5, 1.))
 
-    ax.add_patch(p_fancy)
+fig, axs = plt.subplots(2, 2, figsize=(8, 8))
 
-    ax.text(0.1, 0.8,
-            ' boxstyle="round,pad=0.1"\n mutation_scale=2',
-            size=10, transform=ax.transAxes)
-
-    # draws control points for the fancy box.
-    # l = p_fancy.get_path().vertices
-    # ax.plot(l[:,0], l[:,1], ".")
-
-    draw_bbox(ax, bb)
-
-
-def test4(ax):
-
-    # When the aspect ratio of the axes is not 1, the fancy box may
-    # not be what you expected (green)
-
-    p_fancy = FancyBboxPatch((bb.xmin, bb.ymin),
-                             abs(bb.width), abs(bb.height),
-                             boxstyle="round,pad=0.2",
-                             fc="none",
-                             ec=(0., .5, 0.), zorder=4)
-
-    ax.add_patch(p_fancy)
-
-    # You can compensate this by setting the mutation_aspect (pink).
-    p_fancy = FancyBboxPatch((bb.xmin, bb.ymin),
-                             abs(bb.width), abs(bb.height),
-                             boxstyle="round,pad=0.3",
-                             mutation_aspect=.5,
-                             fc=(1., 0.8, 1.),
-                             ec=(1., 0.5, 1.))
-
-    ax.add_patch(p_fancy)
-
-    ax.text(0.1, 0.8,
-            ' boxstyle="round,pad=0.3"\n mutation_aspect=.5',
-            size=10, transform=ax.transAxes)
-
-    draw_bbox(ax, bb)
-
-
-def test_all():
-    plt.clf()
-
-    ax = plt.subplot(2, 2, 1)
-    test1(ax)
-    ax.set_xlim(0., 1.)
-    ax.set_ylim(0., 1.)
-    ax.set_title("test1")
-    ax.set_aspect(1.)
-
-    ax = plt.subplot(2, 2, 2)
-    ax.set_title("test2")
-    test2(ax)
-    ax.set_xlim(0., 1.)
-    ax.set_ylim(0., 1.)
-    ax.set_aspect(1.)
-
-    ax = plt.subplot(2, 2, 3)
-    ax.set_title("test3")
-    test3(ax)
-    ax.set_xlim(0., 1.)
-    ax.set_ylim(0., 1.)
-    ax.set_aspect(1)
-
-    ax = plt.subplot(2, 2, 4)
-    ax.set_title("test4")
-    test4(ax)
-    ax.set_xlim(-0.5, 1.5)
-    ax.set_ylim(0., 1.)
-    ax.set_aspect(2.)
+# Bbox object around which the fancy box will be drawn.
+bb = mtransforms.Bbox([[0.3, 0.4], [0.7, 0.6]])
 
-    plt.show()
+ax = axs[0, 0]
+# a fancy box with round corners. pad=0.1
+fancy = add_fancy_patch_around(ax, bb, boxstyle="round,pad=0.1")
+ax.set(xlim=(0, 1), ylim=(0, 1), aspect=1,
+       title='boxstyle="round,pad=0.1"')
+
+ax = axs[0, 1]
+# bbox=round has two optional arguments: pad and rounding_size.
+# They can be set during the initialization.
+fancy = add_fancy_patch_around(ax, bb, boxstyle="round,pad=0.1")
+# The boxstyle and its argument can be later modified with set_boxstyle().
+# Note that the old attributes are simply forgotten even if the boxstyle name
+# is same.
+fancy.set_boxstyle("round,pad=0.1,rounding_size=0.2")
+# or: fancy.set_boxstyle("round", pad=0.1, rounding_size=0.2)
+ax.set(xlim=(0, 1), ylim=(0, 1), aspect=1,
+       title='boxstyle="round,pad=0.1,rounding_size=0.2"')
+
+ax = axs[1, 0]
+# mutation_scale determines the overall scale of the mutation, i.e. both pad
+# and rounding_size is scaled according to this value.
+fancy = add_fancy_patch_around(
+    ax, bb, boxstyle="round,pad=0.1", mutation_scale=2)
+ax.set(xlim=(0, 1), ylim=(0, 1), aspect=1,
+       title='boxstyle="round,pad=0.1"\n mutation_scale=2')
+
+ax = axs[1, 1]
+# When the aspect ratio of the axes is not 1, the fancy box may not be what you
+# expected (green).
+fancy = add_fancy_patch_around(ax, bb, boxstyle="round,pad=0.2")
+fancy.set(facecolor="none", edgecolor="green")
+# You can compensate this by setting the mutation_aspect (pink).
+fancy = add_fancy_patch_around(
+    ax, bb, boxstyle="round,pad=0.3", mutation_aspect=0.5)
+ax.set(xlim=(-.5, 1.5), ylim=(0, 1), aspect=2,
+       title='boxstyle="round,pad=0.3"\nmutation_aspect=.5')
+
+for ax in axs.flat:
+    draw_control_points_for_patches(ax)
+    # Draw the original bbox (using boxstyle=square with pad=0).
+    fancy = add_fancy_patch_around(ax, bb, boxstyle="square,pad=0")
+    fancy.set(edgecolor="black", facecolor="none", zorder=10)
+
+fig.tight_layout()
 
 
-test_all()
+plt.show()
 
 #############################################################################
 #
