From 9e37d77a783f0c142cfc03590c690ab1da5a6464 Mon Sep 17 00:00:00 2001
From: Reinhold Kainhofer <reinhold@kainhofer.com>
Date: Sat, 19 May 2018 13:53:38 +0000
Subject: [PATCH] Add mT.setDimInfo function to set the dimensional infor for
 makeQxDataFrame

---
 R/utilityFunctions.R | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/R/utilityFunctions.R b/R/utilityFunctions.R
index db3a5d0..78365ab 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
+}
+
-- 
GitLab