Update/add on-screen messages

SantanderMetGroup · Jul 21, 2017 · b9cb265 · b9cb265
1 parent b66ec53
commit b9cb265
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/R/petGrid.R b/R/petGrid.R
@@ -79,9 +79,11 @@ petGrid <- function(tasmin = NULL,
                     method = c("thornthwaite", "hargreaves"),
                     ...) {
     method <- match.arg(method, choices = c("thornthwaite", "hargreaves"))
+    message("[", Sys.time(), "] Computing PET-", method, " ...")
     out <- switch(method,
            "thornthwaite" = petGrid.th(tas, ...),
            "hargreaves" = petGrid.har(tasmin, tasmax, pr, ...))
+    message("[", Sys.time(), "] Done")
     ## Recover the grid structure -----------------------
     coords <- getCoordinates(out$ref.grid)
     pet.grid <- out$ref.grid

diff --git a/R/speiGrid.R b/R/speiGrid.R
@@ -50,7 +50,7 @@ speiGrid <- function(pr.grid, et0.grid = NULL, scale = 3, ...) {
     coords <- getCoordinates(pr.grid)
     dimNames <- getDim(pr.grid)
     n.mem <- getShape(pr.grid, "member")
-    method <- ifelse(is.null(et0.grid), "SPI", "SPEI")
+    method <- paste(ifelse(is.null(et0.grid), "SPI", "SPEI"), scale, sep = "-")
     message("[", Sys.time(), "] Computing ", method, " ...")
     spei.list <- lapply(1:n.mem, function(x) {
         pr <- subsetGrid(pr.grid, members = x, drop = TRUE) %>% extract2("Data") %>% array3Dto2Dmat()
@@ -68,8 +68,8 @@ speiGrid <- function(pr.grid, et0.grid = NULL, scale = 3, ...) {
     pr.grid$Data <- do.call("abind", c(spei.list, along = -1L)) %>% unname()
     attr(pr.grid$Data, "dimensions") <- dimNames
     pr.grid <- redim(pr.grid, drop = TRUE)
-    pr.grid$Variable$varName <- paste(method, scale, sep = "_")
-    longname <- ifelse(method == "SPI", "Standardized Precipitation Index", "Standardized Precipitation-Evapotranspiration Index")
+    pr.grid$Variable$varName <- method
+    longname <- ifelse(grepl("SPI", method), "Standardized Precipitation Index", "Standardized Precipitation-Evapotranspiration Index")
     attr(pr.grid$Variable, "longname") <- paste(longname, scale)
     attr(pr.grid$Variable, "units") <- "n/d"
     attr(pr.grid$Variable, "daily_agg_cellfun") <- "sum"