diff --git a/R/VizRobinson.R b/R/VizRobinson.R index 2e00161c4da453ee352d4d53506ea65e6500ff4d..63f5c7faa773e2fbf4d4ce8982362413771f7868 100644 --- a/R/VizRobinson.R +++ b/R/VizRobinson.R @@ -246,12 +246,14 @@ VizRobinson <- function(data, lon, lat, lon_dim = NULL, lat_dim = NULL, if (!identical(dim(mask), dim(data))) { stop("Parameter 'mask' must have the same dimensions as 'data'.") } else if (is.numeric(mask)) { + mask[which(is.na(mask))] <- 0 if (all(mask %in% c(0, 1))) { mask <- array(as.logical(mask), dim = dim(mask)) } else { stop("Parameter 'mask' must have only TRUE/FALSE or 0/1.") } } else if (is.logical(mask)) { + mask[which(is.na(mask))] <- F if (!all(mask %in% c(T, F))) { stop("Parameter 'mask' must have only TRUE/FALSE or 0/1.") }