1662 lines
58 KiB
Python
1662 lines
58 KiB
Python
import os,sys,copy
|
|
from itertools import izip
|
|
import pygtk
|
|
import gobject
|
|
import gtk
|
|
import fluents
|
|
import logger
|
|
import matplotlib
|
|
from matplotlib.backends.backend_gtkagg import FigureCanvasGTKAgg as FigureCanvas
|
|
from matplotlib.backend_bases import NavigationToolbar2,cursors
|
|
from matplotlib.backends.backend_gtk import FileChooserDialog,cursord
|
|
from matplotlib.widgets import SubplotTool,RectangleSelector,Lasso
|
|
from matplotlib.nxutils import points_inside_poly
|
|
from matplotlib.axes import Subplot, AxesImage
|
|
from matplotlib.figure import Figure
|
|
from matplotlib import cm,cbook
|
|
from pylab import Polygon, axis, Circle
|
|
from matplotlib.collections import LineCollection
|
|
from matplotlib.mlab import prctile
|
|
import networkx
|
|
import scipy
|
|
|
|
# global active mode. Used by toolbars to communicate correct mode
|
|
active_mode = 'default'
|
|
|
|
class ObjectTable:
|
|
"""A 2D table of elements.
|
|
An ObjectTable is a resizable two-dimensional array of objects.
|
|
When resized, it will keep all the elements that fit in the new shape,
|
|
and forget about other elements. When new elements are needed to fill
|
|
out the new shape, they will be created.
|
|
"""
|
|
|
|
def __init__(self, xsize=0, ysize=0, creator=None):
|
|
"""Creates a two-dimensional table of objects.
|
|
@param xsize: Width of the object table.
|
|
@param ysize: Height of the object table.
|
|
@param creator: A function that will be used to instantiate new items.
|
|
"""
|
|
self._elements = []
|
|
self._creator = creator or (lambda : None)
|
|
self.xsize = xsize
|
|
self.ysize = ysize
|
|
self.resize(xsize, ysize)
|
|
|
|
def resize(self, xsize, ysize):
|
|
"""Resizes the table by removing and creating elements as required.
|
|
Makes the ObjectTable the specified size by removing elements that
|
|
will not fit in the new shape and appending elements to fill the
|
|
table. New elements will be created with the creator function passed
|
|
as an argument to __init__.
|
|
|
|
@param xsize: The new width.
|
|
@type xsize: int
|
|
@param ysize: The new height.
|
|
@type ysize: int
|
|
"""
|
|
|
|
# Delete or append new rows
|
|
del self._elements[ysize:]
|
|
new_rows = ysize - len(self._elements)
|
|
self._elements.extend([list() for i in range(new_rows)])
|
|
|
|
# Delete or append new columns
|
|
for row in self._elements:
|
|
del row[xsize:]
|
|
new_elems = xsize - len(row)
|
|
row.extend([self._creator() for i in range(new_elems)])
|
|
|
|
self.xsize = xsize
|
|
self.ysize = ysize
|
|
|
|
def __getitem__(self, index):
|
|
x, y = index
|
|
return self._elements[y][x]
|
|
|
|
def __setitem__(self, index, value):
|
|
x, y = index
|
|
self._elements[y][x] = value
|
|
|
|
|
|
class ViewFrame (gtk.Frame):
|
|
"""
|
|
A ViewFrame is a gtk container widget that contains a View.
|
|
The ViewFrame is either active or inactive, and belongs to a group of
|
|
VeiwFrames of which only one can be active at any time.
|
|
"""
|
|
|
|
def __init__(self, view_frames):
|
|
"""Initializes a new ViewFrame.
|
|
@param view_frames: A list of view frames to which the new view
|
|
frame shouls belong. Only one ViewFrame in this list can be
|
|
active at any time.
|
|
"""
|
|
gtk.Frame.__init__(self)
|
|
self.focused = False
|
|
self.view_frames = view_frames
|
|
self.empty_view = EmptyView()
|
|
self._button_event = None
|
|
|
|
## Set up a VBox with a label wrapped in an event box.
|
|
label = gtk.Label()
|
|
ebox = gtk.EventBox()
|
|
ebox.add(label)
|
|
vbox = gtk.VBox()
|
|
vbox.pack_start(ebox, expand=False)
|
|
vbox.pack_start(gtk.HSeparator(), expand=False)
|
|
|
|
self._ebox_button_event = ebox.connect("button-press-event",
|
|
self._on_button_press_event)
|
|
## Keep the references for later use.
|
|
self._vbox = vbox
|
|
self._ebox = ebox
|
|
self._view_title = label
|
|
self.add(vbox)
|
|
|
|
view_frames.append(self)
|
|
if len(view_frames) == 1:
|
|
self.focus()
|
|
else:
|
|
self.focused = True
|
|
self.unfocus()
|
|
|
|
# Get dropped views
|
|
self.drag_dest_set(gtk.DEST_DEFAULT_ALL,
|
|
[("GTK_TREE_MODEL_ROW", gtk.TARGET_SAME_APP, 7)],
|
|
gtk.gdk.ACTION_LINK)
|
|
self.connect("drag-data-received", self.on_drag_data_received)
|
|
|
|
# Set view
|
|
self._view = self.empty_view
|
|
self._view.connect("button-press-event", self._on_button_press_event)
|
|
self._vbox.add(self._view)
|
|
self._view_title.set_text(self._view.title)
|
|
self.show_all()
|
|
self._view.show()
|
|
|
|
def focus(self):
|
|
"""Gets focus and ensures that no other ViewFrame in the same set
|
|
is in focus.
|
|
"""
|
|
if self.focused:
|
|
self.emit('focus-changed', self, True)
|
|
return self
|
|
|
|
for frame in self.view_frames:
|
|
frame.unfocus()
|
|
|
|
self.set_shadow_type(gtk.SHADOW_IN)
|
|
self._ebox.set_state(gtk.STATE_ACTIVE)
|
|
self.focused = True
|
|
self.emit('focus-changed', self, True)
|
|
return self
|
|
|
|
def unfocus(self):
|
|
"""Removes focus from the ViewFrame. Does nothing if unfocused."""
|
|
if not self.focused:
|
|
return
|
|
|
|
self.set_shadow_type(gtk.SHADOW_OUT)
|
|
self._ebox.set_state(gtk.STATE_NORMAL)
|
|
self.focused = False
|
|
self.emit('focus-changed', self, False)
|
|
|
|
def set_view(self, view):
|
|
"""Set view to view or to empty view if parameter is None
|
|
@param view: An instance of View
|
|
"""
|
|
|
|
# if None is passed, use empty view
|
|
if view == None:
|
|
view = self.empty_view
|
|
|
|
# do nothing if the view is already there
|
|
if view == self._view:
|
|
return
|
|
|
|
# detach view from current parent
|
|
if view._view_frame:
|
|
view._view_frame.set_view(None)
|
|
|
|
# switch which widget we are listening to
|
|
if self._button_event:
|
|
self._view.disconnect(self._button_event)
|
|
|
|
self._button_event = view.connect("button-press-event",
|
|
self._on_button_press_event)
|
|
|
|
# remove old view, set new view
|
|
if self._view:
|
|
self._view.hide()
|
|
self._vbox.remove(self._view)
|
|
self._view._view_frame = None
|
|
|
|
self._view_title.set_text(view.title)
|
|
self._vbox.add(view)
|
|
view.show()
|
|
|
|
view._view_frame = self
|
|
self._view = view
|
|
|
|
def get_view(self):
|
|
"""Returns current view, or None if the empty view is set.
|
|
@returns: None if the ViewFrame is currently displaying it's
|
|
EmptyView. Otherwise it returns the currently displeyd View.
|
|
"""
|
|
if self._view == self.empty_view:
|
|
return None
|
|
return self._view
|
|
|
|
def _on_button_press_event(self, widget, event):
|
|
if not self.focused:
|
|
self.focus()
|
|
|
|
def on_drag_data_received(self, widget, drag_context, x, y,
|
|
selection, info, timestamp):
|
|
treestore, path = selection.tree_get_row_drag_data()
|
|
iter = treestore.get_iter(path)
|
|
obj = treestore.get_value(iter, 2)
|
|
|
|
if isinstance(obj, Plot):
|
|
self.set_view(obj)
|
|
self.focus()
|
|
|
|
elif isinstance(obj, fluents.dataset.Dataset):
|
|
view = self.get_view()
|
|
if view.is_mappable_with(obj):
|
|
view._update_color_from_dataset(obj)
|
|
|
|
elif isinstance(obj, fluents.dataset.Selection):
|
|
view = self.get_view()
|
|
if view.is_mappable_with(obj):
|
|
view.selection_changed(self.current_dim, obj)
|
|
|
|
|
|
|
|
class MainView (gtk.Notebook):
|
|
"""The MainView class displays the Views in Fluents.
|
|
|
|
MainView, of which there is normally only one instance, contains a
|
|
gtk.Table that holds all the visible Views, and a single ViewFrame
|
|
that is used for maximizing plots.
|
|
"""
|
|
def __init__(self):
|
|
gtk.Notebook.__init__(self)
|
|
self.set_show_tabs(False)
|
|
self.set_show_border(False)
|
|
|
|
self._view_frames = []
|
|
self._views = ObjectTable(2, 2, lambda : ViewFrame(self._view_frames))
|
|
self._large_view = ViewFrame(list())
|
|
self.update_small_views()
|
|
|
|
for vf in self._view_frames:
|
|
vf.connect('focus-changed', self._on_view_focus_changed)
|
|
|
|
self.append_page(self._small_views)
|
|
self.append_page(self._large_view)
|
|
self.show()
|
|
self.set_current_page(0)
|
|
|
|
def __getitem__(self, x, y):
|
|
return self._views[x, y]
|
|
|
|
def update_small_views(self):
|
|
"""Creates a new gtk.Table to show the views. Called after changes to
|
|
the _views property"""
|
|
|
|
self._small_views = gtk.Table(self._views.ysize, self._views.xsize, True)
|
|
self._small_views.set_col_spacings(4)
|
|
self._small_views.set_row_spacings(4)
|
|
for x in range(self._views.xsize):
|
|
for y in range(self._views.ysize):
|
|
child = self._views[x,y]
|
|
self._small_views.attach(child, x, x+1, y, y+1)
|
|
self._small_views.show_all()
|
|
|
|
def get_active_small_view(self):
|
|
"""Returns the active ViewFrame in the small views table.
|
|
If a view is maximized, the corresponding ViewFrame in the table
|
|
will be returned, not the active maximized ViewFrame"""
|
|
for vf in self._view_frames:
|
|
if vf.focused:
|
|
return vf
|
|
return None
|
|
|
|
def get_active_view_frame(self):
|
|
"""Returns the active view frame."""
|
|
if self.get_current_page() == 0:
|
|
return self.get_active_small_view()
|
|
else:
|
|
return self._large_view
|
|
|
|
def goto_large(self):
|
|
"""Maximize the view in the current ViewFrame.
|
|
|
|
Maximizes the View in the current ViewFrame. The ViewFrame itself is
|
|
not resized or modified, except it's view will be set to None.
|
|
This method will do nothing if a view is currently maximized.
|
|
"""
|
|
if self.get_current_page() == 1:
|
|
return
|
|
|
|
vf = self.get_active_small_view()
|
|
view = vf.get_view()
|
|
vf.set_view(None)
|
|
self._large_view.set_view(view)
|
|
self.set_current_page(1)
|
|
|
|
def goto_small(self):
|
|
"""Goes to the small views page.
|
|
|
|
The maximized View will be given to the active ViewFrame in the view
|
|
table.
|
|
"""
|
|
if self.get_current_page() == 0:
|
|
return
|
|
|
|
vf = self.get_active_small_view()
|
|
view = self._large_view.get_view()
|
|
self._large_view.set_view(None)
|
|
vf.set_view(view)
|
|
self.set_current_page(0)
|
|
|
|
def insert_view(self, view):
|
|
"""Set a view in the currently active ViewFrame.
|
|
@param view: An instance of View.
|
|
"""
|
|
if self.get_current_page() == 0:
|
|
vf = self.get_active_small_view()
|
|
else:
|
|
vf = self._large_view
|
|
vf.set_view(view)
|
|
|
|
def set_all_plots(self, views):
|
|
"""Displays all the views in the list, and hides all other views.
|
|
|
|
Loops through all ViewFrames from top left to bottom right, and sets
|
|
their currently active View to the next view in the list. After the
|
|
last view is placed in a ViewFrame, the remaining ViewFrames are
|
|
emptied.
|
|
|
|
@param views: A list of views to set.
|
|
"""
|
|
for y in range(self._views.ysize):
|
|
for x in range(self._views.xsize):
|
|
if views:
|
|
self._views[x, y].set_view(views.pop())
|
|
else:
|
|
self._views[x, y].set_view(None)
|
|
|
|
def _on_view_focus_changed(self, widget, vf, focused):
|
|
if focused:
|
|
self.emit('view-changed', vf)
|
|
|
|
def resize_table(self, width, height):
|
|
"""Resizes the list of small views.
|
|
"""
|
|
## Hide all plots that will be removed from the screen.
|
|
for x in range(self._views.xsize):
|
|
for y in range(self._views.ysize):
|
|
if x >= width or y >= height:
|
|
self._views[x, y].set_view(None)
|
|
self._view_frames.remove(self._views[x, y])
|
|
|
|
for x in range(self._views.xsize):
|
|
for y in range(self._views.ysize):
|
|
self._small_views.remove(self._views[x, y])
|
|
|
|
self._views.resize(width, height)
|
|
self.update_small_views()
|
|
self.insert_page(self._small_views, gtk.Label("small"), 0)
|
|
self.remove_page(1)
|
|
#self.set_current_page(0)
|
|
self.goto_small()
|
|
|
|
|
|
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, creates a toolbar, and sets up matplotlib Figure and Canvas
|
|
objects.
|
|
"""
|
|
|
|
def __init__(self, title):
|
|
gtk.Frame.__init__(self)
|
|
self.title = title
|
|
self.set_shadow_type(gtk.SHADOW_NONE)
|
|
self._view_frame = None
|
|
|
|
def get_toolbar(self):
|
|
return None
|
|
|
|
def is_mappable_with(self, dataset):
|
|
"""Override in individual plots."""
|
|
return False
|
|
|
|
|
|
class EmptyView (View):
|
|
"""EmptyView is shown in ViewFrames that are unused."""
|
|
def __init__(self):
|
|
View.__init__(self, 'Empty view')
|
|
self.set_label(None)
|
|
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.fig = Figure()
|
|
self.canvas = FigureCanvas(self.fig)
|
|
self._background = None
|
|
self._frozen = False
|
|
self._toolbar = PlotToolbar(self)
|
|
self.canvas.add_events(gtk.gdk.ENTER_NOTIFY_MASK)
|
|
self.current_dim = None
|
|
self._current_selection = None
|
|
|
|
def set_frozen(self, frozen):
|
|
"""A frozen plot will not be updated when the current selection is changed."""
|
|
self._frozen = frozen
|
|
if not frozen and self._current_selection != None:
|
|
self.set_current_selection(self._current_selection)
|
|
|
|
def get_title(self):
|
|
return self.title
|
|
|
|
def get_toolbar(self):
|
|
return self._toolbar
|
|
|
|
def selection_changed(self, dim_name, selection):
|
|
""" Selection observer handle.
|
|
|
|
A selection change in a plot is only drawn if:
|
|
1.) plot is sensitive to selections (not freezed)
|
|
2.) plot is visible (has a view)
|
|
3.) the selections dim_name is the plot's dimension.
|
|
"""
|
|
|
|
self._current_selection = selection
|
|
if self._frozen \
|
|
or not self.get_property('visible') \
|
|
or self.current_dim != dim_name:
|
|
return
|
|
|
|
self.set_current_selection(selection)
|
|
|
|
def set_selection_listener(self, listener):
|
|
"""Allow project to listen to selections.
|
|
|
|
The selection will propagate back to all plots through the
|
|
selection_changed() method. The listener will be called as
|
|
listener(dimension_name, ids).
|
|
"""
|
|
self.selection_listener = listener
|
|
|
|
def update_selection(self, ids, key=None):
|
|
"""Returns updated current selection from ids.
|
|
If a key is pressed we use the appropriate mode.
|
|
|
|
key map:
|
|
shift : union
|
|
control : intersection
|
|
"""
|
|
if key == 'shift':
|
|
ids = set(ids).union(self._current_selection[self.current_dim])
|
|
elif key == 'control':
|
|
ids = set(ids).intersection(self._current_selection[self.current_dim])
|
|
else:
|
|
pass
|
|
|
|
return ids
|
|
|
|
def set_current_selection(self, selection):
|
|
"""Called whenever the plot should change the selection.
|
|
|
|
This method is a dummy method, so that specialized plots that have
|
|
no implemented selection can ignore selections alltogether.
|
|
"""
|
|
pass
|
|
|
|
class LineViewPlot(Plot):
|
|
"""Line view plot with percentiles.
|
|
|
|
A line view of vectors across a specified dimension of input dataset.
|
|
No selection interaction is defined.
|
|
Only support for 2d-arrays.
|
|
input:
|
|
-- major_axis : dim_number for line dim (see scipy.ndarray for axis def.)
|
|
-- minor_axis : needs definition only for higher order arrays
|
|
|
|
fixme: slow (cant get linecollection and blit to work)
|
|
"""
|
|
def __init__(self, dataset, major_axis=1, minor_axis=None, name="Line view"):
|
|
self.dataset = dataset
|
|
self._data = dataset.asarray()
|
|
if len(self._data.shape)==2 and not minor_axis:
|
|
minor_axis = major_axis - 1
|
|
self.major_axis = major_axis
|
|
self.minor_axis = minor_axis
|
|
Plot.__init__(self, name)
|
|
self.use_blit = False #fixme: blitting should work
|
|
self.current_dim = self.dataset.get_dim_name(major_axis)
|
|
|
|
# make axes
|
|
self.ax = self.fig.add_subplot(111)
|
|
|
|
#initial draw
|
|
x_axis = scipy.arange(self._data.shape[minor_axis])
|
|
self.line_segs=[]
|
|
for xi in range(self._data.shape[major_axis]):
|
|
yi = self._data.take([xi], major_axis).ravel()
|
|
self.line_segs.append([(xx,yy) for xx,yy in izip(x_axis, yi)])
|
|
|
|
# draw background
|
|
self._set_background(self.ax)
|
|
|
|
# add canvas
|
|
self.add(self.canvas)
|
|
self.canvas.show()
|
|
|
|
# add toolbar
|
|
#FIXME: Lineview plot cannot do selections -> disable in toolbar
|
|
#self._toolbar = PlotToolbar(self)
|
|
self.canvas.mpl_connect('resize_event', self.clear_background)
|
|
|
|
def _set_background(self, ax):
|
|
"""Add three patches representing [min max],[5,95] and [25,75] percentiles, and a line at the median.
|
|
"""
|
|
if self._data.shape[self.minor_axis]<10:
|
|
return
|
|
# settings
|
|
patch_color = 'b' #blue
|
|
patch_lw = 0 #no edges
|
|
patch_alpha = .15 # transparancy
|
|
median_color = 'b' #blue
|
|
median_width = 1.5 #linewidth
|
|
percentiles = [0, 5, 25, 50, 75, 100]
|
|
|
|
# ordinate
|
|
xax = scipy.arange(self._data.shape[self.minor_axis])
|
|
|
|
#vertices
|
|
verts_0 = [] #100,0
|
|
verts_1 = [] # 90,10
|
|
verts_2 = [] # 75,25
|
|
med = []
|
|
# add top vertices the low vertices (do i need an order?)#background
|
|
for i in xax:
|
|
prct = prctile(self._data.take([i], self.minor_axis), percentiles)
|
|
verts_0.append((i, prct[0]))
|
|
verts_1.append((i, prct[1]))
|
|
verts_2.append((i, prct[2]))
|
|
med.append(prct[3])
|
|
for i in xax[::-1]:
|
|
prct = prctile(self._data.take([i], self.minor_axis), percentiles)
|
|
verts_0.append((i, prct[-1]))
|
|
verts_1.append((i, prct[-2]))
|
|
verts_2.append((i, prct[-3]))
|
|
|
|
# make polygons from vertices
|
|
bck0 = Polygon(verts_0, alpha=patch_alpha, lw=patch_lw,
|
|
facecolor=patch_color)
|
|
bck1 = Polygon(verts_1, alpha=patch_alpha, lw=patch_lw,
|
|
facecolor=patch_color)
|
|
bck2 = Polygon(verts_2, alpha=patch_alpha, lw=patch_lw,
|
|
facecolor=patch_color)
|
|
|
|
# add polygons to axes
|
|
ax.add_patch(bck0)
|
|
ax.add_patch(bck1)
|
|
ax.add_patch(bck2)
|
|
# median line
|
|
ax.plot(xax, med, median_color, linewidth=median_width)
|
|
|
|
# Disable selection modes
|
|
btn = self._toolbar.get_button('select')
|
|
btn.set_sensitive(False)
|
|
btn = self._toolbar.get_button('lassoselect')
|
|
btn.set_sensitive(False)
|
|
self._toolbar.freeze_button.set_sensitive(False)
|
|
|
|
def clear_background(self, event):
|
|
"""Callback on resize event. Clears the background.
|
|
"""
|
|
self._background = None
|
|
|
|
def set_current_selection(self, selection):
|
|
"""Draws the current selection.
|
|
"""
|
|
ids = selection[self.current_dim] # current identifiers
|
|
index = self.dataset.get_indices(self.current_dim, ids)
|
|
if len(index)==0: # do we have a selection
|
|
return
|
|
if self.use_blit:
|
|
if self._background is None:
|
|
self._bbox = self.ax.bbox.deepcopy()
|
|
self._background = self.canvas.copy_from_bbox(self.ax.bbox)
|
|
self.canvas.restore_region(self._background)
|
|
|
|
if len(self.ax.collections)>0: # remove old linecollection
|
|
self.ax.collections = []
|
|
segs = [self.line_segs[i] for i in index]
|
|
line_coll = LineCollection(segs, colors=(1,0,0,1))
|
|
line_coll.set_clip_box(self.ax.bbox)
|
|
self.ax.update_datalim(line_coll.get_verts(self.ax.transData))
|
|
#draw
|
|
if self.use_blit:
|
|
self.ax.draw_artist(line_coll)
|
|
self.canvas.blit()
|
|
else:
|
|
self.ax.add_collection(line_coll)
|
|
self.canvas.draw()
|
|
|
|
class ScatterMarkerPlot(Plot):
|
|
"""The ScatterMarkerPlot is faster than regular scatterplot, but
|
|
has no color and size options."""
|
|
|
|
def __init__(self, dataset_1, dataset_2, id_dim, sel_dim,
|
|
id_1, id_2, s=6, name="Scatter plot"):
|
|
Plot.__init__(self, name)
|
|
self.use_blit = False
|
|
self._background = None
|
|
self.ax = self.fig.add_subplot(111)
|
|
self.ax.axhline(0, color='k', lw=1., zorder=1)
|
|
self.ax.axvline(0, color='k', lw=1., zorder=1)
|
|
self.current_dim = id_dim
|
|
self.dataset_1 = dataset_1
|
|
self.ms = s
|
|
self._selection_line = None
|
|
x_index = dataset_1[sel_dim][id_1]
|
|
y_index = dataset_2[sel_dim][id_2]
|
|
self.xaxis_data = dataset_1._array[:, x_index]
|
|
self.yaxis_data = dataset_2._array[:, y_index]
|
|
self.ax.plot(self.xaxis_data, self.yaxis_data, 'o', markeredgewidth=0, markersize=s)
|
|
#self.ax.set_title(self.get_title())
|
|
self.add(self.canvas)
|
|
self.canvas.show()
|
|
|
|
def rectangle_select_callback(self, x1, y1, x2, y2, key):
|
|
ydata = self.yaxis_data
|
|
xdata = self.xaxis_data
|
|
|
|
# find indices of selected area
|
|
if x1>x2:
|
|
x1, x2 = x2, x1
|
|
if y1>y2:
|
|
y1, y2 = y2, y1
|
|
assert x1<=x2
|
|
assert y1<=y2
|
|
index = scipy.nonzero((xdata>x1) & (xdata<x2) & (ydata>y1) & (ydata<y2))[0]
|
|
ids = self.dataset_1.get_identifiers(self.current_dim, index)
|
|
ids = self.update_selection(ids, key)
|
|
self.selection_listener(self.current_dim, ids)
|
|
|
|
def lasso_select_callback(self, verts, key=None):
|
|
self.canvas.draw_idle()
|
|
xys = scipy.c_[self.xaxis_data[:,scipy.newaxis], self.yaxis_data[:,scipy.newaxis]]
|
|
index = scipy.nonzero(points_inside_poly(xys, verts))[0]
|
|
ids = self.dataset_1.get_identifiers(self.current_dim, index)
|
|
ids = self.update_selection(ids, key)
|
|
self.selection_listener(self.current_dim, ids)
|
|
self.canvas.widgetlock.release(self._lasso)
|
|
|
|
def set_current_selection(self, selection):
|
|
ids = selection[self.current_dim] # current identifiers
|
|
index = self.dataset_1.get_indices(self.current_dim, ids)
|
|
if self.use_blit:
|
|
if self._background is None:
|
|
self._background = self.canvas.copy_from_bbox(self.ax.bbox)
|
|
self.canvas.restore_region(self._background)
|
|
if not len(index)>0:
|
|
return
|
|
xdata_new = self.xaxis_data.take(index) #take data
|
|
ydata_new = self.yaxis_data.take(index)
|
|
#remove old selection
|
|
if self._selection_line:
|
|
self.ax.lines.remove(self._selection_line)
|
|
|
|
self._selection_line, = self.ax.plot(xdata_new, ydata_new,marker='o', markersize=self.ms, linestyle=None, markerfacecolor='r')
|
|
|
|
# self._toolbar.forward() #update data lims before draw
|
|
if self.use_blit:
|
|
self.ax.draw_artist(self._selection_line)
|
|
self.canvas.blit()
|
|
else:
|
|
self.canvas.draw()
|
|
|
|
|
|
class ScatterPlot(Plot):
|
|
"""The ScatterPlot is slower than scattermarker, but has size option."""
|
|
def __init__(self, dataset_1, dataset_2, id_dim, sel_dim, id_1, id_2, c='b', s=30, sel_dim_2=None, name="Scatter plot"):
|
|
Plot.__init__(self, name)
|
|
self.use_blit = False
|
|
self.ax = self.fig.add_subplot(111)
|
|
self._clean_bck = self.canvas.copy_from_bbox(self.ax.bbox)
|
|
self.current_dim = id_dim
|
|
self.dataset_1 = dataset_1
|
|
x_index = dataset_1[sel_dim][id_1]
|
|
if sel_dim_2:
|
|
y_index = dataset_2[sel_dim_2][id_2]
|
|
else:
|
|
y_index = dataset_2[sel_dim][id_2]
|
|
self.xaxis_data = dataset_1._array[:, x_index]
|
|
self.yaxis_data = dataset_2._array[:, y_index]
|
|
lw = scipy.zeros(self.xaxis_data.shape)
|
|
self.sc = sc = self.ax.scatter(self.xaxis_data, self.yaxis_data,
|
|
s=s, c=c, linewidth=lw)
|
|
self.ax.axhline(0, color='k', lw=1., zorder=1)
|
|
self.ax.axvline(0, color='k', lw=1., zorder=1)
|
|
|
|
# labels
|
|
self._text_labels = None
|
|
|
|
# add canvas to widget
|
|
self.add(self.canvas)
|
|
self.canvas.show()
|
|
|
|
def is_mappable_with(self, obj):
|
|
"""Returns True if dataset/selection is mappable with this plot.
|
|
"""
|
|
if isinstance(obj, fluents.dataset.Dataset):
|
|
if self.current_dim in obj.get_dim_name() and obj.asarray().shape[0] == self.xaxis_data.shape[0]:
|
|
return True
|
|
elif isinstance(obj, fluents.dataset.Selection):
|
|
if self.current_dim in obj.get_dim_name():
|
|
print "Selection is mappable"
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def _update_color_from_dataset(self, data):
|
|
"""Updates the facecolors from a dataset.
|
|
"""
|
|
array = data.asarray()
|
|
#only support for 2d-arrays:
|
|
try:
|
|
m, n = array.shape
|
|
except:
|
|
raise ValueError, "No support for more than 2 dimensions."
|
|
# is dataset a vector or matrix?
|
|
if not n==1:
|
|
# we have a category dataset
|
|
if isinstance(data, fluents.dataset.CategoryDataset):
|
|
map_vec = scipy.dot(array, scipy.diag(scipy.arange(n))).sum(1)
|
|
else:
|
|
map_vec = array.sum(1)
|
|
else:
|
|
map_vec = array.ravel()
|
|
|
|
# update facecolors
|
|
self.sc.set_array(map_vec)
|
|
self.sc.set_clim(map_vec.min(), map_vec.max())
|
|
self.sc.update_scalarmappable() #sets facecolors from array
|
|
self.canvas.draw()
|
|
|
|
def rectangle_select_callback(self, x1, y1, x2, y2, key):
|
|
ydata = self.yaxis_data
|
|
xdata = self.xaxis_data
|
|
|
|
# find indices of selected area
|
|
if x1>x2:
|
|
x1, x2 = x2, x1
|
|
if y1>y2:
|
|
y1, y2 = y2, y1
|
|
assert x1<=x2
|
|
assert y1<=y2
|
|
|
|
index = scipy.nonzero((xdata>x1) & (xdata<x2) & (ydata>y1) & (ydata<y2))[0]
|
|
ids = self.dataset_1.get_identifiers(self.current_dim, index)
|
|
ids = self.update_selection(ids, key)
|
|
self.selection_listener(self.current_dim, ids)
|
|
|
|
def lasso_select_callback(self, verts, key=None):
|
|
self.canvas.draw_idle()
|
|
xys = scipy.c_[self.xaxis_data[:,scipy.newaxis], self.yaxis_data[:,scipy.newaxis]]
|
|
index = scipy.nonzero(points_inside_poly(xys, verts))[0]
|
|
ids = self.dataset_1.get_identifiers(self.current_dim, index)
|
|
ids = self.update_selection(ids, key)
|
|
self.selection_listener(self.current_dim, ids)
|
|
|
|
def set_current_selection(self, selection):
|
|
ids = selection[self.current_dim] # current identifiers
|
|
if len(ids)==0:
|
|
return
|
|
index = self.dataset_1.get_indices(self.current_dim, ids)
|
|
if self.use_blit:
|
|
if self._background is None:
|
|
self._background = self.canvas.copy_from_bbox(self.ax.bbox)
|
|
self.canvas.restore_region(self._background)
|
|
lw = scipy.zeros(self.xaxis_data.shape, 'f')
|
|
if len(index)>0:
|
|
lw.put(2., index)
|
|
self.sc.set_linewidth(lw)
|
|
|
|
if self.use_blit:
|
|
self.ax.draw_artist(self.sc)
|
|
self.canvas.blit()
|
|
else:
|
|
self.canvas.draw()
|
|
|
|
|
|
class ImagePlot(Plot):
|
|
def __init__(self, dataset, **kw):
|
|
self.dataset = dataset
|
|
self.keywords = kw
|
|
|
|
Plot.__init__(self, kw['name'])
|
|
|
|
self.ax = self.fig.add_subplot(111)
|
|
self.ax.set_xticks([])
|
|
self.ax.set_yticks([])
|
|
self.ax.grid(False)
|
|
|
|
# Initial draw
|
|
self.ax.imshow(dataset.asarray(), interpolation='nearest', aspect='auto')
|
|
|
|
# Add canvas and show
|
|
self.add(self.canvas)
|
|
self.canvas.show()
|
|
|
|
# Disable selection modes
|
|
btn = self._toolbar.get_button('select')
|
|
btn.set_sensitive(False)
|
|
btn = self._toolbar.get_button('lassoselect')
|
|
btn.set_sensitive(False)
|
|
self._toolbar.freeze_button.set_sensitive(False)
|
|
|
|
def get_toolbar(self):
|
|
return self._toolbar
|
|
|
|
|
|
class HistogramPlot(Plot):
|
|
def __init__(self, dataset, **kw):
|
|
Plot.__init__(self, kw['name'])
|
|
|
|
self.ax = self.fig.add_subplot(111)
|
|
self.ax.grid(False)
|
|
|
|
# Initial draw
|
|
self.ax.hist(dataset.asarray(), bins=20)
|
|
|
|
# Add canvas and show
|
|
self.add(self.canvas)
|
|
self.canvas.show()
|
|
|
|
# Disable selection modes
|
|
btn = self._toolbar.get_button('select')
|
|
btn.set_sensitive(False)
|
|
btn = self._toolbar.get_button('lassoselect')
|
|
btn.set_sensitive(False)
|
|
self._toolbar.freeze_button.set_sensitive(False)
|
|
|
|
def get_toolbar(self):
|
|
return self._toolbar
|
|
|
|
|
|
class BarPlot(Plot):
|
|
"""Bar plot.
|
|
|
|
Ordinary bar plot for (column) vectors.
|
|
For matrices there is one color for each row.
|
|
"""
|
|
def __init__(self, dataset, name):
|
|
self.dataset = dataset
|
|
n, m = dataset.shape
|
|
Plot.__init__(self, name)
|
|
self.ax = self.fig.add_subplot(111)
|
|
self.ax.grid(False)
|
|
|
|
# Initial draw
|
|
if m>1:
|
|
sm = cm.ScalarMappable()
|
|
clrs = sm.to_rgba(range(n))
|
|
for i, row in enumerate(dataset.asarray()):
|
|
left = scipy.arange(i+1, m*n+1, n)
|
|
height = row
|
|
color = clrs[i]
|
|
c = (color[0], color[1], color[2])
|
|
self.ax.bar(left, height,color=c)
|
|
else:
|
|
height = dataset.asarray().ravel()
|
|
left = scipy.arange(1, n, 1)
|
|
self.ax.bar(left, height)
|
|
|
|
# Add canvas and show
|
|
self.add(self.canvas)
|
|
self.canvas.show()
|
|
|
|
# Disable selection modes
|
|
btn = self._toolbar.get_button('select')
|
|
btn.set_sensitive(False)
|
|
btn = self._toolbar.get_button('lassoselect')
|
|
btn.set_sensitive(False)
|
|
self._toolbar.freeze_button.set_sensitive(False)
|
|
|
|
def get_toolbar(self):
|
|
return self._toolbar
|
|
|
|
|
|
class NetworkPlot(Plot):
|
|
def __init__(self, dataset, **kw):
|
|
# Set member variables and call superclass' constructor
|
|
self.graph = dataset.asnetworkx()
|
|
self.dataset = dataset
|
|
self.keywords = kw
|
|
self.dim_name = self.dataset.get_dim_name(0)
|
|
|
|
if not kw.has_key('with_labels'):
|
|
kw['with_labels'] = False
|
|
if not kw.has_key('name'):
|
|
kw['name'] = self.dataset.get_name()
|
|
if not kw.has_key('prog'):
|
|
kw['prog'] = 'neato'
|
|
if not kw.has_key('pos') or kw['pos']:
|
|
kw['pos'] = networkx.graphviz_layout(self.graph, kw['prog'])
|
|
if not kw.has_key('nodelist'):
|
|
kw['nodelist'] = self.dataset.get_identifiers(self.dim_name, sorted=True)
|
|
Plot.__init__(self, kw['name'])
|
|
self.current_dim = self.dim_name
|
|
|
|
# Keep node size and color as dicts for fast lookup
|
|
self.node_size = {}
|
|
if kw.has_key('node_size') and cbook.iterable(kw['node_size']):
|
|
kw.remove('node_size')
|
|
for id, size in zip(self.dataset[self.dim_name], kw['node_size']):
|
|
self.node_size[id] = size
|
|
else:
|
|
for id in dataset[self.dim_name]:
|
|
self.node_size[id] = 30
|
|
|
|
self.node_color = {}
|
|
if kw.has_key('node_color') and cbook.iterable(kw['node_color']):
|
|
kw.remove('node_color')
|
|
for id, color in zip(self.dataset[self.dim_name], kw['node_color']):
|
|
self.node_color[id] = color
|
|
else:
|
|
self.node_color = None
|
|
# for id in self.dataset[self.dim_name]:
|
|
# self.node_color[id] = 'red'
|
|
|
|
if kw.has_key('node_color'):
|
|
kw.pop('node_color')
|
|
|
|
self.ax = self.fig.add_subplot(111)
|
|
self.ax.set_xticks([])
|
|
self.ax.set_yticks([])
|
|
self.ax.grid(False)
|
|
self.ax.set_frame_on(False)
|
|
# FIXME: ax shouldn't be in kw at all
|
|
if kw.has_key('ax'):
|
|
kw.pop('ax')
|
|
|
|
# Add canvas and show
|
|
self.add(self.canvas)
|
|
self.canvas.show()
|
|
|
|
# Initial draw
|
|
networkx.draw_networkx(self.graph, ax=self.ax, **kw)
|
|
del kw['nodelist']
|
|
|
|
def get_toolbar(self):
|
|
return self._toolbar
|
|
|
|
def rectangle_select_callback(self, x1, y1, x2, y2, key):
|
|
pos = self.keywords['pos']
|
|
ydata = scipy.zeros((len(pos),), 'l')
|
|
xdata = scipy.zeros((len(pos),), 'l')
|
|
node_ids = []
|
|
c = 0
|
|
for name,(x,y) in pos.items():
|
|
node_ids.append(name)
|
|
xdata[c] = x
|
|
ydata[c] = y
|
|
c+=1
|
|
|
|
# find indices of selected area
|
|
if x1 > x2:
|
|
x1, x2 = x2, x1
|
|
if y1 > y2:
|
|
y1, y2 = y2, y1
|
|
index = scipy.nonzero((xdata>x1) & (xdata<x2) & (ydata>y1) & (ydata<y2))[0]
|
|
ids = [node_ids[i] for i in index]
|
|
ids = self.update_selection(ids, key)
|
|
self.selection_listener(self.current_dim, ids)
|
|
|
|
def lasso_select_callback(self, verts, key=None):
|
|
pos = self.keywords['pos']
|
|
xys = []
|
|
node_ids = []
|
|
c = 0
|
|
for name,(x,y) in pos.items():
|
|
node_ids.append(name)
|
|
xys.append((x,y))
|
|
c+=1
|
|
|
|
self.canvas.draw_idle()
|
|
index = scipy.nonzero(points_inside_poly(xys, verts))[0]
|
|
ids = [node_ids[i] for i in index]
|
|
ids = self.update_selection(ids, key)
|
|
self.selection_listener(self.current_dim, ids)
|
|
self.canvas.widgetlock.release(self._lasso)
|
|
|
|
def set_current_selection(self, selection):
|
|
ids = selection[self.current_dim] # current identifiers
|
|
node_set = set(self.graph.nodes())
|
|
|
|
selected_nodes = list(ids.intersection(node_set))
|
|
unselected_nodes = list(node_set.difference(ids))
|
|
|
|
if self.node_color:
|
|
unselected_colors = [self.node_color[x] for x in unselected_nodes]
|
|
else:
|
|
unselected_colors = 'gray'
|
|
|
|
if self.node_size:
|
|
unselected_sizes = [self.node_size[x] for x in unselected_nodes]
|
|
selected_sizes = [self.node_size[x] for x in selected_nodes]
|
|
|
|
self.ax.collections=[]
|
|
networkx.draw_networkx_edges(self.graph,
|
|
edge_list=self.graph.edges(),
|
|
ax=self.ax,
|
|
**self.keywords)
|
|
|
|
networkx.draw_networkx_labels(self.graph,**self.keywords)
|
|
|
|
if unselected_nodes:
|
|
networkx.draw_networkx_nodes(self.graph, nodelist=unselected_nodes, \
|
|
node_color='gray', node_size=unselected_sizes, ax=self.ax, **self.keywords)
|
|
|
|
if selected_nodes:
|
|
networkx.draw_networkx_nodes(self.graph, nodelist=selected_nodes, \
|
|
node_color='r', node_size=selected_sizes, ax=self.ax, **self.keywords)
|
|
self.ax.collections[-1].set_zorder(3)
|
|
|
|
self.canvas.draw()
|
|
|
|
|
|
class VennPlot(Plot):
|
|
def __init__(self, name="Venn diagram"):
|
|
Plot.__init__(self, name)
|
|
self._ax = self.fig.add_subplot(111)
|
|
self._ax.grid(0)
|
|
self._init_bck()
|
|
for c in self._venn_patches:
|
|
self._ax.add_patch(c)
|
|
for mrk in self._markers:
|
|
self._ax.add_patch(mrk)
|
|
self._ax.set_xlim([-3, 3])
|
|
self._ax.set_ylim([-2.5, 3.5])
|
|
self._last_active = set()
|
|
self._ax.set_xticks([])
|
|
self._ax.set_yticks([])
|
|
self._ax.grid(0)
|
|
self._ax.axis('equal')
|
|
self._ax.set_frame_on(False)
|
|
# add canvas to widget
|
|
self.add(self.canvas)
|
|
self.canvas.show()
|
|
|
|
def _init_bck(self):
|
|
res = 50
|
|
a = .5
|
|
r = 1.5
|
|
mr = .2
|
|
self.c1 = c1 = Circle((-1,0), radius=r, alpha=a, facecolor='b')
|
|
self.c2 = c2 = Circle((1,0), radius=r, alpha=a, facecolor='r')
|
|
self.c3 = c3 = Circle((0, scipy.sqrt(3)), radius=r, alpha=a, facecolor='g')
|
|
|
|
self.c1marker = Circle((-1.25, -.25), radius=mr, facecolor='y', alpha=0)
|
|
self.c2marker = Circle((1.25, -.25), radius=mr, facecolor='y', alpha=0)
|
|
self.c3marker = Circle((0, scipy.sqrt(3)+.25), radius=mr, facecolor='y', alpha=0)
|
|
self.c1c2marker = Circle((0, -.15), radius=mr, facecolor='y', alpha=0)
|
|
|
|
self.c1c3marker = Circle((-scipy.sqrt(2)/2, 1), radius=mr, facecolor='y', alpha=0)
|
|
self.c2c3marker = Circle((scipy.sqrt(2)/2, 1), radius=mr, facecolor='y', alpha=0)
|
|
self.c1c2c3marker = Circle((0, .6), radius=mr, facecolor='y', alpha=0)
|
|
|
|
c1.elements = set(['a', 'b', 'c', 'f'])
|
|
c2.elements = set(['a', 'c', 'd', 'e'])
|
|
c3.elements = set(['a', 'e', 'f', 'g'])
|
|
self.active_elements = set()
|
|
self.all_elements = c1.elements.union(c2.elements).union(c3.elements)
|
|
|
|
c1.active = False
|
|
c2.active = False
|
|
c3.active = False
|
|
|
|
c1.name = 'Blue'
|
|
c2.name = 'Red'
|
|
c3.name = 'Green'
|
|
|
|
self._venn_patches = [c1, c2, c3]
|
|
self._markers = [self.c1marker, self.c2marker, self.c3marker,
|
|
self.c1c2marker, self.c1c3marker,
|
|
self.c2c3marker, self.c1c2c3marker]
|
|
|
|
self._tot_label = 'Tot: ' + str(len(self.all_elements))
|
|
self._sel_label = 'Sel: ' + str(len(self.active_elements))
|
|
self._legend = self._ax.legend((self._tot_label, self._sel_label),
|
|
loc='upper right')
|
|
|
|
def set_selection(self, selection, patch=None):
|
|
if patch:
|
|
patch.selection = selection
|
|
else:
|
|
selection_set = False
|
|
for patch in self._venn_patches:
|
|
if len(patch.elements)==0:
|
|
patch.elements = selection
|
|
selection_set = True
|
|
if not selection_set:
|
|
self.venn_patches[0].elements = selection
|
|
|
|
def lasso_select_callback(self, verts, key=None):
|
|
if verts==None:
|
|
print "ks"
|
|
verts = (self._event.xdata, self._event.ydata)
|
|
if key!='shift':
|
|
for m in self._markers:
|
|
m.set_alpha(0)
|
|
|
|
self._patches_within_verts(verts, key)
|
|
active = [i.active for i in self._venn_patches]
|
|
if active==[True, False, False]:
|
|
self.c1marker.set_alpha(1)
|
|
self.active_elements = self.c1.elements.difference(self.c2.elements.union(self.c3.elements))
|
|
elif active== [False, True, False]:
|
|
self.c2marker.set_alpha(1)
|
|
self.active_elements = self.c2.elements.difference(self.c1.elements.union(self.c3.elements))
|
|
elif active== [False, False, True]:
|
|
self.c3marker.set_alpha(1)
|
|
self.active_elements = self.c3.elements.difference(self.c2.elements.union(self.c1.elements))
|
|
elif active==[True, True, False]:
|
|
self.c1c2marker.set_alpha(1)
|
|
self.active_elements = self.c1.elements.intersection(self.c2.elements)
|
|
elif active==[True, False, True]:
|
|
self.c1c3marker.set_alpha(1)
|
|
self.active_elements = self.c1.elements.intersection(self.c3.elements)
|
|
elif active==[False, True, True]:
|
|
self.c2c3marker.set_alpha(1)
|
|
self.active_elements = self.c2.elements.intersection(self.c3.elements)
|
|
elif active==[True, True, True]:
|
|
self.c1c2c3marker.set_alpha(1)
|
|
self.active_elements = self.c1.elements.intersection(self.c3.elements).intersection(self.c2.elements)
|
|
|
|
if key=='shift':
|
|
self.active_elements = self.active_elements.union(self._last_active)
|
|
self._last_active = self.active_elements.copy()
|
|
self._sel_label = 'Sel: ' + str(len(self.active_elements))
|
|
self._legend.texts[1].set_text(self._sel_label)
|
|
self.canvas.widgetlock.release(self._lasso)
|
|
del self._lasso
|
|
self._ax.figure.canvas.draw()
|
|
|
|
def rectangle_select_callback(self, x1, y1, x2, y2, key):
|
|
"""event1 and event2 are the press and release events"""
|
|
#x1, y1 = event1.xdata, event1.ydata
|
|
#x2, y2 = event2.xdata, event2.ydata
|
|
#key = event1.key
|
|
|
|
verts = [(x1, y1), (x2, y2)]
|
|
if key!='shift':
|
|
for m in self._markers:
|
|
m.set_alpha(0)
|
|
|
|
self._patches_within_verts(verts, key)
|
|
active = [i.active for i in self._venn_patches]
|
|
if active==[True, False, False]:
|
|
self.c1marker.set_alpha(1)
|
|
self.active_elements = self.c1.elements.difference(self.c2.elements.union(self.c3.elements))
|
|
elif active== [False, True, False]:
|
|
self.c2marker.set_alpha(1)
|
|
self.active_elements = self.c2.elements.difference(self.c1.elements.union(self.c3.elements))
|
|
elif active== [False, False, True]:
|
|
self.c3marker.set_alpha(1)
|
|
self.active_elements = self.c3.elements.difference(self.c2.elements.union(self.c1.elements))
|
|
elif active==[True, True, False]:
|
|
self.c1c2marker.set_alpha(1)
|
|
self.active_elements = self.c1.elements.intersection(self.c2.elements)
|
|
elif active==[True, False, True]:
|
|
self.c1c3marker.set_alpha(1)
|
|
self.active_elements = self.c1.elements.intersection(self.c3.elements)
|
|
elif active==[False, True, True]:
|
|
self.c2c3marker.set_alpha(1)
|
|
self.active_elements = self.c2.elements.intersection(self.c3.elements)
|
|
elif active==[True, True, True]:
|
|
self.c1c2c3marker.set_alpha(1)
|
|
self.active_elements = self.c1.elements.intersection(self.c3.elements).intersection(self.c2.elements)
|
|
|
|
if key=='shift':
|
|
self.active_elements = self.active_elements.union(self._last_active)
|
|
self._last_active = self.active_elements.copy()
|
|
self._sel_label = 'Sel: ' + str(len(self.active_elements))
|
|
self._legend.texts[1].set_text(self._sel_label)
|
|
self._ax.figure.canvas.draw()
|
|
|
|
def _patches_within_verts(self, verts, key):
|
|
xy = scipy.array(verts).mean(0)
|
|
for venn_patch in self._venn_patches:
|
|
venn_patch.active = False
|
|
if self._distance(venn_patch.center,xy)<venn_patch.radius:
|
|
venn_patch.active = True
|
|
|
|
def _distance(self, (x1,y1),(x2,y2)):
|
|
return scipy.sqrt( (x2-x1)**2 + (y2-y1)**2 )
|
|
|
|
|
|
class PlotMode:
|
|
"""A PlotMode object corresponds to a mouse mode in a plot.
|
|
|
|
When a mode is selected in the toolbar, the PlotMode corresponding
|
|
to the toolbar button is activated by calling setup(ax) for the axis
|
|
system ax.
|
|
"""
|
|
def __init__(self, plot, name, tooltip, image_file):
|
|
self.name = name
|
|
self.tooltip = tooltip
|
|
self.image_file = image_file
|
|
self.plot = plot
|
|
self.canvas = plot.canvas
|
|
|
|
def get_icon(self):
|
|
"""Returns the icon for the PlotMode"""
|
|
image = gtk.Image()
|
|
image.set_from_pixbuf(fluents.icon_factory.get(self.image_file))
|
|
return image
|
|
|
|
def activate(self):
|
|
"""Subclasses of PlotMode should do their initialization here.
|
|
|
|
The activate method is called when a mode is activated, and is
|
|
used primarily to set up callback functions on events in the
|
|
canvas.
|
|
"""
|
|
pass
|
|
|
|
def deactivate(self):
|
|
"""Subclasses of PlotMode should do their cleanup here.
|
|
|
|
The deactivate method is primarily by subclasses of PlotMode to
|
|
remove any callbacks they might have on the matplotlib canvas.
|
|
"""
|
|
pass
|
|
|
|
def _mpl_disconnect_all(self):
|
|
"""Disconnects all matplotlib callbacks defined on the canvas.
|
|
|
|
This is a hack because the RectangleSelector in matplotlib does
|
|
not store its callbacks, so we need a workaround to remove them.
|
|
"""
|
|
callbacks = self.plot.canvas.callbacks
|
|
|
|
for callbackd in callbacks.values():
|
|
for c in callbackd.keys():
|
|
del callbackd[c]
|
|
|
|
|
|
class DefaultPlotMode (PlotMode):
|
|
def __init__(self, plot):
|
|
PlotMode.__init__(self, plot, 'default', 'Default mode', 'cursor')
|
|
|
|
|
|
class PanPlotMode (PlotMode):
|
|
def __init__(self, plot):
|
|
PlotMode.__init__(self, plot, 'pan',
|
|
'Pan axes with left mouse, zoom with right',
|
|
'move')
|
|
|
|
# Holds handler IDs for callbacks.
|
|
self._button_press = None
|
|
self._button_release = None
|
|
self._motion_notify = None
|
|
self._xypress = None
|
|
self._button_pressed = None
|
|
|
|
def activate(self):
|
|
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._drag = self.canvas.mpl_connect(
|
|
# 'mouse_drag_event', self._on_drag)
|
|
|
|
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)
|
|
|
|
def _on_button_press(self, event):
|
|
|
|
if event.button == 1:
|
|
self._button_pressed = 1
|
|
elif event.button == 3:
|
|
self._button_pressed = 3
|
|
else:
|
|
self._button_pressed=None
|
|
return
|
|
|
|
x, y = event.x, event.y
|
|
|
|
# push the current view to define home if stack is empty
|
|
# if self._views.empty(): self.push_current()
|
|
|
|
self._xypress=[]
|
|
for i, a in enumerate(self.canvas.figure.get_axes()):
|
|
if x is not None and y is not None and a.in_axes(x, y) \
|
|
and a.get_navigate():
|
|
xmin, xmax = a.get_xlim()
|
|
ymin, ymax = a.get_ylim()
|
|
lim = xmin, xmax, ymin, ymax
|
|
self._xypress.append((x, y, a, i, lim,a.transData.deepcopy()))
|
|
self.canvas.mpl_disconnect(self._motion_notify)
|
|
|
|
cid = self.canvas.mpl_connect('motion_notify_event',
|
|
self._on_motion_notify)
|
|
self._motion_notify = cid
|
|
|
|
def _on_motion_notify(self, event):
|
|
"""The drag callback in pan/zoom mode"""
|
|
|
|
def format_deltas(event, dx, dy):
|
|
"""Returns the correct dx and dy based on the modifier keys"""
|
|
if event.key=='control':
|
|
if(abs(dx)>abs(dy)):
|
|
dy = dx
|
|
else:
|
|
dx = dy
|
|
elif event.key=='x':
|
|
dy = 0
|
|
elif event.key=='y':
|
|
dx = 0
|
|
elif event.key=='shift':
|
|
if 2*abs(dx) < abs(dy):
|
|
dx=0
|
|
elif 2*abs(dy) < abs(dx):
|
|
dy=0
|
|
elif(abs(dx)>abs(dy)):
|
|
dy=dy/abs(dy)*abs(dx)
|
|
else:
|
|
dx=dx/abs(dx)*abs(dy)
|
|
return (dx,dy)
|
|
|
|
for cur_xypress in self._xypress:
|
|
lastx, lasty, a, ind, lim, trans = cur_xypress
|
|
xmin, xmax, ymin, ymax = lim
|
|
|
|
#safer to use the recorded button at the press than current button:
|
|
#multiple button can get pressed during motion...
|
|
if self._button_pressed==1:
|
|
lastx, lasty = trans.inverse_xy_tup( (lastx, lasty) )
|
|
x, y = trans.inverse_xy_tup( (event.x, event.y) )
|
|
if a.get_xscale()=='log':
|
|
dx=1-lastx/x
|
|
else:
|
|
dx=x-lastx
|
|
if a.get_yscale()=='log':
|
|
dy=1-lasty/y
|
|
else:
|
|
dy=y-lasty
|
|
|
|
dx,dy=format_deltas(event,dx,dy)
|
|
|
|
if a.get_xscale()=='log':
|
|
xmin *= 1-dx
|
|
xmax *= 1-dx
|
|
else:
|
|
xmin -= dx
|
|
xmax -= dx
|
|
if a.get_yscale()=='log':
|
|
ymin *= 1-dy
|
|
ymax *= 1-dy
|
|
else:
|
|
ymin -= dy
|
|
ymax -= dy
|
|
|
|
elif self._button_pressed==3:
|
|
try:
|
|
dx=(lastx-event.x)/float(a.bbox.width())
|
|
dy=(lasty-event.y)/float(a.bbox.height())
|
|
dx,dy=format_deltas(event,dx,dy)
|
|
if a.get_aspect() != 'auto':
|
|
dx = 0.5*(dx + dy)
|
|
dy = dx
|
|
alphax = pow(10.0,dx)
|
|
alphay = pow(10.0,dy)
|
|
lastx, lasty = trans.inverse_xy_tup( (lastx, lasty) )
|
|
if a.get_xscale()=='log':
|
|
xmin = lastx*(xmin/lastx)**alphax
|
|
xmax = lastx*(xmax/lastx)**alphax
|
|
else:
|
|
xmin = lastx+alphax*(xmin-lastx)
|
|
xmax = lastx+alphax*(xmax-lastx)
|
|
if a.get_yscale()=='log':
|
|
ymin = lasty*(ymin/lasty)**alphay
|
|
ymax = lasty*(ymax/lasty)**alphay
|
|
else:
|
|
ymin = lasty+alphay*(ymin-lasty)
|
|
ymax = lasty+alphay*(ymax-lasty)
|
|
|
|
except OverflowError:
|
|
warnings.warn('Overflow while panning')
|
|
return
|
|
|
|
a.set_xlim(xmin, xmax)
|
|
a.set_ylim(ymin, ymax)
|
|
|
|
self.canvas.draw()
|
|
|
|
def _on_button_release(self, event):
|
|
'the release mouse button callback in pan/zoom mode'
|
|
self.canvas.mpl_disconnect(self._motion_notify)
|
|
if not self._xypress: return
|
|
self._xypress = None
|
|
self._button_pressed = None
|
|
self.canvas.draw()
|
|
|
|
|
|
class ZoomPlotMode (PlotMode):
|
|
def __init__(self, plot):
|
|
PlotMode.__init__(self, plot, 'zoom',
|
|
'Zoom to rectangle','zoom_to_rect')
|
|
self._selectors = {}
|
|
|
|
def activate(self):
|
|
for ax in self.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.canvas.draw()
|
|
self._selectors[rs] = ax
|
|
|
|
def deactivate(self):
|
|
self._mpl_disconnect_all()
|
|
self._selectors = {}
|
|
|
|
def _on_select(self, start, end):
|
|
ax = start.inaxes
|
|
|
|
ax.set_xlim((min(start.xdata, end.xdata), max(start.xdata, end.xdata)))
|
|
ax.set_ylim((min(start.ydata, end.ydata), max(start.ydata, end.ydata)))
|
|
self.canvas.draw()
|
|
|
|
|
|
class SelectPlotMode2 (PlotMode):
|
|
def __init__(self, plot):
|
|
PlotMode.__init__(self, plot, 'lassoselect',
|
|
'Select within lasso', 'lasso')
|
|
|
|
def activate(self):
|
|
self._button_press = self.canvas.mpl_connect(
|
|
'button_press_event', self._on_select)
|
|
|
|
def deactivate(self):
|
|
self._mpl_disconnect_all()
|
|
self.plot._lasso = None
|
|
|
|
def _on_select(self, event):
|
|
if event.inaxes is None:
|
|
logger.log('debug', 'Lasso select not in axes')
|
|
return
|
|
self.plot._lasso = Lasso(event.inaxes, (event.xdata, event.ydata), self.lasso_callback)
|
|
self.plot._lasso.line.set_linewidth(1)
|
|
self.plot._lasso.line.set_linestyle('--')
|
|
self._event = event
|
|
|
|
def lasso_callback(self, verts):
|
|
self.plot.lasso_select_callback(verts, self._event.key)
|
|
|
|
|
|
class SelectPlotMode (PlotMode):
|
|
def __init__(self, plot):
|
|
PlotMode.__init__(self, plot, 'select',
|
|
'Select within rectangle', 'select')
|
|
self._selectors = {}
|
|
|
|
def activate(self):
|
|
for ax in self.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.canvas.draw()
|
|
self._selectors[rs] = ax
|
|
|
|
def deactivate(self):
|
|
self._mpl_disconnect_all()
|
|
self._selectors = {}
|
|
|
|
def _on_select(self, start, end):
|
|
self.plot.rectangle_select_callback(start.xdata, start.ydata,
|
|
end.xdata, end.ydata, end.key)
|
|
|
|
|
|
class PlotToolbar(gtk.Toolbar):
|
|
|
|
def __init__(self, plot):
|
|
gtk.Toolbar.__init__(self)
|
|
self.plot = plot
|
|
self.canvas = plot.canvas
|
|
self._current_mode = None
|
|
self.tooltips = gtk.Tooltips()
|
|
|
|
## Maps toolbar buttons to PlotMode objects.
|
|
self._mode_buttons = {}
|
|
self.set_property('show-arrow', False)
|
|
self.canvas.connect('enter-notify-event', self.on_enter_notify)
|
|
self.show()
|
|
|
|
self.add_mode(DefaultPlotMode(self.plot))
|
|
self.add_mode(PanPlotMode(self.plot))
|
|
self.add_mode(ZoomPlotMode(self.plot))
|
|
self.add_mode(SelectPlotMode(self.plot))
|
|
self.add_mode(SelectPlotMode2(self.plot))
|
|
|
|
self.insert(gtk.SeparatorToolItem(), -1)
|
|
self.set_style(gtk.TOOLBAR_ICONS)
|
|
|
|
# Set up freeze button
|
|
btn = gtk.ToolButton(gtk.STOCK_SAVE)
|
|
|
|
#btn.set_icon_widget(image)
|
|
btn.connect('clicked', self._on_save_clicked)
|
|
self.insert(btn, -1)
|
|
self.save_button = btn
|
|
|
|
# Set up freeze button
|
|
btn = gtk.ToggleToolButton()
|
|
|
|
image = gtk.Image()
|
|
image.set_from_pixbuf(fluents.icon_factory.get('freeze'))
|
|
|
|
btn.set_icon_widget(image)
|
|
btn.connect('toggled', self._on_freeze_toggle)
|
|
self.insert(btn, -1)
|
|
self.freeze_button = btn
|
|
|
|
self.show_all()
|
|
|
|
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.name == 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._current_mode = self.get_mode_by_name(mode_name)
|
|
else:
|
|
logger.log('warning', 'No such mode: %s' % mode_name)
|
|
|
|
if self.get_button(mode_name) and \
|
|
not self.get_button(mode_name).get_active():
|
|
self.get_button(mode_name).set_active(True)
|
|
|
|
globals()['active_mode'] = 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 _on_save_clicked(self, button):
|
|
"""Pops up a file dialog saver, and saves current plot"""
|
|
dialog = gtk.FileChooserDialog('Save plot')
|
|
dialog.set_action(gtk.FILE_CHOOSER_ACTION_SAVE)
|
|
dialog.add_buttons(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_SAVE, gtk.RESPONSE_OK)
|
|
default_name = "fluents_image"
|
|
dialog.set_current_name("%s.png" % default_name)
|
|
retval = dialog.run()
|
|
if retval in [gtk.RESPONSE_CANCEL, gtk.RESPONSE_DELETE_EVENT]:
|
|
logger.log("debug", "Cancelled save plot")
|
|
elif retval == gtk.RESPONSE_OK:
|
|
logger.log("debug", "Saving plot as: %s" % dialog.get_filename())
|
|
self.canvas.print_figure(dialog.get_filename())
|
|
else:
|
|
print "unknown; ", retval
|
|
dialog.destroy()
|
|
|
|
def _on_freeze_toggle(self, button):
|
|
self.plot.set_frozen(button.get_active())
|
|
|
|
def on_enter_notify(self, widget, event):
|
|
self.set_mode(active_mode)
|
|
# need views (plots) to grab key-events
|
|
widget.grab_focus()
|
|
|
|
# Create a view-changed signal that should be emitted every time
|
|
# the active view changes.
|
|
gobject.signal_new('view-changed', MainView, gobject.SIGNAL_RUN_LAST,
|
|
gobject.TYPE_NONE,
|
|
(gobject.TYPE_PYOBJECT,))
|
|
|
|
# Create focus-changed signal
|
|
gobject.signal_new('focus-changed', ViewFrame, gobject.SIGNAL_RUN_LAST,
|
|
gobject.TYPE_NONE,
|
|
(gobject.TYPE_PYOBJECT, gobject.TYPE_BOOLEAN,))
|
|
|