geotrellis.raster.op.zonal.summary.Sum.scala Maven / Gradle / Ivy
The newest version!
package geotrellis.raster.op.zonal.summary
import geotrellis._
import geotrellis.source._
import geotrellis.feature._
import geotrellis.feature.rasterize._
object Sum extends TileSummary[Long,Long,ValueSource[Long]] {
def handlePartialTile[D](pt:PartialTileIntersection[D]):Long = {
val PartialTileIntersection(r,polygons) = pt
var sum: Long = 0L
for(p <- polygons.asInstanceOf[List[Polygon[D]]]) {
Rasterizer.foreachCellByFeature(p, r.rasterExtent)(
new Callback[Geometry,D] {
def apply(col:Int, row:Int, g:Geometry[D]) {
val z = r.get(col,row)
if (isData(z)) { sum = sum + z }
}
}
)
}
sum
}
def handleFullTile(ft:FullTileIntersection):Long = {
var s = 0L
ft.tile.foreach((x:Int) => if (isData(x)) s = s + x)
s
}
def converge(ds:DataSource[Long,_]) =
ds.reduce(_+_)
}
object SumDouble extends TileSummary[Double,Double,ValueSource[Double]] {
def handlePartialTile[D](pt:PartialTileIntersection[D]):Double = {
val PartialTileIntersection(r,polygons) = pt
var sum = 0.0
for(p <- polygons.asInstanceOf[List[Polygon[D]]]) {
Rasterizer.foreachCellByFeature(p, r.rasterExtent)(
new Callback[Geometry,D] {
def apply(col:Int, row:Int, g:Geometry[D]) {
val z = r.getDouble(col,row)
if(isData(z)) { sum = sum + z }
}
}
)
}
sum
}
def handleFullTile(ft:FullTileIntersection):Double = {
var s = 0.0
ft.tile.foreachDouble((x:Double) => if (isData(x)) s = s + x)
s
}
def converge(ds:DataSource[Double,_]) =
ds.reduce(_+_)
}