fix suffix for TIN Mesh creation

qgis · Nov 14, 2021 · 3e28fb0 · 3e28fb0
1 parent 637e5ca
commit 3e28fb0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/analysis/processing/qgsalgorithmtinmeshcreation.cpp b/src/analysis/processing/qgsalgorithmtinmeshcreation.cpp
@@ -71,9 +71,13 @@ void QgsTinMeshCreationAlgorithm::initAlgorithm( const QVariantMap &configuratio
   if ( meta )
     driverList = meta->meshDriversMetadata();
 
-  for ( const QgsMeshDriverMetadata &driverMeta : driverList )
+  for ( const QgsMeshDriverMetadata &driverMeta : std::as_const( driverList ) )
     if ( driverMeta.capabilities() & QgsMeshDriverMetadata::CanWriteMeshData )
-      mAvailableFormat.append( driverMeta.name() );
+    {
+      const QString name = driverMeta.name();
+      mDriverSuffix[name] = driverMeta.writeMeshFrameOnFileSuffix();
+      mAvailableFormat.append( name );
+    }
 
   addParameter( new QgsProcessingParameterEnum( QStringLiteral( "MESH_FORMAT" ), QObject::tr( "Output format" ), mAvailableFormat, false, 0 ) );
   addParameter( new QgsProcessingParameterCrs( QStringLiteral( "CRS_OUTPUT" ), QObject::tr( "Output coordinate system" ), QVariant(), true ) );
@@ -160,7 +164,7 @@ QVariantMap QgsTinMeshCreationAlgorithm::processAlgorithm( const QVariantMap &pa
   if ( feedback && feedback->isCanceled() )
     return QVariantMap();
 
-  const QString fileName = parameterAsFile( parameters, QStringLiteral( "OUTPUT_MESH" ), context );
+  QString fileName = parameterAsFile( parameters, QStringLiteral( "OUTPUT_MESH" ), context );
   const int driverIndex = parameterAsEnum( parameters, QStringLiteral( "MESH_FORMAT" ), context );
   const QString driver = mAvailableFormat.at( driverIndex );
   if ( feedback )
@@ -172,6 +176,12 @@ QVariantMap QgsTinMeshCreationAlgorithm::processAlgorithm( const QVariantMap &pa
 
   const QgsProviderMetadata *providerMetadata = QgsProviderRegistry::instance()->providerMetadata( QStringLiteral( "mdal" ) );
 
+  QFileInfo fileInfo( fileName );
+  QString suffix = fileInfo.suffix();
+
+  if ( suffix.isEmpty() )
+    fileName.append( QStringLiteral( ".%1" ).arg( mDriverSuffix.value( driver ) ) );
+
   if ( feedback )
     feedback->setProgressText( QObject::tr( "Saving mesh to file" ) );
   if ( providerMetadata )

diff --git a/src/analysis/processing/qgsalgorithmtinmeshcreation.h b/src/analysis/processing/qgsalgorithmtinmeshcreation.h
@@ -46,6 +46,7 @@ class QgsTinMeshCreationAlgorithm: public QgsProcessingAlgorithm
 
   private:
     QStringList mAvailableFormat;
+    QMap<QString, QString> mDriverSuffix;
 
     struct Layer
     {