qgsalgorithmextractspecificvertices.cpp

Go to the documentation of this file.

/***************************************************************************
                         qgsalgorithmextractspecificvertices.cpp
                         --------------------------
    begin                : November 2019
    copyright            : (C) 2019 by Alexander Bruy
    email                : alexander dot bruy at gmail dot com
 ***************************************************************************/
 
/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 
#include "qgsalgorithmextractspecificvertices.h"
 
#include "qgsabstractgeometry.h"
#include "qgsgeometryutils.h"
 
#include <QString>
 
using namespace Qt::StringLiterals;

 
QString QgsExtractSpecificVerticesAlgorithm::name() const
{
  return u"extractspecificvertices"_s;
}
 
QString QgsExtractSpecificVerticesAlgorithm::displayName() const
{
  return QObject::tr( "Extract specific vertices" );
}
 
QStringList QgsExtractSpecificVerticesAlgorithm::tags() const
{
  return QObject::tr( "points,vertex,nodes" ).split( ',' );
}
 
QString QgsExtractSpecificVerticesAlgorithm::group() const
{
  return QObject::tr( "Vector geometry" );
}
 
QString QgsExtractSpecificVerticesAlgorithm::groupId() const
{
  return u"vectorgeometry"_s;
}
 
QString QgsExtractSpecificVerticesAlgorithm::shortHelpString() const
{
  return QObject::tr(
           "This algorithm takes a vector layer and generates a point layer with points "
           "representing specific vertices in the input geometries. For instance, this algorithm "
           "can be used to extract the first or last vertices in the geometry. The attributes associated "
           "to each point are the same ones associated to the feature that the point belongs to."
         )
         + u"\n\n"_s
         + QObject::tr(
           "The vertex indices parameter accepts a comma separated string specifying the indices of the "
           "vertices to extract. The first vertex corresponds to an index of 0, the second vertex has an "
           "index of 1, etc. Negative indices can be used to find vertices at the end of the geometry, "
           "e.g., an index of -1 corresponds to the last vertex, -2 corresponds to the second last vertex, etc."
         )
         + u"\n\n"_s
         + QObject::tr(
           "Additional fields are added to the points indicating the specific vertex position (e.g., 0, -1, etc), "
           "the original vertex index, the vertex’s part and its index within the part (as well as its ring for "
           "polygons), distance along the original geometry and bisector angle of vertex for the original geometry."
         );
}
 
QString QgsExtractSpecificVerticesAlgorithm::shortDescription() const
{
  return QObject::tr( "Generates a point layer with points representing specific vertices in the input geometries." );
}
 
Qgis::ProcessingAlgorithmDocumentationFlags QgsExtractSpecificVerticesAlgorithm::documentationFlags() const
{
  return Qgis::ProcessingAlgorithmDocumentationFlag::RegeneratesPrimaryKey;
}
 
QString QgsExtractSpecificVerticesAlgorithm::outputName() const
{
  return QObject::tr( "Vertices" );
}
 
QgsExtractSpecificVerticesAlgorithm *QgsExtractSpecificVerticesAlgorithm::createInstance() const
{
  return new QgsExtractSpecificVerticesAlgorithm();
}
 
Qgis::ProcessingSourceType QgsExtractSpecificVerticesAlgorithm::outputLayerType() const
{
  return Qgis::ProcessingSourceType::VectorPoint;
}
 
QgsFields QgsExtractSpecificVerticesAlgorithm::outputFields( const QgsFields &inputFields ) const
{
  QgsFields newFields;
  newFields.append( QgsField( u"vertex_pos"_s, QMetaType::Type::Int ) );
  newFields.append( QgsField( u"vertex_index"_s, QMetaType::Type::Int ) );
  newFields.append( QgsField( u"vertex_part"_s, QMetaType::Type::Int ) );
  if ( mGeometryType == Qgis::GeometryType::Polygon )
  {
    newFields.append( QgsField( u"vertex_part_ring"_s, QMetaType::Type::Int ) );
  }
  newFields.append( QgsField( u"vertex_part_index"_s, QMetaType::Type::Int ) );
  newFields.append( QgsField( u"distance"_s, QMetaType::Type::Double ) );
  newFields.append( QgsField( u"angle"_s, QMetaType::Type::Double ) );
 
  return QgsProcessingUtils::combineFields( inputFields, newFields );
}
 
Qgis::WkbType QgsExtractSpecificVerticesAlgorithm::outputWkbType( Qgis::WkbType inputWkbType ) const
{
  Qgis::WkbType outputWkbType = Qgis::WkbType::Point;
  if ( QgsWkbTypes::hasM( inputWkbType ) )
  {
    outputWkbType = QgsWkbTypes::addM( outputWkbType );
  }
  if ( QgsWkbTypes::hasZ( inputWkbType ) )
  {
    outputWkbType = QgsWkbTypes::addZ( outputWkbType );
  }
 
  return outputWkbType;
}
 
Qgis::ProcessingFeatureSourceFlags QgsExtractSpecificVerticesAlgorithm::sourceFlags() const
{
  return Qgis::ProcessingFeatureSourceFlag::SkipGeometryValidityChecks;
}
 
QgsFeatureSink::SinkFlags QgsExtractSpecificVerticesAlgorithm::sinkFlags() const
{
  return QgsFeatureSink::RegeneratePrimaryKey;
}
 
void QgsExtractSpecificVerticesAlgorithm::initParameters( const QVariantMap & )
{
  addParameter( new QgsProcessingParameterString( u"VERTICES"_s, QObject::tr( "Vertex indices" ), u"0"_s ) );
}
 
bool QgsExtractSpecificVerticesAlgorithm::prepareAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback * )
{
  std::unique_ptr<QgsProcessingFeatureSource> source( parameterAsSource( parameters, u"INPUT"_s, context ) );
  if ( !source )
    throw QgsProcessingException( invalidSourceError( parameters, u"INPUT"_s ) );
 
  mGeometryType = QgsWkbTypes::geometryType( source->wkbType() );
 
  const QString verticesString = parameterAsString( parameters, u"VERTICES"_s, context );
  const QStringList verticesList = verticesString.split( ',', Qt::SkipEmptyParts );
  for ( const QString &vertex : verticesList )
  {
    bool ok = false;
    const int i = vertex.toInt( &ok );
    if ( ok )
    {
      mIndices << i;
    }
    else
    {
      throw QgsProcessingException( QObject::tr( "'%1' is not a valid vertex index" ).arg( vertex ) );
    }
  }
 
  return true;
}
 
QgsFeatureList QgsExtractSpecificVerticesAlgorithm::processFeature( const QgsFeature &feature, QgsProcessingContext &, QgsProcessingFeedback * )
{
  QgsFeatureList outputFeatures;
 
  QgsFeature f = feature;
  const QgsGeometry inputGeom = f.geometry();
  if ( inputGeom.isEmpty() )
  {
    QgsAttributes attrs = f.attributes();
    attrs << QVariant() << QVariant() << QVariant();
    if ( mGeometryType == Qgis::GeometryType::Polygon )
    {
      attrs << QVariant();
    }
    attrs << QVariant() << QVariant() << QVariant();
 
    f.clearGeometry();
    f.setAttributes( attrs );
    outputFeatures << f;
  }
  else
  {
    int vertexIndex;
    const int totalVertices = inputGeom.constGet()->nCoordinates();
    for ( const int vertex : mIndices )
    {
      if ( vertex < 0 )
      {
        vertexIndex = totalVertices + vertex;
      }
      else
      {
        vertexIndex = vertex;
      }
 
      if ( vertexIndex < 0 || vertexIndex >= totalVertices )
        continue;
 
      QgsVertexId vertexId;
      inputGeom.vertexIdFromVertexNr( vertexIndex, vertexId );
 
      const double distance = inputGeom.distanceToVertex( vertexIndex );
      const double angle = inputGeom.angleAtVertex( vertexIndex ) * 180 / M_PI;
 
      QgsFeature outFeature = QgsFeature();
      QgsAttributes attrs = f.attributes();
      attrs << vertex << vertexIndex << vertexId.part;
      if ( mGeometryType == Qgis::GeometryType::Polygon )
      {
        attrs << vertexId.ring;
      }
      attrs << vertexId.vertex << distance << angle;
 
      outFeature.setAttributes( attrs );
      const QgsPoint point = inputGeom.vertexAt( vertexIndex );
      outFeature.setGeometry( QgsGeometry( point.clone() ) );
      outputFeatures << outFeature;
    }
  }
 
  return outputFeatures;
}