Multiselection included. (Without input validation).

system/navigator.py

@@ -15,34 +15,41 @@ class NavigatorView (gtk.TreeView):
         
         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)
+
+        # Setting up TextRenderers etc
         self.connect('cursor_changed', self.cursor_changed_handler)
         self.connect('row_activated', self.row_activated_handler)
 
-        #pixrenderer  = gtk.CellRendererPixbuf()
         textrenderer = gtk.CellRendererText()
 
         self.object_col = gtk.TreeViewColumn('Object')
-        #self.object_col.pack_start(pixrenderer)
         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)
 
-        # drag'n'drop
-        self.enable_model_drag_source(gtk.gdk.BUTTON1_MASK,[("GTK_TREE_MODEL_ROW",gtk.TARGET_SAME_APP,7)], gtk.gdk.ACTION_LINK)
+        # 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)
+    
+        # receive MOVE drag'n'drops from itself
+        self.enable_model_drag_dest([("GTK_TREE_MODEL_ROW",gtk.TARGET_SAME_WIDGET,138)],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, iter     = treeselection.get_selected()
-        path            = model.get_path(iter)
-    
-        self.data_tree.drag_data_get(path, selection)
+        model, paths     = treeselection.get_selected_rows()
+        if paths:    
+            self.data_tree.drag_data_get(paths[0], selection)
 
 
     def add_project(self, project):
@@ -50,18 +57,44 @@ class NavigatorView (gtk.TreeView):
         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 cursor_changed_handler(self, widget):
-        selection = widget.get_selection()
-        model, tree_iter = selection.get_selected()
-        obj = self.data_tree.get_value(tree_iter, 2)
         
-        if isinstance(obj, dataset.Dataset):
+        
+
+        selection = widget.get_selection()
+        model, paths = selection.get_selected_rows()
+        
+        objs = [self.data_tree.get_value(self.data_tree.get_iter(path),2) for path in paths]
+        
+        if objs and isinstance(objs[0], dataset.Dataset):
             logger.log('debug', 'Selecting dataset')
-            self.project.set_current_data(obj)
-        else: 
-            t = type(obj)
-            logger.log('debug', 'Selected datatype was %s. Don\'t know what to do.' % t)
+            self.project.set_current_data(objs)
+
+
+        # current object hasn't been added to selection yet, so we're adding it in
+        cur_path, column = widget.get_cursor()
+
+        obj  = model.get_value(model.get_iter(cur_path),2)
+        objs = [model.get_value(model.get_iter(path), 2) for path in paths]
+
+        if (not obj in objs) and isinstance(obj, dataset.Dataset):
+            objs += obj
+
+        if objs: 
+            if isinstance(objs[0], dataset.Dataset):
+                logger.log('debug', 'Selecting dataset')
+                self.project.set_current_data(objs)
 
     # TreeView changed. Set correct focus and colours
     def row_changed_handler(self, treestore, pos, iter):
@@ -75,8 +108,8 @@ class NavigatorView (gtk.TreeView):
 
         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)