sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-10-15 00:02:52 -04:00
Code Issues Packages Projects Releases Wiki Activity

do not create spatial index if preserving attributes is not needed

Browse Source
This commit is contained in:
Alexander Bruy 2023-08-09 13:58:09 +03:00 committed by Martin Dobias
parent 514e8b5b24
commit ea6a706c01
2 changed files with 106 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

135
src/analysis/processing/qgsalgorithmvoronoipolygons.cpp
View File

@ -66,46 +66,58 @@ void QgsVoronoiPolygonsAlgorithm::initAlgorithm( const QVariantMap & )
addParameter( new QgsProcessingParameterFeatureSink( QStringLiteral( "OUTPUT" ), QObject::tr( "Voronoi polygons" ), QgsProcessing::TypeVectorPolygon ) );
}
QVariantMap QgsVoronoiPolygonsAlgorithm::processAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
bool QgsVoronoiPolygonsAlgorithm::prepareAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
{
std::unique_ptr< QgsProcessingFeatureSource > source( parameterAsSource( parameters, QStringLiteral( "INPUT" ), context ) );
if ( !source )
Q_UNUSED( feedback );
mSource.reset( parameterAsSource( parameters, QStringLiteral( "INPUT" ), context ) );
if ( !mSource )
throw QgsProcessingException( invalidSourceError( parameters, QStringLiteral( "INPUT" ) ) );
if ( source->featureCount() < 3 )
if ( mSource->featureCount() < 3 )
throw QgsProcessingException( QObject::tr( "Input layer should contain at least 3 points." ) );
const double buffer = parameterAsDouble( parameters, QStringLiteral( "BUFFER" ), context );
const double tolerance = parameterAsDouble( parameters, QStringLiteral( "TOLERANCE" ), context );
const bool copyAttributes = parameterAsBool( parameters, QStringLiteral( "COPY_ATTRIBUTES" ), context );
mBuffer = parameterAsDouble( parameters, QStringLiteral( "BUFFER" ), context );
mTolerance = parameterAsDouble( parameters, QStringLiteral( "TOLERANCE" ), context );
mCopyAttributes = parameterAsBool( parameters, QStringLiteral( "COPY_ATTRIBUTES" ), context );
QgsFields fields;
if ( copyAttributes )
return true;
}
QVariantMap QgsVoronoiPolygonsAlgorithm::processAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
{
QString dest;
if ( mCopyAttributes )
{
fields = source->fields();
dest = voronoiWithAttributes( parameters, context, feedback );
}
else
{
fields.append( QgsField( QStringLiteral( "id" ), QVariant::LongLong ) );
dest = voronoiWithoutAttributes( parameters, context, feedback );
}
QVariantMap outputs;
outputs.insert( QStringLiteral( "OUTPUT" ), dest );
return outputs;
}
QString QgsVoronoiPolygonsAlgorithm::voronoiWithAttributes( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
{
QgsFields fields = mSource->fields();
QString dest;
std::unique_ptr< QgsFeatureSink > sink( parameterAsSink( parameters, QStringLiteral( "OUTPUT" ), context, dest, fields, Qgis::WkbType::Polygon, source->sourceCrs() ) );
std::unique_ptr< QgsFeatureSink > sink( parameterAsSink( parameters, QStringLiteral( "OUTPUT" ), context, dest, fields, Qgis::WkbType::Polygon, mSource->sourceCrs() ) );
if ( !sink )
throw QgsProcessingException( invalidSinkError( parameters, QStringLiteral( "OUTPUT" ) ) );
QgsFeatureRequest request;
if ( !copyAttributes )
{
request.setSubsetOfAttributes( QList< int >() );
}
QgsFeatureIterator it = source->getFeatures( request, QgsProcessingFeatureSource::FlagSkipGeometryValidityChecks );
QgsFeatureIterator it = mSource->getFeatures( request, QgsProcessingFeatureSource::FlagSkipGeometryValidityChecks );
QgsGeometry allPoints;
QHash< QgsFeatureId, QgsAttributes > attributeCache;
long i = 0;
double step = source->featureCount() > 0 ? 50.0 / source->featureCount() : 1;
double step = mSource->featureCount() > 0 ? 50.0 / mSource->featureCount() : 1;
const QgsSpatialIndex index( it, [&]( const QgsFeature & f )->bool
{
@ -137,16 +149,16 @@ QVariantMap QgsVoronoiPolygonsAlgorithm::processAlgorithm( const QVariantMap &pa
return true;
}, QgsSpatialIndex::FlagStoreFeatureGeometries );
QgsRectangle extent = source->sourceExtent();
double delta = extent.width() * buffer / 100.0;
QgsRectangle extent = mSource->sourceExtent();
double delta = extent.width() * mBuffer / 100.0;
extent.setXMinimum( extent.xMinimum() - delta );
extent.setXMaximum( extent.xMaximum() + delta );
delta = extent.height() * buffer / 100.0;
delta = extent.height() * mBuffer / 100.0;
extent.setYMinimum( extent.yMinimum() - delta );
extent.setYMaximum( extent.yMaximum() + delta );
QgsGeometry clippingGeom = QgsGeometry::fromRect( extent );
QgsGeometry voronoiDiagram = allPoints.voronoiDiagram( clippingGeom, tolerance );
QgsGeometry voronoiDiagram = allPoints.voronoiDiagram( clippingGeom, mTolerance );
if ( !voronoiDiagram.isEmpty() )
{
@ -162,33 +174,78 @@ QVariantMap QgsVoronoiPolygonsAlgorithm::processAlgorithm( const QVariantMap &pa
QgsFeature f;
f.setFields( fields );
f.setGeometry( QgsGeometry( extentEngine->intersection( collection[i].constGet() ) ) );
if ( copyAttributes )
const QList< QgsFeatureId > intersected = index.intersects( collection[i].boundingBox() );
engine.reset( QgsGeometry::createGeometryEngine( collection[i].constGet() ) );
engine->prepareGeometry();
for ( const QgsFeatureId id : intersected )
{
const QList< QgsFeatureId > intersected = index.intersects( collection[i].boundingBox() );
engine.reset( QgsGeometry::createGeometryEngine( collection[i].constGet() ) );
engine->prepareGeometry();
for ( const QgsFeatureId id : intersected )
if ( engine->intersects( index.geometry( id ).constGet() ) )
{
if ( engine->intersects( index.geometry( id ).constGet() ) )
{
f.setAttributes( attributeCache.value( id ) );
break;
}
f.setAttributes( attributeCache.value( id ) );
break;
}
}
else
{
f.setAttributes( QgsAttributes() << i );
}
if ( !sink->addFeature( f, QgsFeatureSink::FastInsert ) )
throw QgsProcessingException( writeFeatureError( sink.get(), parameters, QStringLiteral( "OUTPUT" ) ) );
feedback->setProgress( 50 + i * step );
}
}
QVariantMap outputs;
outputs.insert( QStringLiteral( "OUTPUT" ), dest );
return outputs;
return dest;
}
QString QgsVoronoiPolygonsAlgorithm::voronoiWithoutAttributes( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
{
QgsFields fields;
fields.append( QgsField( QStringLiteral( "id" ), QVariant::LongLong ) );
QString dest;
std::unique_ptr< QgsFeatureSink > sink( parameterAsSink( parameters, QStringLiteral( "OUTPUT" ), context, dest, fields, Qgis::WkbType::Polygon, mSource->sourceCrs() ) );
if ( !sink )
throw QgsProcessingException( invalidSinkError( parameters, QStringLiteral( "OUTPUT" ) ) );
QgsFeatureRequest request;
request.setSubsetOfAttributes( QList< int >() );
QgsFeatureIterator it = mSource->getFeatures( request, QgsProcessingFeatureSource::FlagSkipGeometryValidityChecks );
QgsGeometry allPoints;
QHash< QgsFeatureId, QgsAttributes > attributeCache;
double step = mSource->featureCount() > 0 ? 100.0 / mSource->featureCount() : 1;
QgsRectangle extent = mSource->sourceExtent();
double delta = extent.width() * mBuffer / 100.0;
extent.setXMinimum( extent.xMinimum() - delta );
extent.setXMaximum( extent.xMaximum() + delta );
delta = extent.height() * mBuffer / 100.0;
extent.setYMinimum( extent.yMinimum() - delta );
extent.setYMaximum( extent.yMaximum() + delta );
QgsGeometry clippingGeom = QgsGeometry::fromRect( extent );
QgsGeometry voronoiDiagram = allPoints.voronoiDiagram( clippingGeom, mTolerance );
if ( !voronoiDiagram.isEmpty() )
{
std::unique_ptr< QgsGeometryEngine > engine;
std::unique_ptr< QgsGeometryEngine > extentEngine( QgsGeometry::createGeometryEngine( clippingGeom.constGet() ) );
QVector< QgsGeometry > collection = voronoiDiagram.asGeometryCollection();
for ( int i = 0; i < collection.length(); i++ )
{
if ( feedback->isCanceled() )
{
break;
}
QgsFeature f;
f.setFields( fields );
f.setGeometry( QgsGeometry( extentEngine->intersection( collection[i].constGet() ) ) );
f.setAttributes( QgsAttributes() << i );
if ( !sink->addFeature( f, QgsFeatureSink::FastInsert ) )
throw QgsProcessingException( writeFeatureError( sink.get(), parameters, QStringLiteral( "OUTPUT" ) ) );
feedback->setProgress( i * step );
}
}
return dest;
}
///@endcond

10
src/analysis/processing/qgsalgorithmvoronoipolygons.h
View File

@ -47,8 +47,18 @@ class QgsVoronoiPolygonsAlgorithm : public QgsProcessingAlgorithm
protected:
bool prepareAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback ) override;
QVariantMap processAlgorithm( const QVariantMap &parameters,
QgsProcessingContext &context, QgsProcessingFeedback *feedback ) override;
private:
QString voronoiWithAttributes( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback );
QString voronoiWithoutAttributes( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback );
std::unique_ptr< QgsProcessingFeatureSource > mSource;
double mBuffer;
double mTolerance;
bool mCopyAttributes;
};
///@endcond PRIVATE