laydi/system/selections.py

from system import logger, dataset

import pygtk
import gtk
import gtk.gdk
import gtk.glade
import gnome
import gnome.ui
import gobject

class SelectionTree(gtk.TreeView):
    
    def __init__(self):
        self.store = gtk.TreeStore(gobject.TYPE_STRING)
        gtk.TreeView.__init__(self, self.store)

        renderer = gtk.CellRendererText()
        sel_column = gtk.TreeViewColumn('Selection', renderer, text=0)
        self.insert_column(sel_column, 0)

        self.connect('row-activated', self.on_row_activated)

        self._identifier_list = None
        self._dim_list = {}

        # A mapping of selection names to selection lines.
        self._selections = {}

        self.set_headers_visible(True)
        self._current_dim = None

    def set_identifier_list(self, identifier_list):
        self._identifier_list = identifier_list

    def set_project(self, project):
        self.project = project
        project.add_selection_observer(self)
        project.add_dataset_observer(self)
        self.update_dims(project.dim_names)
        
    def selection_changed(self, selection):
        self.update_dims(selection.keys())
        self._update_current_dim()
        
    def dataset_changed(self):
        self.selection_changed(self.project.get_selection())

    def update_dims(self, dim_list):
        for dim in dim_list:
            if not self._dim_list.has_key(dim):
                d = self.store.insert_after(None, None, (dim,))            
                self._dim_list[dim] = d
                self.store.insert_before(d, None, ('current selection',))

    def on_row_activated(self, treeview, path, column):
        i = self.store.get_iter(path)
        p = self.store.iter_parent(i)
        self._current_dim = self.store.get_value(p, 0)
        self._update_current_dim()
        
    def _update_current_dim(self):
        if not self._current_dim:
            return
        id_list = self.project.get_selection()[self._current_dim]
        self._identifier_list.set_identifiers(id_list)

class IdentifierList(gtk.TreeView):

    def __init__(self):
        self.store = gtk.ListStore(gobject.TYPE_STRING)
        gtk.TreeView.__init__(self, self.store)

        renderer = gtk.CellRendererText()
        ids_column = gtk.TreeViewColumn('Identifiers', renderer, text=0)
        self.insert_column(ids_column, 0)

        self.set_headers_visible(True)

    def set_identifiers(self, identifiers):
        self.store.clear()
        for e in identifiers:
            self.store.append((e,))