Add set/get for ellipse width/height (#16395)

LauLauThom · web-flow · commit 182c2f25617c · 2020-02-07T09:56:59.000Z
* Add set/get for ellipse width/height

* add property for ellipse width/height

* fix identation

* Use Flake8 formatting

* add set/get angle
diff --git a/lib/matplotlib/patches.py b/lib/matplotlib/patches.py
@@ -1396,8 +1396,8 @@ def __init__(self, xy, width, height, angle=0, **kwargs):
         Patch.__init__(self, **kwargs)
 
         self._center = xy
-        self.width, self.height = width, height
-        self.angle = angle
+        self._width, self._height = width, height
+        self._angle = angle
         self._path = Path.unit_circle()
         # Note: This cannot be calculated until this is added to an Axes
         self._patch_transform = transforms.IdentityTransform()
@@ -1413,8 +1413,8 @@ def _recompute_transform(self):
         """
         center = (self.convert_xunits(self._center[0]),
                   self.convert_yunits(self._center[1]))
-        width = self.convert_xunits(self.width)
-        height = self.convert_yunits(self.height)
+        width = self.convert_xunits(self._width)
+        height = self.convert_yunits(self._height)
         self._patch_transform = transforms.Affine2D() \
             .scale(width * 0.5, height * 0.5) \
             .rotate_deg(self.angle) \
@@ -1443,12 +1443,69 @@ def set_center(self, xy):
 
     def get_center(self):
         """
-        Return the center of the ellipse
+        Return the center of the ellipse.
         """
         return self._center
 
     center = property(get_center, set_center)
 
+    def set_width(self, width):
+        """
+        Set the width of the ellipse.
+
+        Parameters
+        ----------
+        width : float
+        """
+        self._width = width
+        self.stale = True
+
+    def get_width(self):
+        """
+        Return the width of the ellipse.
+        """
+        return self._width
+
+    width = property(get_width, set_width)
+
+    def set_height(self, height):
+        """
+        Set the height of the ellipse.
+
+        Parameters
+        ----------
+        height : float
+        """
+        self._height = height
+        self.stale = True
+
+    def get_height(self):
+        """
+        Return the height of the ellipse.
+        """
+        return self._height
+
+    height = property(get_height, set_height)
+
+    def set_angle(self, angle):
+        """
+        Set the angle of the ellipse.
+
+        Parameters
+        ----------
+        angle : float
+        """
+        self._angle = angle
+        self.stale = True
+
+    def get_angle(self):
+        """
+        Return the angle of the ellipse.
+        """
+        return self._angle
+
+    angle = property(get_angle, set_angle)
+
 
 class Circle(Ellipse):
     """
