catch and explain GDAL calc errors

databrickslabs · Nov 29, 2024 · ccc98c8 · ccc98c8
1 parent 58c3ba5
commit ccc98c8
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/main/scala/com/databricks/labs/mosaic/core/raster/operator/gdal/GDALCalc.scala b/src/main/scala/com/databricks/labs/mosaic/core/raster/operator/gdal/GDALCalc.scala
@@ -5,6 +5,9 @@ import com.databricks.labs.mosaic.core.raster.gdal.{MosaicRasterGDAL, MosaicRast
 import com.databricks.labs.mosaic.utils.SysUtils
 import org.gdal.gdal.gdal
 
+import java.nio.file.{Files, Paths}
+import scala.util.Try
+
 /** GDALCalc is a helper object for executing GDAL Calc commands. */
 object GDALCalc {
 
@@ -35,6 +38,14 @@ object GDALCalc {
         val toRun = effectiveCommand.replace("gdal_calc", gdal_calc)
         val commandRes = SysUtils.runCommand(s"python3 $toRun")
         val errorMsg = gdal.GetLastErrorMsg
+        val size = Try(Files.size(Paths.get(resultPath))).getOrElse(
+            {
+                val msg = "Error during GDAL calc operation: " +
+                    s"file $resultPath " +
+                    s"with command '$effectiveCommand'. GDAL returned error: $errorMsg"
+                throw new Exception(msg)
+            }
+        )
         val result = GDAL.raster(resultPath, resultPath, None)
         val createInfo = Map(
           "path" -> resultPath,