QGIS API Documentation 3.37.0-Master (fdefdf9c27f)
qgsalgorithmintersection.cpp
Go to the documentation of this file.
1/***************************************************************************
2 qgsalgorithmintersection.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
17
19#include "qgsgeometryengine.h"
20#include "qgsoverlayutils.h"
21
23
24
25QString QgsIntersectionAlgorithm::name() const
26{
27 return QStringLiteral( "intersection" );
28}
29
30QString QgsIntersectionAlgorithm::displayName() const
31{
32 return QObject::tr( "Intersection" );
33}
34
35QString QgsIntersectionAlgorithm::group() const
36{
37 return QObject::tr( "Vector overlay" );
38}
39
40QString QgsIntersectionAlgorithm::groupId() const
41{
42 return QStringLiteral( "vectoroverlay" );
43}
44
45QString QgsIntersectionAlgorithm::shortHelpString() const
46{
47 return QObject::tr( "This algorithm extracts the overlapping portions of features in the Input and Overlay layers. "
48 "Features in the output Intersection layer are assigned the attributes of the overlapping features "
49 "from both the Input and Overlay layers." );
50}
51
52QgsProcessingAlgorithm *QgsIntersectionAlgorithm::createInstance() const
53{
54 return new QgsIntersectionAlgorithm();
55}
56
57void QgsIntersectionAlgorithm::initAlgorithm( const QVariantMap & )
58{
59 addParameter( new QgsProcessingParameterFeatureSource( QStringLiteral( "INPUT" ), QObject::tr( "Input layer" ) ) );
60 addParameter( new QgsProcessingParameterFeatureSource( QStringLiteral( "OVERLAY" ), QObject::tr( "Overlay layer" ) ) );
61
62 addParameter( new QgsProcessingParameterField(
63 QStringLiteral( "INPUT_FIELDS" ),
64 QObject::tr( "Input fields to keep (leave empty to keep all fields)" ), QVariant(),
65 QStringLiteral( "INPUT" ), Qgis::ProcessingFieldParameterDataType::Any, true, true ) );
66 addParameter( new QgsProcessingParameterField(
67 QStringLiteral( "OVERLAY_FIELDS" ),
68 QObject::tr( "Overlay fields to keep (leave empty to keep all fields)" ), QVariant(),
69 QStringLiteral( "OVERLAY" ), Qgis::ProcessingFieldParameterDataType::Any, true, true ) );
70
71 std::unique_ptr< QgsProcessingParameterString > prefix = std::make_unique< QgsProcessingParameterString >( QStringLiteral( "OVERLAY_FIELDS_PREFIX" ), QObject::tr( "Overlay fields prefix" ), QString(), false, true );
72 prefix->setFlags( prefix->flags() | Qgis::ProcessingParameterFlag::Advanced );
73 addParameter( prefix.release() );
74
75 addParameter( new QgsProcessingParameterFeatureSink( QStringLiteral( "OUTPUT" ), QObject::tr( "Intersection" ) ) );
76
77 std::unique_ptr< QgsProcessingParameterNumber > gridSize = std::make_unique< QgsProcessingParameterNumber >( QStringLiteral( "GRID_SIZE" ),
78 QObject::tr( "Grid size" ), Qgis::ProcessingNumberParameterType::Double, QVariant(), true, 0 );
79 gridSize->setFlags( gridSize->flags() | Qgis::ProcessingParameterFlag::Advanced );
80 addParameter( gridSize.release() );
81}
82
83
84QVariantMap QgsIntersectionAlgorithm::processAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
85{
86 std::unique_ptr< QgsFeatureSource > sourceA( parameterAsSource( parameters, QStringLiteral( "INPUT" ), context ) );
87 if ( !sourceA )
88 throw QgsProcessingException( invalidSourceError( parameters, QStringLiteral( "INPUT" ) ) );
89
90 std::unique_ptr< QgsFeatureSource > sourceB( parameterAsSource( parameters, QStringLiteral( "OVERLAY" ), context ) );
91 if ( !sourceB )
92 throw QgsProcessingException( invalidSourceError( parameters, QStringLiteral( "OVERLAY" ) ) );
93
94 const Qgis::WkbType geomType = QgsWkbTypes::multiType( sourceA->wkbType() );
95
96 const QStringList fieldsA = parameterAsStrings( parameters, QStringLiteral( "INPUT_FIELDS" ), context );
97 const QStringList fieldsB = parameterAsStrings( parameters, QStringLiteral( "OVERLAY_FIELDS" ), context );
98
99 const QList<int> fieldIndicesA = QgsProcessingUtils::fieldNamesToIndices( fieldsA, sourceA->fields() );
100 const QList<int> fieldIndicesB = QgsProcessingUtils::fieldNamesToIndices( fieldsB, sourceB->fields() );
101
102 const QString overlayFieldsPrefix = parameterAsString( parameters, QStringLiteral( "OVERLAY_FIELDS_PREFIX" ), context );
103 const QgsFields outputFields = QgsProcessingUtils::combineFields(
104 QgsProcessingUtils::indicesToFields( fieldIndicesA, sourceA->fields() ),
105 QgsProcessingUtils::indicesToFields( fieldIndicesB, sourceB->fields() ),
106 overlayFieldsPrefix );
107
108 QString dest;
109 std::unique_ptr< QgsFeatureSink > sink( parameterAsSink( parameters, QStringLiteral( "OUTPUT" ), context, dest, outputFields, geomType, sourceA->sourceCrs(), QgsFeatureSink::RegeneratePrimaryKey ) );
110 if ( !sink )
111 throw QgsProcessingException( invalidSinkError( parameters, QStringLiteral( "OUTPUT" ) ) );
112
113 QVariantMap outputs;
114 outputs.insert( QStringLiteral( "OUTPUT" ), dest );
115
116 long count = 0;
117 const long total = sourceA->featureCount();
118
119 QgsGeometryParameters geometryParameters;
120 if ( parameters.value( QStringLiteral( "GRID_SIZE" ) ).isValid() )
121 {
122 geometryParameters.setGridSize( parameterAsDouble( parameters, QStringLiteral( "GRID_SIZE" ), context ) );
123 }
124
125 QgsOverlayUtils::intersection( *sourceA, *sourceB, *sink, context, feedback, count, total, fieldIndicesA, fieldIndicesB, geometryParameters );
126
127 return outputs;
128}
129
WkbType
The WKB type describes the number of dimensions a geometry has.
Definition: qgis.h:182
@ 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...
Container of fields for a vector layer.
Definition: qgsfields.h:45
Encapsulates parameters under which a geometry operation is performed.
Definition: qgsgeometry.h:109
void setGridSize(double size)
Sets the grid size which will be used to snap vertices of a geometry.
Definition: qgsgeometry.h:134
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.
Definition: qgsexception.h:83
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.
A vector layer or feature source field parameter for processing algorithms.
static QgsFields indicesToFields(const QList< int > &indices, const QgsFields &fields)
Returns a subset of fields based on the indices of desired fields.
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.
Definition: qgswkbtypes.h:200