diff --git a/R/utilityFunctions.R b/R/utilityFunctions.R index db3a5d0898a823b78dd928a19226a579ab3830ab..78365ab8acfbeff3e598375e2381de9c9b634b69 100644 --- a/R/utilityFunctions.R +++ b/R/utilityFunctions.R @@ -183,6 +183,9 @@ mT.extrapolateProbsExp = function(table, age, up = TRUE) { mT.fitExtrapolationLaw = function(table, method = "LF2", law = "HP", fit = 75:99, extrapolate = 80:120, fadeIn = 80:90, fadeOut = NULL) { + if (!is(table, "mortalityTable")) + stop("First argument must be a mortalityTable.") + ages = ages(table) if (!is.null(table@exposures) && !is.na(table@exposures)) { Ex = table@exposures @@ -215,3 +218,16 @@ mT.fitExtrapolationLaw = function(table, method = "LF2", law = "HP", table } +#' @export +mT.setDimInfo = function(table, ..., append = TRUE) { + if (!is(table, "mortalityTable")) + stop("First argument must be a mortalityTable.") + + if (append) { + table@data$dim[names(list(...))] = list(...) + } else { + table@data$dim = list(...) + } + table +} +