Update RasterizeRDD Methods

locationtech · Aug 31, 2017 · 3698a39 · 3698a39
1 parent 769a73d
commit 3698a39
Show file tree

Hide file tree

Showing 6 changed files with 363 additions and 363 deletions.
diff --git a/spark/src/main/scala/geotrellis/spark/rasterize/FeatureRDDRasterizeMethods.scala b/spark/src/main/scala/geotrellis/spark/rasterize/FeatureRDDRasterizeMethods.scala
diff --git a/spark/src/main/scala/geotrellis/spark/rasterize/GeometryRDDRasterizeMethods.scala b/spark/src/main/scala/geotrellis/spark/rasterize/GeometryRDDRasterizeMethods.scala
diff --git a/spark/src/main/scala/geotrellis/spark/rasterize/Implicits.scala b/spark/src/main/scala/geotrellis/spark/rasterize/Implicits.scala
@@ -16,12 +16,19 @@
 
 package geotrellis.spark.rasterize
 
-import geotrellis.vector.Geometry
+import geotrellis.vector.{Geometry, Feature}
+import geotrellis.raster.rasterize.CellValue
 import org.apache.spark.rdd.RDD
 
 object Implicits extends Implicits
 
 trait Implicits {
   implicit class withGeometryRDDRasterizeMethods[G <: Geometry](val self: RDD[G])
       extends GeometryRDDRasterizeMethods[G]
+
+  implicit class withFeatureRDDRasterizeMethods[G <: Geometry](val self: RDD[Feature[G, Double]])
+      extends FeatureRDDRasterizeMethods[G]
+
+  implicit class withCellValueFeatureRDDRasterizeMethods[G <: Geometry](val self: RDD[Feature[G, CellValue]])
+      extends CellValueFeatureRDDRasterizeMethods[G]
 }
diff --git a/spark/src/main/scala/geotrellis/spark/rasterize/RasterizeFeaturesRDD.scala b/spark/src/main/scala/geotrellis/spark/rasterize/RasterizeFeaturesRDD.scala