QGIS API Documentation 3.40.0-Bratislava (b56115d8743)
Loading...
Searching...
No Matches
qgsalgorithmextractvertices.cpp
Go to the documentation of this file.
1/***************************************************************************
2 qgsalgorithmextractvertices.cpp
3 --------------------------
4 begin : November 2017
5 copyright : (C) 2017 by Mathieu Pellerin
6 email : nirvn dot asia at gmail dot com
7 ***************************************************************************/
8
9/***************************************************************************
10 * *
11 * This program is free software; you can redistribute it and/or modify *
12 * it under the terms of the GNU General Public License as published by *
13 * the Free Software Foundation; either version 2 of the License, or *
14 * (at your option) any later version. *
15 * *
16 ***************************************************************************/
17
19
20#include "qgsabstractgeometry.h"
21#include "qgsgeometryutils.h"
22
24
25QString QgsExtractVerticesAlgorithm::name() const
26{
27 return QStringLiteral( "extractvertices" );
28}
29
30QString QgsExtractVerticesAlgorithm::displayName() const
31{
32 return QObject::tr( "Extract vertices" );
33}
34
35QStringList QgsExtractVerticesAlgorithm::tags() const
36{
37 return QObject::tr( "points,vertex,nodes" ).split( ',' );
38}
39
40QString QgsExtractVerticesAlgorithm::group() const
41{
42 return QObject::tr( "Vector geometry" );
43}
44
45QString QgsExtractVerticesAlgorithm::groupId() const
46{
47 return QStringLiteral( "vectorgeometry" );
48}
49
50QString QgsExtractVerticesAlgorithm::shortHelpString() const
51{
52 return QObject::tr( "This algorithm takes a vector layer and generates a point layer with points representing the vertices in the input geometries. The attributes associated to each point are the same ones associated to the feature that the point belongs to." ) +
53 QStringLiteral( "\n\n" ) +
54 QObject::tr( "Additional fields are added to the point indicating the vertex index (beginning at 0), the vertex’s part and its index within the part (as well as its ring for polygons), distance along original geometry and bisector angle of vertex for original geometry." );
55}
56
57Qgis::ProcessingAlgorithmDocumentationFlags QgsExtractVerticesAlgorithm::documentationFlags() const
58{
60}
61
62QString QgsExtractVerticesAlgorithm::outputName() const
63{
64 return QObject::tr( "Vertices" );
65}
66
67QgsExtractVerticesAlgorithm *QgsExtractVerticesAlgorithm::createInstance() const
68{
69 return new QgsExtractVerticesAlgorithm();
70}
71
72Qgis::ProcessingSourceType QgsExtractVerticesAlgorithm::outputLayerType() const
73{
75}
76
77QgsFields QgsExtractVerticesAlgorithm::outputFields( const QgsFields &inputFields ) const
78{
79 QgsFields outputFields = inputFields;
80 outputFields.append( QgsField( QStringLiteral( "vertex_index" ), QMetaType::Type::Int, QString(), 10, 0 ) );
81 outputFields.append( QgsField( QStringLiteral( "vertex_part" ), QMetaType::Type::Int, QString(), 10, 0 ) );
82 if ( mGeometryType == Qgis::GeometryType::Polygon )
83 {
84 outputFields.append( QgsField( QStringLiteral( "vertex_part_ring" ), QMetaType::Type::Int, QString(), 10, 0 ) );
85 }
86 outputFields.append( QgsField( QStringLiteral( "vertex_part_index" ), QMetaType::Type::Int, QString(), 10, 0 ) );
87 outputFields.append( QgsField( QStringLiteral( "distance" ), QMetaType::Type::Double, QString(), 20, 14 ) );
88 outputFields.append( QgsField( QStringLiteral( "angle" ), QMetaType::Type::Double, QString(), 20, 14 ) );
89
90 return outputFields;
91}
92
93Qgis::WkbType QgsExtractVerticesAlgorithm::outputWkbType( Qgis::WkbType inputWkbType ) const
94{
95 Qgis::WkbType outputWkbType = Qgis::WkbType::Point;
96 if ( QgsWkbTypes::hasM( inputWkbType ) )
97 {
98 outputWkbType = QgsWkbTypes::addM( outputWkbType );
99 }
100 if ( QgsWkbTypes::hasZ( inputWkbType ) )
101 {
102 outputWkbType = QgsWkbTypes::addZ( outputWkbType );
103 }
104
105 return outputWkbType;
106}
107
108Qgis::ProcessingFeatureSourceFlags QgsExtractVerticesAlgorithm::sourceFlags() const
109{
111}
112
113QgsFeatureSink::SinkFlags QgsExtractVerticesAlgorithm::sinkFlags() const
114{
116}
117
118bool QgsExtractVerticesAlgorithm::prepareAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback * )
119{
120 std::unique_ptr< QgsProcessingFeatureSource > source( parameterAsSource( parameters, QStringLiteral( "INPUT" ), context ) );
121 if ( !source )
122 throw QgsProcessingException( invalidSourceError( parameters, QStringLiteral( "INPUT" ) ) );
123
124 mGeometryType = QgsWkbTypes::geometryType( source->wkbType() );
125 return true;
126}
127
128QgsFeatureList QgsExtractVerticesAlgorithm::processFeature( const QgsFeature &feature, QgsProcessingContext &, QgsProcessingFeedback * )
129{
130 QgsFeatureList outputFeatures;
131
132 QgsFeature f = feature;
133 const QgsGeometry inputGeom = f.geometry();
134 if ( inputGeom.isEmpty() )
135 {
136 QgsAttributes attrs = f.attributes();
137 attrs << QVariant()
138 << QVariant();
139 if ( mGeometryType == Qgis::GeometryType::Polygon )
140 {
141 attrs << QVariant();
142 }
143 attrs << QVariant()
144 << QVariant()
145 << QVariant();
146
147 f.clearGeometry();
148 f.setAttributes( attrs );
149 outputFeatures << f;
150 }
151 else
152 {
154 double cumulativeDistance = 0.0;
155 int vertexPos = 0;
156 while ( vi != inputGeom.constGet()->vertices_end() )
157 {
158 const QgsVertexId vertexId = vi.vertexId();
159 const double angle = inputGeom.constGet()->vertexAngle( vertexId ) * 180 / M_PI;
160 QgsAttributes attrs = f.attributes();
161 attrs << vertexPos
162 << vertexId.part;
163 if ( mGeometryType == Qgis::GeometryType::Polygon )
164 {
165 attrs << vertexId.ring;
166 }
167 attrs << vertexId.vertex
168 << cumulativeDistance
169 << angle;
170
171 QgsFeature outputFeature = QgsFeature();
172 outputFeature.setAttributes( attrs );
173 outputFeature.setGeometry( QgsGeometry( ( *vi ).clone() ) );
174 outputFeatures << outputFeature;
175 vi++;
176 vertexPos++;
177
178 // calculate distance to next vertex
179 const double distanceToNext = inputGeom.constGet()->segmentLength( vertexId );
180 cumulativeDistance += distanceToNext;
181 }
182 }
183
184 return outputFeatures;
185}
186
ProcessingSourceType
Processing data source types.
Definition qgis.h:3262
@ VectorPoint
Vector point layers.
@ Polygon
Polygons.
@ RegeneratesPrimaryKey
Algorithm always drops any existing primary keys or FID values and regenerates them in outputs.
QFlags< ProcessingAlgorithmDocumentationFlag > ProcessingAlgorithmDocumentationFlags
Flags describing algorithm behavior for documentation purposes.
Definition qgis.h:3359
@ SkipGeometryValidityChecks
Invalid geometry checks should always be skipped. This flag can be useful for algorithms which always...
WkbType
The WKB type describes the number of dimensions a geometry has.
Definition qgis.h:256
QFlags< ProcessingFeatureSourceFlag > ProcessingFeatureSourceFlags
Flags which control how QgsProcessingFeatureSource fetches features.
Definition qgis.h:3436
The vertex_iterator class provides STL-style iterator for vertices.
QgsVertexId vertexId() const
Returns vertex ID of the current item.
virtual double vertexAngle(QgsVertexId vertex) const =0
Returns approximate angle at a vertex.
vertex_iterator vertices_end() const
Returns STL-style iterator pointing to the imaginary vertex after the last vertex of the geometry.
vertex_iterator vertices_begin() const
Returns STL-style iterator pointing to the first vertex of the geometry.
virtual double segmentLength(QgsVertexId startVertex) const =0
Returns the length of the segment of the geometry which begins at startVertex.
A vector of attributes.
QFlags< SinkFlag > SinkFlags
@ RegeneratePrimaryKey
This flag indicates, that a primary key field cannot be guaranteed to be unique and the sink should i...
The feature class encapsulates a single feature including its unique ID, geometry and a list of field...
Definition qgsfeature.h:58
QgsAttributes attributes
Definition qgsfeature.h:67
void setAttributes(const QgsAttributes &attrs)
Sets the feature's attributes.
QgsGeometry geometry
Definition qgsfeature.h:69
void clearGeometry()
Removes any geometry associated with the feature.
void setGeometry(const QgsGeometry &geometry)
Set the feature's geometry.
Encapsulate a field in an attribute table or data source.
Definition qgsfield.h:53
Container of fields for a vector layer.
Definition qgsfields.h:46
bool append(const QgsField &field, Qgis::FieldOrigin origin=Qgis::FieldOrigin::Provider, int originIndex=-1)
Appends a field.
Definition qgsfields.cpp:69
A geometry is the spatial representation of a feature.
const QgsAbstractGeometry * constGet() const
Returns a non-modifiable (const) reference to the underlying abstract geometry primitive.
bool isEmpty() const
Returns true if the geometry is empty (eg a linestring with no vertices, or a collection with no geom...
Contains information about the context in which a processing algorithm is executed.
Custom exception class for processing related exceptions.
Base class for providing feedback from a processing algorithm.
static Qgis::GeometryType geometryType(Qgis::WkbType type)
Returns the geometry type for a WKB type, e.g., both MultiPolygon and CurvePolygon would have a Polyg...
static Qgis::WkbType addM(Qgis::WkbType type)
Adds the m dimension to a WKB type and returns the new type.
static Qgis::WkbType addZ(Qgis::WkbType type)
Adds the z dimension to a WKB type and returns the new type.
static bool hasZ(Qgis::WkbType type)
Tests whether a WKB type contains the z-dimension.
static bool hasM(Qgis::WkbType type)
Tests whether a WKB type contains m values.
double ANALYSIS_EXPORT angle(QgsPoint *p1, QgsPoint *p2, QgsPoint *p3, QgsPoint *p4)
Calculates the angle between two segments (in 2 dimension, z-values are ignored)
QList< QgsFeature > QgsFeatureList
Utility class for identifying a unique vertex within a geometry.
Definition qgsvertexid.h:30
int vertex
Vertex number.
Definition qgsvertexid.h:94
int part
Part number.
Definition qgsvertexid.h:88
int ring
Ring number.
Definition qgsvertexid.h:91