Comment: Use Overload

locationtech · Jun 15, 2017 · d91a876 · d91a876
1 parent b25bc70
commit d91a876
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/spark/src/main/scala/geotrellis/spark/viewshed/IterativeViewshed.scala b/spark/src/main/scala/geotrellis/spark/viewshed/IterativeViewshed.scala
@@ -162,18 +162,13 @@ object IterativeViewshed {
     val (p, index) = pi
     val md = rdd.metadata
 
-    val p2 = new jts.Coordinate(p.x, p.y)
-    val bounds = md.layout.mapTransform(p2.envelope)
-    require(bounds.colMin == bounds.colMax)
-    require(bounds.rowMin == bounds.rowMax)
-
     val cols = md.layout.tileCols
     val rows = md.layout.tileRows
-    val key = SpatialKey(bounds.colMin, bounds.rowMin)
+    val key = md.layout.mapTransform(Point(p.x, p.y))
     val extent = md.mapTransform(key)
     val re = RasterExtent(extent, cols, rows)
-    val col = re.mapXToGrid(p2.x)
-    val row = re.mapYToGrid(p2.y)
+    val col = re.mapXToGrid(p.x)
+    val row = re.mapYToGrid(p.y)
     val viewHeight = p.viewHeight
 
     PointInfo(