QGIS API Documentation 3.40.0-Bratislava (b56115d8743)
Loading...
Searching...
No Matches
qgsalgorithmunion.cpp
Go to the documentation of this file.
1/***************************************************************************
2 qgsalgorithmunion.cpp
3 ---------------------
4 Date : April 2018
5 Copyright : (C) 2018 by Martin Dobias
6 Email : wonder dot sk at gmail dot com
7 ***************************************************************************
8 * *
9 * This program is free software; you can redistribute it and/or modify *
10 * it under the terms of the GNU General Public License as published by *
11 * the Free Software Foundation; either version 2 of the License, or *
12 * (at your option) any later version. *
13 * *
14 ***************************************************************************/
15
16#include "qgsalgorithmunion.h"
17
18#include "qgsoverlayutils.h"
19
21
22
23QString QgsUnionAlgorithm::name() const
24{
25 return QStringLiteral( "union" );
26}
27
28QString QgsUnionAlgorithm::displayName() const
29{
30 return QObject::tr( "Union" );
31}
32
33QString QgsUnionAlgorithm::group() const
34{
35 return QObject::tr( "Vector overlay" );
36}
37
38QString QgsUnionAlgorithm::groupId() const
39{
40 return QStringLiteral( "vectoroverlay" );
41}
42
43QString QgsUnionAlgorithm::shortHelpString() const
44{
45 return QObject::tr( "This algorithm checks overlaps between features within the Input layer and creates separate features for overlapping "
46 "and non-overlapping parts. The area of overlap will create as many identical overlapping features as there are "
47 "features that participate in that overlap." )
48 + QStringLiteral( "\n\n" )
49 + QObject::tr( "An Overlay layer can also be used, in which case features from each layer are split at their overlap with features from "
50 "the other one, creating a layer containing all the portions from both Input and Overlay layers. "
51 "The attribute table of the Union layer is filled with attribute values from the respective original layer "
52 "for non-overlapping features, and attribute values from both layers for overlapping features." );
53}
54
55Qgis::ProcessingAlgorithmDocumentationFlags QgsUnionAlgorithm::documentationFlags() const
56{
58}
59
60QgsProcessingAlgorithm *QgsUnionAlgorithm::createInstance() const
61{
62 return new QgsUnionAlgorithm();
63}
64
65void QgsUnionAlgorithm::initAlgorithm( const QVariantMap & )
66{
67 addParameter( new QgsProcessingParameterFeatureSource( QStringLiteral( "INPUT" ), QObject::tr( "Input layer" ) ) );
68 addParameter( new QgsProcessingParameterFeatureSource( QStringLiteral( "OVERLAY" ), QObject::tr( "Overlay layer" ), QList< int >(), QVariant(), true ) );
69
70 std::unique_ptr< QgsProcessingParameterString > prefix = std::make_unique< QgsProcessingParameterString >( QStringLiteral( "OVERLAY_FIELDS_PREFIX" ), QObject::tr( "Overlay fields prefix" ), QString(), false, true );
71 prefix->setFlags( prefix->flags() | Qgis::ProcessingParameterFlag::Advanced );
72 addParameter( prefix.release() );
73
74 addParameter( new QgsProcessingParameterFeatureSink( QStringLiteral( "OUTPUT" ), QObject::tr( "Union" ) ) );
75
76 std::unique_ptr< QgsProcessingParameterNumber > gridSize = std::make_unique< QgsProcessingParameterNumber >( QStringLiteral( "GRID_SIZE" ),
77 QObject::tr( "Grid size" ), Qgis::ProcessingNumberParameterType::Double, QVariant(), true, 0 );
78 gridSize->setFlags( gridSize->flags() | Qgis::ProcessingParameterFlag::Advanced );
79 addParameter( gridSize.release() );
80}
81
82QVariantMap QgsUnionAlgorithm::processAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
83{
84 std::unique_ptr< QgsFeatureSource > sourceA( parameterAsSource( parameters, QStringLiteral( "INPUT" ), context ) );
85 if ( !sourceA )
86 throw QgsProcessingException( invalidSourceError( parameters, QStringLiteral( "INPUT" ) ) );
87
88 std::unique_ptr< QgsFeatureSource > sourceB( parameterAsSource( parameters, QStringLiteral( "OVERLAY" ), context ) );
89 if ( parameters.value( QStringLiteral( "OVERLAY" ) ).isValid() && !sourceB )
90 throw QgsProcessingException( invalidSourceError( parameters, QStringLiteral( "OVERLAY" ) ) );
91
92 const Qgis::WkbType geomType = QgsWkbTypes::multiType( sourceA->wkbType() );
93
94 const QString overlayFieldsPrefix = parameterAsString( parameters, QStringLiteral( "OVERLAY_FIELDS_PREFIX" ), context );
95 const QgsFields fields = sourceB ? QgsProcessingUtils::combineFields( sourceA->fields(), sourceB->fields(), overlayFieldsPrefix ) : sourceA->fields();
96
97 QString dest;
98 std::unique_ptr< QgsFeatureSink > sink( parameterAsSink( parameters, QStringLiteral( "OUTPUT" ), context, dest, fields, geomType, sourceA->sourceCrs(), QgsFeatureSink::RegeneratePrimaryKey ) );
99 if ( !sink )
100 throw QgsProcessingException( invalidSinkError( parameters, QStringLiteral( "OUTPUT" ) ) );
101
102 QVariantMap outputs;
103 outputs.insert( QStringLiteral( "OUTPUT" ), dest );
104
105 if ( !sourceB )
106 {
107 // we are doing single layer union
108 QgsOverlayUtils::resolveOverlaps( *sourceA, *sink, feedback );
109 return outputs;
110 }
111
112 const QList<int> fieldIndicesA = QgsProcessingUtils::fieldNamesToIndices( QStringList(), sourceA->fields() );
113 const QList<int> fieldIndicesB = QgsProcessingUtils::fieldNamesToIndices( QStringList(), sourceB->fields() );
114
115 long count = 0;
116 const long total = sourceA->featureCount() * 2 + sourceB->featureCount();
117
118 QgsGeometryParameters geometryParameters;
119 if ( parameters.value( QStringLiteral( "GRID_SIZE" ) ).isValid() )
120 {
121 geometryParameters.setGridSize( parameterAsDouble( parameters, QStringLiteral( "GRID_SIZE" ), context ) );
122 }
123
124 QgsOverlayUtils::intersection( *sourceA, *sourceB, *sink, context, feedback, count, total, fieldIndicesA, fieldIndicesB, geometryParameters );
125 if ( feedback->isCanceled() )
126 return outputs;
127
128 QgsOverlayUtils::difference( *sourceA, *sourceB, *sink, context, feedback, count, total, QgsOverlayUtils::OutputAB, geometryParameters );
129 if ( feedback->isCanceled() )
130 return outputs;
131
132 QgsOverlayUtils::difference( *sourceB, *sourceA, *sink, context, feedback, count, total, QgsOverlayUtils::OutputBA, geometryParameters );
133
134 return outputs;
135}
136
@ 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
WkbType
The WKB type describes the number of dimensions a geometry has.
Definition qgis.h:256
@ Advanced
Parameter is an advanced parameter which should be hidden from users by default.
@ RegeneratePrimaryKey
This flag indicates, that a primary key field cannot be guaranteed to be unique and the sink should i...
bool isCanceled() const
Tells whether the operation has been canceled already.
Definition qgsfeedback.h:53
Container of fields for a vector layer.
Definition qgsfields.h:46
Encapsulates parameters under which a geometry operation is performed.
void setGridSize(double size)
Sets the grid size which will be used to snap vertices of a geometry.
Abstract base class for processing algorithms.
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.
A feature sink output for processing algorithms.
An input feature source (such as vector layers) parameter for processing algorithms.
static QList< int > fieldNamesToIndices(const QStringList &fieldNames, const QgsFields &fields)
Returns a list of field indices parsed from the given list of field names.
static QgsFields combineFields(const QgsFields &fieldsA, const QgsFields &fieldsB, const QString &fieldsBPrefix=QString())
Combines two field lists, avoiding duplicate field names (in a case-insensitive manner).
static Qgis::WkbType multiType(Qgis::WkbType type)
Returns the multi type for a WKB type.