sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-12-30 00:29:39 -05:00
Code Issues Packages Projects Releases Wiki Activity

[processing] restore polygonize algorithm

Browse Source 
Also expose band parameter
This commit is contained in:
Alexander Bruy 2017-08-31 13:15:29 +03:00
parent c53ff66caf
commit b1ff4d774a
2 changed files with 39 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/plugins/processing/algs/gdal/GdalAlgorithmProvider.py
View File

@ -41,13 +41,13 @@ from .hillshade import hillshade
from .information import information
from .nearblack import nearblack
from .pct2rgb import pct2rgb
from .polygonize import polygonize
from .rgb2pct import rgb2pct
from .translate import translate
from .tri import tri
from .warp import warp
# from .merge import merge
# from .polygonize import polygonize
# from .gdaladdo import gdaladdo
# from .ClipByExtent import ClipByExtent
# from .ClipByMask import ClipByMask
@ -142,12 +142,12 @@ class GdalAlgorithmProvider(QgsProcessingProvider):
information(),
nearblack(),
pct2rgb(),
polygonize(),
rgb2pct(),
translate(),
tri(),
warp(),
# merge(),
# polygonize(),
# gdaladdo(),
# ClipByExtent(),
# ClipByMask(),

55
python/plugins/processing/algs/gdal/polygonize.py
View File

@ -30,10 +30,13 @@ import os
from qgis.PyQt.QtGui import QIcon
from qgis.PyQt.QtCore import QFileInfo
from qgis.core import (QgsProcessing,
QgsProcessingParameterRasterLayer,
QgsProcessingParameterBand,
QgsProcessingParameterString,
QgsProcessingParameterBoolean,
QgsProcessingParameterVectorDestination)
from processing.algs.gdal.GdalAlgorithm import GdalAlgorithm
from processing.core.parameters import ParameterRaster
from processing.core.parameters import ParameterString
from processing.core.outputs import OutputVector
from processing.tools.system import isWindows
from processing.algs.gdal.GdalUtils import GdalUtils
@ -43,21 +46,24 @@ pluginPath = os.path.split(os.path.split(os.path.dirname(__file__))[0])[0]
class polygonize(GdalAlgorithm):
INPUT = 'INPUT'
OUTPUT = 'OUTPUT'
BAND = 'BAND'
FIELD = 'FIELD'
def icon(self):
return QIcon(os.path.join(pluginPath, 'images', 'gdaltools', 'polygonize.png'))
EIGHT_CONNECTEDNESS = 'EIGHT_CONNECTEDNESS'
OUTPUT = 'OUTPUT'
def __init__(self):
super().__init__()
def initAlgorithm(self, config=None):
self.addParameter(ParameterRaster(polygonize.INPUT,
self.tr('Input layer'), False))
self.addParameter(ParameterString(polygonize.FIELD,
self.tr('Output field name'), 'DN'))
self.addOutput(OutputVector(polygonize.OUTPUT, self.tr('Vectorized')))
self.addParameter(QgsProcessingParameterRasterLayer(self.INPUT, self.tr('Input layer')))
self.addParameter(QgsProcessingParameterBand(
self.BAND, self.tr('Band number'), parentLayerParameterName=self.INPUT))
self.addParameter(QgsProcessingParameterString(
self.FIELD, self.tr('Name of the field to create'), defaultValue='DN'))
self.addParameter(QgsProcessingParameterBoolean(
self.EIGHT_CONNECTEDNESS, self.tr('Use 8-connectedness'), defaultValue=False))
self.addParameter(QgsProcessingParameterVectorDestination(
self.OUTPUT, self.tr('Vectorized'), QgsProcessing.TypeVectorPolygon))
def name(self):
return 'polygonize'
@ -68,16 +74,29 @@ class polygonize(GdalAlgorithm):
def group(self):
return self.tr('Raster conversion')
def getConsoleCommands(self, parameters, context, feedback):
output = self.getOutputValue(polygonize.OUTPUT)
def icon(self):
return QIcon(os.path.join(pluginPath, 'images', 'gdaltools', 'polygonize.png'))
def getConsoleCommands(self, parameters, context, feedback):
arguments = []
arguments.append(self.getParameterValue(polygonize.INPUT))
arguments.append('-f')
arguments.append(GdalUtils.getVectorDriverFromFileName(output))
inLayer = self.parameterAsRasterLayer(parameters, self.INPUT, context)
arguments.append(inLayer.source())
outFile = self.parameterAsOutputLayer(parameters, self.OUTPUT, context)
output, format = GdalUtils.ogrConnectionStringAndFormat(outFile, context)
arguments.append(output)
if self.parameterAsBool(parameters, self.EIGHT_CONNECTEDNESS, context):
arguments.append('-8')
arguments.append('-b')
arguments.append(str(self.parameterAsInt(parameters, self.BAND, context)))
if format:
arguments.append('-f {}'.format(format))
arguments.append(QFileInfo(output).baseName())
arguments.append(self.getParameterValue(polygonize.FIELD))
arguments.append(self.parameterAsString(parameters, self.FIELD, context))
commands = []
if isWindows():