R/ifelse.R

In magclass: Data Class and Tools for Handling Spatial-Temporal Data

#' @importFrom methods new
#' @exportMethod ifelse
setMethod("ifelse",
          signature(test = "magpie"),
          function(test, yes, no) {
            if (!all(suppressWarnings(as.integer(test)) %in% c(0, 1))) {
              stop("'test' must only contain booleans!")
            }
            yes <- test * yes
            no <- (1 - test) * no
            yes[!test] <- 0
            no[test] <- 0
            return(yes + no)
          }
)