• Home
• com.azavea.geotrellis
• geotrellis_2.10
• 0.9.2
• source code
• conditional.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

geotrellis.raster.op.local.conditional.scala Maven / Gradle / Ivy

package geotrellis.raster.op.local

import geotrellis._
import geotrellis.source._

/**
 * Maps all cells matching `cond` to Int `trueValue`.
 */
object IfCell extends Serializable {
  /**
   * Maps all cells matching `cond` to Int `trueValue`.
   */
  def apply(r:Op[Raster], cond:Int => Boolean, trueValue:Int) = 
    r.map(_.dualMap(z => if (cond(z)) trueValue else z)
                   ({z:Double => if (cond(d2i(z))) i2d(trueValue) else z}))
     .withName("IfCell")

  /**
   * Maps all cells matching `cond` to Double `trueValue`.
   */
  def apply(r:Op[Raster], cond:Double => Boolean, trueValue:Double) = 
    r.map(_.dualMap({z:Int => if (cond(i2d(z))) d2i(trueValue) else z})
                                 (z => if (cond(z)) trueValue else z))
     .withName("IfCell")

  /** Set all values of output raster to one value or another based on whether a
   * condition is true or false.  */
  def apply(r:Op[Raster], cond:Int => Boolean, trueValue:Int, falseValue: Int) = 
    r.map(_.dualMap(z => if (cond(z)) trueValue else falseValue)
                   ({z:Double => if (cond(d2i(z))) i2d(trueValue) else i2d(falseValue)}))
     .withName("IfCell")

  /** Set all values of output raster to one value or another based on whether a
   * condition is true or false for Double values.  */
  def apply(r:Op[Raster], cond:Double => Boolean, trueValue:Double, falseValue: Double) = 
    r.map(_.dualMap({z:Int => if (cond(i2d(z))) d2i(trueValue) else d2i(falseValue)}) (z => if (cond(z)) trueValue else falseValue))
     .withName("IfCell")

  /** Given a condition over two rasters, set the value of each cell in the output
   * to a specified value if the condition is true given the corresponding values in
   * each of the two input rasters. */
  def apply(r1:Op[Raster], r2:Op[Raster], cond:(Int, Int) => Boolean, trueValue:Int) = 
    (r1,r2).map( (a,b) => a.dualCombine(b)((z1,z2) => if (cond(z1, z2)) trueValue else z1)
                                          ((z1,z2) => if (cond(d2i(z1), d2i(z2))) i2d(trueValue) else z1))
           .withName("BinaryIfCell")

  /** Given a Double condition over two rasters, set the value of each cell in the output
   * to a specified value if the condition is true given the corresponding values in
   * each of the two input rasters. */
  def apply(r1:Op[Raster], r2:Op[Raster], cond:(Double, Double) => Boolean, trueValue:Double) = 
    (r1,r2).map( (a,b) => a.dualCombine(b)((z1,z2) => if (cond(i2d(z1), i2d(z2))) d2i(trueValue) else z1)
                                          ((z1,z2) => if (cond(z1, z2)) trueValue else z1))
           .withName("BinaryIfCell")

  /** Given a condition over two rasters, set the value of each cell in the output
   * to a specified true or false value after testing the specified condition on 
   * the corresponding values in each of the two input rasters.*/
  def apply(r1:Op[Raster], r2:Op[Raster], cond:(Int, Int) => Boolean, trueValue:Int, falseValue:Int) = 
    (r1,r2).map( (a,b) => a.dualCombine(b)((z1,z2) => if(cond(z1,z2)) trueValue else falseValue)
                                          ((z1,z2) => if (cond(d2i(z1), d2i(z2))) i2d(trueValue) else i2d(falseValue) ))
           .withName("BinaryIfElseCell")

  /** Given a Double condition over two rasters, set the value of each cell in the output
   * to a specified true or false value after testing the specified condition on 
   * the corresponding values in each of the two input rasters.*/
  def apply(r1:Op[Raster], r2:Op[Raster], cond:(Double, Double) => Boolean, trueValue:Double, falseValue:Double) = 
    (r1,r2).map( (a,b) => a.dualCombine(b)((z1,z2) => if(cond(i2d(z1),i2d(z2))) d2i(trueValue) else d2i(falseValue))
                                          ((z1,z2) => if (cond(z1, z2)) trueValue else falseValue))
           .withName("BinaryIfElseCell")
}

trait ConditionalOpMethods[+Repr <: RasterSource] { self: Repr =>
  def localIf(cond:Int => Boolean,trueValue:Int) = 
    self.mapOp(IfCell(_,cond,trueValue))
  def localIf(cond:Double => Boolean,trueValue:Double) = 
    self.mapOp(IfCell(_,cond,trueValue))
  def localIf(cond:Int => Boolean,trueValue:Int,falseValue:Int) = 
    self.mapOp(IfCell(_,cond,trueValue,falseValue))
  def localIf(cond:Double => Boolean,trueValue:Double,falseValue:Double) = 
    self.mapOp(IfCell(_,cond,trueValue,falseValue))
  def localIf(rs:RasterSource,cond:(Int,Int)=>Boolean,trueValue:Int) = 
    self.combineOp(rs)(IfCell(_,_,cond,trueValue))
  def localIf(rs:RasterSource,cond:(Double,Double)=>Boolean,trueValue:Double) = 
    self.combineOp(rs)(IfCell(_,_,cond,trueValue))
  def localIf(rs:RasterSource,cond:(Int,Int)=>Boolean,trueValue:Int,falseValue:Int) = 
    self.combineOp(rs)(IfCell(_,_,cond,trueValue,falseValue))
  def localIf(rs:RasterSource,cond:(Double,Double)=>Boolean,trueValue:Double,falseValue:Double) = 
    self.combineOp(rs)(IfCell(_,_,cond,trueValue,falseValue))
}