QGIS API Documentation 3.37.0-Master (fdefdf9c27f)
qgsalgorithmextractbyextent.cpp
Go to the documentation of this file.
1/***************************************************************************
2 qgsalgorithmextractbyextent.cpp
3 ---------------------
4 begin : April 2017
5 copyright : (C) 2017 by Nyall Dawson
6 email : nyall dot dawson 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
21
22QString QgsExtractByExtentAlgorithm::name() const
23{
24 return QStringLiteral( "extractbyextent" );
25}
26
27QString QgsExtractByExtentAlgorithm::displayName() const
28{
29 return QObject::tr( "Extract/clip by extent" );
30}
31
32QStringList QgsExtractByExtentAlgorithm::tags() const
33{
34 return QObject::tr( "clip,extract,intersect,intersection,mask,extent" ).split( ',' );
35}
36
37QString QgsExtractByExtentAlgorithm::group() const
38{
39 return QObject::tr( "Vector overlay" );
40}
41
42QString QgsExtractByExtentAlgorithm::groupId() const
43{
44 return QStringLiteral( "vectoroverlay" );
45}
46void QgsExtractByExtentAlgorithm::initAlgorithm( const QVariantMap & )
47{
48 addParameter( new QgsProcessingParameterFeatureSource( QStringLiteral( "INPUT" ), QObject::tr( "Input layer" ) ) );
49 addParameter( new QgsProcessingParameterExtent( QStringLiteral( "EXTENT" ), QObject::tr( "Extent" ) ) );
50 addParameter( new QgsProcessingParameterBoolean( QStringLiteral( "CLIP" ), QObject::tr( "Clip features to extent" ), false ) );
51 addParameter( new QgsProcessingParameterFeatureSink( QStringLiteral( "OUTPUT" ), QObject::tr( "Extracted" ) ) );
52}
53
54QString QgsExtractByExtentAlgorithm::shortHelpString() const
55{
56 return QObject::tr( "This algorithm creates a new vector layer that only contains features which fall within a specified extent. "
57 "Any features which intersect the extent will be included.\n\n"
58 "Optionally, feature geometries can also be clipped to the extent. If this option is selected, then the output "
59 "geometries will automatically be converted to multi geometries to ensure uniform output geometry types." );
60}
61
62QgsExtractByExtentAlgorithm *QgsExtractByExtentAlgorithm::createInstance() const
63{
64 return new QgsExtractByExtentAlgorithm();
65}
66
67QVariantMap QgsExtractByExtentAlgorithm::processAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
68{
69 std::unique_ptr< QgsFeatureSource > featureSource( parameterAsSource( parameters, QStringLiteral( "INPUT" ), context ) );
70 if ( !featureSource )
71 throw QgsProcessingException( invalidSourceError( parameters, QStringLiteral( "INPUT" ) ) );
72
73 if ( featureSource->hasSpatialIndex() == Qgis::SpatialIndexPresence::NotPresent )
74 feedback->pushWarning( QObject::tr( "No spatial index exists for input layer, performance will be severely degraded" ) );
75
76 const QgsRectangle extent = parameterAsExtent( parameters, QStringLiteral( "EXTENT" ), context, featureSource->sourceCrs() );
77 const bool clip = parameterAsBoolean( parameters, QStringLiteral( "CLIP" ), context );
78
79 // if clipping, we force multi output
80 const Qgis::WkbType outType = clip ? QgsWkbTypes::promoteNonPointTypesToMulti( featureSource->wkbType() ) : featureSource->wkbType();
81
82 QString dest;
83 std::unique_ptr< QgsFeatureSink > sink( parameterAsSink( parameters, QStringLiteral( "OUTPUT" ), context, dest, featureSource->fields(), outType, featureSource->sourceCrs() ) );
84
85 if ( !sink )
86 throw QgsProcessingException( invalidSinkError( parameters, QStringLiteral( "OUTPUT" ) ) );
87
88 const QgsGeometry clipGeom = parameterAsExtentGeometry( parameters, QStringLiteral( "EXTENT" ), context, featureSource->sourceCrs() );
89
90 const double step = featureSource->featureCount() > 0 ? 100.0 / featureSource->featureCount() : 1;
91 QgsFeatureIterator inputIt = featureSource->getFeatures( QgsFeatureRequest().setFilterRect( extent ).setFlags( Qgis::FeatureRequestFlag::ExactIntersect ) );
92 QgsFeature f;
93 int i = -1;
94 while ( inputIt.nextFeature( f ) )
95 {
96 i++;
97 if ( feedback->isCanceled() )
98 {
99 break;
100 }
101
102 if ( clip )
103 {
104 QgsGeometry g = f.geometry().intersection( clipGeom );
105
106 if ( g.type() != Qgis::GeometryType::Point )
107 {
108 // some data providers are picky about the geometries we pass to them: we can't add single-part geometries
109 // when we promised multi-part geometries, so ensure we have the right type
111 }
112
113 f.setGeometry( g );
114 }
115
116 if ( !sink->addFeature( f, QgsFeatureSink::FastInsert ) )
117 throw QgsProcessingException( writeFeatureError( sink.get(), parameters, QStringLiteral( "OUTPUT" ) ) );
118 feedback->setProgress( i * step );
119 }
120
121 QVariantMap outputs;
122 outputs.insert( QStringLiteral( "OUTPUT" ), dest );
123 return outputs;
124}
125
127
@ NotPresent
No spatial index exists for the source.
@ ExactIntersect
Use exact geometry intersection (slower) instead of bounding boxes.
WkbType
The WKB type describes the number of dimensions a geometry has.
Definition: qgis.h:182
Wrapper for iterator of features from vector data provider or vector layer.
bool nextFeature(QgsFeature &f)
Fetch next feature and stores in f, returns true on success.
This class wraps a request for features to a vector layer (or directly its vector data provider).
@ FastInsert
Use faster inserts, at the cost of updating the passed features to reflect changes made at the provid...
The feature class encapsulates a single feature including its unique ID, geometry and a list of field...
Definition: qgsfeature.h:56
QgsGeometry geometry
Definition: qgsfeature.h:67
void setGeometry(const QgsGeometry &geometry)
Set the feature's geometry.
Definition: qgsfeature.cpp:167
bool isCanceled() const
Tells whether the operation has been canceled already.
Definition: qgsfeedback.h:53
void setProgress(double progress)
Sets the current progress for the feedback object.
Definition: qgsfeedback.h:61
A geometry is the spatial representation of a feature.
Definition: qgsgeometry.h:162
Qgis::GeometryType type
Definition: qgsgeometry.h:165
QgsGeometry intersection(const QgsGeometry &geometry, const QgsGeometryParameters &parameters=QgsGeometryParameters()) const
Returns a geometry representing the points shared by this geometry and other.
bool convertToMultiType()
Converts single type geometry into multitype geometry e.g.
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.
virtual void pushWarning(const QString &warning)
Pushes a warning informational message from the algorithm.
A boolean parameter for processing algorithms.
A rectangular map extent parameter for processing algorithms.
A feature sink output for processing algorithms.
An input feature source (such as vector layers) parameter for processing algorithms.
A rectangle specified with double values.
Definition: qgsrectangle.h:42
static Qgis::WkbType promoteNonPointTypesToMulti(Qgis::WkbType type)
Promotes a WKB geometry type to its multi-type equivalent, with the exception of point geometry types...
Definition: qgswkbtypes.h:347