Avoid GeoAttrsError by using more robust target extent calculation (#3572)

* avoid GeoAttrsError by using more robust target extent calculation in case a LayoutDefinition is available
#3559

* Code cleanup

---------

Co-authored-by: Jeroen Dries <jeroen.dries@vito.be>

Author: pomadchin

CHANGELOG.md

@@ -15,6 +15,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - GeoTiffReader Unknown tags skip fix [#3557](https://github.com/locationtech/geotrellis/pull/3557)
 - Fix implicitNotFound error and rename RGBAMethods implicit class [#3563](https://github.com/locationtech/geotrellis/pull/3563)
 - Set 'role' attribute in GDALMetadata tag written to tiff header, to match GDAL behaviour. [#3496](https://github.com/locationtech/geotrellis/issues/3496)
+- Avoid GeoAttrsError by using more robust target extent calculation [#3572](https://github.com/locationtech/geotrellis/pull/3572)
 
 ## [3.7.1] - 2024-01-08
 

spark/src/main/scala/geotrellis/spark/reproject/TileRDDReproject.scala

@@ -66,12 +66,17 @@ object TileRDDReproject {
   ): (Int, RDD[(K, V)] with Metadata[TileLayerMetadata[K]]) = {
     val crs: CRS = metadata.crs
     val layout = metadata.layout
-    val tileLayout: TileLayout = layout.tileLayout
     implicit val sc = bufferedTiles.context
 
     val sourceDataGridExtent = metadata.layout.createAlignedGridExtent(metadata.extent)
-    val passthroughGridExtent = ReprojectRasterExtent(sourceDataGridExtent, metadata.crs, destCrs)
-    val targetDataExtent = passthroughGridExtent.extent
+    val (targetCellSize, targetDataExtent) =
+      targetLayout match {
+        case Right(l) =>
+          l.cellSize -> l.createAlignedGridExtent(ProjectedExtent(metadata.extent,metadata.crs).reproject(destCrs)).extent
+        case Left(_) =>
+          val passthroughGridExtent = ReprojectRasterExtent(sourceDataGridExtent, metadata.crs, destCrs)
+          passthroughGridExtent.cellSize -> passthroughGridExtent.extent
+      }
 
     val targetPartitioner: Option[Partitioner] = partitioner.orElse(bufferedTiles.partitioner)
 
@@ -112,7 +117,7 @@ object TileRDDReproject {
                   layoutScheme.levelFor(targetDataExtent, ct)
 
                 case None =>
-                  layoutScheme.levelFor(targetDataExtent, passthroughGridExtent.cellSize)
+                  layoutScheme.levelFor(targetDataExtent, targetCellSize)
               }
           }
         }

spark/src/test/scala/geotrellis/spark/reproject/TileRDDReprojectSpec.scala

@@ -26,10 +26,8 @@ import geotrellis.spark.reproject.Reproject.Options
 import geotrellis.spark.testkit._
 import geotrellis.vector._
 import geotrellis.proj4._
-
 import spire.syntax.cfor._
 import org.apache.spark._
-
 import org.scalatest.funspec.AnyFunSpec
 
 class TileRDDReprojectSpec extends AnyFunSpec with TestEnvironment {
@@ -328,4 +326,22 @@ class TileRDDReprojectSpec extends AnyFunSpec with TestEnvironment {
       afterMetadata.layout should be (beforeMetadata.layout)
     }
   }
+
+  describe("Very small cube at high latitude") {
+    it("should reproject without error") {
+      val layout = LayoutDefinition(RasterExtent(Extent(-120.550992, 71.82628788888888, -120.5421031111111, 71.83517677777778), 32, 32), 32, 32)
+      val targetLayout = LayoutDefinition(RasterExtent(Extent(585180.0, 7971640.0, 585820.0, 7972280.0), 32, 32), 32, 32)
+      val extent = Extent(-120.550992, 71.834899, -120.550792, 71.835099)
+      val cube: TileLayerRDD[SpatialKey] = TileLayerRDDBuilders.createTileLayerRDD(sc, ByteArrayTile.fill(1, 32, 32), layout.tileLayout, LatLng)
+
+      val utm = CRS.fromEpsgCode(32610)
+
+      val badMetadata = cube.metadata.copy(crs = LatLng, extent = extent, layout = layout)
+      val result = TileRDDReproject(ContextRDD(cube, badMetadata), utm, Right(targetLayout), 16, Reproject.Options.DEFAULT, None)
+      //should not throw an error
+      val resultMetadata = result._2.metadata
+      targetLayout should be(resultMetadata.layout)
+      utm should be(resultMetadata.crs)
+    }
+  }
 }