QGIS/python/plugins/processing/modeler/ModelerParametersDialog.py

797 lines
34 KiB
Python
Raw Normal View History

2012-10-04 19:33:47 +02:00
# -*- coding: utf-8 -*-
"""
***************************************************************************
ModelerParametersDialog.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. *
* *
***************************************************************************
"""
2016-09-21 18:24:26 +02:00
from builtins import str
from builtins import range
2012-10-04 19:33:47 +02:00
__author__ = 'Victor Olaya'
__date__ = 'August 2012'
__copyright__ = '(C) 2012, Victor Olaya'
2012-10-04 19:33:47 +02:00
# This will get replaced with a git SHA1 when you do a git archive
2012-10-04 19:33:47 +02:00
__revision__ = '$Format:%H$'
2016-04-22 10:38:48 +02:00
from qgis.PyQt.QtCore import Qt, QUrl, QMetaObject
from qgis.PyQt.QtWidgets import (QDialog, QDialogButtonBox, QLabel, QLineEdit,
QFrame, QPushButton, QSizePolicy, QVBoxLayout,
QHBoxLayout, QTabWidget, QWidget, QScrollArea,
QComboBox, QTableWidgetItem, QMessageBox,
QTextBrowser)
from qgis.PyQt.QtNetwork import QNetworkRequest, QNetworkReply
from qgis.core import QgsNetworkAccessManager
from qgis.gui import QgsMessageBar
2013-08-12 20:44:27 +02:00
from processing.gui.CrsSelectionPanel import CrsSelectionPanel
from processing.gui.MultipleInputPanel import MultipleInputPanel
from processing.gui.FixedTablePanel import FixedTablePanel
from processing.gui.RangePanel import RangePanel
from processing.gui.GeometryPredicateSelectionPanel import \
GeometryPredicateSelectionPanel
from processing.core.parameters import (ParameterExtent,
ParameterRaster,
ParameterVector,
ParameterBoolean,
ParameterTable,
ParameterFixedTable,
ParameterMultipleInput,
ParameterSelection,
ParameterRange,
ParameterNumber,
ParameterString,
ParameterCrs,
ParameterTableField,
ParameterTableMultipleField,
ParameterFile,
ParameterPoint,
ParameterGeometryPredicate)
from processing.core.outputs import (OutputRaster,
OutputVector,
OutputTable,
OutputHTML,
OutputFile,
OutputDirectory,
OutputNumber,
OutputString,
OutputExtent,
OutputCrs)
from processing.modeler.ModelerAlgorithm import (ValueFromInput,
ValueFromOutput,
Algorithm,
ModelerOutput)
from processing.modeler.MultilineTextPanel import MultilineTextPanel
from processing.tools import dataobjects
2012-09-15 18:25:25 +03:00
class ModelerParametersDialog(QDialog):
2012-09-15 18:25:25 +03:00
ENTER_NAME = '[Enter name if this is a final result]'
NOT_SELECTED = '[Not selected]'
USE_MIN_COVERING_EXTENT = '[Use min covering extent]'
2012-09-15 18:25:25 +03:00
def __init__(self, alg, model, algName=None):
QDialog.__init__(self)
2012-09-15 18:25:25 +03:00
self.setModal(True)
# The algorithm to define in this dialog. It is an instance of GeoAlgorithm
self._alg = alg
# The resulting algorithm after the user clicks on OK. it is an instance of the container Algorithm class
self.alg = None
# The model this algorithm is going to be added to
2012-09-15 18:25:25 +03:00
self.model = model
# The name of the algorithm in the model, in case we are editing it and not defining it for the first time
self._algName = algName
2012-09-15 18:25:25 +03:00
self.setupUi()
self.params = None
def setupUi(self):
self.labels = {}
self.widgets = {}
self.checkBoxes = {}
self.showAdvanced = False
2012-09-15 18:25:25 +03:00
self.valueItems = {}
self.dependentItems = {}
self.resize(650, 450)
self.buttonBox = QDialogButtonBox()
self.buttonBox.setOrientation(Qt.Horizontal)
self.buttonBox.setStandardButtons(QDialogButtonBox.Cancel
| QDialogButtonBox.Ok)
tooltips = self._alg.getParameterDescriptions()
self.setSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Expanding)
self.verticalLayout = QVBoxLayout()
self.verticalLayout.setSpacing(5)
self.verticalLayout.setMargin(20)
2014-07-02 07:46:03 +02:00
self.bar = QgsMessageBar()
self.bar.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Fixed)
self.verticalLayout.addWidget(self.bar)
hLayout = QHBoxLayout()
hLayout.setSpacing(5)
hLayout.setMargin(0)
descriptionLabel = QLabel(self.tr("Description"))
self.descriptionBox = QLineEdit()
self.descriptionBox.setText(self._alg.name)
hLayout.addWidget(descriptionLabel)
hLayout.addWidget(self.descriptionBox)
self.verticalLayout.addLayout(hLayout)
line = QFrame()
line.setFrameShape(QFrame.HLine)
line.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(line)
2014-07-02 07:46:03 +02:00
for param in self._alg.parameters:
if param.isAdvanced:
self.advancedButton = QPushButton()
self.advancedButton.setText(self.tr('Show advanced parameters'))
self.advancedButton.clicked.connect(
self.showAdvancedParametersClicked)
advancedButtonHLayout = QHBoxLayout()
advancedButtonHLayout.addWidget(self.advancedButton)
advancedButtonHLayout.addStretch()
self.verticalLayout.addLayout(advancedButtonHLayout)
break
for param in self._alg.parameters:
if param.hidden:
continue
desc = param.description
if isinstance(param, ParameterExtent):
desc += self.tr('(xmin, xmax, ymin, ymax)')
if isinstance(param, ParameterPoint):
desc += self.tr('(x, y)')
label = QLabel(desc)
self.labels[param.name] = label
widget = self.getWidgetFromParameter(param)
self.valueItems[param.name] = widget
2016-09-21 18:24:26 +02:00
if param.name in list(tooltips.keys()):
tooltip = tooltips[param.name]
else:
tooltip = param.description
label.setToolTip(tooltip)
widget.setToolTip(tooltip)
if param.isAdvanced:
label.setVisible(self.showAdvanced)
widget.setVisible(self.showAdvanced)
self.widgets[param.name] = widget
self.verticalLayout.addWidget(label)
self.verticalLayout.addWidget(widget)
for output in self._alg.outputs:
if output.hidden:
continue
if isinstance(output, (OutputRaster, OutputVector, OutputTable,
2015-12-20 10:12:18 +01:00
OutputHTML, OutputFile, OutputDirectory)):
label = QLabel(output.description + '<'
+ output.__class__.__name__ + '>')
item = QLineEdit()
if hasattr(item, 'setPlaceholderText'):
item.setPlaceholderText(ModelerParametersDialog.ENTER_NAME)
self.verticalLayout.addWidget(label)
self.verticalLayout.addWidget(item)
self.valueItems[output.name] = item
2013-02-07 01:09:39 +01:00
label = QLabel(' ')
2013-02-03 10:26:43 +01:00
self.verticalLayout.addWidget(label)
label = QLabel(self.tr('Parent algorithms'))
2013-02-07 01:09:39 +01:00
self.dependenciesPanel = self.getDependenciesPanel()
2013-02-03 10:26:43 +01:00
self.verticalLayout.addWidget(label)
self.verticalLayout.addWidget(self.dependenciesPanel)
2013-02-07 01:09:39 +01:00
self.verticalLayout.addStretch(1000)
self.setLayout(self.verticalLayout)
2012-09-15 18:25:25 +03:00
self.setPreviousValues()
self.setWindowTitle(self._alg.name)
self.verticalLayout2 = QVBoxLayout()
self.verticalLayout2.setSpacing(2)
self.verticalLayout2.setMargin(0)
self.tabWidget = QTabWidget()
2012-12-10 00:12:07 +01:00
self.tabWidget.setMinimumWidth(300)
self.paramPanel = QWidget()
self.paramPanel.setLayout(self.verticalLayout)
self.scrollArea = QScrollArea()
self.scrollArea.setWidget(self.paramPanel)
self.scrollArea.setWidgetResizable(True)
self.tabWidget.addTab(self.scrollArea, self.tr('Parameters'))
self.txtHelp = QTextBrowser()
2014-04-19 22:04:24 +02:00
2012-09-15 18:25:25 +03:00
html = None
isText, algHelp = self._alg.help()
if algHelp is not None:
algHelp = algHelp if isText else QUrl(algHelp)
try:
if isText:
self.txtHelp.setHtml(algHelp)
else:
html = self.tr('<p>Downloading algorithm help... Please wait.</p>')
self.txtHelp.setHtml(html)
self.reply = QgsNetworkAccessManager.instance().get(QNetworkRequest(algHelp))
self.reply.finished.connect(self.requestFinished)
except:
self.txtHelp.setHtml(self.tr('<h2>No help available for this algorithm</h2>'))
self.tabWidget.addTab(self.txtHelp, 'Help')
self.verticalLayout2.addWidget(self.tabWidget)
self.verticalLayout2.addWidget(self.buttonBox)
self.setLayout(self.verticalLayout2)
self.buttonBox.accepted.connect(self.okPressed)
self.buttonBox.rejected.connect(self.cancelPressed)
QMetaObject.connectSlotsByName(self)
2012-09-15 18:25:25 +03:00
def requestFinished(self):
"""Change the webview HTML content"""
reply = self.sender()
if reply.error() != QNetworkReply.NoError:
html = self.tr('<h2>No help available for this algorithm</h2><p>{}</p>'.format(reply.errorString()))
else:
2016-09-21 18:24:26 +02:00
html = str(reply.readAll())
reply.deleteLater()
self.txtHelp.setHtml(html)
2013-02-07 01:09:39 +01:00
def getAvailableDependencies(self):
if self._algName is None:
2013-02-03 10:26:43 +01:00
dependent = []
else:
dependent = self.model.getDependentAlgorithms(self._algName)
2013-02-07 01:09:39 +01:00
opts = []
2016-09-21 18:24:26 +02:00
for alg in list(self.model.algs.values()):
if alg.name not in dependent:
opts.append(alg)
2013-02-03 10:26:43 +01:00
return opts
def getDependenciesPanel(self):
return MultipleInputPanel([alg.algorithm.name for alg in self.getAvailableDependencies()])
2013-02-07 01:09:39 +01:00
def showAdvancedParametersClicked(self):
self.showAdvanced = not self.showAdvanced
if self.showAdvanced:
self.advancedButton.setText(self.tr('Hide advanced parameters'))
else:
self.advancedButton.setText(self.tr('Show advanced parameters'))
for param in self._alg.parameters:
if param.isAdvanced:
self.labels[param.name].setVisible(self.showAdvanced)
self.widgets[param.name].setVisible(self.showAdvanced)
2012-12-10 00:12:07 +01:00
def getAvailableValuesOfType(self, paramType, outType=None, dataType=None):
values = []
inputs = self.model.inputs
2016-09-21 18:24:26 +02:00
for i in list(inputs.values()):
param = i.param
if isinstance(param, paramType):
if dataType is not None and param.datatype in dataType:
values.append(ValueFromInput(param.name))
else:
values.append(ValueFromInput(param.name))
if outType is None:
return values
if self._algName is None:
2012-09-15 18:25:25 +03:00
dependent = []
else:
dependent = self.model.getDependentAlgorithms(self._algName)
2016-09-21 18:24:26 +02:00
for alg in list(self.model.algs.values()):
if alg.name not in dependent:
for out in alg.algorithm.outputs:
if isinstance(out, outType):
if dataType is not None and out.datatype in dataType:
values.append(ValueFromOutput(alg.name, out.name))
else:
values.append(ValueFromOutput(alg.name, out.name))
return values
2014-07-02 07:46:03 +02:00
def resolveValueDescription(self, value):
if isinstance(value, ValueFromInput):
return self.model.inputs[value.name].param.description
2012-09-15 18:25:25 +03:00
else:
alg = self.model.algs[value.alg]
return self.tr("'%s' from algorithm '%s'") % (alg.algorithm.getOutputFromName(value.output).description, alg.description)
2012-09-15 18:25:25 +03:00
def getWidgetFromParameter(self, param):
if isinstance(param, ParameterRaster):
item = QComboBox()
layers = self.getAvailableValuesOfType(ParameterRaster, OutputRaster)
if param.optional:
2012-09-15 18:25:25 +03:00
item.addItem(self.NOT_SELECTED, None)
for layer in layers:
item.addItem(self.resolveValueDescription(layer), layer)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterVector):
item = QComboBox()
layers = self.getAvailableValuesOfType(ParameterVector, OutputVector)
if param.optional:
2012-09-15 18:25:25 +03:00
item.addItem(self.NOT_SELECTED, None)
for layer in layers:
item.addItem(self.resolveValueDescription(layer), layer)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterTable):
item = QComboBox()
tables = self.getAvailableValuesOfType(ParameterTable, OutputTable)
layers = self.getAvailableValuesOfType(ParameterVector, OutputVector)
if param.optional:
2012-09-15 18:25:25 +03:00
item.addItem(self.NOT_SELECTED, None)
for table in tables:
item.addItem(self.resolveValueDescription(table), table)
2012-09-15 18:25:25 +03:00
for layer in layers:
item.addItem(self.resolveValueDescription(layer), layer)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterBoolean):
item = QComboBox()
item.addItem('Yes')
item.addItem('No')
bools = self.getAvailableValuesOfType(ParameterBoolean, None)
2012-09-15 18:25:25 +03:00
for b in bools:
item.addItem(self.resolveValueDescription(b), b)
if param.default:
item.setCurrentIndex(0)
else:
item.setCurrentIndex(1)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterSelection):
item = QComboBox()
2012-09-15 18:25:25 +03:00
item.addItems(param.options)
item.setCurrentIndex(param.default or 0)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterFixedTable):
item = FixedTablePanel(param)
elif isinstance(param, ParameterRange):
item = RangePanel(param)
elif isinstance(param, ParameterMultipleInput):
if param.datatype == dataobjects.TYPE_VECTOR_ANY:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector)
elif param.datatype == dataobjects.TYPE_VECTOR_POINT:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector, [dataobjects.TYPE_VECTOR_POINT, dataobjects.TYPE_VECTOR_ANY])
elif param.datatype == dataobjects.TYPE_VECTOR_LINE:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector, [dataobjects.TYPE_VECTOR_LINE, dataobjects.TYPE_VECTOR_ANY])
elif param.datatype == dataobjects.TYPE_VECTOR_POLYGON:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector, [dataobjects.TYPE_VECTOR_POLYGON, dataobjects.TYPE_VECTOR_ANY])
elif param.datatype == dataobjects.TYPE_RASTER:
options = self.getAvailableValuesOfType(ParameterRaster, OutputRaster)
else:
options = self.getAvailableValuesOfType(ParameterFile, OutputFile)
2012-09-15 18:25:25 +03:00
opts = []
for opt in options:
opts.append(self.resolveValueDescription(opt))
2012-09-15 18:25:25 +03:00
item = MultipleInputPanel(opts)
elif isinstance(param, ParameterString):
strings = self.getAvailableValuesOfType(ParameterString, OutputString)
options = [(self.resolveValueDescription(s), s) for s in strings]
2013-02-03 10:26:43 +01:00
if param.multiline:
item = MultilineTextPanel(options)
2016-09-21 18:24:26 +02:00
item.setText(str(param.default or ""))
2013-02-03 10:26:43 +01:00
else:
item = QComboBox()
2013-02-07 01:09:39 +01:00
item.setEditable(True)
for desc, val in options:
item.addItem(desc, val)
2016-09-21 18:24:26 +02:00
item.setEditText(str(param.default or ""))
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterTableField):
item = QComboBox()
2012-09-15 18:25:25 +03:00
item.setEditable(True)
fields = self.getAvailableValuesOfType(ParameterTableField, None)
2012-09-15 18:25:25 +03:00
for f in fields:
item.addItem(self.resolveValueDescription(f), f)
elif isinstance(param, ParameterTableMultipleField):
item = QComboBox()
item.setEditable(True)
fields = self.getAvailableValuesOfType(ParameterTableMultipleField, None)
for f in fields:
item.addItem(self.resolveValueDescription(f), f)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterNumber):
item = QComboBox()
2012-09-15 18:25:25 +03:00
item.setEditable(True)
numbers = self.getAvailableValuesOfType(ParameterNumber, OutputNumber)
2012-09-15 18:25:25 +03:00
for n in numbers:
item.addItem(self.resolveValueDescription(n), n)
2016-09-21 18:24:26 +02:00
item.setEditText(str(param.default))
elif isinstance(param, ParameterCrs):
item = QComboBox()
values = self.getAvailableValuesOfType(ParameterCrs, OutputCrs)
for v in values:
item.addItem(self.resolveValueDescription(v), v)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterExtent):
item = QComboBox()
2012-09-15 18:25:25 +03:00
item.setEditable(True)
extents = self.getAvailableValuesOfType(ParameterExtent, OutputExtent)
if self.canUseAutoExtent():
item.addItem(self.USE_MIN_COVERING_EXTENT, None)
2012-09-15 18:25:25 +03:00
for ex in extents:
item.addItem(self.resolveValueDescription(ex), ex)
if not self.canUseAutoExtent():
2016-09-21 18:24:26 +02:00
item.setEditText(str(param.default))
elif isinstance(param, ParameterPoint):
item = QComboBox()
item.setEditable(True)
points = self.getAvailableValuesOfType(ParameterPoint)
for p in points:
item.addItem(self.resolveValueDescription(p), p)
2016-09-21 18:24:26 +02:00
item.setEditText(str(param.default))
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterFile):
item = QComboBox()
2012-09-15 18:25:25 +03:00
item.setEditable(True)
2014-07-04 22:35:41 +02:00
files = self.getAvailableValuesOfType(ParameterFile, OutputFile)
2012-09-15 18:25:25 +03:00
for f in files:
item.addItem(self.resolveValueDescription(f), f)
elif isinstance(param, ParameterGeometryPredicate):
item = GeometryPredicateSelectionPanel(param.enabledPredicates)
2012-09-15 18:25:25 +03:00
else:
item = QLineEdit()
2012-09-15 18:25:25 +03:00
try:
2016-09-21 18:24:26 +02:00
item.setText(str(param.default))
2012-09-15 18:25:25 +03:00
except:
pass
return item
def canUseAutoExtent(self):
for param in self._alg.parameters:
if isinstance(param, (ParameterRaster, ParameterVector, ParameterMultipleInput)):
return True
return False
2012-09-15 18:25:25 +03:00
def setTableContent(self):
params = self._alg.parameters
outputs = self._alg.outputs
2014-07-02 07:46:03 +02:00
visibleParams = [p for p in params if not p.hidden]
visibleOutputs = [p for o in outputs if not o.hidden]
self.tableWidget.setRowCount(len(visibleParams) + len(visibleOutputs))
2014-07-02 07:46:03 +02:00
for i, param in visibleParams:
item = QTableWidgetItem(param.description)
item.setFlags(Qt.ItemIsEnabled)
self.tableWidget.setItem(i, 0, item)
item = self.getWidgetFromParameter(param)
self.valueItems[param.name] = item
self.tableWidget.setCellWidget(i, 1, item)
self.tableWidget.setRowHeight(i, 22)
2014-07-02 07:46:03 +02:00
for i, output in visibleOutputs:
item = QTableWidgetItem(output.description + '<'
+ output.__module__.split('.')[-1] + '>')
item.setFlags(Qt.ItemIsEnabled)
self.tableWidget.setItem(i, 0, item)
item = QLineEdit()
if hasattr(item, 'setPlaceholderText'):
item.setPlaceholderText(ModelerParametersDialog.ENTER_NAME)
self.valueItems[output.name] = item
self.tableWidget.setCellWidget(i, 1, item)
self.tableWidget.setRowHeight(i, 22)
2012-09-15 18:25:25 +03:00
def setComboBoxValue(self, combo, value, param):
if isinstance(value, list):
value = value[0]
items = [combo.itemData(i) for i in range(combo.count())]
try:
2014-07-02 07:46:03 +02:00
idx = items.index(value)
combo.setCurrentIndex(idx)
2014-07-02 07:46:03 +02:00
return
except ValueError:
pass
2014-07-02 07:46:03 +02:00
if combo.isEditable():
if value is not None:
2016-09-21 18:24:26 +02:00
combo.setEditText(str(value))
2014-07-02 07:46:03 +02:00
elif isinstance(param, ParameterSelection):
2012-09-15 18:25:25 +03:00
combo.setCurrentIndex(int(value))
2014-07-02 07:46:03 +02:00
elif isinstance(param, ParameterBoolean):
2012-09-15 18:25:25 +03:00
if value:
combo.setCurrentIndex(0)
else:
combo.setCurrentIndex(1)
2014-07-02 07:46:03 +02:00
def setPreviousValues(self):
if self._algName is not None:
alg = self.model.algs[self._algName]
self.descriptionBox.setText(alg.description)
for param in alg.algorithm.parameters:
if param.hidden:
continue
widget = self.valueItems[param.name]
if param.name in alg.params:
value = alg.params[param.name]
else:
value = param.default
if isinstance(param, (
ParameterRaster,
ParameterVector,
ParameterTable,
ParameterTableField,
ParameterSelection,
ParameterNumber,
ParameterBoolean,
ParameterExtent,
ParameterFile,
ParameterPoint,
ParameterCrs
)):
2012-09-15 18:25:25 +03:00
self.setComboBoxValue(widget, value, param)
2013-02-03 10:26:43 +01:00
elif isinstance(param, ParameterString):
if param.multiline:
widget.setValue(value)
else:
2013-02-07 01:09:39 +01:00
self.setComboBoxValue(widget, value, param)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterFixedTable):
pass # TODO!
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterMultipleInput):
if param.datatype == dataobjects.TYPE_VECTOR_ANY:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector)
elif param.datatype == dataobjects.TYPE_VECTOR_POINT:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector, [dataobjects.TYPE_VECTOR_POINT, dataobjects.TYPE_VECTOR_ANY])
elif param.datatype == dataobjects.TYPE_VECTOR_LINE:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector, [dataobjects.TYPE_VECTOR_LINE, dataobjects.TYPE_VECTOR_ANY])
elif param.datatype == dataobjects.TYPE_VECTOR_POLYGON:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector, [dataobjects.TYPE_VECTOR_POLYGON, dataobjects.TYPE_VECTOR_ANY])
elif param.datatype == dataobjects.TYPE_RASTER:
options = self.getAvailableValuesOfType(ParameterRaster, OutputRaster)
else:
options = self.getAvailableValuesOfType(ParameterFile, OutputFile)
selected = []
for i, opt in enumerate(options):
if opt in value:
selected.append(i)
widget.setSelectedItems(selected)
elif isinstance(param, ParameterGeometryPredicate):
widget.setValue(value)
2012-09-15 18:25:25 +03:00
2016-09-21 18:24:26 +02:00
for name, out in alg.outputs.items():
2014-07-02 07:46:03 +02:00
widget = self.valueItems[name].setText(out.description)
2013-02-07 01:09:39 +01:00
2013-02-03 10:26:43 +01:00
selected = []
dependencies = self.getAvailableDependencies()
for idx, dependency in enumerate(dependencies):
if dependency.name in alg.dependencies:
selected.append(idx)
2013-02-07 01:09:39 +01:00
2013-02-03 10:26:43 +01:00
self.dependenciesPanel.setSelectedItems(selected)
2012-09-15 18:25:25 +03:00
def createAlgorithm(self):
alg = Algorithm(self._alg.commandLineName())
alg.setName(self.model)
2014-07-02 07:46:03 +02:00
alg.description = self.descriptionBox.text()
params = self._alg.parameters
outputs = self._alg.outputs
2012-09-15 18:25:25 +03:00
for param in params:
if param.hidden:
continue
if not self.setParamValue(alg, param, self.valueItems[param.name]):
self.bar.pushMessage("Error", "Wrong or missing value for parameter '%s'" % param.description,
level=QgsMessageBar.WARNING)
return None
2012-09-15 18:25:25 +03:00
for output in outputs:
if not output.hidden:
2016-09-21 18:24:26 +02:00
name = str(self.valueItems[output.name].text())
if name.strip() != '' and name != ModelerParametersDialog.ENTER_NAME:
alg.outputs[output.name] = ModelerOutput(name)
2013-02-07 01:09:39 +01:00
2013-02-03 10:26:43 +01:00
selectedOptions = self.dependenciesPanel.selectedoptions
availableDependencies = self.getAvailableDependencies()
for selected in selectedOptions:
alg.dependencies.append(availableDependencies[selected].name)
2013-02-07 01:09:39 +01:00
return alg
2012-09-15 18:25:25 +03:00
def setParamValueLayerOrTable(self, alg, param, widget):
idx = widget.currentIndex()
2012-09-15 18:25:25 +03:00
if idx < 0:
return False
2012-09-15 18:25:25 +03:00
else:
value = widget.itemData(widget.currentIndex())
alg.params[param.name] = value
return True
2012-09-15 18:25:25 +03:00
def setParamTableFieldValue(self, alg, param, widget):
2012-09-15 18:25:25 +03:00
idx = widget.findText(widget.currentText())
2014-07-02 07:46:03 +02:00
if idx < 0:
2016-09-21 18:24:26 +02:00
s = str(widget.currentText()).strip()
if s == '':
if param.optional:
alg.params[param.name] = None
return True
else:
return False
2014-07-02 07:46:03 +02:00
else:
2014-10-06 07:13:32 +02:00
alg.params[param.name] = s
2012-09-15 18:25:25 +03:00
return True
else:
alg.params[param.name] = widget.itemData(widget.currentIndex())
2012-09-15 18:25:25 +03:00
return True
def setParamStringValue(self, alg, param, widget):
2014-07-02 07:46:03 +02:00
if param.multiline:
2013-02-03 10:26:43 +01:00
value = widget.getValue()
option = widget.getOption()
2013-04-15 07:16:20 +02:00
if option == MultilineTextPanel.USE_TEXT:
if value == '':
if param.optional:
alg.params[param.name] = None
2013-04-15 07:16:20 +02:00
return True
else:
return False
2013-04-15 07:16:20 +02:00
else:
alg.params[param.name] = value
2013-02-07 01:09:39 +01:00
else:
alg.params[param.name] = value
2014-07-02 07:46:03 +02:00
else:
2013-02-03 10:26:43 +01:00
idx = widget.findText(widget.currentText())
2014-07-02 07:46:03 +02:00
if idx < 0:
value = widget.currentText().strip()
if value == '':
if param.optional:
alg.params[param.name] = None
return True
else:
return False
else:
alg.params[param.name] = value
2013-02-03 10:26:43 +01:00
else:
2014-07-02 07:46:03 +02:00
alg.params[param.name] = widget.itemData(widget.currentIndex())
2012-09-15 18:25:25 +03:00
return True
def setParamFileValue(self, alg, param, widget):
2012-09-15 18:25:25 +03:00
idx = widget.findText(widget.currentText())
2014-07-02 07:46:03 +02:00
if idx < 0:
value = widget.currentText()
2012-09-15 18:25:25 +03:00
else:
value = widget.itemData(widget.currentIndex())
alg.params[param.name] = value
2012-09-15 18:25:25 +03:00
return True
def setParamNumberValue(self, alg, param, widget):
2012-09-15 18:25:25 +03:00
idx = widget.findText(widget.currentText())
2014-07-02 07:46:03 +02:00
if idx < 0:
s = widget.currentText().strip()
if s:
try:
value = float(s)
except:
return False
elif param.optional:
value = None
else:
2012-09-15 18:25:25 +03:00
return False
else:
value = widget.itemData(widget.currentIndex())
2014-07-02 07:46:03 +02:00
alg.params[param.name] = value
2012-09-15 18:25:25 +03:00
return True
def setParamExtentValue(self, alg, param, widget):
2012-09-15 18:25:25 +03:00
idx = widget.findText(widget.currentText())
2014-07-02 07:46:03 +02:00
if idx < 0:
2016-09-21 18:24:26 +02:00
s = str(widget.currentText()).strip()
if s:
try:
tokens = s.split(',')
if len(tokens) != 4:
return False
for token in tokens:
float(token)
except:
2012-09-15 18:25:25 +03:00
return False
elif param.optional:
s = None
else:
2014-07-02 07:46:03 +02:00
return False
alg.params[param.name] = [s]
2012-09-15 18:25:25 +03:00
else:
value = widget.itemData(widget.currentIndex())
alg.params[param.name] = value
2012-09-15 18:25:25 +03:00
return True
def setParamPointValue(self, alg, param, widget):
idx = widget.findText(widget.currentText())
if idx < 0:
2016-09-21 18:24:26 +02:00
s = str(widget.currentText()).strip()
if s:
try:
tokens = s.split(',')
if len(tokens) != 2:
return False
for token in tokens:
float(token)
except:
return False
elif param.optional:
s = None
else:
return False
alg.params[param.name] = [s]
else:
value = widget.itemData(widget.currentIndex())
alg.params[param.name] = value
return True
def setParamCrsValue(self, alg, param, widget):
idx = widget.currentIndex()
if idx < 0:
return False
else:
value = widget.itemData(widget.currentIndex())
alg.params[param.name] = value
return True
def setParamValue(self, alg, param, widget):
if isinstance(param, (ParameterRaster, ParameterVector,
2015-12-20 10:12:18 +01:00
ParameterTable)):
return self.setParamValueLayerOrTable(alg, param, widget)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterBoolean):
if widget.currentIndex() < 2:
value = widget.currentIndex() == 0
else:
value = widget.itemData(widget.currentIndex())
alg.params[param.name] = value
return True
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterString):
return self.setParamStringValue(alg, param, widget)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterNumber):
return self.setParamNumberValue(alg, param, widget)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterExtent):
return self.setParamExtentValue(alg, param, widget)
elif isinstance(param, ParameterPoint):
return self.setParamPointValue(alg, param, widget)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterFile):
return self.setParamFileValue(alg, param, widget)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterSelection):
alg.params[param.name] = widget.currentIndex()
2014-07-02 07:46:03 +02:00
return True
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterRange):
alg.params[param.name] = widget.getValue()
2014-07-02 07:46:03 +02:00
return True
elif isinstance(param, ParameterCrs):
return self.setParamCrsValue(alg, param, widget)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterFixedTable):
table = widget.table
if not bool(table) and not param.optional:
return False
alg.params[param.name] = ParameterFixedTable.tableToString(table)
2014-07-02 07:46:03 +02:00
return True
elif isinstance(param, (ParameterTableField,
ParameterTableMultipleField)):
return self.setParamTableFieldValue(alg, param, widget)
2012-09-15 18:25:25 +03:00
elif isinstance(param, ParameterMultipleInput):
if param.datatype == dataobjects.TYPE_VECTOR_ANY:
2014-07-02 07:46:03 +02:00
options = self.getAvailableValuesOfType(ParameterVector, OutputVector)
elif param.datatype == dataobjects.TYPE_VECTOR_POINT:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector, [dataobjects.TYPE_VECTOR_POINT, dataobjects.TYPE_VECTOR_ANY])
elif param.datatype == dataobjects.TYPE_VECTOR_LINE:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector, [dataobjects.TYPE_VECTOR_LINE, dataobjects.TYPE_VECTOR_ANY])
elif param.datatype == dataobjects.TYPE_VECTOR_POLYGON:
options = self.getAvailableValuesOfType(ParameterVector, OutputVector, [dataobjects.TYPE_VECTOR_POLYGON, dataobjects.TYPE_VECTOR_ANY])
elif param.datatype == dataobjects.TYPE_RASTER:
options = self.getAvailableValuesOfType(ParameterRaster, OutputRaster)
else:
options = self.getAvailableValuesOfType(ParameterFile, OutputFile)
2014-07-02 07:46:03 +02:00
values = [options[i] for i in widget.selectedoptions]
2012-09-15 18:25:25 +03:00
if len(values) == 0 and not param.optional:
2014-07-02 07:46:03 +02:00
return False
alg.params[param.name] = values
2012-09-15 18:25:25 +03:00
return True
elif isinstance(param, ParameterGeometryPredicate):
alg.params[param.name] = widget.value()
return True
2012-09-15 18:25:25 +03:00
else:
2016-09-21 18:24:26 +02:00
alg.params[param.name] = str(widget.text())
2012-09-15 18:25:25 +03:00
return True
def okPressed(self):
self.alg = self.createAlgorithm()
if self.alg is not None:
2012-09-15 18:25:25 +03:00
self.close()
2012-09-15 18:25:25 +03:00
def cancelPressed(self):
self.alg = None
2012-09-15 18:25:25 +03:00
self.close()