qgsalgorithmtotalcurvature.cpp

Go to the documentation of this file.

/***************************************************************************
                         qgsalgorithmtotalcurvature.cpp
                         ---------------------
    begin                : December 2025
    copyright            : (C) 2025 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 "qgsalgorithmtotalcurvature.h"
 
#include "qgsrasterfilewriter.h"
#include "qgstotalcurvaturefilter.h"
 
#include <QString>
 
using namespace Qt::StringLiterals;

 
QString QgsTotalCurvatureAlgorithm::name() const
{
  return u"totalcurvature"_s;
}
 
QString QgsTotalCurvatureAlgorithm::displayName() const
{
  return QObject::tr( "Total curvature" );
}
 
QStringList QgsTotalCurvatureAlgorithm::tags() const
{
  return QObject::tr( "dem,total curvature,terrain" ).split( ',' );
}
 
QString QgsTotalCurvatureAlgorithm::group() const
{
  return QObject::tr( "Raster terrain analysis" );
}
 
QString QgsTotalCurvatureAlgorithm::groupId() const
{
  return u"rasterterrainanalysis"_s;
}
 
QString QgsTotalCurvatureAlgorithm::shortHelpString() const
{
  return QObject::tr( "This algorithm calculates total curvature of the input raster as described by Wilson, Gallant (2000): terrain analysis." );
}
 
QString QgsTotalCurvatureAlgorithm::shortDescription() const
{
  return QObject::tr( "Calculates total curvature as described by Wilson, Gallant (2000): terrain analysis." );
}
 
QgsTotalCurvatureAlgorithm *QgsTotalCurvatureAlgorithm::createInstance() const
{
  return new QgsTotalCurvatureAlgorithm();
}
 
void QgsTotalCurvatureAlgorithm::initAlgorithm( const QVariantMap & )
{
  addParameter( new QgsProcessingParameterRasterLayer( u"INPUT"_s, QObject::tr( "Elevation layer" ) ) );
 
  auto zFactorParam = std::make_unique<QgsProcessingParameterNumber>( u"Z_FACTOR"_s, QObject::tr( "Z factor" ), Qgis::ProcessingNumberParameterType::Double, 1.0, false, 0.0 );
  zFactorParam->setHelp( QObject::tr( "Multiplication factor to convert vertical Z units to horizontal XY units." ) );
  zFactorParam->setMetadata(
    { QVariantMap( { { u"widget_wrapper"_s, QVariantMap( { { u"decimals"_s, 12 } } ) } } )
    }
  );
  addParameter( zFactorParam.release() );
 
  auto outputNodataParam = std::make_unique<QgsProcessingParameterNumber>( u"NODATA"_s, QObject::tr( "Output NoData value" ), Qgis::ProcessingNumberParameterType::Double, -9999.0 );
  outputNodataParam->setFlags( outputNodataParam->flags() | Qgis::ProcessingParameterFlag::Advanced );
  addParameter( outputNodataParam.release() );
 
  auto creationOptsParam = std::make_unique<QgsProcessingParameterString>( u"CREATION_OPTIONS"_s, QObject::tr( "Creation options" ), QVariant(), false, true );
  creationOptsParam->setMetadata( QVariantMap( { { u"widget_wrapper"_s, QVariantMap( { { u"widget_type"_s, u"rasteroptions"_s } } ) } } ) );
  creationOptsParam->setFlags( creationOptsParam->flags() | Qgis::ProcessingParameterFlag::Advanced );
  addParameter( creationOptsParam.release() );
 
  addParameter( new QgsProcessingParameterRasterDestination( u"OUTPUT"_s, QObject::tr( "Total curvature" ) ) );
}
 
bool QgsTotalCurvatureAlgorithm::prepareAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback * )
{
  QgsRasterLayer *layer = parameterAsRasterLayer( parameters, u"INPUT"_s, context );
  if ( !layer )
  {
    throw QgsProcessingException( invalidRasterError( parameters, u"INPUT"_s ) );
  }
 
  mLayerSource = layer->source();
  return true;
}
 
QVariantMap QgsTotalCurvatureAlgorithm::processAlgorithm( const QVariantMap &parameters, QgsProcessingContext &context, QgsProcessingFeedback *feedback )
{
  const double zFactor = parameterAsDouble( parameters, u"Z_FACTOR"_s, context );
  const QString creationOptions = parameterAsString( parameters, u"CREATION_OPTIONS"_s, context ).trimmed();
  const double outputNodata = parameterAsDouble( parameters, u"NODATA"_s, context );
 
  const QString outputFile = parameterAsOutputLayer( parameters, u"OUTPUT"_s, context );
  const QString outputFormat = parameterAsOutputRasterFormat( parameters, u"OUTPUT"_s, context );
 
  QgsTotalCurvatureFilter curvature( mLayerSource, outputFile, outputFormat );
  curvature.setZFactor( zFactor );
  if ( !creationOptions.isEmpty() )
  {
    curvature.setCreationOptions( creationOptions.split( '|' ) );
  }
  curvature.setOutputNodataValue( outputNodata );
  curvature.processRaster( feedback );
 
  QVariantMap outputs;
  outputs.insert( u"OUTPUT"_s, outputFile );
  return outputs;
}