Add getters to EllipseColletion

ericpre · ericpre · commit 45a6a6f2f026 · 2024-03-28T14:40:21.000Z
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -1805,20 +1805,32 @@ def _set_angles(self, angles):
         self._angles = np.deg2rad(angles).ravel()
 
     def set_widths(self, widths):
-        """Set the lengths of the first axes (e.g., major axis lengths)."""
+        """Set the lengths of the first axes (e.g., major axis)."""
         self._set_widths(widths)
         self.stale = True
 
     def set_heights(self, heights):
-        """Set the lengths of second axes.."""
+        """Set the lengths of second axes (e.g., minor axes)."""
         self._set_heights(heights)
         self.stale = True
 
+    def get_widths(self):
+        """Get the lengths of the first axes (e.g., major axis)."""
+        return self._widths * 2
+
+    def get_heights(self):
+        """Set the lengths of second axes (e.g., minor axes)."""
+        return self._heights * 2
+
     def set_angles(self, angles):
         """Set the angles of the first axes, degrees CCW from the x-axis."""
         self._set_angles(angles)
         self.stale = True
 
+    def get_angles(self):
+        """Get the angles of the first axes, degrees CCW from the x-axis."""
+        return np.rad2deg(self._angles)
+
     @artist.allow_rasterization
     def draw(self, renderer):
         self._set_transforms()
diff --git a/lib/matplotlib/collections.pyi b/lib/matplotlib/collections.pyi
@@ -181,8 +181,11 @@ class EllipseCollection(Collection):
         **kwargs
     ) -> None: ...
     def set_widths(self, widths: ArrayLike) -> None: ...
+    def get_widths(self) -> ArrayLike: ...
     def set_heights(self, heights: ArrayLike) -> None: ...
+    def get_heights(self) -> ArrayLike: ...
     def set_angles(self, angles: ArrayLike) -> None: ...
+    def get_angles(self) -> ArrayLike: ...
 
 class PatchCollection(Collection):
     def __init__(
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -408,7 +408,7 @@ def test_EllipseCollection():
     ax.autoscale_view()
 
 
-def test_EllipseCollection_setter():
+def test_EllipseCollection_setter_getter():
     # Test widths, heights and angle setter
     rng = np.random.default_rng(0)
 
@@ -432,6 +432,10 @@ def test_EllipseCollection_setter():
     assert_array_almost_equal(ec._heights, np.array(heights).ravel() * 0.5)
     assert_array_almost_equal(ec._angles, np.deg2rad(angles).ravel())
 
+    assert_array_almost_equal(ec.get_widths(), widths)
+    assert_array_almost_equal(ec.get_heights(), heights)
+    assert_array_almost_equal(ec.get_angles(), angles)
+
     ax.add_collection(ec)
     ax.set_xlim(-2, 12)
     ax.set_ylim(-2, 12)
@@ -442,9 +446,9 @@ def test_EllipseCollection_setter():
 
     ec.set(widths=new_widths, heights=new_heights, angles=new_angles)
 
-    assert_array_almost_equal(ec._widths, np.array(new_widths).ravel() * 0.5)
-    assert_array_almost_equal(ec._heights, np.array(new_heights).ravel() * 0.5)
-    assert_array_almost_equal(ec._angles, np.deg2rad(new_angles).ravel())
+    assert_array_almost_equal(ec.get_widths(), new_widths.ravel())
+    assert_array_almost_equal(ec.get_heights(), new_heights.ravel())
+    assert_array_almost_equal(ec.get_angles(), new_angles.ravel())
 
 
 @image_comparison(['polycollection_close.png'], remove_text=True, style='mpl20')
