Cleaner logic in _onmove

dstansby · dstansby · commit 8229572e0de8 · 2021-12-10T19:26:34.000Z
diff --git a/lib/matplotlib/widgets.py b/lib/matplotlib/widgets.py
@@ -2985,6 +2985,7 @@ def _press(self, event):
             self.update()
 
         if self._active_handle is None and not self.ignore_event_outside:
+            # Start drawing a new rectangle
             x = event.xdata
             y = event.ydata
             self.visible = False
@@ -3051,17 +3052,26 @@ def _release(self, event):
         return False
 
     def _onmove(self, event):
-        """Motion notify event handler."""
+        """
+        Motion notify event handler.
 
-        state = self._state
-        rotate = ('rotate' in state and
-                  self._active_handle in self._corner_order)
+        This can do one of four things:
+        - Translate
+        - Rotate
+        - Re-size
+        - Contine the creation of a new shape
+        """
         xy = np.array([event.xdata, event.ydata])
         xy_press = np.array([self._eventpress.xdata, self._eventpress.ydata])
 
         # The calculations are done for rotation at zero: we apply inverse
         # transformation to events except when we rotate and move
+        state = self._state
+        rotate = ('rotate' in state and
+                  self._active_handle in self._corner_order)
         move = self._active_handle == 'C'
+        resize = self._active_handle and not move
+
         if not move and not rotate:
             inv_tr = self._get_rotation_transform().inverted()
             xy = inv_tr.transform(xy)
@@ -3089,8 +3099,7 @@ def _onmove(self, event):
                      np.arctan2(a[1]-b[1], a[0]-b[0]))
             self.rotation = np.rad2deg(self._rotation_on_press + angle)
 
-        # resize an existing shape
-        elif self._active_handle and self._active_handle != 'C':
+        elif resize:
             size_on_press = [x1 - x0, y1 - y0]
             center = [x0 + size_on_press[0] / 2, y0 + size_on_press[1] / 2]
 
@@ -3143,16 +3152,15 @@ def _onmove(self, event):
                         x1 = x0 + sign * abs(y1 - y0) * \
                             self._aspect_ratio_correction
 
-        # move existing shape
-        elif self._active_handle == 'C':
+        elif move:
             x0, x1, y0, y1 = self._extents_on_press
             x0 += dxy[0]
             x1 += dxy[0]
             y0 += dxy[1]
             y1 += dxy[1]
 
-        # new shape
         else:
+            # Create a new shape
             self._rotation = 0
             # Don't create a new rectangle if there is already one when
             # ignore_event_outside=True
