[processing] add GDAL rasterize algorithm (fix #5854)

alexbruy · alexbruy · commit e7ae5d2c683f · 2013-09-14T18:24:39.000+03:00
diff --git a/python/plugins/processing/gdal/GdalOgrAlgorithmProvider.py b/python/plugins/processing/gdal/GdalOgrAlgorithmProvider.py
@@ -46,6 +46,7 @@
 from processing.gdal.ClipByExtent import ClipByExtent
 from processing.gdal.ClipByMask import ClipByMask
 from processing.gdal.contour import contour
+from processing.gdal.rasterize import rasterize
 
 from processing.gdal.ogr2ogr import Ogr2Ogr
 from processing.gdal.ogrinfo import OgrInfo
@@ -95,7 +96,7 @@ def createAlgsList(self):
         self.preloadedAlgs = [nearblack(), information(), warp(), translate(),
                               rgb2pct(), pct2rgb(), merge(), polygonize(),
                               gdaladdo(), ClipByExtent(), ClipByMask(),
-                              contour(),
+                              contour(), rasterize(),
                               OgrInfo(), Ogr2Ogr(), OgrSql()]
 
         #And then we add those that are created as python scripts
diff --git a/python/plugins/processing/gdal/rasterize.py b/python/plugins/processing/gdal/rasterize.py
@@ -0,0 +1,88 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    rasterize.py
+    ---------------------
+    Date                 : September 2013
+    Copyright            : (C) 2013 by Alexander Bruy
+    Email                : alexander dot bruy 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__ = 'Alexander Bruy'
+__date__ = 'September 2013'
+__copyright__ = '(C) 2013, Alexander Bruy'
+# This will get replaced with a git SHA1 when you do a git archive
+__revision__ = '$Format:%H$'
+
+import os
+from PyQt4 import QtGui, QtCore
+
+from processing.core.GeoAlgorithm import GeoAlgorithm
+from processing.tools.system import *
+
+from processing.parameters.ParameterVector import ParameterVector
+from processing.parameters.ParameterTableField import ParameterTableField
+from processing.parameters.ParameterSelection import ParameterSelection
+from processing.parameters.ParameterNumber import ParameterNumber
+
+from processing.outputs.OutputRaster import OutputRaster
+
+from processing.gdal.GdalUtils import GdalUtils
+
+class rasterize(GeoAlgorithm):
+
+    INPUT = "INPUT"
+    FIELD = "FIELD"
+    DIMENSIONS = "DIMENSIONS"
+    WIDTH = "WIDTH"
+    HEIGHT = "HEIGHT"
+    OUTPUT = "OUTPUT"
+
+
+
+    def getIcon(self):
+        filepath = os.path.dirname(__file__) + "/icons/rasterize.png"
+        return QtGui.QIcon(filepath)
+
+    def defineCharacteristics(self):
+        self.name = "Rasterize"
+        self.group = "[GDAL] Conversion"
+        self.addParameter(ParameterVector(self.INPUT, "Input layer"))
+        self.addParameter(ParameterTableField(self.FIELD, "Attribute field", self.INPUT))
+        self.addParameter(ParameterSelection(self.DIMENSIONS, "Set output raster size", ["Output size in pixels", "Output resolution in map units per pixel"], 0))
+        self.addParameter(ParameterNumber(self.WIDTH, "Horizontal", 0.0, 99999999.999999, 3000.0))
+        self.addParameter(ParameterNumber(self.HEIGHT, "Vertical", 0.0, 99999999.999999, 3000.0))
+
+        self.addOutput(OutputRaster(self.OUTPUT, "Output layer"))
+
+    def processAlgorithm(self, progress):
+        arguments = []
+        arguments.append("-a")
+        arguments.append(str(self.getParameterValue(self.FIELD)))
+
+        dimType = self.getParameterValue(self.DIMENSIONS)
+        if dimType == 0:
+            # size in pixels
+            arguments.append("-ts")
+        else:
+            # resolution in map units per pixel
+            arguments.append("-tr")
+        arguments.append(str(self.getParameterValue(self.WIDTH)))
+        arguments.append(str(self.getParameterValue(self.HEIGHT)))
+
+        arguments.append("-l")
+        arguments.append(os.path.basename(os.path.splitext(unicode(self.getParameterValue(self.INPUT)))[0]))
+        arguments.append(unicode(self.getParameterValue(self.INPUT)))
+
+        arguments.append(unicode(self.getOutputValue(self.OUTPUT)))
+
+        GdalUtils.runGdal(["gdal_rasterize", GdalUtils.escapeAndJoin(arguments)], progress)
