Add GeoTiffSegmentLayout.intersectingSegments

locationtech · Dec 9, 2016 · 1ca071e · 1ca071e
1 parent e0cac50
commit 1ca071e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/raster/src/main/scala/geotrellis/raster/io/geotiff/GeoTiffSegmentLayout.scala b/raster/src/main/scala/geotrellis/raster/io/geotiff/GeoTiffSegmentLayout.scala
@@ -190,6 +190,8 @@ case class GeoTiffSegmentLayout(totalCols: Int, totalRows: Int, tileLayout: Tile
    * Generate layout for cropped a cropped region of this raster.
    * Preserves the structure of parent layout as best possible,
    * using same storageMethod and segment size.
+   *
+   * @param bounds Pixel bounds inside the current segment layout
    */
   def crop(bounds: GridBounds): GeoTiffSegmentLayout = {
     if(isTiled) {
@@ -212,6 +214,19 @@ case class GeoTiffSegmentLayout(totalCols: Int, totalRows: Int, tileLayout: Tile
       GeoTiffSegmentLayout(bounds.width, bounds.height, newLayout, isTiled = false)
     }
   }
+
+  /** Returns all segment indices which intersect given pixel grid bounds */
+  def intersectingSegments(bounds: GridBounds): Array[Int] = {
+    val tc = tileLayout.tileCols
+    val tr = tileLayout.tileRows
+    val ab = ArrayBuffer[Int]()
+    for (layoutCol <- (bounds.colMin / tc) to (bounds.colMax / tc)) {
+      for (layoutRow <- (bounds.rowMin / tr) to (bounds.rowMax / tr)) {
+        ab += (layoutRow * tileLayout.layoutCols) + layoutCol
+      }
+    }
+    ab.toArray
+  }
 }
 
 /**