mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-26 00:02:08 -05:00
421 lines
16 KiB
Python
421 lines
16 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
ProcessingToolbox.py
|
|
---------------------
|
|
Date : August 2012
|
|
Copyright : (C) 2012 by Victor Olaya
|
|
Email : volayaf at gmail dot com
|
|
***************************************************************************
|
|
* *
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation; either version 2 of the License, or *
|
|
* (at your option) any later version. *
|
|
* *
|
|
***************************************************************************
|
|
"""
|
|
|
|
__author__ = 'Victor Olaya'
|
|
__date__ = 'August 2012'
|
|
__copyright__ = '(C) 2012, Victor Olaya'
|
|
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
|
|
__revision__ = '$Format:%H$'
|
|
|
|
import os
|
|
|
|
from PyQt4 import uic
|
|
from PyQt4.QtCore import Qt, QSettings, QCoreApplication
|
|
from PyQt4.QtGui import QMenu, QAction, QTreeWidgetItem
|
|
from qgis.utils import iface
|
|
|
|
from processing.modeler.ModelerUtils import ModelerUtils
|
|
from processing.core.Processing import Processing
|
|
from processing.core.ProcessingLog import ProcessingLog
|
|
from processing.core.ProcessingConfig import ProcessingConfig
|
|
from processing.core.GeoAlgorithm import GeoAlgorithm
|
|
from processing.gui.MessageDialog import MessageDialog
|
|
from processing.gui import AlgorithmClassification
|
|
from processing.gui.AlgorithmDialog import AlgorithmDialog
|
|
from processing.gui.BatchAlgorithmDialog import BatchAlgorithmDialog
|
|
from processing.gui.EditRenderingStylesDialog import EditRenderingStylesDialog
|
|
|
|
pluginPath = os.path.split(os.path.dirname(__file__))[0]
|
|
WIDGET, BASE = uic.loadUiType(
|
|
os.path.join(pluginPath, 'ui', 'ProcessingToolbox.ui'))
|
|
|
|
|
|
class ProcessingToolbox(BASE, WIDGET):
|
|
|
|
USE_CATEGORIES = '/Processing/UseSimplifiedInterface'
|
|
|
|
updateAlgList = True
|
|
|
|
def __init__(self):
|
|
super(ProcessingToolbox, self).__init__(None)
|
|
self.setupUi(self)
|
|
self.setAllowedAreas(Qt.LeftDockWidgetArea | Qt.RightDockWidgetArea)
|
|
|
|
self.modeComboBox.clear()
|
|
self.modeComboBox.addItems([self.tr('Simplified interface'),
|
|
self.tr('Advanced interface')])
|
|
settings = QSettings()
|
|
if not settings.contains(self.USE_CATEGORIES):
|
|
settings.setValue(self.USE_CATEGORIES, True)
|
|
useCategories = settings.value(self.USE_CATEGORIES, type=bool)
|
|
if useCategories:
|
|
self.modeComboBox.setCurrentIndex(0)
|
|
else:
|
|
self.modeComboBox.setCurrentIndex(1)
|
|
self.modeComboBox.currentIndexChanged.connect(self.modeHasChanged)
|
|
|
|
self.searchBox.textChanged.connect(self.textChanged)
|
|
self.algorithmTree.customContextMenuRequested.connect(
|
|
self.showPopupMenu)
|
|
self.algorithmTree.doubleClicked.connect(self.executeAlgorithm)
|
|
|
|
if hasattr(self.searchBox, 'setPlaceholderText'):
|
|
self.searchBox.setPlaceholderText(self.tr('Search...'))
|
|
|
|
self.fillTree()
|
|
|
|
def textChanged(self):
|
|
text = self.searchBox.text().strip(' ').lower()
|
|
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, TreeActionItem)):
|
|
hide = bool(text) and (text not in item.text(0).lower())
|
|
item.setHidden(hide)
|
|
return not hide
|
|
else:
|
|
item.setHidden(True)
|
|
return False
|
|
|
|
def modeHasChanged(self):
|
|
idx = self.modeComboBox.currentIndex()
|
|
settings = QSettings()
|
|
if idx == 0:
|
|
# Simplified
|
|
settings.setValue(self.USE_CATEGORIES, True)
|
|
else:
|
|
settings.setValue(self.USE_CATEGORIES, False)
|
|
|
|
self.fillTree()
|
|
|
|
def algsListHasChanged(self):
|
|
if self.updateAlgList:
|
|
self.fillTree()
|
|
|
|
def updateProvider(self, providerName, updateAlgsList=True):
|
|
if updateAlgsList:
|
|
self.updateAlgList = False
|
|
Processing.updateAlgsList()
|
|
self.updateAlgList = True
|
|
for i in xrange(self.algorithmTree.invisibleRootItem().childCount()):
|
|
child = self.algorithmTree.invisibleRootItem().child(i)
|
|
if isinstance(child, TreeProviderItem):
|
|
if child.providerName == providerName:
|
|
child.refresh()
|
|
# sort categories and items in categories
|
|
child.sortChildren(0, Qt.AscendingOrder)
|
|
for i in xrange(child.childCount()):
|
|
child.child(i).sortChildren(0, Qt.AscendingOrder)
|
|
break
|
|
self.addRecentAlgorithms(True)
|
|
|
|
def showPopupMenu(self, point):
|
|
item = self.algorithmTree.itemAt(point)
|
|
if isinstance(item, TreeAlgorithmItem):
|
|
alg = item.alg
|
|
popupmenu = QMenu()
|
|
executeAction = QAction(self.tr('Execute'), self.algorithmTree)
|
|
executeAction.triggered.connect(self.executeAlgorithm)
|
|
popupmenu.addAction(executeAction)
|
|
if alg.canRunInBatchMode and not alg.allowOnlyOpenedLayers:
|
|
executeBatchAction = QAction(
|
|
self.tr('Execute as batch process'),
|
|
self.algorithmTree)
|
|
executeBatchAction.triggered.connect(
|
|
self.executeAlgorithmAsBatchProcess)
|
|
popupmenu.addAction(executeBatchAction)
|
|
popupmenu.addSeparator()
|
|
editRenderingStylesAction = QAction(
|
|
self.tr('Edit rendering styles for outputs'),
|
|
self.algorithmTree)
|
|
editRenderingStylesAction.triggered.connect(
|
|
self.editRenderingStyles)
|
|
popupmenu.addAction(editRenderingStylesAction)
|
|
actions = Processing.contextMenuActions
|
|
if len(actions) > 0:
|
|
popupmenu.addSeparator()
|
|
for action in actions:
|
|
action.setData(alg, self)
|
|
if action.isEnabled():
|
|
contextMenuAction = QAction(action.name,
|
|
self.algorithmTree)
|
|
contextMenuAction.triggered.connect(action.execute)
|
|
popupmenu.addAction(contextMenuAction)
|
|
|
|
popupmenu.exec_(self.algorithmTree.mapToGlobal(point))
|
|
|
|
def editRenderingStyles(self):
|
|
item = self.algorithmTree.currentItem()
|
|
if isinstance(item, TreeAlgorithmItem):
|
|
alg = Processing.getAlgorithm(item.alg.commandLineName())
|
|
dlg = EditRenderingStylesDialog(alg)
|
|
dlg.exec_()
|
|
|
|
def executeAlgorithmAsBatchProcess(self):
|
|
item = self.algorithmTree.currentItem()
|
|
if isinstance(item, TreeAlgorithmItem):
|
|
alg = Processing.getAlgorithm(item.alg.commandLineName())
|
|
alg = alg.getCopy()
|
|
dlg = BatchAlgorithmDialog(alg)
|
|
dlg.show()
|
|
dlg.exec_()
|
|
|
|
def executeAlgorithm(self):
|
|
item = self.algorithmTree.currentItem()
|
|
if isinstance(item, TreeAlgorithmItem):
|
|
alg = Processing.getAlgorithm(item.alg.commandLineName())
|
|
message = alg.checkBeforeOpeningParametersDialog()
|
|
if message:
|
|
dlg = MessageDialog()
|
|
dlg.setTitle(self.tr('Missing dependency'))
|
|
dlg.setMessage(
|
|
self.tr('<h3>Missing dependency. This algorithm cannot '
|
|
'be run :-( </h3>\n%s') % message)
|
|
dlg.exec_()
|
|
return
|
|
alg = alg.getCopy()
|
|
dlg = alg.getCustomParametersDialog()
|
|
if not dlg:
|
|
dlg = AlgorithmDialog(alg)
|
|
canvas = iface.mapCanvas()
|
|
prevMapTool = canvas.mapTool()
|
|
dlg.show()
|
|
dlg.exec_()
|
|
if canvas.mapTool() != prevMapTool:
|
|
try:
|
|
canvas.mapTool().reset()
|
|
except:
|
|
pass
|
|
canvas.setMapTool(prevMapTool)
|
|
if dlg.executed:
|
|
showRecent = ProcessingConfig.getSetting(
|
|
ProcessingConfig.SHOW_RECENT_ALGORITHMS)
|
|
if showRecent:
|
|
self.addRecentAlgorithms(True)
|
|
if isinstance(item, TreeActionItem):
|
|
action = item.action
|
|
action.setData(self)
|
|
action.execute()
|
|
|
|
def fillTree(self):
|
|
settings = QSettings()
|
|
useCategories = settings.value(self.USE_CATEGORIES, type=bool)
|
|
if useCategories:
|
|
self.fillTreeUsingCategories()
|
|
else:
|
|
self.fillTreeUsingProviders()
|
|
self.algorithmTree.sortItems(0, Qt.AscendingOrder)
|
|
self.addRecentAlgorithms(False)
|
|
|
|
def addRecentAlgorithms(self, updating):
|
|
showRecent = ProcessingConfig.getSetting(
|
|
ProcessingConfig.SHOW_RECENT_ALGORITHMS)
|
|
if showRecent:
|
|
recent = ProcessingLog.getRecentAlgorithms()
|
|
if len(recent) != 0:
|
|
found = False
|
|
if updating:
|
|
recentItem = self.algorithmTree.topLevelItem(0)
|
|
treeWidget = recentItem.treeWidget()
|
|
treeWidget.takeTopLevelItem(
|
|
treeWidget.indexOfTopLevelItem(recentItem))
|
|
|
|
recentItem = QTreeWidgetItem()
|
|
recentItem.setText(0, self.tr('Recently used algorithms'))
|
|
for algname in recent:
|
|
alg = Processing.getAlgorithm(algname)
|
|
if alg is not None:
|
|
algItem = TreeAlgorithmItem(alg)
|
|
recentItem.addChild(algItem)
|
|
found = True
|
|
if found:
|
|
self.algorithmTree.insertTopLevelItem(0, recentItem)
|
|
recentItem.setExpanded(True)
|
|
|
|
self.algorithmTree.setWordWrap(True)
|
|
|
|
def fillTreeUsingCategories(self):
|
|
providersToExclude = ['model', 'script']
|
|
self.algorithmTree.clear()
|
|
text = unicode(self.searchBox.text())
|
|
groups = {}
|
|
for providerName in Processing.algs.keys():
|
|
provider = Processing.algs[providerName]
|
|
name = 'ACTIVATE_' + providerName.upper().replace(' ', '_')
|
|
if not ProcessingConfig.getSetting(name):
|
|
continue
|
|
if providerName in providersToExclude or \
|
|
len(ModelerUtils.providers[providerName].actions) != 0:
|
|
continue
|
|
algs = provider.values()
|
|
|
|
# add algorithms
|
|
|
|
for alg in algs:
|
|
if not alg.showInToolbox:
|
|
continue
|
|
altgroup, altsubgroup = AlgorithmClassification.getClassification(alg)
|
|
if altgroup is None:
|
|
continue
|
|
algName = AlgorithmClassification.getDisplayName(alg)
|
|
if text == '' or text.lower() in algName.lower():
|
|
if altgroup not in groups:
|
|
groups[altgroup] = {}
|
|
group = groups[altgroup]
|
|
if altsubgroup not in group:
|
|
groups[altgroup][altsubgroup] = []
|
|
subgroup = groups[altgroup][altsubgroup]
|
|
subgroup.append(alg)
|
|
|
|
if len(groups) > 0:
|
|
mainItem = QTreeWidgetItem()
|
|
mainItem.setText(0, 'Geoalgorithms')
|
|
mainItem.setIcon(0, GeoAlgorithm.getDefaultIcon())
|
|
mainItem.setToolTip(0, mainItem.text(0))
|
|
for (groupname, group) in groups.items():
|
|
groupItem = QTreeWidgetItem()
|
|
groupItem.setText(0, groupname)
|
|
groupItem.setIcon(0, GeoAlgorithm.getDefaultIcon())
|
|
groupItem.setToolTip(0, groupItem.text(0))
|
|
mainItem.addChild(groupItem)
|
|
for (subgroupname, subgroup) in group.items():
|
|
subgroupItem = QTreeWidgetItem()
|
|
subgroupItem.setText(0, subgroupname)
|
|
subgroupItem.setIcon(0, GeoAlgorithm.getDefaultIcon())
|
|
subgroupItem.setToolTip(0, subgroupItem.text(0))
|
|
groupItem.addChild(subgroupItem)
|
|
for alg in subgroup:
|
|
algItem = TreeAlgorithmItem(alg)
|
|
subgroupItem.addChild(algItem)
|
|
|
|
self.algorithmTree.addTopLevelItem(mainItem)
|
|
|
|
for providerName in Processing.algs.keys():
|
|
if providerName not in providersToExclude:
|
|
continue
|
|
name = 'ACTIVATE_' + providerName.upper().replace(' ', '_')
|
|
if not ProcessingConfig.getSetting(name):
|
|
continue
|
|
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):
|
|
|
|
def __init__(self, alg):
|
|
settings = QSettings()
|
|
useCategories = settings.value(ProcessingToolbox.USE_CATEGORIES,
|
|
type=bool)
|
|
QTreeWidgetItem.__init__(self)
|
|
self.alg = alg
|
|
icon = alg.getIcon()
|
|
if useCategories:
|
|
icon = GeoAlgorithm.getDefaultIcon()
|
|
name = AlgorithmClassification.getDisplayName(alg)
|
|
self.setIcon(0, icon)
|
|
self.setToolTip(0, name)
|
|
self.setText(0, name)
|
|
|
|
|
|
class TreeActionItem(QTreeWidgetItem):
|
|
|
|
def __init__(self, action):
|
|
QTreeWidgetItem.__init__(self)
|
|
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()
|
|
name = alg.i18n_group or alg.group
|
|
groupItem.setText(0, name)
|
|
groupItem.setToolTip(0, name)
|
|
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()
|
|
+ QCoreApplication.translate("TreeProviderItem", " [{0} geoalgorithms]").format(count))
|
|
self.setToolTip(0, self.text(0))
|
|
for groupItem in groups.values():
|
|
self.addChild(groupItem)
|