geotrellis.spark.partition.SpatiallyPartition.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of geotrellis-spark_2.10 Show documentation

GeoTrellis is an open source geographic data processing engine for high performance applications.

The newest version!

package geotrellis.spark.partition

import geotrellis.spark._
import geotrellis.util._

import org.apache.spark.rdd._

import scala.reflect.ClassTag

object SpatiallyPartition {
  def apply[
    K: Boundable: PartitionerIndex: ClassTag,
    V: ClassTag,
    M: GetComponent[?, Bounds[K]]
  ](rdd: RDD[(K, V)] with Metadata[M]): RDD[(K, V)] with Metadata[Bounds[K]] =
    apply(rdd, rdd.metadata.getComponent[Bounds[K]])

  def apply[
    K: Boundable: PartitionerIndex: ClassTag,
    V: ClassTag,
    M: GetComponent[?, Bounds[K]]
  ](rdd: RDD[(K, V)] with Metadata[M], filterBounds: Bounds[K]): RDD[(K, V)] with Metadata[Bounds[K]] = {
    val partitioner = SpacePartitioner(filterBounds)
    partitioner(rdd)
  }
}