From 3947731a5a8c3598271b26f5201dea4bcb13ef6d Mon Sep 17 00:00:00 2001 From: Johannes Ranke Date: Thu, 19 Jan 2017 10:47:09 +0100 Subject: Fix one_box for ini = 1, use in sawtooth examples --- R/twa.R | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) (limited to 'R/twa.R') diff --git a/R/twa.R b/R/twa.R index 88cfec3..0e03baf 100644 --- a/R/twa.R +++ b/R/twa.R @@ -43,7 +43,7 @@ #' # Use a model with more than one observed variable #' m_2 <- mkinmod(parent = mkinsub("SFO", "m1"), m1 = mkinsub("SFO")) #' fit_2 <- mkinfit(m_2, FOCUS_2006_D, quiet = TRUE) -#' pred_2 <- one_box(fit_2) +#' pred_2 <- one_box(fit_2, ini = "model") #' plot(pred_2) one_box <- function(x, ini, ..., t_end = 100, res = 0.01) @@ -99,11 +99,10 @@ one_box.character <- function(x, ini = 1, parms, ..., #' @export one_box.mkinfit <- function(x, ini = c("model", 1), ..., t_end = 100, res = 0.01) { fit <- x - - ini = match.arg(ini) - if (ini == "model") { + if (ini[1] == "model") { odeini = x$bparms.state } else { + if (ini[1] != 1) stop ("Argument ini can only be 'model' or 1") odeini <- c(1, rep(0, length(fit$mkinmod$spec) - 1)) names(odeini) <- names(fit$mkinmod$spec) } @@ -142,7 +141,7 @@ one_box.mkinfit <- function(x, ini = c("model", 1), ..., t_end = 100, res = 0.01 #' # Use a fitted mkinfit model #' m_2 <- mkinmod(parent = mkinsub("SFO", "m1"), m1 = mkinsub("SFO")) #' fit_2 <- mkinfit(m_2, FOCUS_2006_D, quiet = TRUE) -#' pred_2 <- one_box(fit_2) +#' pred_2 <- one_box(fit_2, ini = 1) #' pred_2_saw <- sawtooth(pred_2, 2, 7) #' plot(pred_2_saw, max_twa = 21, max_twa_var = "m1") plot.one_box <- function(x, @@ -184,7 +183,7 @@ plot.one_box <- function(x, #' #' m_2 <- mkinmod(parent = mkinsub("SFO", "m1"), m1 = mkinsub("SFO")) #' fit_2 <- mkinfit(m_2, FOCUS_2006_D, quiet = TRUE) -#' pred_2 <- one_box(fit_2) +#' pred_2 <- one_box(fit_2, ini = 1) #' pred_2_saw <- sawtooth(pred_2, 2, 7) #' plot(pred_2_saw, max_twa = 21, max_twa_var = "m1") #' -- cgit v1.2.1