qgsprocessingparametermeshdataset.cpp

Go to the documentation of this file.

/***************************************************************************
  qgsprocessingparametermeshdataset.cpp
  ---------------------
  Date                 : October 2020
  Copyright            : (C) 2020 by Vincent Cloarec
  Email                : vcloarec 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 "qgsprocessingparametermeshdataset.h"
 
#include <QString>
 
using namespace Qt::StringLiterals;

QgsProcessingParameterMeshDatasetGroups::QgsProcessingParameterMeshDatasetGroups(
  const QString &name, const QString &description, const QString &meshLayerParameterName, const QSet<int> supportedDataType, bool optional
)
  : QgsProcessingParameterDefinition( name, description, QVariantList(), optional, QString() )
  , mMeshLayerParameterName( meshLayerParameterName )
  , mSupportedDataType( supportedDataType )
{}
 
QgsProcessingParameterDefinition *QgsProcessingParameterMeshDatasetGroups::clone() const
{
  return new QgsProcessingParameterMeshDatasetGroups( name(), description(), mMeshLayerParameterName, mSupportedDataType );
}
 
QString QgsProcessingParameterMeshDatasetGroups::type() const
{
  return typeName();
}
 
bool QgsProcessingParameterMeshDatasetGroups::checkValueIsAcceptable( const QVariant &input, QgsProcessingContext *context ) const
{
  Q_UNUSED( context );
  return valueIsAcceptable( input, mFlags & Qgis::ProcessingParameterFlag::Optional );
}
 
QString QgsProcessingParameterMeshDatasetGroups::valueAsPythonString( const QVariant &value, QgsProcessingContext &context ) const
{
  Q_UNUSED( context );
  QStringList parts;
  const QList<int> groups = valueAsDatasetGroup( value );
  for ( const int g : groups )
    parts.append( QString::number( g ) );
 
  return parts.join( ',' ).prepend( '[' ).append( ']' );
}
 
QString QgsProcessingParameterMeshDatasetGroups::asPythonString( QgsProcessing::PythonOutputType outputType ) const
{
  switch ( outputType )
  {
    case QgsProcessing::PythonOutputType::PythonQgsProcessingAlgorithmSubclass:
    {
      QString code = u"QgsProcessingParameterMeshDatasetGroups('%1', %2"_s.arg( name(), QgsProcessingUtils::stringToPythonLiteral( description() ) );
      if ( !mMeshLayerParameterName.isEmpty() )
        code += u", meshLayerParameterName=%1"_s.arg( QgsProcessingUtils::stringToPythonLiteral( mMeshLayerParameterName ) );
 
      QStringList dt;
      if ( mSupportedDataType.contains( QgsMeshDatasetGroupMetadata::DataOnFaces ) )
        dt.append( u"QgsMeshDatasetGroupMetadata.DataOnFaces"_s );
      if ( mSupportedDataType.contains( QgsMeshDatasetGroupMetadata::DataOnVertices ) )
        dt.append( u"QgsMeshDatasetGroupMetadata.DataOnVertices"_s );
      if ( mSupportedDataType.contains( QgsMeshDatasetGroupMetadata::DataOnVolumes ) )
        dt.append( u"QgsMeshDatasetGroupMetadata.DataOnVolumes"_s );
      if ( mSupportedDataType.contains( QgsMeshDatasetGroupMetadata::DataOnEdges ) )
        dt.append( u"QgsMeshDatasetGroupMetadata.DataOnEdges"_s );
      if ( !dt.isEmpty() )
      {
        code += ", supportedDataType=["_L1;
        code += dt.join( ',' );
        code += ']';
      }
 
      if ( mFlags & Qgis::ProcessingParameterFlag::Optional )
        code += ", optional=True"_L1;
      code += ')';
      return code;
    }
  }
  return QString();
}
 
QStringList QgsProcessingParameterMeshDatasetGroups::dependsOnOtherParameters() const
{
  if ( mMeshLayerParameterName.isEmpty() )
    return QStringList();
  else
    return QStringList() << mMeshLayerParameterName;
}
 
QString QgsProcessingParameterMeshDatasetGroups::meshLayerParameterName() const
{
  return mMeshLayerParameterName;
}
 
bool QgsProcessingParameterMeshDatasetGroups::isDataTypeSupported( QgsMeshDatasetGroupMetadata::DataType dataType ) const
{
  return mSupportedDataType.contains( dataType );
}
 
QList<int> QgsProcessingParameterMeshDatasetGroups::valueAsDatasetGroup( const QVariant &value )
{
  if ( !valueIsAcceptable( value, true ) )
    return QList<int>();
 
  QList<int> ret;
 
  if ( value.isValid() )
  {
    if ( value.userType() == QMetaType::Type::QVariantList )
    {
      const QVariantList varList = value.toList();
      for ( const QVariant &v : varList )
        ret << v.toInt();
    }
    else
    {
      ret << value.toInt();
    }
  }
 
  return ret;
}
 
QVariantMap QgsProcessingParameterMeshDatasetGroups::toVariantMap() const
{
  QVariantMap map = QgsProcessingParameterDefinition::toVariantMap();
  map.insert( u"mesh_layer"_s, mMeshLayerParameterName );
  QVariantList dataType;
  for ( int v : mSupportedDataType )
    dataType.append( v );
  map.insert( u"supported_data_type"_s, dataType );
  return map;
}
 
bool QgsProcessingParameterMeshDatasetGroups::fromVariantMap( const QVariantMap &map )
{
  QgsProcessingParameterDefinition::fromVariantMap( map );
  mMeshLayerParameterName = map.value( u"mesh_layer"_s ).toString();
  const QVariantList dataType = map.value( u"supported_data_type"_s ).toList();
  mSupportedDataType.clear();
  for ( const QVariant &var : dataType )
    mSupportedDataType.insert( var.toInt() );
  return true;
}
 
bool QgsProcessingParameterMeshDatasetGroups::valueIsAcceptable( const QVariant &input, bool allowEmpty )
{
  if ( !input.isValid() )
    return allowEmpty;
 
  if ( input.userType() != QMetaType::Type::QVariantList )
  {
    bool ok = false;
    input.toInt( &ok );
    return ok;
  }
  const QVariantList list = input.toList();
 
  if ( !allowEmpty && list.isEmpty() )
    return false;
 
  for ( const QVariant &var : list )
  {
    bool ok = false;
    var.toInt( &ok );
    if ( !ok )
      return false;
  }
 
  return true;
}
 
QgsProcessingParameterMeshDatasetTime::QgsProcessingParameterMeshDatasetTime( const QString &name, const QString &description, const QString &meshLayerParameterName, const QString &datasetGroupParameterName )
  : QgsProcessingParameterDefinition( name, description, QVariant() )
  , mMeshLayerParameterName( meshLayerParameterName )
  , mDatasetGroupParameterName( datasetGroupParameterName )
{}
 
QgsProcessingParameterDefinition *QgsProcessingParameterMeshDatasetTime::clone() const
{
  return new QgsProcessingParameterMeshDatasetTime( name(), description(), mMeshLayerParameterName, mDatasetGroupParameterName );
}
 
QString QgsProcessingParameterMeshDatasetTime::type() const
{
  return typeName();
}
 
bool QgsProcessingParameterMeshDatasetTime::checkValueIsAcceptable( const QVariant &input, QgsProcessingContext *context ) const
{
  Q_UNUSED( context );
  return valueIsAcceptable( input, mFlags & Qgis::ProcessingParameterFlag::Optional );
}
 
QString QgsProcessingParameterMeshDatasetTime::valueAsPythonString( const QVariant &value, QgsProcessingContext &context ) const
{
  Q_UNUSED( context );
  QStringList parts;
  const QString type = QgsProcessingUtils::variantToPythonLiteral( valueAsTimeType( value ) );
  parts << u"'type': "_s + type;
 
  if ( value.toDateTime().isValid() )
  {
    QDateTime dateTime = value.toDateTime();
    dateTime.setTimeSpec( Qt::UTC );
    parts << u"'value': "_s + QgsProcessingUtils::variantToPythonLiteral( dateTime );
  }
  else
  {
    const QVariantMap variantTimeDataset = value.toMap();
    if ( variantTimeDataset.value( u"type"_s ) == "dataset-time-step"_L1 )
    {
      const QVariantList datasetIndex = variantTimeDataset.value( u"value"_s ).toList();
      parts << u"'value': "_s + QString( "[%1,%2]" ).arg( datasetIndex.at( 0 ).toString(), datasetIndex.at( 1 ).toString() );
    }
    else if ( variantTimeDataset.value( u"type"_s ) == "defined-date-time"_L1 )
    {
      parts << u"'value': "_s + QgsProcessingUtils::variantToPythonLiteral( variantTimeDataset.value( u"value"_s ) );
    }
  }
 
  return parts.join( ',' ).prepend( '{' ).append( '}' );
}
 
QString QgsProcessingParameterMeshDatasetTime::asPythonString( QgsProcessing::PythonOutputType outputType ) const
{
  switch ( outputType )
  {
    case QgsProcessing::PythonOutputType::PythonQgsProcessingAlgorithmSubclass:
    {
      QString code = u"QgsProcessingParameterMeshDatasetTime('%1', '%2'"_s.arg( name(), description() );
      if ( !mMeshLayerParameterName.isEmpty() )
        code += u", meshLayerParameterName=%1"_s.arg( QgsProcessingUtils::stringToPythonLiteral( mMeshLayerParameterName ) );
 
      if ( !mDatasetGroupParameterName.isEmpty() )
        code += u", datasetGroupParameterName=%1"_s.arg( QgsProcessingUtils::stringToPythonLiteral( mDatasetGroupParameterName ) );
 
      if ( mFlags & Qgis::ProcessingParameterFlag::Optional )
        code += ", optional=True"_L1;
      code += ')';
      return code;
    }
  }
  return QString();
}
 
QStringList QgsProcessingParameterMeshDatasetTime::dependsOnOtherParameters() const
{
  QStringList otherParameters;
  if ( !mMeshLayerParameterName.isEmpty() )
    otherParameters << mMeshLayerParameterName;
 
  if ( !mDatasetGroupParameterName.isEmpty() )
    otherParameters << mMeshLayerParameterName << mDatasetGroupParameterName;
 
  return otherParameters;
}
 
QVariantMap QgsProcessingParameterMeshDatasetTime::toVariantMap() const
{
  QVariantMap map = QgsProcessingParameterDefinition::toVariantMap();
  map.insert( u"mesh_layer"_s, mMeshLayerParameterName );
  map.insert( u"dataset_groups"_s, mDatasetGroupParameterName );
  return map;
}
 
bool QgsProcessingParameterMeshDatasetTime::fromVariantMap( const QVariantMap &map )
{
  QgsProcessingParameterDefinition::fromVariantMap( map );
  mMeshLayerParameterName = map.value( u"mesh_layer"_s ).toString();
  mDatasetGroupParameterName = map.value( u"dataset_groups"_s ).toString();
  return true;
}
 
QString QgsProcessingParameterMeshDatasetTime::meshLayerParameterName() const
{
  return mMeshLayerParameterName;
}
 
QString QgsProcessingParameterMeshDatasetTime::datasetGroupParameterName() const
{
  return mDatasetGroupParameterName;
}
 
QString QgsProcessingParameterMeshDatasetTime::valueAsTimeType( const QVariant &value )
{
  if ( !valueIsAcceptable( value, false ) )
    return QString();
 
  if ( value.toDateTime().isValid() )
    return u"defined-date-time"_s;
 
  return value.toMap().value( u"type"_s ).toString();
}
 
QgsMeshDatasetIndex QgsProcessingParameterMeshDatasetTime::timeValueAsDatasetIndex( const QVariant &value )
{
  if ( !valueIsAcceptable( value, false ) || valueAsTimeType( value ) != "dataset-time-step"_L1 )
    return QgsMeshDatasetIndex( -1, -1 );
 
  const QVariantList list = value.toMap().value( u"value"_s ).toList();
  return QgsMeshDatasetIndex( list.at( 0 ).toInt(), list.at( 1 ).toInt() );
}
 
QDateTime QgsProcessingParameterMeshDatasetTime::timeValueAsDefinedDateTime( const QVariant &value )
{
  if ( value.toDateTime().isValid() )
  {
    QDateTime dateTime = value.toDateTime();
    dateTime.setTimeSpec( Qt::UTC );
    return dateTime;
  }
 
  if ( !valueIsAcceptable( value, false ) && valueAsTimeType( value ) != "defined-date-time"_L1 )
    return QDateTime();
 
  return value.toMap().value( u"value"_s ).toDateTime();
}
 
bool QgsProcessingParameterMeshDatasetTime::valueIsAcceptable( const QVariant &input, bool allowEmpty )
{
  if ( !input.isValid() )
    return allowEmpty;
 
  if ( input.toDateTime().isValid() )
    return true;
 
  if ( input.userType() != QMetaType::Type::QVariantMap )
    return false;
 
  const QVariantMap map = input.toMap();
 
  if ( map.isEmpty() )
    return allowEmpty;
 
  if ( !map.contains( u"type"_s ) )
    return false;
 
  const QString type = map.value( u"type"_s ).toString();
  const QVariant value = map.value( u"value"_s );
 
  if ( type == "static"_L1 || type == "current-context-time"_L1 )
    return true;
 
  if ( type == "dataset-time-step"_L1 )
  {
    if ( value.userType() != QMetaType::Type::QVariantList )
      return false;
    const QVariantList list = value.toList();
    if ( value.toList().count() != 2 )
      return false;
    if ( list.at( 0 ).userType() != QMetaType::Type::Int || list.at( 1 ).userType() != QMetaType::Type::Int )
      return false;
  }
  else if ( type == "defined-date-time"_L1 )
  {
    if ( value.userType() != QMetaType::Type::QDateTime )
      return false;
  }
  else
    return false;
 
  return true;
}