Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Lasso selector #730

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 13 commits into from
Mar 17, 2012
Merged

Lasso selector #730

Changes from 1 commit
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
8af797d
Change cursors and selectors to subclass Widget.
tonysyu Feb 27, 2012
5b35427
Remove call to `new_axes` in SpanSelector.
tonysyu Feb 27, 2012
e37c135
Add AxesWidget class and let it initialize `ax` and `canvas` attributes.
tonysyu Feb 27, 2012
21bca27
Add `connect_event` and `disconnect_events` to AxesWidget.
tonysyu Feb 27, 2012
c8f152a
Add default ignore method and check it in callbacks.
tonysyu Feb 28, 2012
6baa65a
Fix: save correct callback ids.
tonysyu Feb 28, 2012
0f0286e
Remove duplicate code in `SpanSelector.__init__`.
tonysyu Feb 28, 2012
8345d4c
Add LassoSelector widget with demo.
tonysyu Mar 1, 2012
adb1754
DOC: Fix sphinx class references.
tonysyu Mar 2, 2012
f169460
Fix and improve docstring.
tonysyu Mar 2, 2012
c668798
Merge branch 'base-widget' into lasso-selector
tonysyu Mar 2, 2012
fe3e2b8
Use `Path.contains_point` instead of `nxutils` function.
tonysyu Mar 12, 2012
78798f1
Add docstrings for `Lasso` and `LassoSelector`.
tonysyu Mar 12, 2012
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Add connect_event and disconnect_events to AxesWidget.
  • Loading branch information
@tonysyu
tonysyu committed Feb 28, 2012
commit 21bca27fc3df736561c47fe74c01845f39757ed3
67 changes: 36 additions & 31 deletions lib/matplotlib/widgets.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -55,14 +55,14 @@ def locked(self):
return self._owner is not None



class Widget(object):
"""
Abstract base class for GUI neutral widgets
"""
drawon = True
eventson = True


class AxesWidget(Widget):
"""
Widget that is connected to a single :class:`Axes`.
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Does this render correctly? It is certainly safer to do :class:~matplotlib.axes.Axes. (the tilde shortens the link to Axes)

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I didn't know about this. Thanks for the tip.

Expand All @@ -77,6 +77,15 @@ class AxesWidget(Widget):
def __init__(self, ax):
self.ax = ax
self.canvas = ax.figure.canvas
self.cids = []

def connect_event(self, event, callback):
self.canvas.mpl_connect(event, callback)
self.cids.append(callback)

def disconnect_events(self):
for c in self.cids:
self.canvas.mpl_disconnect(c)


class Button(AxesWidget):
Expand Down Expand Up @@ -133,9 +142,9 @@ def __init__(self, ax, label, image=None,
self.cnt = 0
self.observers = {}

ax.figure.canvas.mpl_connect('button_press_event', self._click)
ax.figure.canvas.mpl_connect('button_release_event', self._release)
ax.figure.canvas.mpl_connect('motion_notify_event', self._motion)
self.connect_event('button_press_event', self._click)
self.connect_event('button_release_event', self._release)
self.connect_event('motion_notify_event', self._motion)
ax.set_navigate(False)
ax.set_axis_bgcolor(color)
ax.set_xticks([])
Expand Down Expand Up @@ -270,10 +279,10 @@ def __init__(self, ax, label, valmin, valmax, valinit=0.5, valfmt='%1.2f',
ax.set_xticks([])
ax.set_navigate(False)

ax.figure.canvas.mpl_connect('button_press_event', self._update)
ax.figure.canvas.mpl_connect('button_release_event', self._update)
self.connect_event('button_press_event', self._update)
self.connect_event('button_release_event', self._update)
if dragging:
ax.figure.canvas.mpl_connect('motion_notify_event', self._update)
self.connect_event('motion_notify_event', self._update)
self.label = ax.text(-0.02, 0.5, label, transform=ax.transAxes,
verticalalignment='center',
horizontalalignment='right')
Expand Down Expand Up @@ -446,7 +455,7 @@ def __init__(self, ax, labels, actives):
ax.add_line(l2)
cnt += 1

ax.figure.canvas.mpl_connect('button_press_event', self._clicked)
self.connect_event('button_press_event', self._clicked)

self.cnt = 0
self.observers = {}
Expand Down Expand Up @@ -557,7 +566,7 @@ def __init__(self, ax, labels, active=0, activecolor='blue'):
ax.add_patch(p)
cnt += 1

ax.figure.canvas.mpl_connect('button_press_event', self._clicked)
self.connect_event('button_press_event', self._clicked)

self.cnt = 0
self.observers = {}
Expand Down Expand Up @@ -770,8 +779,8 @@ def __init__(self, ax, useblit=False, **lineprops):
# TODO: Is the GTKAgg limitation still true?
AxesWidget.__init__(self, ax)

self.canvas.mpl_connect('motion_notify_event', self.onmove)
self.canvas.mpl_connect('draw_event', self.clear)
self.connect_event('motion_notify_event', self.onmove)
self.connect_event('draw_event', self.clear)

self.visible = True
self.horizOn = True
Expand Down Expand Up @@ -951,7 +960,6 @@ def __init__(self, ax, onselect, direction, minspan=None, useblit=False,
self.direction = direction

self.visible = True
self.cids=[]

self.rect = None
self.background = None
Expand All @@ -967,10 +975,10 @@ def __init__(self, ax, onselect, direction, minspan=None, useblit=False,
self.buttonDown = False
self.prev = (0, 0)

self.cids.append(self.canvas.mpl_connect('motion_notify_event', self.onmove))
self.cids.append(self.canvas.mpl_connect('button_press_event', self.press))
self.cids.append(self.canvas.mpl_connect('button_release_event', self.release))
self.cids.append(self.canvas.mpl_connect('draw_event', self.update_background))
self.connect_event('motion_notify_event', self.onmove)
self.connect_event('button_press_event', self.press)
self.connect_event('button_release_event', self.release)
self.connect_event('draw_event', self.update_background)

if self.direction == 'horizontal':
trans = blended_transform_factory(self.ax.transData, self.ax.transAxes)
Expand All @@ -989,15 +997,14 @@ def __init__(self, ax, onselect, direction, minspan=None, useblit=False,
def new_axes(self,ax):
self.ax = ax
if self.canvas is not ax.figure.canvas:
for cid in self.cids:
self.canvas.mpl_disconnect(cid)
self.disconnect_events()

self.canvas = ax.figure.canvas
self.connect_event('motion_notify_event', self.onmove)
self.connect_event('button_press_event', self.press)
self.connect_event('button_release_event', self.release)
self.connect_event('draw_event', self.update_background)

self.cids.append(self.canvas.mpl_connect('motion_notify_event', self.onmove))
self.cids.append(self.canvas.mpl_connect('button_press_event', self.press))
self.cids.append(self.canvas.mpl_connect('button_release_event', self.release))
self.cids.append(self.canvas.mpl_connect('draw_event', self.update_background))
if self.direction == 'horizontal':
trans = blended_transform_factory(self.ax.transData, self.ax.transAxes)
w,h = 0,1
Expand Down Expand Up @@ -1193,10 +1200,10 @@ def __init__(self, ax, onselect, drawtype='box',
AxesWidget.__init__(self, ax)

self.visible = True
self.canvas.mpl_connect('motion_notify_event', self.onmove)
self.canvas.mpl_connect('button_press_event', self.press)
self.canvas.mpl_connect('button_release_event', self.release)
self.canvas.mpl_connect('draw_event', self.update_background)
self.connect_event('motion_notify_event', self.onmove)
self.connect_event('button_press_event', self.press)
self.connect_event('button_release_event', self.release)
self.connect_event('draw_event', self.update_background)

self.active = True # for activation / deactivation
self.to_draw = None
Expand Down Expand Up @@ -1400,9 +1407,8 @@ def __init__(self, ax, xy, callback=None, useblit=True):
self.line = Line2D([x], [y], linestyle='-', color='black', lw=2)
self.ax.add_line(self.line)
self.callback = callback
self.cids = []
self.cids.append(self.canvas.mpl_connect('button_release_event', self.onrelease))
self.cids.append(self.canvas.mpl_connect('motion_notify_event', self.onmove))
self.connect_event('button_release_event', self.onrelease)
self.connect_event('motion_notify_event', self.onmove)

def onrelease(self, event):
if self.verts is not None:
Expand All @@ -1411,8 +1417,7 @@ def onrelease(self, event):
self.callback(self.verts)
self.ax.lines.remove(self.line)
self.verts = None
for cid in self.cids:
self.canvas.mpl_disconnect(cid)
self.disconnect_events()

def onmove(self, event):
if self.verts is None: return
Expand Down