mirror of
https://github.com/qgis/QGIS.git
synced 2025-03-03 00:02:25 -05:00
146 lines
6.4 KiB
Python
146 lines
6.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
Postprocessing.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
|
|
import traceback
|
|
from qgis.PyQt.QtWidgets import QApplication
|
|
from qgis.PyQt.QtCore import QCoreApplication
|
|
from qgis.core import (Qgis,
|
|
QgsProject,
|
|
QgsProcessingFeedback,
|
|
QgsProcessingUtils,
|
|
QgsMapLayerType,
|
|
QgsWkbTypes,
|
|
QgsMessageLog,
|
|
QgsProviderRegistry,
|
|
QgsExpressionContext,
|
|
QgsExpressionContextScope)
|
|
|
|
from processing.core.ProcessingConfig import ProcessingConfig
|
|
from processing.gui.RenderingStyles import RenderingStyles
|
|
|
|
|
|
def set_layer_name(layer, context_layer_details):
|
|
"""
|
|
Sets the name for the given layer, either using the layer's file name
|
|
(or database layer name), or the name specified by the parameter definition.
|
|
"""
|
|
use_filename_as_layer_name = ProcessingConfig.getSetting(ProcessingConfig.USE_FILENAME_AS_LAYER_NAME)
|
|
|
|
if use_filename_as_layer_name or not context_layer_details.name:
|
|
source_parts = QgsProviderRegistry.instance().decodeUri(layer.dataProvider().name(), layer.source())
|
|
layer_name = source_parts.get('layerName', '')
|
|
# if source layer name exists, use that -- else use
|
|
if layer_name:
|
|
layer.setName(layer_name)
|
|
else:
|
|
path = source_parts.get('path', '')
|
|
if path:
|
|
layer.setName(os.path.splitext(os.path.basename(path))[0])
|
|
elif context_layer_details.name:
|
|
# fallback to parameter's name -- shouldn't happen!
|
|
layer.setName(context_layer_details.name)
|
|
else:
|
|
layer.setName(context_layer_details.name)
|
|
|
|
|
|
def handleAlgorithmResults(alg, context, feedback=None, showResults=True, parameters={}):
|
|
wrongLayers = []
|
|
if feedback is None:
|
|
feedback = QgsProcessingFeedback()
|
|
feedback.setProgressText(QCoreApplication.translate('Postprocessing', 'Loading resulting layers'))
|
|
i = 0
|
|
|
|
for l, details in context.layersToLoadOnCompletion().items():
|
|
if feedback.isCanceled():
|
|
return False
|
|
|
|
if len(context.layersToLoadOnCompletion()) > 2:
|
|
# only show progress feedback if we're loading a bunch of layers
|
|
feedback.setProgress(100 * i / float(len(context.layersToLoadOnCompletion())))
|
|
try:
|
|
layer = QgsProcessingUtils.mapLayerFromString(l, context, typeHint=details.layerTypeHint)
|
|
if layer is not None:
|
|
set_layer_name(layer, details)
|
|
|
|
'''If running a model, the execution will arrive here when an algorithm that is part of
|
|
that model is executed. We check if its output is a final otuput of the model, and
|
|
adapt the output name accordingly'''
|
|
outputName = details.outputName
|
|
expcontext = QgsExpressionContext()
|
|
scope = QgsExpressionContextScope()
|
|
expcontext.appendScope(scope)
|
|
for out in alg.outputDefinitions():
|
|
if out.name() not in parameters:
|
|
continue
|
|
outValue = parameters[out.name()]
|
|
if hasattr(outValue, "sink"):
|
|
outValue = outValue.sink.valueAsString(expcontext)[0]
|
|
else:
|
|
outValue = str(outValue)
|
|
if outValue == l:
|
|
outputName = out.name()
|
|
break
|
|
style = None
|
|
if outputName:
|
|
style = RenderingStyles.getStyle(alg.id(), outputName)
|
|
if style is None:
|
|
if layer.type() == QgsMapLayerType.RasterLayer:
|
|
style = ProcessingConfig.getSetting(ProcessingConfig.RASTER_STYLE)
|
|
else:
|
|
if layer.geometryType() == QgsWkbTypes.PointGeometry:
|
|
style = ProcessingConfig.getSetting(ProcessingConfig.VECTOR_POINT_STYLE)
|
|
elif layer.geometryType() == QgsWkbTypes.LineGeometry:
|
|
style = ProcessingConfig.getSetting(ProcessingConfig.VECTOR_LINE_STYLE)
|
|
else:
|
|
style = ProcessingConfig.getSetting(ProcessingConfig.VECTOR_POLYGON_STYLE)
|
|
if style:
|
|
layer.loadNamedStyle(style)
|
|
|
|
details.project.addMapLayer(context.temporaryLayerStore().takeMapLayer(layer))
|
|
|
|
if details.postProcessor():
|
|
details.postProcessor().postProcessLayer(layer, context, feedback)
|
|
|
|
else:
|
|
wrongLayers.append(str(l))
|
|
except Exception:
|
|
QgsMessageLog.logMessage(QCoreApplication.translate('Postprocessing', "Error loading result layer:") + "\n" + traceback.format_exc(), 'Processing', Qgis.Critical)
|
|
wrongLayers.append(str(l))
|
|
i += 1
|
|
|
|
feedback.setProgress(100)
|
|
|
|
if wrongLayers:
|
|
msg = QCoreApplication.translate('Postprocessing', "The following layers were not correctly generated.")
|
|
msg += "<ul>" + "".join(["<li>%s</li>" % lay for lay in wrongLayers]) + "</ul>"
|
|
msg += QCoreApplication.translate('Postprocessing', "You can check the 'Log Messages Panel' in QGIS main window to find more information about the execution of the algorithm.")
|
|
feedback.reportError(msg)
|
|
|
|
return len(wrongLayers) == 0
|