@@ -24,8 +24,8 @@ package astraea.spark.rasterframes.expressions
2424import java .net .URI
2525
2626import astraea .spark .rasterframes .encoders .CatalystSerializer ._
27+ import astraea .spark .rasterframes .ref .RasterSource .ReadCallback
2728import astraea .spark .rasterframes .ref .{RasterRef , RasterSource }
28- import astraea .spark .rasterframes .util ._
2929import com .typesafe .scalalogging .LazyLogging
3030import org .apache .spark .sql .catalyst .expressions .codegen .CodegenFallback
3131import org .apache .spark .sql .catalyst .expressions .{ExpectsInputTypes , Expression , UnaryExpression }
@@ -41,7 +41,7 @@ import org.apache.spark.unsafe.types.UTF8String
4141 *
4242 * @since 5/4/18
4343 */
44- case class URIToRasterSource (override val child : Expression , accumulator : Option [ReadAccumulator ])
44+ case class URIToRasterSource (override val child : Expression , accumulator : Option [ReadCallback ])
4545 extends UnaryExpression with ExpectsInputTypes with CodegenFallback with LazyLogging {
4646
4747 override def nodeName : String = " uri_to_raster_source"
@@ -61,8 +61,8 @@ case class URIToRasterSource(override val child: Expression, accumulator: Option
6161object URIToRasterSource {
6262 def apply (rasterURI : Column ): TypedColumn [Any , RasterRef ] =
6363 new Column (new URIToRasterSource (rasterURI.expr, None )).as[RasterRef ]
64- def apply (rasterURI : Column , accumulator : ReadAccumulator ): TypedColumn [Any , RasterRef ] =
64+ def apply (rasterURI : Column , accumulator : ReadCallback ): TypedColumn [Any , RasterRef ] =
6565 new Column (new URIToRasterSource (rasterURI.expr, Option (accumulator))).as[RasterRef ]
66- def apply (rasterURI : Column , accumulator : Option [ReadAccumulator ]): TypedColumn [Any , RasterRef ] =
66+ def apply (rasterURI : Column , accumulator : Option [ReadCallback ]): TypedColumn [Any , RasterRef ] =
6767 new Column (new URIToRasterSource (rasterURI.expr, accumulator)).as[RasterRef ]
6868}
0 commit comments