fix no pi or N case

IvanWilli · Nov 19, 2022 · 76df243 · 76df243
1 parent 9b8c0db
commit 76df243
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/R/kin_time_variant.R b/R/kin_time_variant.R
@@ -42,8 +42,10 @@ kin_time_variant <- function(U = NULL, f = NULL, N = NULL, pi = NULL,
     if(is.null(N)){
       # create pi and fill it during the loop
       message("Stable assumption was made for calculating pi on each year because no input data.")
+      pi_N_null_flag <- TRUE
       pi <- matrix(0, nrow=ages, ncol=n_years_data)
     }else{
+      pi_N_null_flag <-FALSE
       pi <- rbind(t(t(N * f)/colSums(N * f)), matrix(0,ages,length(years_data)))
     }
   }
@@ -72,7 +74,7 @@ kin_time_variant <- function(U = NULL, f = NULL, N = NULL, pi = NULL,
     # print(iyear)
     Ut <- as.matrix(U[[iyear]])
     ft <- as.matrix(f[[iyear]])
-    if(is.null(pi)){
+    if(pi_N_null_flag){
       A <- Ut[1:ages,1:ages] + ft[1:ages,1:ages]
       A_decomp = eigen(A)
       w <- as.double(A_decomp$vectors[,1])/sum(as.double(A_decomp$vectors[,1]))