Allow Rectangle and Ellipse selectors to be rotated

dstansby · dstansby · commit ca2abb31840d · 2021-05-08T15:10:17.000+01:00
diff --git a/doc/users/next_whats_new/selector_rotate.rst b/doc/users/next_whats_new/selector_rotate.rst
@@ -0,0 +1,7 @@
+Rotating selectors
+------------------
+
+The `~matplotlib.widgets.RectangleSelector` and
+`~matplotlib.widgets.EllipseSelector` can now be rotated by pressing the 'r' key,
+and dragging one of their control points. The rotation is done about the first
+vertex placed when drawing the selector.
diff --git a/lib/matplotlib/tests/test_widgets.py b/lib/matplotlib/tests/test_widgets.py
@@ -189,6 +189,31 @@ def onselect(epress, erelease):
         tool._corner_handles.artist.get_markeredgecolor(), 'b')
 
 
+def test_rectangle_rotate():
+    ax = get_ax()
+
+    def onselect(epress, erelease):
+        pass
+
+    tool = widgets.RectangleSelector(ax, onselect=onselect, interactive=True)
+    # Draw rectangle
+    do_event(tool, 'press', xdata=100, ydata=100)
+    do_event(tool, 'onmove', xdata=130, ydata=140)
+    do_event(tool, 'release', xdata=130, ydata=140)
+    assert tool.extents == (100, 130, 100, 140)
+
+    # Rotate by 45 degrees anticlockwise
+    do_event(tool, 'press', xdata=130, ydata=140, key='r')
+    do_event(tool, 'onmove', xdata=100, ydata=150, key='r')
+    do_event(tool, 'release', xdata=100, ydata=200, key='r')
+    # Extents shouldn't change (as shape of rectangle hasn't changed)
+    assert tool.extents == (100, 130, 100, 140)
+    # Corners should move
+    # The third corner is at (100, 150), as the diagonal of the rectangle has
+    # length 50
+    assert tool.corners == ((100, 124, 100, 76), (100, 118, 150, 132))
+
+
 def check_span(*args, **kwargs):
     ax = get_ax()
 
diff --git a/lib/matplotlib/widgets.py b/lib/matplotlib/widgets.py
