[processing] restore clip vector by extent algorithm

qgis · Oct 11, 2017 · 679cc2c · 679cc2c
1 parent 3750a5b
commit 679cc2c
Show file tree

Hide file tree

Showing 3 changed files with 105 additions and 110 deletions.
diff --git a/python/plugins/processing/algs/gdal/ClipVectorByExtent.py b/python/plugins/processing/algs/gdal/ClipVectorByExtent.py
@@ -0,0 +1,103 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    ClipVectorByExtent.py
+    ---------------------
+    Date                 : November 2012
+    Copyright            : (C) 2012 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__ = 'November 2012'
+__copyright__ = '(C) 2012, Victor Olaya'
+
+# This will get replaced with a git SHA1 when you do a git archive
+
+__revision__ = '$Format:%H$'
+
+from qgis.core import (QgsProcessing,
+                       QgsProcessingParameterDefinition,
+                       QgsProcessingParameterFeatureSource,
+                       QgsProcessingParameterExtent,
+                       QgsProcessingParameterString,
+                       QgsProcessingParameterVectorDestination)
+from processing.algs.gdal.GdalAlgorithm import GdalAlgorithm
+from processing.algs.gdal.GdalUtils import GdalUtils
+
+
+class ClipVectorByExtent(GdalAlgorithm):
+
+    INPUT = 'INPUT'
+    EXTENT = 'EXTENT'
+    OPTIONS = 'OPTIONS'
+    OUTPUT = 'OUTPUT'
+
+    def __init__(self):
+        super().__init__()
+
+    def initAlgorithm(self, config=None):
+        self.addParameter(QgsProcessingParameterFeatureSource(self.INPUT,
+                                                              self.tr('Input layer')))
+        self.addParameter(QgsProcessingParameterExtent(self.EXTENT,
+                                                       self.tr('Clipping extent')))
+
+        options_param = QgsProcessingParameterString(self.OPTIONS,
+                                                     self.tr('Additional creation options'),
+                                                     defaultValue='',
+                                                     optional=True)
+        options_param.setFlags(options_param.flags() | QgsProcessingParameterDefinition.FlagAdvanced)
+        self.addParameter(options_param)
+
+        self.addParameter(QgsProcessingParameterVectorDestination(self.OUTPUT,
+                                                                  self.tr('Clipped (extent)'),
+                                                                  QgsProcessing.TypeVectorPolygon))
+
+    def name(self):
+        return 'clipvectorbyextent'
+
+    def displayName(self):
+        return self.tr('Clip vector by extent')
+
+    def group(self):
+        return self.tr('Vector geoprocessing')
+
+    def commandName(self):
+        return 'ogr2ogr'
+
+    def getConsoleCommands(self, parameters, context, feedback):
+        ogrLayer, layerName = self.getOgrCompatibleSource(self.INPUT, parameters, context, feedback)
+        extent = self.parameterAsExtent(parameters, self.EXTENT, context)
+        options = self.parameterAsString(parameters, self.OPTIONS, context)
+        outFile = self.parameterAsOutputLayer(parameters, self.OUTPUT, context)
+
+        output, outputFormat = GdalUtils.ogrConnectionStringAndFormat(outFile, context)
+
+        arguments = []
+        arguments.append('-spat')
+        arguments.append(str(extent.xMinimum()))
+        arguments.append(str(extent.yMaximum()))
+        arguments.append(str(extent.xMaximum()))
+        arguments.append(str(extent.yMinimum()))
+        arguments.append('-clipsrc spat_extent')
+
+        arguments.append(output)
+        arguments.append(ogrLayer)
+        arguments.append(layerName)
+
+        if options:
+            arguments.append(options)
+
+        if outputFormat:
+            arguments.append('-f {}'.format(outputFormat))
+
+        return ['ogr2ogr', GdalUtils.escapeAndJoin(arguments)]
diff --git a/python/plugins/processing/algs/gdal/GdalAlgorithmProvider.py b/python/plugins/processing/algs/gdal/GdalAlgorithmProvider.py
@@ -74,6 +74,7 @@
 # from .rasterize_over import rasterize_over
 
 from .Buffer import Buffer
+from .ClipVectorByExtent import ClipVectorByExtent
 from .OffsetCurve import OffsetCurve
 from .ogrinfo import ogrinfo
 from .OgrToPostGis import OgrToPostGis
@@ -82,7 +83,6 @@
 
 # from .ogr2ogr import Ogr2Ogr
 # from .ogr2ogrclip import Ogr2OgrClip
-# from .ogr2ogrclipextent import Ogr2OgrClipExtent
 # from .ogr2ogrtopostgislist import Ogr2OgrToPostGisList
 # from .ogr2ogrdissolve import Ogr2OgrDissolve
 # from .ogr2ogrtabletopostgislist import Ogr2OgrTableToPostGisList
@@ -176,14 +176,14 @@ def loadAlgorithms(self):
             # rasterize_over(),
             # ----- OGR tools -----
             Buffer(),
+            ClipVectorByExtent(),
             OffsetCurve(),
             ogrinfo(),
             OgrToPostGis(),
             OneSideBuffer(),
             PointsAlongLines(),
             # Ogr2Ogr(),
             # Ogr2OgrClip(),
-            # Ogr2OgrClipExtent(),
             # Ogr2OgrToPostGisList(),
             # Ogr2OgrDissolve(),
             # Ogr2OgrTableToPostGisList(),

diff --git a/python/plugins/processing/algs/gdal/ogr2ogrclipextent.py b/python/plugins/processing/algs/gdal/ogr2ogrclipextent.py