From f08ad7191bbc2995b610c016b400a077e7844986 Mon Sep 17 00:00:00 2001
From: Reinhold Kainhofer <reinhold@kainhofer.com>
Date: Mon, 26 Oct 2020 16:56:28 +0100
Subject: [PATCH] getPeriodTable: Allow giving desired ages; use dimensional
 data of original table (overwrite only the year)

---
 R/getPeriodTable.R | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/R/getPeriodTable.R b/R/getPeriodTable.R
index d76cca9..972064e 100644
--- a/R/getPeriodTable.R
+++ b/R/getPeriodTable.R
@@ -23,15 +23,22 @@ setGeneric("getPeriodTable",
 #' @describeIn getPeriodTable Return the period life table as a
 #'             \code{mortalityTable.period} object
 setMethod("getPeriodTable","mortalityTable",
-          function (object, Period, ...) {
+          function (object, Period, ages = NULL, ...) {
               if(missing(Period)) {
                   Period = baseYear(object)
               }
+              if (missing(ages) | is.null(ages)) {
+                  ages = ages(object)
+              }
+              data = object@data
+              data$dim$Period = Period
+              data$dim$year = Period
               mortalityTable.period(
                   name = paste0(object@name, ", Period ", Period),
                   baseYear = Period,
-                  ages = ages(object),
-                  deathProbs = periodDeathProbabilities(object, Period = Period, ...)
+                  ages = ages,
+                  deathProbs = periodDeathProbabilities(object, Period = Period, ages = ages, ...),
+                  data = data
               )
           })
 
-- 
GitLab