QGIS API Documentation  3.14.0-Pi (9f7028fd23)
qgsalgorithmlineintersection.cpp
Go to the documentation of this file.
1 /***************************************************************************
2  qgsalgorithmlineintersection.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 #include "qgsgeometryengine.h"
20 
22 
23 QString QgsLineIntersectionAlgorithm::name() const
24 {
25  return QStringLiteral( "lineintersections" );
26 }
27 
28 QString QgsLineIntersectionAlgorithm::displayName() const
29 {
30  return QObject::tr( "Line intersections" );
31 }
32 
33 QStringList QgsLineIntersectionAlgorithm::tags() const
34 {
35  return QObject::tr( "line,intersection" ).split( ',' );
36 }
37 
38 QString QgsLineIntersectionAlgorithm::group() const
39 {
40  return QObject::tr( "Vector overlay" );
41 }
42 
43 QString QgsLineIntersectionAlgorithm::groupId() const
44 {
45  return QStringLiteral( "vectoroverlay" );
46 }
47 
48 void QgsLineIntersectionAlgorithm::initAlgorithm( const QVariantMap & )
49 {
50  addParameter( new QgsProcessingParameterFeatureSource( QStringLiteral( "INPUT" ),
51  QObject::tr( "Input layer" ), QList< int >() << QgsProcessing::TypeVectorLine ) );
52  addParameter( new QgsProcessingParameterFeatureSource( QStringLiteral( "INTERSECT" ),
53  QObject::tr( "Intersect layer" ), QList< int >() << QgsProcessing::TypeVectorLine ) );
54 
55  addParameter( new QgsProcessingParameterField(
56  QStringLiteral( "INPUT_FIELDS" ),
57  QObject::tr( "Input fields to keep (leave empty to keep all fields)" ), QVariant(),
58  QStringLiteral( "INPUT" ), QgsProcessingParameterField::Any,
59  true, true ) );
60  addParameter( new QgsProcessingParameterField(
61  QStringLiteral( "INTERSECT_FIELDS" ),
62  QObject::tr( "Intersect fields to keep (leave empty to keep all fields)" ), QVariant(),
63  QStringLiteral( "INTERSECT" ), QgsProcessingParameterField::Any,
64  true, true ) );
65 
66  std::unique_ptr< QgsProcessingParameterString > prefix = qgis::make_unique< QgsProcessingParameterString >( QStringLiteral( "INTERSECT_FIELDS_PREFIX" ), QObject::tr( "Intersect fields prefix" ), QString(), false, true );
67  prefix->setFlags( prefix->flags() | QgsProcessingParameterDefinition::FlagAdvanced );
68  addParameter( prefix.release() );
69 
70  addParameter( new QgsProcessingParameterFeatureSink( QStringLiteral( "OUTPUT" ), QObject::tr( "Intersections" ), QgsProcessing::TypeVectorPoint ) );
71 }
72 
73 QString QgsLineIntersectionAlgorithm::shortHelpString() const
74 {
75  return QObject::tr( "This algorithm creates point features where the lines in the Intersect layer intersect the lines in the Input layer." );
76 }
77 
78 QgsLineIntersectionAlgorithm *QgsLineIntersectionAlgorithm::createInstance() const
79 {
80  return new QgsLineIntersectionAlgorithm();
81 }
82 
83 QVariantMap QgsLineIntersectionAlgorithm::processAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
84 {
85  std::unique_ptr< QgsFeatureSource > sourceA( parameterAsSource( parameters, QStringLiteral( "INPUT" ), context ) );
86  if ( !sourceA )
87  throw QgsProcessingException( invalidSourceError( parameters, QStringLiteral( "INPUT" ) ) );
88 
89  std::unique_ptr< QgsFeatureSource > sourceB( parameterAsSource( parameters, QStringLiteral( "INTERSECT" ), context ) );
90  if ( !sourceB )
91  throw QgsProcessingException( invalidSourceError( parameters, QStringLiteral( "INTERSECT" ) ) );
92 
93  const QStringList fieldsA = parameterAsFields( parameters, QStringLiteral( "INPUT_FIELDS" ), context );
94  const QStringList fieldsB = parameterAsFields( parameters, QStringLiteral( "INTERSECT_FIELDS" ), context );
95 
96  QgsAttributeList fieldIndicesA = QgsProcessingUtils::fieldNamesToIndices( fieldsA, sourceA->fields() );
97  QgsAttributeList fieldIndicesB = QgsProcessingUtils::fieldNamesToIndices( fieldsB, sourceB->fields() );
98 
99  QString intersectFieldsPrefix = parameterAsString( parameters, QStringLiteral( "INTERSECT_FIELDS_PREFIX" ), context );
101  QgsProcessingUtils::indicesToFields( fieldIndicesA, sourceA->fields() ),
102  QgsProcessingUtils::indicesToFields( fieldIndicesB, sourceB->fields() ),
103  intersectFieldsPrefix );
104 
105  QString dest;
106  std::unique_ptr< QgsFeatureSink > sink( parameterAsSink( parameters, QStringLiteral( "OUTPUT" ), context, dest, outFields, QgsWkbTypes::Point, sourceA->sourceCrs(), QgsFeatureSink::RegeneratePrimaryKey ) );
107  if ( !sink )
108  throw QgsProcessingException( invalidSinkError( parameters, QStringLiteral( "OUTPUT" ) ) );
109 
110  QgsSpatialIndex spatialIndex( sourceB->getFeatures( QgsFeatureRequest().setNoAttributes().setDestinationCrs( sourceA->sourceCrs(), context.transformContext() ) ), feedback );
111  QgsFeature outFeature;
112  QgsFeatureIterator features = sourceA->getFeatures( QgsFeatureRequest().setSubsetOfAttributes( fieldIndicesA ) );
113  double step = sourceA->featureCount() > 0 ? 100.0 / sourceA->featureCount() : 1;
114  int i = 0;
115  QgsFeature inFeatureA;
116  while ( features.nextFeature( inFeatureA ) )
117  {
118  i++;
119  if ( feedback->isCanceled() )
120  {
121  break;
122  }
123 
124  if ( !inFeatureA.hasGeometry() )
125  continue;
126 
127  QgsGeometry inGeom = inFeatureA.geometry();
128  QgsFeatureIds lines = qgis::listToSet( spatialIndex.intersects( inGeom.boundingBox() ) );
129  if ( !lines.empty() )
130  {
131  // use prepared geometries for faster intersection tests
132  std::unique_ptr< QgsGeometryEngine > engine( QgsGeometry::createGeometryEngine( inGeom.constGet() ) );
133  engine->prepareGeometry();
134 
136  request.setDestinationCrs( sourceA->sourceCrs(), context.transformContext() );
137  request.setSubsetOfAttributes( fieldIndicesB );
138 
139  QgsFeature inFeatureB;
140  QgsFeatureIterator featuresB = sourceB->getFeatures( request );
141  while ( featuresB.nextFeature( inFeatureB ) )
142  {
143  if ( feedback->isCanceled() )
144  {
145  break;
146  }
147 
148  QgsGeometry tmpGeom = inFeatureB.geometry();
149  if ( engine->intersects( tmpGeom.constGet() ) )
150  {
151  QgsMultiPointXY points;
152  QgsGeometry intersectGeom = inGeom.intersection( tmpGeom );
153  QgsAttributes outAttributes;
154  for ( int a : qgis::as_const( fieldIndicesA ) )
155  {
156  outAttributes.append( inFeatureA.attribute( a ) );
157  }
158  for ( int b : qgis::as_const( fieldIndicesB ) )
159  {
160  outAttributes.append( inFeatureB.attribute( b ) );
161  }
163  {
164  const QVector<QgsGeometry> geomCollection = intersectGeom.asGeometryCollection();
165  for ( const QgsGeometry &part : geomCollection )
166  {
167  if ( part.type() == QgsWkbTypes::PointGeometry )
168  {
169  if ( part.isMultipart() )
170  {
171  points = part.asMultiPoint();
172  }
173  else
174  {
175  points.append( part.asPoint() );
176  }
177  }
178  }
179  }
180  else if ( intersectGeom.type() == QgsWkbTypes::PointGeometry )
181  {
182  if ( intersectGeom.isMultipart() )
183  {
184  points = intersectGeom.asMultiPoint();
185  }
186  else
187  {
188  points.append( intersectGeom.asPoint() );
189  }
190  }
191  for ( const QgsPointXY &j : qgis::as_const( points ) )
192  {
193  outFeature.setGeometry( QgsGeometry::fromPointXY( j ) );
194  outFeature.setAttributes( outAttributes );
195  sink->addFeature( outFeature, QgsFeatureSink::FastInsert );
196  }
197  }
198  }
199  }
200 
201  feedback->setProgress( i * step );
202 
203  }
204 
205  QVariantMap outputs;
206  outputs.insert( QStringLiteral( "OUTPUT" ), dest );
207  return outputs;
208 }
209 
211 
212 
QgsGeometry::fromPointXY
static QgsGeometry fromPointXY(const QgsPointXY &point)
Creates a new geometry from a QgsPointXY object.
Definition: qgsgeometry.cpp:164
qgsalgorithmlineintersection.h
QgsFeedback::setProgress
void setProgress(double progress)
Sets the current progress for the feedback object.
Definition: qgsfeedback.h:75
QgsWkbTypes::Point
@ Point
Definition: qgswkbtypes.h:71
QgsProcessingFeedback
Definition: qgsprocessingfeedback.h:37
QgsGeometry::isMultipart
bool isMultipart() const
Returns true if WKB of the geometry is of WKBMulti* type.
Definition: qgsgeometry.cpp:377
QgsFields
Definition: qgsfields.h:44
QgsFeature::geometry
QgsGeometry geometry
Definition: qgsfeature.h:71
QgsProcessingUtils::indicesToFields
static QgsFields indicesToFields(const QList< int > &indices, const QgsFields &fields)
Returns a subset of fields based on the indices of desired fields.
Definition: qgsprocessingutils.cpp:1172
QgsGeometry::wkbType
QgsWkbTypes::Type wkbType() const
Returns type of the geometry as a WKB type (point / linestring / polygon etc.)
Definition: qgsgeometry.cpp:345
QgsProcessingParameterDefinition::FlagAdvanced
@ FlagAdvanced
Parameter is an advanced parameter which should be hidden from users by default.
Definition: qgsprocessingparameters.h:419
QgsProcessing::TypeVectorLine
@ TypeVectorLine
Vector line layers.
Definition: qgsprocessing.h:49
QgsProcessingParameterFeatureSource
Definition: qgsprocessingparameters.h:2612
QgsFeatureRequest::setSubsetOfAttributes
QgsFeatureRequest & setSubsetOfAttributes(const QgsAttributeList &attrs)
Set a subset of attributes that will be fetched.
Definition: qgsfeaturerequest.cpp:190
QgsProcessing::TypeVectorPoint
@ TypeVectorPoint
Vector point layers.
Definition: qgsprocessing.h:48
QgsAttributeList
QList< int > QgsAttributeList
Definition: qgsfield.h:26
QgsMultiPointXY
QVector< QgsPointXY > QgsMultiPointXY
A collection of QgsPoints that share a common collection of attributes.
Definition: qgsgeometry.h:80
QgsGeometry::asGeometryCollection
QVector< QgsGeometry > asGeometryCollection() const
Returns contents of the geometry as a list of geometries.
Definition: qgsgeometry.cpp:2526
QgsProcessingUtils::combineFields
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).
Definition: qgsprocessingutils.cpp:1113
QgsGeometry::intersection
QgsGeometry intersection(const QgsGeometry &geometry) const
Returns a geometry representing the points shared by this geometry and other.
Definition: qgsgeometry.cpp:2395
QgsProcessingParameterFeatureSink
Definition: qgsprocessingparameters.h:2773
QgsFeatureRequest
Definition: qgsfeaturerequest.h:75
qgsgeometryengine.h
QgsFeature::setGeometry
void setGeometry(const QgsGeometry &geometry)
Set the feature's geometry.
Definition: qgsfeature.cpp:137
QgsProcessingContext
Definition: qgsprocessingcontext.h:43
QgsProcessingUtils::fieldNamesToIndices
static QList< int > fieldNamesToIndices(const QStringList &fieldNames, const QgsFields &fields)
Returns a list of field indices parsed from the given list of field names.
Definition: qgsprocessingutils.cpp:1149
QgsWkbTypes::GeometryCollection
@ GeometryCollection
Definition: qgswkbtypes.h:78
QgsFeature::attribute
QVariant attribute(const QString &name) const
Lookup attribute value from attribute name.
Definition: qgsfeature.cpp:262
QgsSpatialIndex
Definition: qgsspatialindex.h:67
QgsFeatureRequest::setFilterFids
QgsFeatureRequest & setFilterFids(const QgsFeatureIds &fids)
Sets feature IDs that should be fetched.
Definition: qgsfeaturerequest.cpp:110
QgsProcessingContext::transformContext
QgsCoordinateTransformContext transformContext() const
Returns the coordinate transform context.
Definition: qgsprocessingcontext.h:135
QgsGeometry::constGet
const QgsAbstractGeometry * constGet() const
Returns a non-modifiable (const) reference to the underlying abstract geometry primitive.
Definition: qgsgeometry.cpp:128
QgsFeatureIds
QSet< QgsFeatureId > QgsFeatureIds
Definition: qgsfeatureid.h:34
QgsFeatureSink::RegeneratePrimaryKey
@ RegeneratePrimaryKey
This flag indicates, that a primary key field cannot be guaranteed to be unique and the sink should i...
Definition: qgsfeaturesink.h:55
QgsPointXY
Definition: qgspointxy.h:43
QgsFeedback::isCanceled
bool isCanceled() const
Tells whether the operation has been canceled already.
Definition: qgsfeedback.h:66
QgsGeometry::asPoint
QgsPointXY asPoint() const
Returns the contents of the geometry as a 2-dimensional point.
Definition: qgsgeometry.cpp:1559
QgsGeometry::createGeometryEngine
static QgsGeometryEngine * createGeometryEngine(const QgsAbstractGeometry *geometry)
Creates and returns a new geometry engine.
Definition: qgsgeometry.cpp:3659
QgsWkbTypes::PointGeometry
@ PointGeometry
Definition: qgswkbtypes.h:141
QgsFeatureIterator::nextFeature
bool nextFeature(QgsFeature &f)
Definition: qgsfeatureiterator.h:373
QgsGeometry
Definition: qgsgeometry.h:122
QgsFeature::hasGeometry
bool hasGeometry() const
Returns true if the feature has an associated geometry.
Definition: qgsfeature.cpp:197
QgsGeometry::asMultiPoint
QgsMultiPointXY asMultiPoint() const
Returns the contents of the geometry as a multi-point.
Definition: qgsgeometry.cpp:1655
QgsGeometry::boundingBox
QgsRectangle boundingBox() const
Returns the bounding box of the geometry.
Definition: qgsgeometry.cpp:962
QgsAttributes
Definition: qgsattributes.h:57
QgsFeature
Definition: qgsfeature.h:55
QgsFeatureRequest::setDestinationCrs
QgsFeatureRequest & setDestinationCrs(const QgsCoordinateReferenceSystem &crs, const QgsCoordinateTransformContext &context)
Sets the destination crs for feature's geometries.
Definition: qgsfeaturerequest.cpp:263
QgsFeature::setAttributes
void setAttributes(const QgsAttributes &attrs)
Sets the feature's attributes.
Definition: qgsfeature.cpp:127
QgsProcessingParameterField::Any
@ Any
Accepts any field.
Definition: qgsprocessingparameters.h:2502
QgsGeometry::type
QgsWkbTypes::GeometryType type
Definition: qgsgeometry.h:126
QgsFeatureIterator
Definition: qgsfeatureiterator.h:263
QgsProcessingException
Definition: qgsexception.h:82
QgsProcessingParameterField
Definition: qgsprocessingparameters.h:2495
QgsWkbTypes::flatType
static Type flatType(Type type)
Returns the flat type for a WKB type.
Definition: qgswkbtypes.h:701
QgsFeatureSink::FastInsert
@ FastInsert
Use faster inserts, at the cost of updating the passed features to reflect changes made at the provid...
Definition: qgsfeaturesink.h:70