LayerToGeoTIFF.scala

import geotrellis.raster.io.geotiff.SinglebandGeoTiff
import org.apache.spark.SparkFiles
import org.apache.spark.sql.SparkSession
import org.locationtech.rasterframes._
import org.locationtech.rasterframes.datasource.geotiff._


object LayerToGeoTiff {
  def main(args: Array[String]): Unit = {
    implicit val spark = new SparkSession.Builder()
      .master("local[*]")
      .withKryoSerialization
      .getOrCreate()
      .withRasterFrames

    val imageName = "LC08_B7_Memphis_COG.tiff"

    spark.sparkContext.addFile("https://raw.githubusercontent.com/locationtech/rasterframes/develop/core/src/test/resources/" + imageName)

    val tiff = SinglebandGeoTiff(SparkFiles.get(imageName))

    val layer = tiff.projectedRaster.toLayer(64, 64)

    // Could write the layer out if we wanted to
    // import org.locationtech.rasterframes.datasource.geotrellis._
    //val layerDest = Files.createTempFile("gt-", "-layer")
    //layer.write.geotrellis.asLayer(Layer(layerDest.toUri, "test", 0))

    layer.write.geotiff.save("former-layer.tif")

    spark.stop()
  }
}