290 lines
8.8 KiB
Python
290 lines
8.8 KiB
Python
|
|
import pygtk
|
|
import gtk
|
|
import matplotlib
|
|
import project
|
|
from matplotlib.backends.backend_gtkagg import FigureCanvasGTKAgg as FigureCanvas
|
|
from matplotlib.axes import Subplot
|
|
from matplotlib.figure import Figure
|
|
from matplotlib.numerix import arange, sin, pi
|
|
from matplotlib.backends.backend_gtk import NavigationToolbar2GTK as NavigationToolbar
|
|
|
|
from matplotlib.widgets import RectangleSelector
|
|
|
|
import scipy
|
|
import logger
|
|
import project
|
|
|
|
class MainView (gtk.Notebook):
|
|
def __init__(self):
|
|
gtk.Notebook.__init__(self)
|
|
self.set_show_tabs(False)
|
|
self.set_show_border(False)
|
|
|
|
# Add a multiple pane view and a single pane view.
|
|
self.small_view = SmallView()
|
|
self.small_view.show()
|
|
self.large_view = LargeView()
|
|
self.large_view.show()
|
|
|
|
self.append_page(self.small_view)
|
|
self.append_page(self.large_view)
|
|
self.set_current_page(0)
|
|
|
|
# Set current view to the upper left view.
|
|
self.current_view = self.small_view.get_view(0, 0)
|
|
self.current_view.mark_active(True)
|
|
|
|
def goto_small(self):
|
|
if self.get_current_page() == 0:
|
|
return None
|
|
self.set_current_page(0)
|
|
view = self.large_view.remove_child()
|
|
self.small_view.return_current(view)
|
|
|
|
def goto_large(self):
|
|
if self.get_current_page() == 1:
|
|
return None
|
|
self.set_current_page(1)
|
|
view = self.small_view.borrow_current()
|
|
self.large_view.set_child(view)
|
|
|
|
def show(self):
|
|
gtk.Notebook.show(self)
|
|
|
|
def insert_view(self, view):
|
|
self.small_view.insert_view(view)
|
|
|
|
class SmallView (gtk.Table):
|
|
def __init__(self):
|
|
gtk.Table.__init__(self, 2, 2, True)
|
|
|
|
self.child_views = [[EmptyView(), EmptyView()],
|
|
[EmptyView(), EmptyView()]]
|
|
self.cols = 2
|
|
self.rows = 2
|
|
|
|
self.active_x = 0
|
|
self.active_y = 0
|
|
|
|
self.set_row_spacings(3)
|
|
self.set_col_spacings(3)
|
|
|
|
for x in range(self.cols):
|
|
for y in range(self.rows):
|
|
child = self.child_views[x][y]
|
|
child.parent_signalling = child.connect('button_press_event', self.__view_button_event__)
|
|
self.attach(child, x, x+1, y, y+1)
|
|
|
|
def set_child(self, child, col, row):
|
|
cur_widget = self.child_views[col][row]
|
|
cur_widget.disconnect(cur_widget.parent_signalling)
|
|
self.remove(cur_widget)
|
|
self.attach(child, col, col+1, row, row+1)
|
|
child.parent_signalling = child.connect('button_press_event', self.__view_button_event__)
|
|
self.child_views[col][row] = child
|
|
child.show()
|
|
|
|
def borrow_current(self):
|
|
self.borrowed = self.child_views[self.active_x][self.active_y]
|
|
self.remove_child(self.active_x, self.active_y)
|
|
return self.borrowed
|
|
|
|
def return_current(self, view):
|
|
self.set_child(view, self.active_x, self.active_y)
|
|
|
|
def remove_child(self, col, row):
|
|
self.remove(self.child_views[col][row])
|
|
self.attach(EmptyView(), col, col+1, row, row+1)
|
|
|
|
def insert_view(self, child):
|
|
if not self.find_child(child):
|
|
self.set_child(child, self.active_x, self.active_y)
|
|
|
|
def show(self):
|
|
for x in self.child_views:
|
|
for y in x:
|
|
y.show()
|
|
gtk.Table.show(self)
|
|
|
|
def hide(self):
|
|
for x in self.child_views:
|
|
for y in x:
|
|
y.hide()
|
|
gtk.Table.hide(self)
|
|
|
|
def set_active(self, x, y):
|
|
old_focus = self.child_views[self.active_x][self.active_y]
|
|
new_focus = self.child_views[x][y]
|
|
old_focus.mark_active(False)
|
|
new_focus.mark_active(True)
|
|
self.active_x = x
|
|
self.active_y = y
|
|
|
|
def find_child(self, child):
|
|
for i, row in enumerate(self.child_views):
|
|
for j, v in enumerate(row):
|
|
if v == child:
|
|
return (i, j)
|
|
return None
|
|
|
|
def __view_button_event__(self, view, *rest):
|
|
loc = self.find_child(view)
|
|
if loc:
|
|
self.set_active(loc[0], loc[1])
|
|
|
|
def get_view(self, x, y):
|
|
return self.child_views[x][y]
|
|
|
|
class LargeView (gtk.Frame):
|
|
def __init__(self):
|
|
gtk.Frame.__init__(self)
|
|
self.child_view = EmptyView()
|
|
self.add(self.child_view)
|
|
|
|
def set_child(self, child):
|
|
self.remove(self.child_view)
|
|
self.child_view.hide()
|
|
self.add(child)
|
|
self.child_view = child
|
|
child.show()
|
|
|
|
def hide(self):
|
|
self.child_view.hide()
|
|
gtk.Frame.hide(self)
|
|
|
|
def show(self):
|
|
self.child_view.show()
|
|
gtk.Frame.show(self)
|
|
|
|
def remove_child(self):
|
|
child = self.child_view
|
|
child.hide()
|
|
self.remove(child)
|
|
return child
|
|
|
|
class Plot (gtk.Frame):
|
|
|
|
def __init__(self):
|
|
gtk.Frame.__init__(self)
|
|
self.mark_active(False)
|
|
self.connect('button_press_event', self.on_button_press)
|
|
self.sel_obj = None
|
|
project.c_p.attach(self)
|
|
|
|
def on_button_press(self, *rest):
|
|
logger.log('debug', 'button pressed in plot')
|
|
self.mark_active(True)
|
|
|
|
def get_toolbar(self, window):
|
|
return None
|
|
|
|
def mark_active(self, active):
|
|
if active:
|
|
self.set_shadow_type(gtk.SHADOW_IN)
|
|
else:
|
|
self.set_shadow_type(gtk.SHADOW_OUT)
|
|
|
|
def update_selection(self,project):
|
|
self.selection_object = project.get_selection()
|
|
|
|
class EmptyView (Plot):
|
|
def __init__(self):
|
|
Plot.__init__(self)
|
|
|
|
label = gtk.Label('No view')
|
|
ebox = gtk.EventBox()
|
|
ebox.add(label)
|
|
self.add(ebox)
|
|
label.connect('button_press_event', self.on_button_press)
|
|
|
|
self.label = label
|
|
self.ebox = ebox
|
|
self.show()
|
|
|
|
def show(self):
|
|
self.ebox.show()
|
|
self.label.show()
|
|
Plot.show(self)
|
|
|
|
def hide(self):
|
|
self.label.hide()
|
|
self.ebox.hide()
|
|
Plot.hide(self)
|
|
|
|
class SinePlot (Plot):
|
|
|
|
def __init__(self):
|
|
Plot.__init__(self)
|
|
fig = Figure(figsize=(5,4), dpi=72)
|
|
ax = fig.add_subplot(111)
|
|
t = arange(0.0,3.0,0.01)
|
|
s = sin(2*pi*t)
|
|
ax.plot(t,s)
|
|
self.canvas = FigureCanvas(fig)
|
|
self.add(self.canvas)
|
|
self.canvas.show()
|
|
|
|
def get_toolbar(self, window):
|
|
self.toolbar = NavigationToolbar(self.canvas, window)
|
|
self.toolbar.set_property('show-arrow', False)
|
|
return self.toolbar
|
|
|
|
class ScatterPlot (Plot):
|
|
def __init__(self):
|
|
Plot.__init__(self)
|
|
fig = Figure(figsize=(5,4), dpi=72)
|
|
self.ax = ax = fig.add_subplot(111)
|
|
self.x_dataset = project.c_p.datasets[0]
|
|
x = self.x_dataset._data
|
|
self.xaxis_data = xaxis_data = x[:,0]
|
|
self.yaxis_data = yaxis_data = x[:,1]
|
|
ax.plot(xaxis_data,yaxis_data,'og')
|
|
self.canvas = FigureCanvas(fig)
|
|
self.add(self.canvas)
|
|
rectprops = dict(facecolor='blue', edgecolor = 'black',
|
|
alpha=0.2, fill=True)
|
|
sel = RectangleSelector(ax, self.line_select_callback,
|
|
drawtype='box',useblit=True,rectprops=rectprops)
|
|
|
|
self.canvas.show()
|
|
|
|
def line_select_callback(self,event1, event2):
|
|
'event1 and event2 are the press and release events'
|
|
x1, y1 = event1.xdata, event1.ydata
|
|
x2, y2 = event2.xdata, event2.ydata
|
|
logger.log('debug', "(%3.2f, %3.2f) --> (%3.2f, %3.2f)"%(x1,y1,x2,y2))
|
|
logger.log('debug',"The button you used were:%s, %s "%(event1.button, event2.button))
|
|
# get all points within x1, y1, x2, y2
|
|
ydata = self.yaxis_data
|
|
xdata = self.xaxis_data
|
|
if x1>x2:
|
|
if y1<y2:
|
|
index =scipy.nonzero((xdata<x1) & (xdata>x2) & (ydata>y1) & (ydata<y2))
|
|
else:
|
|
index =scipy.nonzero((xdata<x1) & (xdata>x2) & (ydata<y1) & (ydata>y2))
|
|
else:
|
|
if y1<y2:
|
|
index =scipy.nonzero((xdata>x2) & (xdata<x1) & (ydata>y1) & (ydata<y2))
|
|
else:
|
|
index =scipy.nonzero((xdata>x2) & (xdata<x1) & (ydata<y1) & (ydata>y2))
|
|
|
|
if len(index)==0:
|
|
logger.log('debug','No points selected!')
|
|
else:
|
|
logger.log('debug','Selected:\n%s'%index)
|
|
ids = self.x_dataset.extract_id_from_index('samples',index)
|
|
logger.log('debug','Selected identifiers:\n%s'%ids)
|
|
xdata_new = scipy.take(xdata,index)
|
|
ydata_new = scipy.take(ydata,index)
|
|
# self.ax.plot(xdata,ydata,'og')
|
|
self.ax.plot(xdata_new,ydata_new,'or')
|
|
self.canvas.draw()
|
|
|
|
def get_toolbar(self, window):
|
|
self.toolbar = NavigationToolbar(self.canvas, window)
|
|
self.toolbar.set_property('show-arrow', False)
|
|
return self.toolbar
|
|
|
|
|