Projects/laydi
Projects
/
laydi
Archived
7
0
Fork 0

Cleaned up toolbar code, lots of work still to be done.

Does not work.
This commit is contained in:
Einar Ryeng 2006-10-10 21:15:13 +00:00
parent 0ed180d6ab
commit a6571d8c6b
1 changed files with 471 additions and 303 deletions

View File

@ -20,7 +20,7 @@ import networkx
import scipy import scipy
# global active mode. Used by toolbars to communicate correct mode # global active mode. Used by toolbars to communicate correct mode
active_mode = 'DEFAULT' active_mode = 'default'
class ObjectTable: class ObjectTable:
"""A 2D table of elements.""" """A 2D table of elements."""
@ -245,28 +245,51 @@ class MainView (gtk.Notebook):
self.emit('view-changed', vf) self.emit('view-changed', vf)
class Plot (gtk.Frame): class View (gtk.Frame):
"""The base class of everything that is shown in the center view of fluents.
Most views should rather subclass Plot, which automatically handles freezing and
toolbars, and sets up matplotlib Figure and Canvas objects.
"""
def __init__(self, title): def __init__(self, title):
gtk.Frame.__init__(self) gtk.Frame.__init__(self)
self.title = title self.title = title
self.set_shadow_type(gtk.SHADOW_NONE)
def get_toolbar(self):
return None
class EmptyView (View):
"""EmptyView is shown in ViewFrames that are unused."""
def __init__(self):
View.__init__(self, 'Empty view')
label = gtk.Label('No view')
ebox = gtk.EventBox()
ebox.add(label)
self.add(ebox)
label.show()
ebox.show()
self.show()
class Plot (View):
def __init__(self, title):
View.__init__(self, title)
self.selection_listener = None self.selection_listener = None
self.fig = Figure() self.fig = Figure()
self.canvas = FigureCanvas(self.fig) self.canvas = FigureCanvas(self.fig)
self.set_shadow_type(gtk.SHADOW_NONE)
self._background = None self._background = None
self._sel_sensitive = True self._frozen = False
self._toolbar = PlotToolbar(self)
self.canvas.add_events(gtk.gdk.ENTER_NOTIFY_MASK) self.canvas.add_events(gtk.gdk.ENTER_NOTIFY_MASK)
self.current_dim = None self.current_dim = None
def set_selection_sensitive(self,event):
if event:
if event.get_active():
logger.log('debug','Selection freezed')
self._sel_sensitive = False
else:
logger.log('debug','Selections active')
self._sel_sensitive = True
def set_frozen(self, frozen):
"""A frozen plot will not be updated when the current selection is changed."""
self._frozen = frozen
def get_title(self): def get_title(self):
return self.title return self.title
@ -280,9 +303,12 @@ class Plot (gtk.Frame):
3.) the selections dim_name is the plot's dimension. 3.) the selections dim_name is the plot's dimension.
""" """
if not self._sel_sensitive or not self.get_property('visible') or self.current_dim!=dim_name: if not self._sel_sensitive \
or not self.get_property('visible') \
or self.current_dim != dim_name:
print "Ignored a selection changed call in plot: %s" %self.get_title() print "Ignored a selection changed call in plot: %s" %self.get_title()
return return
print "Setting current selection in: %s " %self.get_title() print "Setting current selection in: %s " %self.get_title()
self.set_current_selection(selection) self.set_current_selection(selection)
@ -295,21 +321,7 @@ class Plot (gtk.Frame):
""" """
self.selection_listener = listener self.selection_listener = listener
def get_toolbar(self):
return None
class EmptyView (Plot):
def __init__(self):
Plot.__init__(self, 'Empty view')
label = gtk.Label('No view')
ebox = gtk.EventBox()
ebox.add(label)
self.add(ebox)
label.show()
ebox.show()
self.show()
class LineViewPlot(Plot): class LineViewPlot(Plot):
"""Line view of current selection, no interaction """Line view of current selection, no interaction
Only works on 2d-arrays Only works on 2d-arrays
@ -369,8 +381,7 @@ class LineViewPlot(Plot):
self.canvas.show() self.canvas.show()
#FIXME: Lineview plot cannot do selections -> disable in toolbar #FIXME: Lineview plot cannot do selections -> disable in toolbar
self._toolbar = PlotToolbar(self.canvas,self) self._toolbar = PlotToolbar(self)
self._toolbar.chk.connect ('toggled' , self.set_selection_sensitive)
self.canvas.mpl_connect('resize_event', self.clear_background) self.canvas.mpl_connect('resize_event', self.clear_background)
@ -436,12 +447,6 @@ has no color and size options."""
self.add(self.canvas) self.add(self.canvas)
self.canvas.show() self.canvas.show()
# add toolbar
self._toolbar = PlotToolbar(self.canvas, self)
self._toolbar.chk.connect ('toggled', self.set_selection_sensitive)
self._toolbar.set_property('show-arrow', False)
self._toolbar.set_select_callback(self.rectangle_select_callback)
def get_toolbar(self): def get_toolbar(self):
return self._toolbar return self._toolbar
@ -516,11 +521,6 @@ class ScatterPlot(Plot):
self.add(self.canvas) self.add(self.canvas)
self.canvas.show() self.canvas.show()
# create toolbar
self._toolbar = PlotToolbar(self.canvas, self)
self._toolbar.chk.connect ('toggled' , self.set_selection_sensitive)
self._toolbar.set_select_callback(self.rectangle_select_callback)
def get_toolbar(self): def get_toolbar(self):
return self._toolbar return self._toolbar
@ -620,11 +620,6 @@ class NetworkPlot(Plot):
# Initial draw # Initial draw
networkx.draw_networkx(self.graph, ax=self.ax, **kw) networkx.draw_networkx(self.graph, ax=self.ax, **kw)
# Setup toolbar
self._toolbar = PlotToolbar(self.canvas, self)
self._toolbar.chk.connect ('toggled' , self.set_selection_sensitive)
self._toolbar.set_select_callback(self.rectangle_select_callback)
def get_toolbar(self): def get_toolbar(self):
return self._toolbar return self._toolbar
@ -682,304 +677,477 @@ class NetworkPlot(Plot):
self.canvas.draw() self.canvas.draw()
class PlotToolbar(NavigationToolbar2,gtk.Toolbar): class PlotMode:
# list of toolitems to add to the toolbar, format is: """A PlotMode object corresponds to a mouse mode in a plot.
# text, tooltip_text, image_file, callback(str)
toolitems = (
('Home', 'Reset original view', 'home.png', 'home'),
#('Back', 'Back to previous view','back.png', 'back'),
#('Forward', 'Forward to next view','forward.png', 'forward'),
#('Subplots', 'Configure subplots','subplots.png', 'configure_subplots'),
('Save', 'Save the figure','filesave.png', 'save_figure'),
)
radiobuttons = ( When a mode is selected in the toolbar, the PlotMode corresponding
('DEFAULT', 'Default mode', 'cursor.png', 'default'), to the toolbar button is activated by calling setup(ax) for the axis
('PAN', 'Pan axes with left mouse, zoom with right', 'move.png', 'pan'), system ax.
('ZOOM', 'Zoom to rectangle','zoom_to_rect.png', 'zoom'), """
('SELECT', 'Select within rectangle', 'select.png', 'select'), def __init__(self, name, tooltip, image_file):
) self.name = name
self.tooltip = tooltip
self.image_file = image_file
def __init__(self, canvas, plot): def get_icon(self):
self.win = None """Returns the icon for the PlotMode"""
self.plot = plot fname = os.path.join(fluents.ICONDIR, self.image_file)
image = gtk.Image()
image.set_from_file(fname)
return image
def activate(self, canvas):
pass
def deactivate(self):
pass
class DefaultPlotMode (PlotMode):
def __init__(self):
PlotMode.__init__(self, 'default', 'Default mode', 'cursor.png')
class PanPlotMode (PlotMode):
def __init__(self):
PlotMode.__init__(self, 'pan',
'Pan axes with left mouse, zoom with right',
'move.png')
self._button_press = None
self._button_release = None
def activate(self, canvas):
self._button_press = self.canvas.mpl_connect(
'button_press_event', self._on_button_press)
self._button_relese = self.canvas.mpl_connect(
'button_release_event', self._on_button_release)
self.mode = 'pan/zoom mode'
def deactivate(self):
if self._button_press:
self.canvas.mpl_disconnect(self._button_press)
if self._button_release:
self.canvas.mpl_disconnect(self._button_release)
class ZoomPlotMode (PlotMode):
def __init__(self):
PlotMode.__init__(self, 'zoom',
'Zoom to rectangle','zoom_to_rect.png')
def activate(self, canvas):
self._button_press = self.canvas.mpl_connect(
'button_press_event', self._on_button_press)
self._button_relese = self.canvas.mpl_connect(
'button_release_event', self._on_button_release)
self.mode = 'pan/zoom mode'
def deactivate(self):
if self._button_press:
self.canvas.mpl_disconnect(self._button_press)
if self._button_release:
self.canvas.mpl_disconnect(self._button_release)
class SelectPlotMode (PlotMode):
def __init__(self):
PlotMode.__init__(self, 'select',
'Select within rectangle', 'select.png')
self._selectors = []
def activate(self, canvas):
for ax in canvas.figure.get_axes():
props = dict(facecolor='blue', edgecolor = 'black',
alpha=0.3, fill=True)
rs = RectangleSelector(ax, self.on_select, drawtype='box',
useblit=True, rectprops = props)
self._selectors.append(rs)
def deactivate(self):
for sel in self.selectors:
self.canvas.mpl_disconnect(sel.onmove)
self.canvas.mpl_disconnect(sel.press)
self.canvas.mpl_disconnect(sel.release)
self.canvas.mpl_disconnect(sel.update_background)
class PlotToolbar(gtk.Toolbar):
def __init__(self, plot):
gtk.Toolbar.__init__(self) gtk.Toolbar.__init__(self)
NavigationToolbar2.__init__(self, canvas) self.plot = plot
self._idleId = 0 self.canvas = plot.canvas
self._select_callback = None self._current_mode = None
canvas.connect('enter-notify-event', self.on_enter_notify)
def _init_toolbar(self):
self.set_style(gtk.TOOLBAR_ICONS)
self.tooltips = gtk.Tooltips() self.tooltips = gtk.Tooltips()
self._states = {}
self._selector = None ## Maps toolbar buttons to PlotMode objects.
self._mode_buttons = {}
self.set_property('show-arrow', False) self.set_property('show-arrow', False)
basedir = fluents.ICONDIR
# setup base buttons
for text, tooltip_text, image_file, callback in self.toolitems:
if text is None:
self.insert(gtk.SeparatorToolItem(), -1 )
continue
fname = os.path.join(basedir, image_file)
image = gtk.Image()
image.set_from_file(fname)
tbutton = gtk.ToolButton(image, text)
self.insert(tbutton, -1)
tbutton.connect('clicked', getattr(self, callback))
tbutton.set_tooltip(self.tooltips, tooltip_text, 'Private')
self.insert(gtk.SeparatorToolItem(), -1)
# mode/state buttons
rbutton = None
for text,tooltip_text,image_file,callback in self.radiobuttons:
if text is None:
self.insert(gtk.SeparatorToolItem(), -1 )
continue
fname = os.path.join(basedir, image_file)
image = gtk.Image()
image.set_from_file(fname)
rbutton = gtk.RadioToolButton(rbutton)
rbutton.set_icon_widget(image)
rbutton.connect('toggled', getattr(self, callback))
rbutton.set_tooltip(self.tooltips, tooltip_text, 'Private')
self._states[text] = rbutton
self.insert(rbutton, -1)
self.insert(gtk.SeparatorToolItem(), -1)
toolitem = gtk.ToolItem() #canvas.connect('enter-notify-event', self.on_enter_notify)
self.insert(toolitem, -1) self.show()
self.message = gtk.Label() self.add_mode(DefaultPlotMode())
toolitem.add(self.message) self.add_mode(PanPlotMode())
self.add_mode(ZoomPlotMode())
self.tb_freeze = gtk.ToolItem() self.add_mode(SelectPlotMode())
self.chk = gtk.CheckButton ()
self.chk.set_label ('Freeze')
self.tb_freeze.add (self.chk)
self.tb_freeze.set_tooltip(self.tooltips, 'Freeze current selection')
self.insert(self.tb_freeze,-1)
toolitem = gtk.SeparatorToolItem()
self.insert(toolitem, -1)
self.show_all() self.show_all()
self.fileselect = FileChooserDialog(title='Save the figure', parent=self.win,) def add_mode(self, mode):
"""Adds a new mode to the toolbar."""
if len(self._mode_buttons) > 0:
other = self._mode_buttons.keys()[0]
else:
other = None
btn = gtk.RadioToolButton(other)
btn.set_icon_widget(mode.get_icon())
btn.set_tooltip(self.tooltips, mode.tooltip, 'Private')
btn.connect('toggled', self._on_mode_toggle)
self._mode_buttons[btn] = mode
self.insert(btn, -1)
if self._current_mode == None:
self._current_mode = mode
def get_mode(self):
"""Returns the active mode name."""
if self._current_mode:
return self._current_mode.name
return None
def get_mode_by_name(self, mode_name):
"""Returns the mode with the given name or None."""
for m in self._mode_buttons.values():
if m.name == mode_name:
return m
return None
def get_button(self, mode_name):
"""Returns the button that corresponds to a mode name."""
for b, m in self._mode_buttons.items():
if m == mode_name:
return b
return None
def set_mode(self, mode_name):
"""Sets a mode by name. Returns the mode or None"""
if mode_name == self._current_mode.name:
return None
if self._current_mode:
self._current_mode.deactivate()
new_mode = self.get_mode_by_name(mode_name)
if new_mode:
new_mode.activate(self.canvas)
self._current_mode = self.get_mode_by_name(mode_name)
else:
logger.log('warning', 'No such mode: %s' % mode_name)
return self._current_mode
def _on_mode_toggle(self, button):
if button.get_active():
self.set_mode(self._mode_buttons[button].name)
# def show(self):
# for b in self._mode_buttons.keys():
# b.show()
# gtk.Toolbar.show(self)
# class PlotToolbar(NavigationToolbar2,gtk.Toolbar):
# # list of toolitems to add to the toolbar, format is:
# # text, tooltip_text, image_file, callback(str)
# toolitems = (
# ('Home', 'Reset original view', 'home.png', 'home'),
# #('Back', 'Back to previous view','back.png', 'back'),
# #('Forward', 'Forward to next view','forward.png', 'forward'),
# #('Subplots', 'Configure subplots','subplots.png', 'configure_subplots'),
# ('Save', 'Save the figure','filesave.png', 'save_figure'),
# )
# def __init__(self, canvas, plot):
# self.win = None
# self.plot = plot
# gtk.Toolbar.__init__(self)
# NavigationToolbar2.__init__(self, canvas)
# self._idleId = 0
# self._select_callback = None
# canvas.connect('enter-notify-event', self.on_enter_notify)
# def _init_toolbar(self):
# self.set_style(gtk.TOOLBAR_ICONS)
# self.tooltips = gtk.Tooltips()
# self._modes = {}
# self._selector = None
# self.set_property('show-arrow', False)
# basedir = fluents.ICONDIR
# # setup base buttons
# for text, tooltip_text, image_file, callback in self.toolitems:
# if text is None:
# self.insert(gtk.SeparatorToolItem(), -1 )
# continue
# fname = os.path.join(basedir, image_file)
# image = gtk.Image()
# image.set_from_file(fname)
# tbutton = gtk.ToolButton(image, text)
# self.insert(tbutton, -1)
# tbutton.connect('clicked', getattr(self, callback))
# tbutton.set_tooltip(self.tooltips, tooltip_text, 'Private')
# self.insert(gtk.SeparatorToolItem(), -1)
# # mode/state buttons
# rbutton = None
# for text,tooltip_text,image_file,callback in self.radiobuttons:
# if text is None:
# self.insert(gtk.SeparatorToolItem(), -1 )
# continue
# fname = os.path.join(basedir, image_file)
# image = gtk.Image()
# image.set_from_file(fname)
# rbutton = gtk.RadioToolButton(rbutton)
# rbutton.set_icon_widget(image)
# rbutton.connect('toggled', getattr(self, callback))
# rbutton.set_tooltip(self.tooltips, tooltip_text, 'Private')
# self._modes[text] = rbutton
# a self.insert(rbutton, -1)
# self.insert(gtk.SeparatorToolItem(), -1)
# toolitem = gtk.ToolItem()
# self.insert(toolitem, -1)
# self.message = gtk.Label()
# toolitem.add(self.message)
# self.tb_freeze = gtk.ToolItem()
# self.chk = gtk.CheckButton ()
# self.chk.set_label ('Freeze')
# self.tb_freeze.add (self.chk)
# self.tb_freeze.set_tooltip(self.tooltips, 'Freeze current selection')
# self.insert(self.tb_freeze,-1)
# toolitem = gtk.SeparatorToolItem()
# self.insert(toolitem, -1)
# self.show_all()
# self.fileselect = FileChooserDialog(title='Save the figure', parent=self.win,)
def on_enter_notify(self, widget, event): # def on_enter_notify(self, widget, event):
if self._active != active_mode: # if self._active != active_mode:
self.set_mode(active_mode) # self.set_mode(active_mode)
self.set_mode(active_mode) # self.set_mode(active_mode)
def set_mode(self, active):
print "Set mode called in toolbar from: %s" %self.plot.get_title()
# if state is unkown or not set, set to default
if active == None or active not in self._states.keys():
active = 'DEFAULT'
# remove current Selector: # def set_mode(self, mode):
if self._selector != None: # ## Do nothing if this mode is already active
# problem is ... i have mutliple selectors still connected # if self.mode == mode:
# trying to remove old selectors connections # print "set_mode: already in mode '%s': returning" % mode
# # return mode
# blah
# Her Einar ...
#
self.canvas.mpl_disconnect(self._selector.onmove)
self.canvas.mpl_disconnect(self._selector.press)
self.canvas.mpl_disconnect(self._selector.release)
self.canvas.mpl_disconnect(self._selector.update_background)
print "Removing selector in: %s" %self.plot.get_title()
self._selector = None
# remove current button bindings
if self._idPress != None:
self._idPress = self.canvas.mpl_disconnect(self._idPress)
self.mode = ''
if self._idRelease != None: # ## Remove current Selector if set.
self._idRelease = self.canvas.mpl_disconnect(self._idRelease) # if self._selector != None:
self.mode = '' # self.canvas.mpl_disconnect(self._selector.onmove)
# self.canvas.mpl_disconnect(self._selector.press)
# self.canvas.mpl_disconnect(self._selector.release)
# self.canvas.mpl_disconnect(self._selector.update_background)
# print "set_mode: Removing selector in: %s" %self.plot.get_title()
# self._selector = None
# ## Remove current mouse button bindings
# if self._idPress != None:
# self._idPress = self.canvas.mpl_disconnect(self._idPress)
# self.mode = ''
# if self._idRelease != None:
# self._idRelease = self.canvas.mpl_disconnect(self._idRelease)
# self.mode = ''
# ## Set state, and do state initialization
# self.mode = mode
for state, button in self._states.items():
if state != active:
continue
if state == 'SELECT': # def set_mode(self, active):
for ax in self.canvas.figure.get_axes(): # print "Set mode called in toolbar from: %s" %self.plot.get_title()
props = dict(facecolor='blue', edgecolor = 'black', # # if state is unkown or not set, set to default
alpha=0.3, fill=True) # if active == None or active not in self._modes.keys():
print "creating a selector" # active = 'DEFAULT'
self._selector = RectangleSelector(ax, self.on_select,
drawtype='box', useblit=True,
rectprops=props)
self.mode = 'Select rectangle mode'
elif state == 'PAN':
self._idPress = self.canvas.mpl_connect(
'button_press_event', self.press_pan)
self._idRelease = self.canvas.mpl_connect(
'button_release_event', self.release_pan)
self.mode = 'pan/zoom mode'
elif state == 'ZOOM': # # remove current Selector:
self._idPress = self.canvas.mpl_connect('button_press_event', self.press_zoom) # if self._selector != None:
self._idRelease = self.canvas.mpl_connect('button_release_event', self.release_zoom) # # problem is ... i have mutliple selectors still connected
self.mode = 'Zoom to rect mode' # # trying to remove old selectors connections
# #
# # blah
# # Her Einar ...
# #
# self.canvas.mpl_disconnect(self._selector.onmove)
# self.canvas.mpl_disconnect(self._selector.press)
# self.canvas.mpl_disconnect(self._selector.release)
# self.canvas.mpl_disconnect(self._selector.update_background)
# print "Removing selector in: %s" %self.plot.get_title()
# self._selector = None
# # remove current button bindings
# if self._idPress != None:
# self._idPress = self.canvas.mpl_disconnect(self._idPress)
# self.mode = ''
elif state == 'DEFAULT': # if self._idRelease != None:
pass # self._idRelease = self.canvas.mpl_disconnect(self._idRelease)
else: # self.mode = ''
pass
# self._modes[active]
# for state, button in self._modes.items():
# if state != active:
# continue
if not button.get_active(): # if state == 'SELECT':
button.set_active(True) # for ax in self.canvas.figure.get_axes():
# props = dict(facecolor='blue', edgecolor = 'black',
# alpha=0.3, fill=True)
# print "creating a selector"
# self._selector = RectangleSelector(ax, self.on_select,
# drawtype='box', useblit=True,
# rectprops=props)
# self.mode = 'Select rectangle mode'
for a in self.canvas.figure.get_axes(): # elif state == 'PAN':
a.set_navigate_mode(self._active) # self._idPress = self.canvas.mpl_connect(
# 'button_press_event', self.press_pan)
# self._idRelease = self.canvas.mpl_connect(
# 'button_release_event', self.release_pan)
# self.mode = 'pan/zoom mode'
self.set_message(self.mode) # elif state == 'ZOOM':
# self._idPress = self.canvas.mpl_connect('button_press_event', self.press_zoom)
# self._idRelease = self.canvas.mpl_connect('button_release_event', self.release_zoom)
# self.mode = 'Zoom to rect mode'
self._active = active # elif state == 'DEFAULT':
# update global active_mode (for all toolbar) # pass
globals()['active_mode'] = active # else:
# pass
# if not button.get_active():
# button.set_active(True)
# for a in self.canvas.figure.get_axes():
# a.set_navigate_mode(self._active)
# self.set_message(self.mode)
# self._active = active
# # update global active_mode (for all toolbar)
# globals()['active_mode'] = active
def get_mode(self): # def get_mode(self):
if self._active == None: # if self._active == None:
return 'DEFAULT' # return 'DEFAULT'
return self._active # return self._active
def default(self, button): # def default(self, button):
"""Activates default mode""" # """Activates default mode"""
if not button.get_active(): # if not button.get_active():
return # return
self.set_mode('DEFAULT') # self.set_mode('DEFAULT')
def select(self, button): # def select(self, button):
"""Activate select mode""" # """Activate select mode"""
if not button.get_active(): # if not button.get_active():
return # return
self.set_mode('SELECT') # self.set_mode('SELECT')
def on_select(self,eclick, erelease): # def on_select(self, eclick, erelease, selector):
'eclick and erelease are matplotlib events at press and release' # 'eclick and erelease are matplotlib events at press and release'
if self._select_callback: # if self._select_callback:
print "Onselect called" # print "Onselect called:"
self._select_callback(eclick.xdata, eclick.ydata, erelease.xdata, erelease.ydata) # print " %s" % selector
# self._select_callback(eclick.xdata, eclick.ydata, erelease.xdata, erelease.ydata)
def pan(self, button): # def pan(self, button):
"""Activate the pan/zoom tool. pan with left button, zoom with right""" # """Activate the pan/zoom tool. pan with left button, zoom with right"""
if not button.get_active(): # if not button.get_active():
return # return
self.set_mode('PAN') # self.set_mode('PAN')
def zoom(self, button): # def zoom(self, button):
"""Activate zoom to rect mode""" # """Activate zoom to rect mode"""
if not button.get_active(): # if not button.get_active():
return # return
self.set_mode('ZOOM') # self.set_mode('ZOOM')
def mouse_move(self, event): # def mouse_move(self, event):
"""Extend NavigationToolbar2.mouse_move to provide selection support.""" # """Extend NavigationToolbar2.mouse_move to provide selection support."""
# Only update the rubberband for selection mode when mouse is # # Only update the rubberband for selection mode when mouse is
# within the plotting area. # # within the plotting area.
#if event.inaxes and self._active=='SELECT': # #if event.inaxes and self._active=='SELECT':
# if self._lastCursor != cursors.SELECT_REGION: # # if self._lastCursor != cursors.SELECT_REGION:
# self.set_cursor(cursors.SELECT_REGION) # # self.set_cursor(cursors.SELECT_REGION)
# self._lastCursor = cursors.SELECT_REGION # # self._lastCursor = cursors.SELECT_REGION
# if self._xypress is not None: # # if self._xypress is not None:
# x, y = event.x, event.y # # x, y = event.x, event.y
# lastx, lasty, a, ind, lim, trans= self._xypress # # lastx, lasty, a, ind, lim, trans= self._xypress
# self.draw_rubberband(event, x, y, lastx, lasty) # # self.draw_rubberband(event, x, y, lastx, lasty)
NavigationToolbar2.mouse_move(self, event) # NavigationToolbar2.mouse_move(self, event)
def set_select_callback(self, listener): # def set_select_callback(self, listener):
"""Allow plots to register a callback for selection events. # """Allow plots to register a callback for selection events.
The callback will be called as listener(x1, y1, x2, y2). All # The callback will be called as listener(x1, y1, x2, y2). All
coordinates are in the plot coordinate system, not pixels or # coordinates are in the plot coordinate system, not pixels or
widget coordinates. # widget coordinates.
""" # """
self._select_callback = listener # self._select_callback = listener
def set_cursor(self, cursor): # def set_cursor(self, cursor):
self.canvas.window.set_cursor(cursord[cursor]) # self.canvas.window.set_cursor(cursord[cursor])
def dynamic_update(self): # def dynamic_update(self):
# legacy method; new method is canvas.draw_idle # # legacy method; new method is canvas.draw_idle
self.canvas.draw_idle() # self.canvas.draw_idle()
def draw_rubberband(self, event, x0, y0, x1, y1):
'adapted from http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/189744'
drawable = self.canvas.window
if drawable is None:
return
gc = drawable.new_gc() # def save_figure(self, button):
# fname = self.fileselect.get_filename_from_user()
# if fname:
# self.canvas.print_figure(fname)
height = self.canvas.figure.bbox.height() # def configure_subplots(self, button):
y1 = height - y1 # toolfig = Figure(figsize=(6,3))
y0 = height - y0 # canvas = self._get_canvas(toolfig)
# toolfig.subplots_adjust(top=0.9)
# tool = SubplotTool(self.canvas.figure, toolfig)
w = abs(x1 - x0) # w = int (toolfig.bbox.width())
h = abs(y1 - y0) # h = int (toolfig.bbox.height())
rect = [int(val)for val in min(x0,x1), min(y0, y1), w, h]
try: lastrect, imageBack = self._imageBack
except AttributeError:
#snap image back
if event.inaxes is None:
return
ax = event.inaxes
l,b,w,h = [int(val) for val in ax.bbox.get_bounds()]
b = int(height)-(b+h)
axrect = l,b,w,h
self._imageBack = axrect, drawable.get_image(*axrect)
drawable.draw_rectangle(gc, False, *rect)
self._idleId = 0
else:
def idle_draw(*args):
drawable.draw_image(gc, imageBack, 0, 0, *lastrect)
drawable.draw_rectangle(gc, False, *rect)
self._idleId = 0
return False
if self._idleId == 0:
self._idleId = gobject.idle_add(idle_draw)
def save_figure(self, button):
fname = self.fileselect.get_filename_from_user()
if fname:
self.canvas.print_figure(fname)
def configure_subplots(self, button):
toolfig = Figure(figsize=(6,3))
canvas = self._get_canvas(toolfig)
toolfig.subplots_adjust(top=0.9)
tool = SubplotTool(self.canvas.figure, toolfig)
w = int (toolfig.bbox.width())
h = int (toolfig.bbox.height())
window = gtk.Window() # window = gtk.Window()
window.set_title("Subplot Configuration Tool") # window.set_title("Subplot Configuration Tool")
window.set_default_size(w, h) # window.set_default_size(w, h)
vbox = gtk.VBox() # vbox = gtk.VBox()
window.add(vbox) # window.add(vbox)
vbox.show() # vbox.show()
canvas.show() # canvas.show()
vbox.pack_start(canvas, True, True) # vbox.pack_start(canvas, True, True)
window.show() # window.show()
def _get_canvas(self, fig): # def _get_canvas(self, fig):
return FigureCanvas(fig) # return FigureCanvas(fig)
# Create a view-changed signal that should be emitted every time # Create a view-changed signal that should be emitted every time