Cleanup demo_text_path.

anntzer · anntzer · commit f25b5d29e148 · 2019-07-12T19:58:03.000+02:00
diff --git a/examples/text_labels_and_annotations/demo_text_path.py b/examples/text_labels_and_annotations/demo_text_path.py
@@ -9,21 +9,17 @@
 """
 
 import matplotlib.pyplot as plt
-from matplotlib.image import BboxImage
-import numpy as np
-from matplotlib.transforms import IdentityTransform
-
-import matplotlib.patches as mpatches
-
-from matplotlib.offsetbox import AnnotationBbox,\
-    AnchoredOffsetbox, AuxTransformBox
-
 from matplotlib.cbook import get_sample_data
-
+from matplotlib.image import BboxImage
+from matplotlib.offsetbox import (
+    AnnotationBbox, AnchoredOffsetbox, AuxTransformBox)
+from matplotlib.patches import PathPatch, Shadow
 from matplotlib.text import TextPath
+from matplotlib.transforms import IdentityTransform
+import numpy as np
 
 
-class PathClippedImagePatch(mpatches.PathPatch):
+class PathClippedImagePatch(PathPatch):
     """
     The given image is used to draw the face of the patch. Internally,
     it uses BboxImage whose clippath set to the path of the patch.
@@ -32,31 +28,20 @@ class PathClippedImagePatch(mpatches.PathPatch):
     """
 
     def __init__(self, path, bbox_image, **kwargs):
-        mpatches.PathPatch.__init__(self, path, **kwargs)
-        self._init_bbox_image(bbox_image)
+        super().__init__(path, **kwargs)
+        self.bbox_image = BboxImage(
+            self.get_window_extent, norm=None, origin=None)
+        self.bbox_image.set_data(bbox_image)
 
     def set_facecolor(self, color):
         """Simply ignore facecolor."""
-        mpatches.PathPatch.set_facecolor(self, "none")
-
-    def _init_bbox_image(self, im):
-
-        bbox_image = BboxImage(self.get_window_extent,
-                               norm=None,
-                               origin=None,
-                               )
-        bbox_image.set_transform(IdentityTransform())
-
-        bbox_image.set_data(im)
-        self.bbox_image = bbox_image
+        super().set_facecolor("none")
 
     def draw(self, renderer=None):
-
         # the clip path must be updated every draw. any solution? -JJ
         self.bbox_image.set_clip_path(self._path, self.get_transform())
         self.bbox_image.draw(renderer)
-
-        mpatches.PathPatch.draw(self, renderer)
+        super().draw(renderer)
 
 
 if __name__ == "__main__":
@@ -93,8 +78,7 @@ def draw(self, renderer=None):
     else:
         r = r"textpath supports mathtext & TeX"
 
-    text_path = TextPath((0, 0), r,
-                         size=20, usetex=usetex)
+    text_path = TextPath((0, 0), r, size=20, usetex=usetex)
 
     p1 = PathPatch(text_path, ec="w", lw=3, fc="w", alpha=0.9,
                    transform=IdentityTransform())
@@ -121,7 +105,7 @@ def draw(self, renderer=None):
 
     ax = plt.subplot(212)
 
-    arr = np.arange(256).reshape(1, 256)/256.
+    arr = np.arange(256).reshape(1, 256) / 256
 
     if usetex:
         s = (r"$\displaystyle\left[\sum_{n=1}^\infty"
@@ -132,10 +116,8 @@ def draw(self, renderer=None):
     text_patch = PathClippedImagePatch(text_path, arr, ec="none",
                                        transform=IdentityTransform())
 
-    shadow1 = mpatches.Shadow(text_patch, 1, -1,
-                              props=dict(fc="none", ec="0.6", lw=3))
-    shadow2 = mpatches.Shadow(text_patch, 1, -1,
-                              props=dict(fc="0.3", ec="none"))
+    shadow1 = Shadow(text_patch, 1, -1, props=dict(fc="none", ec="0.6", lw=3))
+    shadow2 = Shadow(text_patch, 1, -1, props=dict(fc="0.3", ec="none"))
 
     # make offset box
     offsetbox = AuxTransformBox(IdentityTransform())
