addded new SEXTANTE python algorithms

python/plugins/sextante/algs/BarPlot.py

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    BarPlot.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 matplotlib.pyplot as plt
+import matplotlib.pylab as lab
+import numpy as np
+from PyQt4.QtCore import *
+from qgis.core import *
+from sextante.parameters.ParameterTable import ParameterTable
+from sextante.parameters.ParameterTableField import ParameterTableField
+from sextante.core.GeoAlgorithm import GeoAlgorithm
+from sextante.outputs.OutputHTML import OutputHTML
+from sextante.tools import *
+from sextante.core.QGisLayers import QGisLayers
+
+class BarPlot(GeoAlgorithm):
+
+    INPUT = "INPUT"
+    OUTPUT = "OUTPUT"
+    NAME_FIELD = "NAME_FIELD"
+    VALUE_FIELD = "VALUE_FIELD"    
+
+    def processAlgorithm(self, progress):
+        uri = self.getParameterValue(self.INPUT)
+        layer = QGisLayers.getObjectFromUri(uri)
+        namefieldname = self.getParameterValue(self.NAME_FIELD)
+        valuefieldname = self.getParameterValue(self.VALUE_FIELD)        
+        output = self.getOutputValue(self.OUTPUT)
+        values = vector.getAttributeValues(layer, namefieldname, valuefieldname)
+        plt.close()
+
+        ind = np.arange(len(values[namefieldname]))
+        width = 0.8
+        plt.bar(ind, values[valuefieldname], width, color='r')
+
+        plt.xticks(ind, values[namefieldname], rotation = 45)       
+        plotFilename = output +".png"
+        lab.savefig(plotFilename)        
+        f = open(output, "w")
+        f.write("<img src=\"" + plotFilename + "\"/>")
+        f.close()
+
+    def defineCharacteristics(self):
+        self.name = "Bar plot"
+        self.group = "Graphics"
+        self.addParameter(ParameterTable(self.INPUT, "Input table"))
+        self.addParameter(ParameterTableField(self.NAME_FIELD, "Category name field", self.INPUT))    
+        self.addParameter(ParameterTableField(self.VALUE_FIELD, "Value field", self.INPUT))                
+        self.addOutput(OutputHTML(self.OUTPUT, "Output"))
+
+
+
+
+

python/plugins/sextante/algs/MeanAndStdDevPlot.py

@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    MeanAndStdDevPlot.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 matplotlib.pyplot as plt
+import matplotlib.pylab as lab
+import numpy as np
+from PyQt4.QtCore import *
+from qgis.core import *
+from sextante.parameters.ParameterTable import ParameterTable
+from sextante.parameters.ParameterTableField import ParameterTableField
+from sextante.core.GeoAlgorithm import GeoAlgorithm
+from sextante.outputs.OutputHTML import OutputHTML
+from sextante.tools import *
+from sextante.core.QGisLayers import QGisLayers
+
+class MeanAndStdDevPlot(GeoAlgorithm):
+
+    INPUT = "INPUT"
+    OUTPUT = "OUTPUT"
+    NAME_FIELD = "NAME_FIELD"
+    MEAN_FIELD = "MEAN_FIELD"
+    STDDEV_FIELD = "STDDEV_FIELD"
+
+
+    def processAlgorithm(self, progress):
+        uri = self.getParameterValue(self.INPUT)
+        layer = QGisLayers.getObjectFromUri(uri)
+        namefieldname = self.getParameterValue(self.NAME_FIELD)
+        meanfieldname = self.getParameterValue(self.MEAN_FIELD)
+        stddevfieldname = self.getParameterValue(self.STDDEV_FIELD)
+        output = self.getOutputValue(self.OUTPUT)
+        values = vector.getAttributeValues(layer, namefieldname, meanfieldname, stddevfieldname)
+        plt.close()
+
+
+        ind = np.arange(len(values[namefieldname]))
+        width = 0.8
+        plt.bar(ind, values[meanfieldname], width,
+                    color='r',
+                    yerr=values[stddevfieldname],
+                    error_kw=dict(ecolor='yellow'))
+
+        plt.xticks(ind, values[namefieldname], rotation = 45)       
+        plotFilename = output +".png"
+        lab.savefig(plotFilename)        
+        f = open(output, "w")
+        f.write("<img src=\"" + plotFilename + "\"/>")
+        f.close()
+
+    def defineCharacteristics(self):
+        self.name = "Mean and standard deviation plot"
+        self.group = "Graphics"
+        self.addParameter(ParameterTable(self.INPUT, "Input table"))
+        self.addParameter(ParameterTableField(self.NAME_FIELD, "Category name field", self.INPUT,ParameterTableField.DATA_TYPE_ANY))    
+        self.addParameter(ParameterTableField(self.MEAN_FIELD, "Mean field", self.INPUT))        
+        self.addParameter(ParameterTableField(self.STDDEV_FIELD, "StdDev field", self.INPUT))
+        self.addOutput(OutputHTML(self.OUTPUT, "Output"))
+
+
+
+
+

python/plugins/sextante/algs/PolarPlot.py

@@ -0,0 +1,97 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    BarPlot.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 matplotlib.pyplot as plt
+import matplotlib.pylab as lab
+import matplotlib.cm as cm
+from matplotlib.pyplot import figure, show, rc
+import numpy as np
+from PyQt4.QtCore import *
+from qgis.core import *
+from sextante.parameters.ParameterTable import ParameterTable
+from sextante.parameters.ParameterTableField import ParameterTableField
+from sextante.core.GeoAlgorithm import GeoAlgorithm
+from sextante.outputs.OutputHTML import OutputHTML
+from sextante.tools import *
+from sextante.core.QGisLayers import QGisLayers
+
+class PolarPlot(GeoAlgorithm):
+
+    INPUT = "INPUT"
+    OUTPUT = "OUTPUT"
+    NAME_FIELD = "NAME_FIELD"
+    VALUE_FIELD = "VALUE_FIELD"    
+
+    def processAlgorithm(self, progress):
+        uri = self.getParameterValue(self.INPUT)
+        layer = QGisLayers.getObjectFromUri(uri)
+        namefieldname = self.getParameterValue(self.NAME_FIELD)
+        valuefieldname = self.getParameterValue(self.VALUE_FIELD)        
+        output = self.getOutputValue(self.OUTPUT)
+        values = vector.getAttributeValues(layer, namefieldname, valuefieldname)
+        plt.close()
+
+        fig = figure(figsize=(8,8))
+        ax = fig.add_axes([0.1, 0.1, 0.8, 0.8], polar=True)
+        N = len(values[valuefieldname])
+        theta = np.arange(0.0, 2*np.pi, 2*np.pi/N)
+        radii = values[valuefieldname]
+        width = 2*np.pi/N
+        ax.bar(theta, radii, width=width, bottom=0.0)   
+        plotFilename = output +".png"
+        lab.savefig(plotFilename)        
+        f = open(output, "w")
+        f.write("<img src=\"" + plotFilename + "\"/>")
+        f.close()
+
+    def defineCharacteristics(self):
+        self.name = "Polar plot"
+        self.group = "Graphics"
+        self.addParameter(ParameterTable(self.INPUT, "Input table"))
+        self.addParameter(ParameterTableField(self.NAME_FIELD, "Category name field", self.INPUT))    
+        self.addParameter(ParameterTableField(self.VALUE_FIELD, "Value field", self.INPUT))                
+        self.addOutput(OutputHTML(self.OUTPUT, "Output"))
+
+
+
+
+
+
+
+
+
+
+# force square figure and square axes looks better for polar, IMO
+fig = figure(figsize=(8,8))
+ax = fig.add_axes([0.1, 0.1, 0.8, 0.8], polar=True)
+
+N = 20
+theta = np.arange(0.0, 2*np.pi, 2*np.pi/N)
+radii = 10*np.random.rand(N)
+width = np.pi/4*np.random.rand(N)
+bars = ax.bar(theta, radii, width=width, bottom=0.0)
+for r,bar in zip(radii, bars):
+    bar.set_facecolor( cm.jet(r/10.))
+    bar.set_alpha(0.5)

python/plugins/sextante/algs/QGISAlgorithmProvider.py

@@ -16,12 +16,21 @@
 *                                                                         *
 ***************************************************************************
 """
+from sextante.algs.MeanAndStdDevPlot import MeanAndStdDevPlot
+from sextante.algs.BarPlot import BarPlot
+from sextante.algs.PolarPlot import PolarPlot
+from sextante.algs.RasterLayerStatistics import RasterLayerStatistics
+
 __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 QtGui
+from sextante.algs.RasterLayerHistogram import RasterLayerHistogram
+from sextante.algs.StatisticsByCategories import StatisticsByCategories
 from sextante.algs.ftools.PointsInPolygon import PointsInPolygon
 from sextante.algs.ftools.PointsInPolygonUnique import PointsInPolygonUnique
 from sextante.algs.ftools.PointsInPolygonWeighted import PointsInPolygonWeighted
@@ -66,10 +75,10 @@
     mmqgisx_merge_algorithm, mmqgisx_select_algorithm,
     mmqgisx_text_to_float_algorithm)
 from sextante.algs.EquivalentNumField import EquivalentNumField
+from sextante.algs.VectorLayerHistogram import VectorLayerHistogram
+from sextante.algs.VectorLayerScatterplot import VectorLayerScatterplot
 from sextante.core.AlgorithmProvider import AlgorithmProvider
 from sextante.algs.AddTableField import AddTableField
-from PyQt4 import QtGui
-import os
 from sextante.algs.FieldsCalculator import FieldsCalculator
 from sextante.algs.SaveSelectedFeatures import SaveSelectedFeatures
 from sextante.algs.Explode import Explode
@@ -84,33 +93,23 @@ class QGISAlgorithmProvider(AlgorithmProvider):
     def __init__(self):
         AlgorithmProvider.__init__(self)
         self.alglist = [AddTableField(), FieldsCalculator(), SaveSelectedFeatures(), JoinAttributes(),
-                        AutoincrementalField(), Explode(), FieldsPyculator(), EquivalentNumField(),
-                        #FTOOLS
+                        AutoincrementalField(), Explode(), FieldsPyculator(), EquivalentNumField(),                        
                         SumLines(), PointsInPolygon(), PointsInPolygonWeighted(), PointsInPolygonUnique(),
                         BasicStatisticsStrings(), BasicStatisticsNumbers(), NearestNeighbourAnalysis(),
                         MeanCoords(), LinesIntersection(), UniqueValues(), PointDistance(), PointsLayerFromTable(),
-                        # data management
-                        ReprojectLayer(),
-                        # geometry
+                        StatisticsByCategories(), ReprojectLayer(),                        
                         ExportGeometryInfo(), Centroids(), Delaunay(), VoronoiPolygons(),
                         SimplifyGeometries(), DensifyGeometries(), DensifyGeometriesInterval(),
                         MultipartToSingleparts(), SinglePartsToMultiparts(), PolygonsToLines(),
-                        LinesToPolygons(), ExtractNodes(),
-                        # geoprocessing
+                        LinesToPolygons(), ExtractNodes(),                        
                         ConvexHull(), FixedDistanceBuffer(), VariableDistanceBuffer(),
-                        Dissolve(), Difference(), Intersection(), Union(), Clip(),
-                        # research
+                        Dissolve(), Difference(), Intersection(), Union(), Clip(),                        
                         ExtentFromLayer(), RandomSelection(), RandomSelectionWithinSubsets(),
                         SelectByLocation(),
                         #MMQGISX
-                        #mmqgisx_attribute_export_algorithm(),
-                        #mmqgisx_attribute_join_algorithm(),
                         mmqgisx_delete_columns_algorithm(),
                         mmqgisx_delete_duplicate_geometries_algorithm(),
-                        #mmqgisx_geocode_google_algorithm(),
                         mmqgisx_geometry_convert_algorithm(),
-                        #mmqgisx_geometry_export_algorithm(),
-                        #mmqgisx_geometry_import_algorithm(),
                         mmqgisx_grid_algorithm(),
                         mmqgisx_gridify_algorithm(),
                         mmqgisx_hub_distance_algorithm(),
@@ -119,7 +118,10 @@ def __init__(self):
                         mmqgisx_select_algorithm(),
                         mmqgisx_text_to_float_algorithm(),
                         #raster
-                        CreateConstantRaster()]
+                        CreateConstantRaster(), RasterLayerStatistics(),
+                        #graphics
+                        VectorLayerHistogram(), VectorLayerScatterplot(), RasterLayerHistogram(),
+                        MeanAndStdDevPlot(), BarPlot(), PolarPlot()]
 
     def initializeSettings(self):
         AlgorithmProvider.initializeSettings(self)