mirror of
https://github.com/qgis/QGIS.git
synced 2025-04-07 00:05:28 -04:00
a27db3533c
execution) Because: 1. It's nicer to show all algorithms on all installs, and give a descriptive error message to users when they try to run algorithms which depend on Plotly if the library is missing. Otherwise on some installs these algorithms are just missing for no apparent reason. 2. The plotly library takes a long time to load on windows (3-4 seconds), so by moving this load to an on-demand load at time of algorithm execution we can shave a few seconds off the QGIS startup time for ALL users.
108 lines
4.6 KiB
Python
108 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
EquivalentNumField.py
|
|
---------------------
|
|
Date : January 2013
|
|
Copyright : (C) 2013 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__ = 'January 2013'
|
|
__copyright__ = '(C) 2013, Victor Olaya'
|
|
|
|
import warnings
|
|
from qgis.core import (QgsProcessingParameterFeatureSource,
|
|
QgsProcessingParameterField,
|
|
QgsProcessingParameterFileDestination,
|
|
QgsProcessingUtils,
|
|
QgsProcessingException)
|
|
from processing.algs.qgis.QgisAlgorithm import QgisAlgorithm
|
|
|
|
from processing.tools import vector
|
|
|
|
|
|
class VectorLayerScatterplot3D(QgisAlgorithm):
|
|
|
|
INPUT = 'INPUT'
|
|
OUTPUT = 'OUTPUT'
|
|
XFIELD = 'XFIELD'
|
|
YFIELD = 'YFIELD'
|
|
ZFIELD = 'ZFIELD'
|
|
|
|
def group(self):
|
|
return self.tr('Graphics')
|
|
|
|
def groupId(self):
|
|
return 'graphics'
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def initAlgorithm(self, config=None):
|
|
self.addParameter(QgsProcessingParameterFeatureSource(self.INPUT,
|
|
self.tr('Input layer')))
|
|
self.addParameter(QgsProcessingParameterField(self.XFIELD,
|
|
self.tr('X attribute'),
|
|
parentLayerParameterName=self.INPUT,
|
|
type=QgsProcessingParameterField.Numeric))
|
|
self.addParameter(QgsProcessingParameterField(self.YFIELD,
|
|
self.tr('Y attribute'),
|
|
parentLayerParameterName=self.INPUT,
|
|
type=QgsProcessingParameterField.Numeric))
|
|
self.addParameter(QgsProcessingParameterField(self.ZFIELD,
|
|
self.tr('Z attribute'),
|
|
parentLayerParameterName=self.INPUT,
|
|
type=QgsProcessingParameterField.Numeric))
|
|
|
|
self.addParameter(QgsProcessingParameterFileDestination(self.OUTPUT, self.tr('Histogram'), self.tr('HTML files (*.html)')))
|
|
|
|
def name(self):
|
|
return 'scatter3dplot'
|
|
|
|
def displayName(self):
|
|
return self.tr('Vector layer scatterplot 3D')
|
|
|
|
def processAlgorithm(self, parameters, context, feedback):
|
|
try:
|
|
# importing plotly throws Python warnings from within the library - filter these out
|
|
with warnings.catch_warnings():
|
|
warnings.filterwarnings("ignore", category=ResourceWarning)
|
|
warnings.filterwarnings("ignore", category=ImportWarning)
|
|
import plotly as plt
|
|
import plotly.graph_objs as go
|
|
except ImportError:
|
|
raise QgsProcessingException(self.tr('This algorithm requires the Python “plotly” library. Please install this library and try again.'))
|
|
|
|
source = self.parameterAsSource(parameters, self.INPUT, context)
|
|
if source is None:
|
|
raise QgsProcessingException(self.invalidSourceError(parameters, self.INPUT))
|
|
|
|
xfieldname = self.parameterAsString(parameters, self.XFIELD, context)
|
|
yfieldname = self.parameterAsString(parameters, self.YFIELD, context)
|
|
zfieldname = self.parameterAsString(parameters, self.ZFIELD, context)
|
|
|
|
output = self.parameterAsFileOutput(parameters, self.OUTPUT, context)
|
|
|
|
values = vector.values(source, xfieldname, yfieldname, zfieldname)
|
|
|
|
data = [go.Scatter3d(
|
|
x=values[xfieldname],
|
|
y=values[yfieldname],
|
|
z=values[zfieldname],
|
|
mode='markers')]
|
|
|
|
plt.offline.plot(data, filename=output, auto_open=False)
|
|
|
|
return {self.OUTPUT: output}
|