2 new plots added

qgis · Apr 12, 2017 · bca0ca2 · bca0ca2
1 parent 0c5b667
commit bca0ca2
Show file tree

Hide file tree

Showing 3 changed files with 214 additions and 1 deletion.
diff --git a/python/plugins/processing/algs/qgis/BoxPlot.py b/python/plugins/processing/algs/qgis/BoxPlot.py
@@ -0,0 +1,110 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    BarPlot.py
+    ---------------------
+    Date                 : March 2015
+    Copyright            : (C) 2017 by Matteo Ghetta
+    Email                : matteo dot ghetta 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__ = 'Matteo Ghetta'
+__date__ = 'March 2017'
+__copyright__ = '(C) 2017, Matteo Ghetta'
+
+# This will get replaced with a git SHA1 when you do a git archive
+
+__revision__ = '$Format:%H$'
+
+import plotly as plt
+import plotly.graph_objs as go
+
+from qgis.core import (QgsApplication)
+from processing.core.parameters import ParameterTable
+from processing.core.parameters import ParameterTableField
+from processing.core.parameters import ParameterSelection
+from processing.core.GeoAlgorithm import GeoAlgorithm
+from processing.core.outputs import OutputHTML
+from processing.tools import vector
+from processing.tools import dataobjects
+
+
+class BoxPlot(GeoAlgorithm):
+
+    INPUT = 'INPUT'
+    OUTPUT = 'OUTPUT'
+    NAME_FIELD = 'NAME_FIELD'
+    VALUE_FIELD = 'VALUE_FIELD'
+    MSD = 'MSD'
+
+    def icon(self):
+        return QgsApplication.getThemeIcon("/providerQgis.svg")
+
+    def svgIconPath(self):
+        return QgsApplication.iconPath("providerQgis.svg")
+
+    def group(self):
+        return self.tr('Graphics')
+
+    def name(self):
+        return 'boxplot'
+
+    def displayName(self):
+        return self.tr('Box plot')
+
+    def defineCharacteristics(self):
+        self.addParameter(ParameterTable(self.INPUT, self.tr('Input table')))
+        self.addParameter(ParameterTableField(self.NAME_FIELD,
+                                              self.tr('Category name field'),
+                                              self.INPUT,
+                                              ParameterTableField.DATA_TYPE_ANY))
+        self.addParameter(ParameterTableField(self.VALUE_FIELD,
+                                              self.tr('Value field'),
+                                              self.INPUT,
+                                              ParameterTableField.DATA_TYPE_NUMBER))
+        msd = [self.tr('Show Mean'),
+               self.tr('Show Standard Deviation'),
+               self.tr('Don\'t show Mean and Standard Deviation')
+               ]
+        self.addParameter(ParameterSelection(
+            self.MSD,
+            self.tr('Additional Statistic Lines'),
+            msd, default=0))
+
+        self.addOutput(OutputHTML(self.OUTPUT, self.tr('Box plot')))
+
+    def processAlgorithm(self, feedback):
+        layer = dataobjects.getLayerFromString(
+            self.getParameterValue(self.INPUT))
+        namefieldname = self.getParameterValue(self.NAME_FIELD)
+        valuefieldname = self.getParameterValue(self.VALUE_FIELD)
+
+        output = self.getOutputValue(self.OUTPUT)
+
+        values = vector.values(layer, valuefieldname)
+
+        x_var = [i[namefieldname] for i in layer.getFeatures()]
+
+        msdIndex = self.getParameterValue(self.MSD)
+        msd = True
+
+        if msdIndex == 1:
+            msd = 'sd'
+        elif msdIndex == 2:
+            msd = False
+
+        data = [go.Box(
+                x=x_var,
+                y=values[valuefieldname],
+                boxmean=msd)]
+
+        plt.offline.plot(data, filename=output, auto_open=False)
diff --git a/python/plugins/processing/algs/qgis/QGISAlgorithmProvider.py b/python/plugins/processing/algs/qgis/QGISAlgorithmProvider.py
@@ -265,10 +265,13 @@ def getAlgs(self):
             from .MeanAndStdDevPlot import MeanAndStdDevPlot
             from .BarPlot import BarPlot
             from .PolarPlot import PolarPlot
+            from .BoxPlot import BoxPlot
+            from .VectorLayerScatterplot3D import VectorLayerScatterplot3D
 
             algs.extend([VectorLayerHistogram(), RasterLayerHistogram(),
                          VectorLayerScatterplot(), MeanAndStdDevPlot(),
-                         BarPlot(), PolarPlot()])
+                         BarPlot(), PolarPlot(), BoxPlot(),
+                         VectorLayerScatterplot3D()])
 
         # to store algs added by 3rd party plugins as scripts
         folder = os.path.join(os.path.dirname(__file__), 'scripts')

diff --git a/python/plugins/processing/algs/qgis/VectorLayerScatterplot3D.py b/python/plugins/processing/algs/qgis/VectorLayerScatterplot3D.py
@@ -0,0 +1,100 @@
+# -*- 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'
+
+# This will get replaced with a git SHA1 when you do a git archive
+
+__revision__ = '$Format:%H$'
+
+import plotly as plt
+import plotly.graph_objs as go
+
+from qgis.core import (QgsApplication)
+from processing.core.GeoAlgorithm import GeoAlgorithm
+from processing.core.parameters import ParameterVector
+from processing.core.parameters import ParameterTableField
+from processing.core.outputs import OutputHTML
+
+from processing.tools import vector
+from processing.tools import dataobjects
+
+
+class VectorLayerScatterplot3D(GeoAlgorithm):
+
+    INPUT = 'INPUT'
+    OUTPUT = 'OUTPUT'
+    XFIELD = 'XFIELD'
+    YFIELD = 'YFIELD'
+    ZFIELD = 'ZFIELD'
+
+    def icon(self):
+        return QgsApplication.getThemeIcon("/providerQgis.svg")
+
+    def svgIconPath(self):
+        return QgsApplication.iconPath("providerQgis.svg")
+
+    def group(self):
+        return self.tr('Graphics')
+
+    def name(self):
+        return 'scatter3dplot'
+
+    def displayName(self):
+        return self.tr('Vector layer scatterplot 3D')
+
+    def defineCharacteristics(self):
+        self.addParameter(ParameterVector(self.INPUT,
+                                          self.tr('Input layer')))
+        self.addParameter(ParameterTableField(self.XFIELD,
+                                              self.tr('X attribute'),
+                                              self.INPUT,
+                                              ParameterTableField.DATA_TYPE_NUMBER))
+        self.addParameter(ParameterTableField(self.YFIELD,
+                                              self.tr('Y attribute'),
+                                              self.INPUT,
+                                              ParameterTableField.DATA_TYPE_NUMBER))
+        self.addParameter(ParameterTableField(self.ZFIELD,
+                                              self.tr('Z attribute'),
+                                              self.INPUT,
+                                              ParameterTableField.DATA_TYPE_NUMBER))
+
+        self.addOutput(OutputHTML(self.OUTPUT, self.tr('Scatterplot 3D')))
+
+    def processAlgorithm(self, feedback):
+
+        layer = dataobjects.getLayerFromString(
+            self.getParameterValue(self.INPUT))
+        xfieldname = self.getParameterValue(self.XFIELD)
+        yfieldname = self.getParameterValue(self.YFIELD)
+        zfieldname = self.getParameterValue(self.ZFIELD)
+
+        output = self.getOutputValue(self.OUTPUT)
+
+        values = vector.values(layer, 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)