laydi/system/navigator.py

import gtk
import gobject
import plots
import time
from system import dataset, logger, plots, project, workflow

class NavigatorView (gtk.TreeView):
    def __init__(self, project, app):
        self.project = project
        self.app = app
        if project:
            self.data_tree = project.data_tree
        else:
            self.data_tree = None
        
        gtk.TreeView.__init__(self) 

        # various properties
        self.set_headers_visible(False)

        # Selection Mode
        self.get_selection().set_mode(gtk.SELECTION_MULTIPLE)
        self.get_selection().set_select_function(self.is_selectable)
        self.get_selection().connect('changed',self.selection_changed_handler)
        self._previous_selection = []

        # Setting up TextRenderers etc
        #        self.connect('cursor_changed', self.cursor_changed_handler)
        self.connect('row_activated', self.row_activated_handler)

        textrenderer = gtk.CellRendererText()

        self.object_col = gtk.TreeViewColumn('Object')
        self.object_col.pack_start(textrenderer)
        self.object_col.set_attributes(textrenderer, cell_background=3, foreground = 4, text=0)
        self.append_column(self.object_col)

        # send events to plots / itself
        self.enable_model_drag_source(gtk.gdk.BUTTON1_MASK,[("GTK_TREE_MODEL_ROW",gtk.TARGET_SAME_APP,7)], gtk.gdk.ACTION_LINK | gtk.gdk.ACTION_MOVE)
    
        self.connect("drag-data-get",self.slot_drag_data)

        logger.log('debug', 'Initializing navigator window.')

   
    # sets data for drag event.
    def slot_drag_data(self,treeview,context,selection,target_id,etime):
        treeselection   = treeview.get_selection()
        model, paths     = treeselection.get_selected_rows()
        if paths:    
            self.data_tree.drag_data_get(paths[0], selection)


    def add_project(self, project):
        self.project = project
        self.data_tree = project.data_tree
        self.set_model(project.data_tree)
        self.data_tree.connect('row-changed',self.row_changed_handler)
    
    def is_selectable(self,path):
        if self.data_tree:
            obj = self.data_tree.get_value(self.data_tree.get_iter(path),2)
            if not obj:
                return False
            if not isinstance(obj, dataset.Dataset):
                return False
        return True

    # selection changed, setting current_data ojbects
    def selection_changed_handler(self, selection):
                
        # update prev selection right away in case of multiple events
        model, paths = selection.get_selected_rows()
        tmp = self._previous_selection 
        self._previous_selection = paths
              
        # set timestamp on newly selected objects
        [self.data_tree.set_value(self.data_tree.get_iter(path),5,time.time()) for path in paths if path not in tmp]

        objs = [self.data_tree.get_iter(path) for path in paths]
        objs = [(self.data_tree.get_value(iter,5), self.data_tree.get_value(iter,2)) for iter in objs]
        objs.sort()
        objs = [obj for timestamp, obj in objs]
        # order dataset

        if objs and isinstance(objs[0], dataset.Dataset):
            logger.log('debug', 'Selecting dataset')
            self.project.set_current_data(objs)
        else:
            logger.log('debug', 'Deselecting dataset')
            self.project.set_current_data([])


    # TreeView changed. Set correct focus and colours
    def row_changed_handler(self, treestore, pos, iter):
        obj  = treestore.get_value(iter,2)
        type_= treestore.get_value(iter,1)

        if not (treestore.get_value(iter,2) or treestore.get_value(iter,1)):
            return

        self.expand_to_path(pos)        

        if isinstance(obj,dataset.Dataset):
            self.set_cursor(pos)
            self.grab_focus()
        
    def display_data_info(self, data):
        dims = zip(data.get_dim_names(), data.shape)
        
        dim_text = ", ".join(["%s (%d)" % dim for dim in dims])

        text = """<span weight="bold">Data:</span> %s

<span weight="bold">Dimensions:</span> %s""" % (data.get_name(), dim_text)

        d = gtk.MessageDialog(flags=(gtk.DIALOG_MODAL | gtk.DIALOG_DESTROY_WITH_PARENT),
                              buttons=gtk.BUTTONS_OK)
        d.set_markup(text)
        d.set_default_response(gtk.BUTTONS_OK)
        d.run()
        d.destroy()        

    def row_activated_handler(self, widget, path, column):
        tree_iter = self.data_tree.get_iter(path)
        obj = self.data_tree.get_value(tree_iter, 2)
        
        if isinstance(obj, plots.Plot):
            logger.log('debug', 'Activating plot')
            self.app.change_plot(obj)
        elif isinstance(obj, dataset.Dataset):
            self.display_data_info(obj)
        else:
            t = type(obj)
            logger.log('debug', 'Activated datatype was %s. Don\'t know what to do.' % t)