Updated GO workflow.

workflows/go_workflow.py

@@ -1,6 +1,6 @@
 import gtk
 from fluents import dataset, logger, plots, workflow, fluents
-#import geneontology
+import geneontology
 #import gostat
 from scipy import array, randn, log, ones
 import networkx
@@ -9,20 +9,59 @@ class GoTermView (gtk.Frame):
 
     def __init__(self):
         gtk.Frame.__init__(self)
-        self.set_label('GO Term')
+        tab = gtk.Table(2, 2, False)
+        self._table = tab
         
+        self._name = gtk.Label('')
+        tab.attach(gtk.Label('Name:'), 0, 1, 0, 1)
+        tab.attach(self._name, 1, 2, 0, 1)
+        
+        self.add(tab)
+        self.set_go_term(None)
+
+    def set_go_term(self, term):
+        if term:
+            self.set_label(term['id'])
+            self._name.set_text(term['name'])
+        else:
+            self.set_label('GO Term')
+            self._name.set_text('')
+
         
 class GeneOntologyTree (gtk.HPaned):
     
-    def __init__(self):
+    def __init__(self, network):
         gtk.HPaned.__init__(self)
-        self._tree_view = gtk.TreeView()
+    
+        treemodel = geneontology.get_go_treestore(network)
+        self._treemodel = treemodel
+        self._tree_view = gtk.TreeView(treemodel)
+
+        renderer = gtk.CellRendererText()
+        go_column = gtk.TreeViewColumn('GO ID', renderer, text=0)
+        self._tree_view.insert_column(go_column, 0)
+
+        renderer = gtk.CellRendererText()
+        go_column = gtk.TreeViewColumn('Name', renderer, text=1)
+        self._tree_view.insert_column(go_column, 1)
+
         self._desc_view = GoTermView()
 
-        self.add1(self._tree_view)
+        self._tree_view.connect('cursor-changed', self._on_cursor_changed)
+
+        scrolled_window = gtk.ScrolledWindow()
+        scrolled_window.add(self._tree_view)
+        self.add1(scrolled_window)
         self.add2(self._desc_view)
         self.show_all()
 
+    def _on_cursor_changed(self, tree):
+        path, col = self._tree_view.get_cursor()
+        current = self._treemodel.get_iter(path)
+        term = self._treemodel.get_value(current, 2)
+        self._desc_view.set_go_term(term)        
+        
+
 class GoWorkflow (workflow.Workflow):
 
     name = 'Gene Ontology'
@@ -44,7 +83,9 @@ class LoadGOFunction(workflow.Function):
         workflow.Function.__init__(self, 'load-go', 'Load Gene Ontology')
 
     def run(self):
-        browser = GeneOntologyTree()
+        global go
+        go = geneontology.read_default_go()
+        browser = GeneOntologyTree(go)
         label = gtk.Label('_Gene Ontology')
         label.set_use_underline(True)
         fluents.app['bottom_notebook'].append_page(browser, label)