mirror of
https://github.com/qgis/QGIS.git
synced 2025-12-15 00:07:25 -05:00
use native algorithm instead of RectanglesOvalsDiamondFixed
This commit is contained in:
Alexander Bruy
parent
9a02e216d7
commit
1fe16b712a
@ -75,7 +75,6 @@ from .RandomSelectionWithinSubsets import RandomSelectionWithinSubsets
|
||||
from .RasterCalculator import RasterCalculator
|
||||
from .RasterLayerHistogram import RasterLayerHistogram
|
||||
from .RasterSampling import RasterSampling
|
||||
from .RectanglesOvalsDiamondsFixed import RectanglesOvalsDiamondsFixed
|
||||
from .RectanglesOvalsDiamondsVariable import RectanglesOvalsDiamondsVariable
|
||||
from .RegularPoints import RegularPoints
|
||||
from .Relief import Relief
|
||||
@ -107,6 +106,7 @@ class QgisAlgorithmProvider(QgsProcessingProvider):
|
||||
super().__init__()
|
||||
self.algs = []
|
||||
self.externalAlgs = []
|
||||
QgsApplication.processingRegistry().addAlgorithmAlias('qgis:rectanglesovalsdiamondsfixed', 'native:rectanglesovalsdiamonds')
|
||||
|
||||
def getAlgs(self):
|
||||
algs = [Aggregate(),
|
||||
@ -156,7 +156,6 @@ class QgisAlgorithmProvider(QgsProcessingProvider):
|
||||
RasterCalculator(),
|
||||
RasterLayerHistogram(),
|
||||
RasterSampling(),
|
||||
RectanglesOvalsDiamondsFixed(),
|
||||
RectanglesOvalsDiamondsVariable(),
|
||||
RegularPoints(),
|
||||
Relief(),
|
||||
|
||||
@ -1,266 +0,0 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
"""
|
||||
***************************************************************************
|
||||
RectanglesOvalsDiamondsFixed.py
|
||||
---------------------
|
||||
Date : April 2016
|
||||
Copyright : (C) 2016 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__ = 'August 2012'
|
||||
__copyright__ = '(C) 2012, Victor Olaya'
|
||||
|
||||
import math
|
||||
|
||||
from qgis.core import (QgsProcessingParameterFeatureSource,
|
||||
QgsProcessingParameterEnum,
|
||||
QgsProcessingParameterNumber,
|
||||
QgsProcessingParameterFeatureSink,
|
||||
QgsProcessingParameterDistance,
|
||||
QgsProcessingException,
|
||||
QgsProcessingAlgorithm,
|
||||
QgsFeature,
|
||||
QgsFeatureSink,
|
||||
QgsGeometry,
|
||||
QgsPointXY,
|
||||
QgsWkbTypes,
|
||||
QgsProcessing)
|
||||
|
||||
from processing.algs.qgis.QgisAlgorithm import QgisAlgorithm
|
||||
|
||||
|
||||
class RectanglesOvalsDiamondsFixed(QgisAlgorithm):
|
||||
|
||||
INPUT = 'INPUT'
|
||||
SHAPE = 'SHAPE'
|
||||
WIDTH = 'WIDTH'
|
||||
HEIGHT = 'HEIGHT'
|
||||
ROTATION = 'ROTATION'
|
||||
SEGMENTS = 'SEGMENTS'
|
||||
OUTPUT = 'OUTPUT'
|
||||
|
||||
def group(self):
|
||||
return self.tr('Vector geometry')
|
||||
|
||||
def groupId(self):
|
||||
return 'vectorgeometry'
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
|
||||
def flags(self):
|
||||
return super().flags() | QgsProcessingAlgorithm.FlagDeprecated
|
||||
|
||||
def initAlgorithm(self, config=None):
|
||||
self.shapes = [self.tr('Rectangles'), self.tr('Diamonds'), self.tr('Ovals')]
|
||||
|
||||
self.addParameter(QgsProcessingParameterFeatureSource(self.INPUT,
|
||||
self.tr('Input layer'),
|
||||
[QgsProcessing.TypeVectorPoint]))
|
||||
self.addParameter(QgsProcessingParameterEnum(self.SHAPE,
|
||||
self.tr('Buffer shape'), options=self.shapes))
|
||||
self.addParameter(QgsProcessingParameterDistance(self.WIDTH, self.tr('Width'), parentParameterName=self.INPUT,
|
||||
minValue=0.0000001, defaultValue=1.0))
|
||||
self.addParameter(QgsProcessingParameterDistance(self.HEIGHT, self.tr('Height'), parentParameterName=self.INPUT,
|
||||
minValue=0.0000001, defaultValue=1.0))
|
||||
self.addParameter(QgsProcessingParameterNumber(self.ROTATION, self.tr('Rotation'), type=QgsProcessingParameterNumber.Double,
|
||||
minValue=0.0, maxValue=360.0, optional=True))
|
||||
self.addParameter(QgsProcessingParameterNumber(self.SEGMENTS,
|
||||
self.tr('Number of segments'),
|
||||
minValue=1,
|
||||
defaultValue=36))
|
||||
|
||||
self.addParameter(QgsProcessingParameterFeatureSink(self.OUTPUT,
|
||||
self.tr('Output'),
|
||||
type=QgsProcessing.TypeVectorPolygon))
|
||||
|
||||
def name(self):
|
||||
return 'rectanglesovalsdiamondsfixed'
|
||||
|
||||
def displayName(self):
|
||||
return self.tr('Rectangles, ovals, diamonds (fixed)')
|
||||
|
||||
def processAlgorithm(self, parameters, context, feedback):
|
||||
source = self.parameterAsSource(parameters, self.INPUT, context)
|
||||
if source is None:
|
||||
raise QgsProcessingException(self.invalidSourceError(parameters, self.INPUT))
|
||||
|
||||
shape = self.parameterAsEnum(parameters, self.SHAPE, context)
|
||||
width = self.parameterAsDouble(parameters, self.WIDTH, context)
|
||||
height = self.parameterAsDouble(parameters, self.HEIGHT, context)
|
||||
rotation = self.parameterAsDouble(parameters, self.ROTATION, context)
|
||||
segments = self.parameterAsInt(parameters, self.SEGMENTS, context)
|
||||
|
||||
(sink, dest_id) = self.parameterAsSink(parameters, self.OUTPUT, context,
|
||||
source.fields(), QgsWkbTypes.Polygon, source.sourceCrs())
|
||||
if sink is None:
|
||||
raise QgsProcessingException(self.invalidSinkError(parameters, self.OUTPUT))
|
||||
|
||||
if shape == 0:
|
||||
self.rectangles(sink, source, width, height, rotation, feedback)
|
||||
elif shape == 1:
|
||||
self.diamonds(sink, source, width, height, rotation, feedback)
|
||||
else:
|
||||
self.ovals(sink, source, width, height, rotation, segments, feedback)
|
||||
|
||||
return {self.OUTPUT: dest_id}
|
||||
|
||||
def rectangles(self, sink, source, width, height, rotation, feedback):
|
||||
|
||||
features = source.getFeatures()
|
||||
ft = QgsFeature()
|
||||
|
||||
xOffset = width / 2.0
|
||||
yOffset = height / 2.0
|
||||
|
||||
total = 100.0 / source.featureCount() if source.featureCount() else 0
|
||||
|
||||
if rotation is not None:
|
||||
phi = rotation * math.pi / 180
|
||||
for current, feat in enumerate(features):
|
||||
if feedback.isCanceled():
|
||||
break
|
||||
|
||||
if not feat.hasGeometry():
|
||||
continue
|
||||
|
||||
point = feat.geometry().asPoint()
|
||||
x = point.x()
|
||||
y = point.y()
|
||||
points = [(-xOffset, -yOffset), (-xOffset, yOffset), (xOffset, yOffset), (xOffset, -yOffset)]
|
||||
polygon = [[QgsPointXY(i[0] * math.cos(phi) + i[1] * math.sin(phi) + x,
|
||||
-i[0] * math.sin(phi) + i[1] * math.cos(phi) + y) for i in points]]
|
||||
|
||||
ft.setGeometry(QgsGeometry.fromPolygonXY(polygon))
|
||||
ft.setAttributes(feat.attributes())
|
||||
sink.addFeature(ft, QgsFeatureSink.FastInsert)
|
||||
|
||||
feedback.setProgress(int(current * total))
|
||||
else:
|
||||
for current, feat in enumerate(features):
|
||||
if feedback.isCanceled():
|
||||
break
|
||||
|
||||
if not feat.hasGeometry():
|
||||
continue
|
||||
|
||||
point = feat.geometry().asPoint()
|
||||
x = point.x()
|
||||
y = point.y()
|
||||
points = [(-xOffset, -yOffset), (-xOffset, yOffset), (xOffset, yOffset), (xOffset, -yOffset)]
|
||||
polygon = [[QgsPointXY(i[0] + x, i[1] + y) for i in points]]
|
||||
|
||||
ft.setGeometry(QgsGeometry.fromPolygonXY(polygon))
|
||||
ft.setAttributes(feat.attributes())
|
||||
sink.addFeature(ft, QgsFeatureSink.FastInsert)
|
||||
|
||||
feedback.setProgress(int(current * total))
|
||||
|
||||
def diamonds(self, sink, source, width, height, rotation, feedback):
|
||||
features = source.getFeatures()
|
||||
ft = QgsFeature()
|
||||
|
||||
xOffset = width / 2.0
|
||||
yOffset = height / 2.0
|
||||
|
||||
total = 100.0 / source.featureCount() if source.featureCount() else 0
|
||||
if rotation is not None:
|
||||
phi = rotation * math.pi / 180
|
||||
for current, feat in enumerate(features):
|
||||
if feedback.isCanceled():
|
||||
break
|
||||
|
||||
if not feat.hasGeometry():
|
||||
continue
|
||||
|
||||
point = feat.geometry().asPoint()
|
||||
x = point.x()
|
||||
y = point.y()
|
||||
points = [(0.0, -yOffset), (-xOffset, 0.0), (0.0, yOffset), (xOffset, 0.0)]
|
||||
polygon = [[QgsPointXY(i[0] * math.cos(phi) + i[1] * math.sin(phi) + x,
|
||||
-i[0] * math.sin(phi) + i[1] * math.cos(phi) + y) for i in points]]
|
||||
|
||||
ft.setGeometry(QgsGeometry.fromPolygonXY(polygon))
|
||||
ft.setAttributes(feat.attributes())
|
||||
sink.addFeature(ft, QgsFeatureSink.FastInsert)
|
||||
feedback.setProgress(int(current * total))
|
||||
else:
|
||||
for current, feat in enumerate(features):
|
||||
if feedback.isCanceled():
|
||||
break
|
||||
|
||||
if not feat.hasGeometry():
|
||||
continue
|
||||
|
||||
point = feat.geometry().asPoint()
|
||||
x = point.x()
|
||||
y = point.y()
|
||||
points = [(0.0, -yOffset), (-xOffset, 0.0), (0.0, yOffset), (xOffset, 0.0)]
|
||||
polygon = [[QgsPointXY(i[0] + x, i[1] + y) for i in points]]
|
||||
|
||||
ft.setGeometry(QgsGeometry.fromPolygonXY(polygon))
|
||||
ft.setAttributes(feat.attributes())
|
||||
sink.addFeature(ft, QgsFeatureSink.FastInsert)
|
||||
feedback.setProgress(int(current * total))
|
||||
|
||||
def ovals(self, sink, source, width, height, rotation, segments, feedback):
|
||||
features = source.getFeatures()
|
||||
ft = QgsFeature()
|
||||
|
||||
xOffset = width / 2.0
|
||||
yOffset = height / 2.0
|
||||
|
||||
total = 100.0 / source.featureCount() if source.featureCount() else 0
|
||||
if rotation is not None:
|
||||
phi = rotation * math.pi / 180
|
||||
for current, feat in enumerate(features):
|
||||
if feedback.isCanceled():
|
||||
break
|
||||
|
||||
if not feat.hasGeometry():
|
||||
continue
|
||||
|
||||
point = feat.geometry().asPoint()
|
||||
x = point.x()
|
||||
y = point.y()
|
||||
points = []
|
||||
for t in [(2 * math.pi) / segments * i for i in range(segments)]:
|
||||
points.append((xOffset * math.cos(t), yOffset * math.sin(t)))
|
||||
polygon = [[QgsPointXY(i[0] * math.cos(phi) + i[1] * math.sin(phi) + x,
|
||||
-i[0] * math.sin(phi) + i[1] * math.cos(phi) + y) for i in points]]
|
||||
|
||||
ft.setGeometry(QgsGeometry.fromPolygonXY(polygon))
|
||||
ft.setAttributes(feat.attributes())
|
||||
sink.addFeature(ft, QgsFeatureSink.FastInsert)
|
||||
feedback.setProgress(int(current * total))
|
||||
else:
|
||||
for current, feat in enumerate(features):
|
||||
if feedback.isCanceled():
|
||||
break
|
||||
|
||||
if not feat.hasGeometry():
|
||||
continue
|
||||
|
||||
point = feat.geometry().asPoint()
|
||||
x = point.x()
|
||||
y = point.y()
|
||||
points = []
|
||||
for t in [(2 * math.pi) / segments * i for i in range(segments)]:
|
||||
points.append((xOffset * math.cos(t), yOffset * math.sin(t)))
|
||||
polygon = [[QgsPointXY(i[0] + x, i[1] + y) for i in points]]
|
||||
|
||||
ft.setGeometry(QgsGeometry.fromPolygonXY(polygon))
|
||||
ft.setAttributes(feat.attributes())
|
||||
sink.addFeature(ft, QgsFeatureSink.FastInsert)
|
||||
feedback.setProgress(int(current * total))
|
||||
Loading…
x
Reference in New Issue
Block a user