/
variables_nat.R
36 lines (27 loc) · 1.15 KB
/
variables_nat.R
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#' @title Create natural immunity variables
#' @param variables a list
#' @param parameters list of model parameters
#' @importFrom individual IntegerVariable
#' @return named list of individual::Variable
#' @export
create_natural_immunity_variables <- function(variables, parameters) {
n <- sum(parameters$population)
variables$inf_num <- IntegerVariable$new(initial_values = rep(0L, n))
variables$inf_time <- IntegerVariable$new(initial_values = rep(-1L, n))
return(variables)
}
#' @title Create independent infection-derived NAT variables
#' @description This creates an independent set of variables to track infection-derived NAT independently
#' from vaccine-derived NAT.
#' @param variables a list
#' @param parameters list of model parameters
#' @importFrom individual DoubleVariable
#' @return named list of individual::Variable
#' @export
create_independent_nat_variables <- function(variables, parameters) {
n <- sum(parameters$population)
variables <- create_natural_immunity_variables(variables = variables, parameters = parameters)
# ab dynamics
variables$ab_titre_inf <- DoubleVariable$new(initial_values = rep(-Inf, n))
return(variables)
}