Enable any valid Gdal file format for inputs and outputs for the different algorithms

Author: XavierCLL

automated_global_coregistration_algorithm.py

@@ -27,6 +27,8 @@
                        QgsProcessingParameterNumber, QgsProcessingParameterDefinition, QgsProcessingParameterEnum,
                        QgsProcessingParameterPoint, QgsProcessingParameterBoolean)
 
+from Coregistration.utils.system_utils import get_raster_driver_name_by_extension
+
 
 class AutomatedGlobalCoregistrationAlgorithm(QgsProcessingAlgorithm):
     """
@@ -257,6 +259,7 @@ def get_inputfilepath(layer):
         resampling_method = self.resampling_methods[self.parameterAsEnum(parameters, self.RESAMPLING, context)][1]
 
         output_file = self.parameterAsOutputLayer(parameters, self.OUTPUT, context)
+        output_driver_name = get_raster_driver_name_by_extension(output_file)
 
         feedback.pushInfo("Image to image Co-Registration:")
         feedback.pushInfo("\nProcessing file: " + img_tgt)
@@ -265,7 +268,7 @@ def get_inputfilepath(layer):
         feedback.pushInfo("\n(To check the complete log of the process, open the Python Console)...\n")
         CR = COREG(img_ref, img_tgt, path_out=output_file, align_grids=align_grids, match_gsd=match_gsd,
                    wp=(wp_x, wp_y), ws=(ws_x, ws_y), resamp_alg_deshift=resampling_method,
-                   max_shift=max_shift, max_iter=15, fmt_out="GTiff", out_crea_options=["WRITE_METADATA=NO"],
+                   max_shift=max_shift, max_iter=15, fmt_out=output_driver_name, out_crea_options=["WRITE_METADATA=NO"],
                    CPUs=1 if platform.system() == "Windows" else None)
         CR.correct_shifts()
 

automated_local_coregistration_algorithm.py

@@ -27,6 +27,8 @@
                        QgsProcessingParameterNumber, QgsProcessingParameterDefinition, QgsProcessingParameterEnum,
                        QgsProcessingParameterBoolean)
 
+from Coregistration.utils.system_utils import get_raster_driver_name_by_extension
+
 
 class AutomatedLocalCoregistrationAlgorithm(QgsProcessingAlgorithm):
     """
@@ -252,6 +254,7 @@ def get_inputfilepath(layer):
         resampling_method = self.resampling_methods[self.parameterAsEnum(parameters, self.RESAMPLING, context)][1]
 
         output_file = self.parameterAsOutputLayer(parameters, self.OUTPUT, context)
+        output_driver_name = get_raster_driver_name_by_extension(output_file)
 
         feedback.pushInfo("Image to image Co-Registration:")
         feedback.pushInfo("\nProcessing file: " + img_tgt)
@@ -264,7 +267,7 @@ def get_inputfilepath(layer):
         CRL = COREG_LOCAL(img_ref, img_tgt, path_out=output_file, align_grids=align_grids, match_gsd=match_gsd,
                           grid_res=grid_res, window_size=(window_size, window_size),
                           resamp_alg_deshift=resampling_method, max_shift=max_shift, max_iter=15,
-                          fmt_out="GTiff", out_crea_options=["WRITE_METADATA=NO"],
+                          fmt_out=output_driver_name, out_crea_options=["WRITE_METADATA=NO"],
                           CPUs=1 if platform.system() == "Windows" else None)
         CRL.correct_shifts()
 

panning_pixel_adjustment_algorithm.py

@@ -187,7 +187,7 @@ def get_inputfilepath(layer):
             output_file = file_in_path
 
         # gdal driver based on the output file
-        gdal_driver = gdal.GetDriverByName(get_raster_driver_name_by_extension(output_file.split(".")[-1]))
+        gdal_driver = gdal.GetDriverByName(get_raster_driver_name_by_extension(output_file))
         gdal_driver.CreateCopy(output_file, input_ds)
         input_ds = None
 

utils/system_utils.py

@@ -18,10 +18,11 @@
  *                                                                         *
  ***************************************************************************/
 """
+import os
 
 
-def get_raster_driver_name_by_extension(file_extension):
-    # Normalize the extension by removing leading dot and converting to lowercase
+def get_raster_driver_name_by_extension(file_path):
+    file_extension = os.path.splitext(file_path)[1]
     ext = file_extension.lower().lstrip('.')
 
     # Dictionary mapping file extensions to GDAL driver names