[processing]do not rebuild thhe whole toolbox tree when filtering or updating

python/plugins/processing/ProcessingPlugin.py

@@ -59,7 +59,7 @@ class ProcessingPlugin:
         self.toolbox = ProcessingToolbox()
         interface.iface.addDockWidget(Qt.RightDockWidgetArea, self.toolbox)
         self.toolbox.hide()
-        Processing.addAlgListListener(self.toolbox)
+        #Processing.addAlgListListener(self.toolbox)
 
         self.menu = QMenu(interface.iface.mainWindow())
         self.menu.setTitle(QCoreApplication.translate('Processing',
@@ -142,7 +142,8 @@ class ProcessingPlugin:
         dlg.show()
         dlg.exec_()
         if dlg.update:
-            self.toolbox.updateTree()
+            Processing.updateAlgsList()
+            self.toolbox.updateProvider('model')
 
     def openResults(self):
         dlg = ResultsDialog()

python/plugins/processing/gui/ContextAction.py

@@ -31,3 +31,11 @@ class ContextAction:
     def setData(self, alg, toolbox):
         self.alg = alg
         self.toolbox = toolbox
+        
+    def updateToolbox(self):
+        '''
+        Updates the list of algorithms and then the toolbox. 
+        It only update the item corresponding to the provider of the algorithm. 
+        To be called after the action is executed, if needed
+        '''        
+        self.toolbox.updateProvider(self.alg.provider.getName())

python/plugins/processing/gui/CreateNewScriptAction.py

@@ -30,7 +30,6 @@ from processing.gui.ToolboxAction import ToolboxAction
 from processing.gui.ScriptEditorDialog import ScriptEditorDialog
 import processing.resources_rc
 
-
 class CreateNewScriptAction(ToolboxAction):
 
     SCRIPT_PYTHON = 0
@@ -55,5 +54,5 @@ class CreateNewScriptAction(ToolboxAction):
             dlg = ScriptEditorDialog(ScriptEditorDialog.SCRIPT_R, None)
         dlg.show()
         dlg.exec_()
-        if dlg.update:
-            self.toolbox.updateTree()
+        if dlg.update: 
+            self.toolbox.updateProvider('model')

python/plugins/processing/gui/ProcessingToolbox.py

@@ -64,7 +64,7 @@ class ProcessingToolbox(QDockWidget, Ui_ProcessingToolbox):
             self.modeComboBox.setCurrentIndex(1)
         self.modeComboBox.currentIndexChanged.connect(self.modeHasChanged)
 
-        self.searchBox.textChanged.connect(self.fillTree)
+        self.searchBox.textChanged.connect(self.textChanged)
         self.algorithmTree.customContextMenuRequested.connect(
                 self.showPopupMenu)
         self.algorithmTree.doubleClicked.connect(self.executeAlgorithm)
@@ -73,6 +73,33 @@ class ProcessingToolbox(QDockWidget, Ui_ProcessingToolbox):
             self.searchBox.setPlaceholderText(self.tr('Search...'))
 
         self.fillTree()
+        
+    def textChanged(self):
+        text = self.searchBox.text().strip(' ')
+        self._filterItem(self.algorithmTree.invisibleRootItem(), text)
+        if text:
+            self.algorithmTree.expandAll()
+        else:
+            self.algorithmTree.collapseAll()
+            self.algorithmTree.invisibleRootItem().child(0).setExpanded(True)
+    
+    def _filterItem(self, item, text):        
+        if (item.childCount() > 0):
+            show = False
+            for i in xrange(item.childCount()):                
+                child = item.child(i)
+                showChild = self._filterItem(child, text)
+                show = showChild or show            
+            item.setHidden(not show)            
+            return show
+        elif isinstance(item, TreeAlgorithmItem):                     
+            hide = bool(text) and (text not in item.text(0))            
+            item.setHidden(hide)            
+            return not hide
+        else:
+            item.setHidden(True)            
+            return False
+                
 
     def modeHasChanged(self):
         idx = self.modeComboBox.currentIndex()
@@ -85,11 +112,20 @@ class ProcessingToolbox(QDockWidget, Ui_ProcessingToolbox):
 
         self.fillTree()
 
-    def algsListHasChanged(self):
-        self.fillTree()
 
     def updateTree(self):
         Processing.updateAlgsList()
+        self.fillTree()
+        
+    def updateProvider(self, providerName, updateAlgsList = True):
+        if updateAlgsList:
+            Processing.updateAlgsList()
+        for i in xrange(self.algorithmTree.invisibleRootItem().childCount()):
+            child = self.algorithmTree.invisibleRootItem().child(i)
+            if isinstance(child, TreeProviderItem):
+                if child.providerName == providerName:
+                    child.refresh()
+                    break
 
     def showPopupMenu(self, point):
         item = self.algorithmTree.itemAt(point)
@@ -267,111 +303,24 @@ class ProcessingToolbox(QDockWidget, Ui_ProcessingToolbox):
             self.algorithmTree.addTopLevelItem(mainItem)
 
         for providerName in Processing.algs.keys():
-            groups = {}
-            provider = Processing.algs[providerName]
-            name = 'ACTIVATE_' + providerName.upper().replace(' ', '_')
-            if not ProcessingConfig.getSetting(name):
-                continue
             if providerName not in providersToExclude:
                 continue
-            algs = provider.values()
-
-            # add algorithms
-
-            for alg in algs:
-                if not alg.showInToolbox:
-                    continue
-                if text == '' or text.lower() in alg.name.lower():
-                    if alg.group in groups:
-                        groupItem = groups[alg.group]
-                    else:
-                        groupItem = QTreeWidgetItem()
-                        groupItem.setText(0, alg.group)
-                        groupItem.setToolTip(0, alg.group)
-                        groups[alg.group] = groupItem
-                    algItem = TreeAlgorithmItem(alg)
-                    groupItem.addChild(algItem)
-
-            actions = Processing.actions[providerName]
-            for action in actions:
-                if text == '' or text.lower() in action.name.lower():
-                    if action.group in groups:
-                        groupItem = groups[action.group]
-                    else:
-                        groupItem = QTreeWidgetItem()
-                        groupItem.setText(0, action.group)
-                        groups[action.group] = groupItem
-                    algItem = TreeActionItem(action)
-                    groupItem.addChild(algItem)
-
-            if len(groups) > 0:
-                providerItem = QTreeWidgetItem()
-                providerItem.setText(0, Processing.getProviderFromName(
-                        providerName).getDescription())
-                providerItem.setIcon(0, Processing.getProviderFromName(
-                        providerName).getIcon())
-                providerItem.setToolTip(0, providerItem.text(0))
-                for groupItem in groups.values():
-                    providerItem.addChild(groupItem)
-                self.algorithmTree.addTopLevelItem(providerItem)
-
-        if text != '':
-            self.algorithmTree.expandAll()
-
-    def fillTreeUsingProviders(self):
-        self.algorithmTree.clear()
-        text = unicode(self.searchBox.text())
-        for providerName in Processing.algs.keys():
-            groups = {}
-            count = 0
-            provider = Processing.algs[providerName]
             name = 'ACTIVATE_' + providerName.upper().replace(' ', '_')
             if not ProcessingConfig.getSetting(name):
                 continue
-            algs = provider.values()
-
-            # Add algorithms
-            for alg in algs:
-                if not alg.showInToolbox:
-                    continue
-                if text == '' or text.lower() in alg.name.lower():
-                    if alg.group in groups:
-                        groupItem = groups[alg.group]
-                    else:
-                        groupItem = QTreeWidgetItem()
-                        groupItem.setText(0, alg.group)
-                        groupItem.setToolTip(0, alg.group)
-                        groups[alg.group] = groupItem
-                    algItem = TreeAlgorithmItem(alg)
-                    groupItem.addChild(algItem)
-                    count += 1
-
-            actions = Processing.actions[providerName]
-            for action in actions:
-                if text == '' or text.lower() in action.name.lower():
-                    if action.group in groups:
-                        groupItem = groups[action.group]
-                    else:
-                        groupItem = QTreeWidgetItem()
-                        groupItem.setText(0, action.group)
-                        groups[action.group] = groupItem
-                    algItem = TreeActionItem(action)
-                    groupItem.addChild(algItem)
-
-            if len(groups) > 0:
-                providerItem = QTreeWidgetItem()
-                providerItem.setText(0, Processing.getProviderFromName(
-                        providerName).getDescription()
-                        + ' [' + str(count) + ' geoalgorithms]')
-                providerItem.setIcon(0, Processing.getProviderFromName(
-                        providerName).getIcon())
-                providerItem.setToolTip(0, providerItem.text(0))
-                for groupItem in groups.values():
-                    providerItem.addChild(groupItem)
-                self.algorithmTree.addTopLevelItem(providerItem)
-                providerItem.setExpanded(text != '')
-                for groupItem in groups.values():
-                    groupItem.setExpanded(text != '')
+            providerItem = TreeProviderItem(providerName)            
+            self.algorithmTree.addTopLevelItem(providerItem)
+            
+    def fillTreeUsingProviders(self):
+        self.algorithmTree.clear()
+        for providerName in Processing.algs.keys():
+            name = 'ACTIVATE_' + providerName.upper().replace(' ', '_')
+            if not ProcessingConfig.getSetting(name):
+                continue
+            providerItem = TreeProviderItem(providerName)
+            self.algorithmTree.addTopLevelItem(providerItem)
+            providerItem.setHidden(providerItem.childCount() == 0)
+            
 
 
 class TreeAlgorithmItem(QTreeWidgetItem):
@@ -399,3 +348,57 @@ class TreeActionItem(QTreeWidgetItem):
         self.action = action
         self.setText(0, action.name)
         self.setIcon(0, action.getIcon())
+        
+class TreeProviderItem(QTreeWidgetItem):
+
+    def __init__(self, providerName):
+        QTreeWidgetItem.__init__(self)
+        self.providerName = providerName                
+        self.provider = Processing.getProviderFromName(providerName)
+        self.setIcon(0, self.provider.getIcon())
+        self.populate()
+        
+    def refresh(self):
+        self.takeChildren()
+        self.populate()
+                
+    def populate(self):
+        groups = {}
+        count = 0
+        provider = Processing.algs[self.providerName]
+        algs = provider.values()
+
+        # Add algorithms
+        for alg in algs:
+            if not alg.showInToolbox:
+                continue
+            if alg.group in groups:
+                groupItem = groups[alg.group]
+            else:
+                groupItem = QTreeWidgetItem()
+                groupItem.setText(0, alg.group)
+                groupItem.setToolTip(0, alg.group)
+                groups[alg.group] = groupItem
+            algItem = TreeAlgorithmItem(alg)
+            groupItem.addChild(algItem)
+            count += 1
+
+        actions = Processing.actions[self.providerName]
+        for action in actions:           
+            if action.group in groups:
+                groupItem = groups[action.group]
+            else:
+                groupItem = QTreeWidgetItem()
+                groupItem.setText(0, action.group)
+                groups[action.group] = groupItem
+            algItem = TreeActionItem(action)
+            groupItem.addChild(algItem)        
+        
+        self.setText(0, self.provider.getDescription()
+                    + ' [' + str(count) + ' geoalgorithms]')
+        self.setToolTip(0, self.text(0))
+        for groupItem in groups.values():
+            self.addChild(groupItem)
+
+        
+

python/plugins/processing/modeler/CreateNewModelAction.py

@@ -30,7 +30,6 @@ from PyQt4 import QtGui
 from processing.gui.ToolboxAction import ToolboxAction
 from processing.modeler.ModelerDialog import ModelerDialog
 
-
 class CreateNewModelAction(ToolboxAction):
 
     def __init__(self):
@@ -44,5 +43,5 @@ class CreateNewModelAction(ToolboxAction):
         dlg = ModelerDialog()
         dlg.show()
         dlg.exec_()
-        if dlg.update:
-            self.toolbox.updateTree()
+        if dlg.update:            
+            self.toolbox.updateProvider('model')

python/plugins/processing/modeler/DeleteModelAction.py

@@ -46,4 +46,4 @@ class DeleteModelAction(ContextAction):
                 QtGui.QMessageBox.No)
         if reply == QtGui.QMessageBox.Yes:
             os.remove(self.alg.descriptionFile)
-            self.toolbox.updateTree()
+            self.updateToolbox()

python/plugins/processing/modeler/EditModelAction.py

@@ -29,7 +29,6 @@ from processing.gui.ContextAction import ContextAction
 from processing.modeler.ModelerAlgorithm import ModelerAlgorithm
 from processing.modeler.ModelerDialog import ModelerDialog
 
-
 class EditModelAction(ContextAction):
 
     def __init__(self):
@@ -42,4 +41,4 @@ class EditModelAction(ContextAction):
         dlg = ModelerDialog(self.alg.getCopy())
         dlg.exec_()
         if dlg.update:
-            self.toolbox.updateTree()
+            self.updateToolbox()

python/plugins/processing/modeler/SaveAsPythonScriptAction.py

@@ -34,7 +34,6 @@ from processing.modeler.ModelerAlgorithm import ModelerAlgorithm, \
 from processing.script.ScriptUtils import ScriptUtils
 from processing.parameters.ParameterMultipleInput import ParameterMultipleInput
 
-
 class SaveAsPythonScriptAction(ContextAction):
 
     def __init__(self):
@@ -56,8 +55,8 @@ class SaveAsPythonScriptAction(ContextAction):
                 fout.write(text)
                 fout.close()
                 if filename.replace('\\', '/').startswith(
-                        ScriptUtils.scriptsFolder().replace('\\', '/')):
-                    self.toolbox.updateTree()
+                        ScriptUtils.scriptsFolder().replace('\\', '/')):                                    
+                    self.toolbox.updateProvider('script')
             except:
                 QMessageBox.warning(self, self.tr('I/O error'),
                         self.tr('Unable to save edits. Reason:\n %s')