Added ability to drag out of axes with span selector. event was not registered before.

cmoad · cmoad · commit 4a595e046cea · 2005-11-18T19:42:48.000Z
svn path=/trunk/matplotlib/; revision=1874
diff --git a/lib/matplotlib/widgets.py b/lib/matplotlib/widgets.py
@@ -732,6 +732,10 @@ def __init__(self, ax, onselect, direction, minspan=None, useblit=False, rectpro
         self.useblit = useblit
         self.minspan = minspan
 
+        # Needed when dragging out of axes
+        self.buttonDown = False
+        self.prev = (0, 0)
+
         if self.direction == 'horizontal':
             trans = blend_xy_sep_transform(self.ax.transData, self.ax.transAxes)
             w,h = 0,1
@@ -760,6 +764,7 @@ def ignore(self, event):
     def press(self, event):
         'on button press event'
         if self.ignore(event): return
+        self.buttonDown = True
         
         self.rect.set_visible(self.visible)
         if self.direction == 'horizontal':
@@ -771,15 +776,17 @@ def press(self, event):
 
     def release(self, event):
         'on button release event'
-        if self.pressv is None or self.ignore(event): return
+        if self.pressv is None or (self.ignore(event) and not self.buttonDown): return
+        self.buttonDown = False
 
         self.rect.set_visible(False)
         self.canvas.draw()
         vmin = self.pressv
         if self.direction == 'horizontal':
-            vmax = event.xdata
+            vmax = event.xdata or self.prev[0]
         else:
-            vmax = event.ydata
+            vmax = event.ydata or self.prev[1]
+
         if vmin>vmax: vmin, vmax = vmax, vmin
         span = vmax - vmin
         if self.minspan is not None and span<self.minspan: return
@@ -802,7 +809,8 @@ def update(self):
     def onmove(self, event):
         'on motion notify event'
         if self.pressv is None or self.ignore(event): return
-        x,y = event.xdata, event.ydata
+        x, y = event.xdata, event.ydata
+        self.prev = x, y
         if self.direction == 'horizontal':
             v = x
         else:
