Added box and bar plots to the right-click menu of datasets.

This commit is contained in:
Einar Ryeng 2008-12-07 23:31:03 +00:00
parent 0a76e9c50f
commit fdf51c7c7f
2 changed files with 56 additions and 0 deletions

View File

@ -255,6 +255,16 @@ class NavigatorMenu(gtk.Menu):
self.plot_menu.append(self.plot_line_item)
self.plot_line_item.show()
self.plot_bar_item = gtk.MenuItem('Bar Plot')
self.plot_bar_item.connect('activate', self.on_plot_bar, navigator)
self.plot_menu.append(self.plot_bar_item)
self.plot_bar_item.show()
self.plot_box_item = gtk.MenuItem('Box Plot')
self.plot_box_item.connect('activate', self.on_plot_box, navigator)
self.plot_menu.append(self.plot_box_item)
self.plot_box_item.show()
self.plot_item = gtk.MenuItem('Plot')
self.append(self.plot_item)
self.plot_item.set_submenu(self.plot_menu)
@ -385,6 +395,28 @@ class NavigatorMenu(gtk.Menu):
plot.set_selection_listener(project.set_selection)
project._selection_observers.append(plot)
def on_plot_bar(self, item, navigator):
project = main.project
ds = self.dataset
dims = ds.get_dim_name()
ids = ds.get_identifiers(dims[1])
plot = plots.BarPlot(ds, name='Bar (%s)' % ds.get_name())
icon = laydi.icon_factory.get("line_plot")
project.data_tree_insert(self.tree_iter, 'Bar plot', plot, None, "black", icon)
plot.set_selection_listener(project.set_selection)
project._selection_observers.append(plot)
def on_plot_box(self, item, navigator):
project = main.project
ds = self.dataset
dims = ds.get_dim_name()
ids = ds.get_identifiers(dims[1])
plot = plots.BoxPlot(ds, name='Box (%s)' % ds.get_name())
icon = laydi.icon_factory.get("line_plot")
project.data_tree_insert(self.tree_iter, 'Box plot', plot, None, "black", icon)
plot.set_selection_listener(project.set_selection)
project._selection_observers.append(plot)
def on_transpose(self, item, navigator):
project = main.project
ds = self.dataset.transpose()

View File

@ -756,6 +756,30 @@ class HistogramPlot(Plot):
best_bin_size = bin_vec[scipy.argmin(cost)]
return best_bin_size
class BoxPlot(Plot):
"""Box plot.
"""
def __init__(self, ds, name="Box Plot"):
Plot.__init__(self, name)
self.axes = self.fig.add_subplot(111)
m = ds.asarray()
box_plot_lines = self.axes.boxplot(m)
self.axes.grid(False)
dim = ds.get_dim_name(1)
labels = ds.get_identifiers(dim, sorted=True)
self.axes.set_xticklabels(labels)
self.add(self.canvas)
self.canvas.show()
def set_current_selection(self, selection):
pass
class BarPlot(Plot):
"""Bar plot.