Skip to content

Instantly share code, notes, and snippets.

@timchurches

timchurches/_icm.control.seiqhrf.R

Last active April 28, 2022 08:43
Show Gist options
  • Save timchurches/92073d0ea75cfbd387f91f7c6e624bd7 to your computer and use it in GitHub Desktop.
Save timchurches/92073d0ea75cfbd387f91f7c6e624bd7 to your computer and use it in GitHub Desktop.
Download ZIP
COVID-19 extensions to EpiModel v1.8
Raw
_icm.control.seiqhrf.R
control.icm <- function(type, nsteps, nsims = 1,
rec.rand = TRUE, quar.rand = TRUE, hosp.rand = TRUE, disch.rand = TRUE,
fat.rand = TRUE, a.rand = TRUE, d.rand = TRUE, initialize.FUN = initialize.icm,
infection.FUN = infection.icm, recovery.FUN = recovery.icm,
departures.FUN = departures.icm, arrivals.FUN = arrivals.icm,
get_prev.FUN = get_prev.icm, verbose = FALSE,
verbose.int = 0, skip.check = FALSE, ncores=1, ...) {
# Get arguments
p <- list()
formal.args <- formals(sys.function())
formal.args[["..."]] <- NULL
for (arg in names(formal.args)) {
if (as.logical(mget(arg) != "")) {
p[arg] <- list(get(arg))
}
}
dot.args <- list(...)
names.dot.args <- names(dot.args)
if (length(dot.args) > 0) {
for (i in 1:length(dot.args)) {
p[[names.dot.args[i]]] <- dot.args[[i]]
}
}
if ("births.FUN" %in% names(dot.args)) {
p$arrivals.FUN <- dot.args$births.FUN
p$births.FUN <- dot.args$births.FUN <- NULL
message("EpiModel 1.7.0 onward renamed the birth function births.FUN to arrivals.FUN. See documentation for details.")
}
if ("deaths.FUN" %in% names(dot.args)) {
p$departures.FUN <- dot.args$deaths.FUN
p$deaths.FUN <- dot.args$deaths.FUN <- NULL
message("EpiModel 1.7.0 onward renamed the death function deaths.FUN to departures.FUN. See documentation for details.")
}
## Module classification
p$bi.mods <- grep(".FUN", names(formal.args), value = TRUE)
p$user.mods <- grep(".FUN", names(dot.args), value = TRUE)
## Defaults and checks
if (is.null(p$type) | !(p$type %in% c("SI", "SIS", "SIR", "SEIR", "SEIQHR", "SEIQHRF"))) {
stop("Specify type as \"SI\", \"SIS\", \"SIR\", \"SEIR\", \"SEIQHR\", or \"SEIQHRF\" ", call. = FALSE)
}
if (is.null(p$nsteps)) {
stop("Specify nsteps", call. = FALSE)
}
## Output
class(p) <- c("control.icm", "list")
return(p)
}
Raw
_icm.icm.seiqhrf.R
merge.seiqhrf.icm <- function(x, y, ...) {
## Check structure
if (length(x) != length(y) || !identical(names(x), names(y))) {
stop("x and y have different structure")
}
if (x$control$nsims > 1 & y$control$nsims > 1 &
!identical(sapply(x, class), sapply(y, class))) {
stop("x and y have different structure")
}
## Check params
check1 <- identical(x$param, y$param)
check2 <- identical(x$control[-which(names(x$control) %in% c("nsims", "currsim"))],
y$control[-which(names(y$control) %in% c("nsims", "currsim"))])
if (check1 == FALSE) {
stop("x and y have different parameters")
}
if (check2 == FALSE) {
stop("x and y have different controls")
}
z <- x
new.range <- (x$control$nsims + 1):(x$control$nsims + y$control$nsims)
# Merge data
for (i in 1:length(x$epi)) {
if (x$control$nsims == 1) {
x$epi[[i]] <- data.frame(x$epi[[i]])
}
if (y$control$nsims == 1) {
y$epi[[i]] <- data.frame(y$epi[[i]])
}
z$epi[[i]] <- cbind(x$epi[[i]], y$epi[[i]])
names(z$epi[[i]])[new.range] <- paste0("sim", new.range)
}
z$control$nsims <- max(new.range)
return(z)
}
icm.seiqhrf <- function(param, init, control) {
crosscheck.icm(param, init, control)
verbose.icm(control, type = "startup")
nsims <- control$nsims
ncores <- ifelse(control$nsims == 1, 1, min(future::availableCores(), control$ncores))
control$ncores <- ncores
if (ncores == 1) {
# Simulation loop start
for (s in 1:control$nsims) {
## Initialization module
if (!is.null(control[["initialize.FUN"]])) {
dat <- do.call(control[["initialize.FUN"]], list(param, init, control))
}
# Timestep loop
for (at in 2:control$nsteps) {
## User Modules
um <- control$user.mods
if (length(um) > 0) {
for (i in 1:length(um)) {
dat <- do.call(control[[um[i]]], list(dat, at))
}
}
## Infection
if (!is.null(control[["infection.FUN"]])) {
dat <- do.call(control[["infection.FUN"]], list(dat, at))
}
## Recovery
if (!is.null(control[["recovery.FUN"]])) {
dat <- do.call(control[["recovery.FUN"]], list(dat, at))
}
## Departure Module
if (!is.null(control[["departures.FUN"]])) {
dat <- do.call(control[["departures.FUN"]], list(dat, at))
}
## Arrival Module
if (!is.null(control[["arrivals.FUN"]])) {
dat <- do.call(control[["arrivals.FUN"]], list(dat, at))
}
## Outputs
if (!is.null(control[["get_prev.FUN"]])) {
dat <- do.call(control[["get_prev.FUN"]], list(dat, at))
}
## Track progress
verbose.icm(dat, type = "progress", s, at)
}
# Set output
if (s == 1) {
out <- saveout.seiqhrf.icm(dat, s)
} else {
out <- saveout.seiqhrf.icm(dat, s, out)
}
} # Simulation loop end
class(out) <- "icm"
} # end of single core execution
if (ncores > 1) {
doParallel::registerDoParallel(ncores)
sout <- foreach(s = 1:nsims) %dopar% {
control$nsims <- 1
control$currsim <- s
## Initialization module
if (!is.null(control[["initialize.FUN"]])) {
dat <- do.call(control[["initialize.FUN"]], list(param, init, control))
}
# Timestep loop
for (at in 2:control$nsteps) {
## User Modules
um <- control$user.mods
if (length(um) > 0) {
for (i in 1:length(um)) {
dat <- do.call(control[[um[i]]], list(dat, at))
}
}
## Infection
if (!is.null(control[["infection.FUN"]])) {
dat <- do.call(control[["infection.FUN"]], list(dat, at))
}
## Recovery
if (!is.null(control[["recovery.FUN"]])) {
dat <- do.call(control[["recovery.FUN"]], list(dat, at))
}
## Departure Module
if (!is.null(control[["departures.FUN"]])) {
dat <- do.call(control[["departures.FUN"]], list(dat, at))
}
## Arrival Module
if (!is.null(control[["arrivals.FUN"]])) {
dat <- do.call(control[["arrivals.FUN"]], list(dat, at))
}
## Outputs
if (!is.null(control[["get_prev.FUN"]])) {
dat <- do.call(control[["get_prev.FUN"]], list(dat, at))
}
## Track progress
verbose.icm(dat, type = "progress", s, at)
}
# Set output
out <- saveout.seiqhrf.icm(dat, s=1)
class(out) <- "icm"
return(out)
}
# aggregate results collected from each thread
collected_times <- list()
# collect the times from sout then delete them
for (i in 1:length(sout)) {
collected_times[[paste0("sim", i)]] <- sout[[i]]$times$sim1
sout[[i]]$times <- NULL
}
# merge $epi structures
merged.out <- sout[[1]]
for (i in 2:length(sout)) {
merged.out <- merge.seiqhrf.icm(merged.out, sout[[i]], param.error = FALSE)
}
out <- merged.out
# add the collected timing data
out$times <- collected_times
class(out) <- "icm"
} # end of parallel execution
return(out)
}
Raw
_icm.mod.init.seiqhrf.R
initialize.icm <- function(param, init, control) {
## Master List for Data ##
dat <- list()
dat$param <- param
dat$init <- init
dat$control <- control
# Set attributes
dat$attr <- list()
numeric.init <- init[which(sapply(init, class) == "numeric")]
n <- do.call("sum", numeric.init)
dat$attr$active <- rep(1, n)
if (dat$param$groups == 1) {
dat$attr$group <- rep(1, n)
} else {
g2inits <- grep(".g2", names(numeric.init))
g1inits <- setdiff(1:length(numeric.init), g2inits)
nG1 <- sum(sapply(g1inits, function(x) init[[x]]))
nG2 <- sum(sapply(g2inits, function(x) init[[x]]))
dat$attr$group <- c(rep(1, nG1), rep(2, max(0, nG2)))
}
# Initialize status and infection time
dat <- init_status.icm(dat)
# Summary out list
dat <- get_prev.icm(dat, at = 1)
return(dat)
}
init_status.icm <- function(dat) {
# Variables ---------------------------------------------------------------
type <- dat$control$type
group <- dat$attr$group
nGroups <- dat$param$groups
nG1 <- sum(group == 1)
nG2 <- sum(group == 2)
e.num <- dat$init$e.num
i.num <- dat$init$i.num
q.num <- dat$init$q.num
h.num <- dat$init$h.num
r.num <- dat$init$r.num
f.num <- dat$init$f.num
e.num.g2 <- dat$init$e.num.g2
i.num.g2 <- dat$init$i.num.g2
q.num.g2 <- dat$init$q.num.g2
h.num.g2 <- dat$init$h.num.g2
r.num.g2 <- dat$init$r.num.g2
f.num.g2 <- dat$init$f.num.g2
# Status ------------------------------------------------------------------
status <- rep("s", nG1 + nG2)
status[sample(which(group == 1), size = i.num)] <- "i"
if (nGroups == 2) {
status[sample(which(group == 2), size = i.num.g2)] <- "i"
}
if (type %in% c("SIR", "SEIR", "SEIQHR", "SEIQHRF")) {
status[sample(which(group == 1 & status == "s"), size = r.num)] <- "r"
if (nGroups == 2) {
status[sample(which(group == 2 & status == "s"), size = r.num.g2)] <- "r"
}
}
if (type %in% c("SEIR", "SEIQHR", "SEIQHRF")) {
status[sample(which(group == 1 & status == "s"), size = e.num)] <- "e"
if (nGroups == 2) {
status[sample(which(group == 2 & status == "s"), size = e.num.g2)] <- "e"
}
}
if (type %in% c("SEIQHR", "SEIQHRF")) {
status[sample(which(group == 1 & status == "s"), size = q.num)] <- "q"
if (nGroups == 2) {
status[sample(which(group == 2 & status == "s"), size = q.num.g2)] <- "q"
}
status[sample(which(group == 1 & status == "s"), size = h.num)] <- "h"
if (nGroups == 2) {
status[sample(which(group == 2 & status == "s"), size = h.num.g2)] <- "h"
}
}
if (type %in% c("SEIQHRF")) {
status[sample(which(group == 1 & status == "s"), size = f.num)] <- "f"
if (nGroups == 2) {
status[sample(which(group == 2 & status == "s"), size = f.num.g2)] <- "f"
}
}
dat$attr$status <- status
# Exposure Time ----------------------------------------------------------
idsExp <- which(status == "e")
expTime <- rep(NA, length(status))
# leave exposure time uninitialised for now, and
# just set to NA at start.
dat$attr$expTime <- expTime # overwritten below
# Infection Time ----------------------------------------------------------
idsInf <- which(status == "i")
infTime <- rep(NA, length(status))
dat$attr$infTime <- infTime # overwritten below
# Recovery Time ----------------------------------------------------------
idsRecov <- which(status == "r")
recovTime <- rep(NA, length(status))
dat$attr$recovTime <- recovTime
# Need for Hospitalisation Time ----------------------------------------------------------
idsHosp <- which(status == "h")
hospTime <- rep(NA, length(status))
dat$attr$hospTime <- hospTime
# Quarantine Time ----------------------------------------------------------
idsQuar <- which(status == "q")
quarTime <- rep(NA, length(status))
dat$attr$quarTime <- quarTime
# Hospital-need cessation Time ----------------------------------------------------------
dischTime <- rep(NA, length(status))
dat$attr$dischTime <- dischTime
# Case-fatality Time ----------------------------------------------------------
fatTime <- rep(NA, length(status))
dat$attr$fatTime <- fatTime
# If vital=TRUE, infTime is a uniform draw over the duration of infection
# note the initial infections may have negative infTime!
if (FALSE) {
# not sure what the following section is trying to do, but it
# mucks up the gamma-distributed incumabtion periods, so set
# infTime for initial infected people to t=1 instead
if (dat$param$vital == TRUE && dat$param$di.rate > 0) {
infTime[idsInf] <- -rgeom(n = length(idsInf), prob = dat$param$di.rate) + 2
} else {
if (dat$control$type == "SI" || dat$param$rec.rate == 0) {
# infTime a uniform draw over the number of sim time steps
infTime[idsInf] <- ssample(1:(-dat$control$nsteps + 2),
length(idsInf), replace = TRUE)
} else {
if (nGroups == 1) {
infTime[idsInf] <- ssample(1:(-round(1 / dat$param$rec.rate) + 2),
length(idsInf), replace = TRUE)
}
if (nGroups == 2) {
infG1 <- which(status == "i" & group == 1)
infTime[infG1] <- ssample(1:(-round(1 / dat$param$rec.rate) + 2),
length(infG1), replace = TRUE)
infG2 <- which(status == "i" & group == 2)
infTime[infG2] <- ssample(1:(-round(1 / dat$param$rec.rate.g2) + 2),
length(infG2), replace = TRUE)
}
}
}
}
expTime[idsExp] <- 1
dat$attr$expTime <- expTime
infTime[idsInf] <- 1
dat$attr$infTime <- infTime
return(dat)
}
Raw
_icm.mod.status.seiqhrf.R
infection.seiqhrf.icm <- function(dat, at) {
type <- dat$control$type
# the following checks need to be moved to control.icm in due course
nsteps <- dat$control$nsteps
act.rate.i <- dat$param$act.rate.i
if (!(length(act.rate.i) == 1 || length(act.rate.i == nsteps))) {
stop("Length of act.rate.i must be 1 or the value of nsteps")
}
act.rate.i.g2 <- dat$param$act.rate.i.g2
if (!is.null(act.rate.i.g2) &&
!(length(act.rate.i.g2) == 1 || length(act.rate.i.g2 == nsteps))) {
stop("Length of act.rate.i.g2 must be 1 or the value of nsteps")
}
inf.prob.i <- dat$param$inf.prob.i
if (!(length(inf.prob.i) == 1 || length(inf.prob.i == nsteps))) {
stop("Length of inf.prob.i must be 1 or the value of nsteps")
}
inf.prob.i.g2 <- dat$param$inf.prob.i.g2
if (!is.null(inf.prob.i.g2) &&
!(length(inf.prob.i.g2) == 1 || length(inf.prob.i.g2 == nsteps))) {
stop("Length of inf.prob.i.g2 must be 1 or the value of nsteps")
}
if (type %in% c("SEIQHR", "SEIQHRF")) {
quar.rate <- dat$param$quar.rate
if (!(length(quar.rate) == 1 || length(quar.rate == nsteps))) {
stop("Length of quar.rate must be 1 or the value of nsteps")
}
quar.rate.g2 <- dat$param$quar.rate.g2
if (!is.null(quar.rate.g2) &&
!(length(quar.rate.g2) == 1 || length(quar.rate.g2 == nsteps))) {
stop("Length of quar.rate.g2 must be 1 or the value of nsteps")
}
disch.rate <- dat$param$disch.rate
if (!(length(disch.rate) == 1 || length(disch.rate == nsteps))) {
stop("Length of disch.rate must be 1 or the value of nsteps")
}
disch.rate.g2 <- dat$param$disch.rate.g2
if (!is.null(disch.rate.g2) &&
!(length(disch.rate.g2) == 1 || length(disch.rate.g2 == nsteps))) {
stop("Length of disch.rate.g2 must be 1 or the value of nsteps")
}
}
if (type %in% c("SEIQHR", "SEIQHRF")) {
act.rate.e <- dat$param$act.rate.e
if (!(length(act.rate.e) == 1 || length(act.rate.e == nsteps))) {
stop("Length of act.rate.e must be 1 or the value of nsteps")
}
act.rate.e.g2 <- dat$param$act.rate.e.g2
if (!is.null(act.rate.e.g2) &&
!(length(act.rate.e.g2) == 1 || length(act.rate.e.g2 == nsteps))) {
stop("Length of act.rate.e.g2 must be 1 or the value of nsteps")
}
inf.prob.e <- dat$param$inf.prob.e
if (!(length(inf.prob.e) == 1 || length(inf.prob.e == nsteps))) {
stop("Length of inf.prob.e must be 1 or the value of nsteps")
}
inf.prob.e.g2 <- dat$param$inf.prob.e.g2
if (!is.null(inf.prob.e.g2) &&
!(length(inf.prob.e.g2) == 1 || length(inf.prob.e.g2 == nsteps))) {
stop("Length of inf.prob.e.g2 must be 1 or the value of nsteps")
}
act.rate.q <- dat$param$act.rate.q
if (!(length(act.rate.q) == 1 || length(act.rate.q == nsteps))) {
stop("Length of act.rate.q must be 1 or the value of nsteps")
}
act.rate.q.g2 <- dat$param$act.rate.q.g2
if (!is.null(act.rate.q.g2) &&
!(length(act.rate.q.g2) == 1 || length(act.rate.q.g2 == nsteps))) {
stop("Length of act.rate.q.g2 must be 1 or the value of nsteps")
}
inf.prob.q <- dat$param$inf.prob.q
if (!(length(inf.prob.q) == 1 || length(inf.prob.q == nsteps))) {
stop("Length of inf.prob.q must be 1 or the value of nsteps")
}
inf.prob.q.g2 <- dat$param$inf.prob.q.g2
if (!is.null(inf.prob.q.g2) &&
!(length(inf.prob.q.g2) == 1 || length(inf.prob.q.g2 == nsteps))) {
stop("Length of inf.prob.q.g2 must be 1 or the value of nsteps")
}
}
# Transmission from infected
## Expected acts
if (dat$param$groups == 1) {
if (length(act.rate.i) > 1) {
acts <- round(act.rate.i[at - 1] * dat$epi$num[at - 1] / 2)
} else {
acts <- round(act.rate.i * dat$epi$num[at - 1] / 2)
}
}
if (dat$param$groups == 2) {
if (dat$param$balance == "g1") {
if (length(act.rate.i) > 1) {
acts <- round(act.rate.i[at - 1] *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
} else {
acts <- round(act.rate.i *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
}
}
if (dat$param$balance == "g2") {
if (length(act.rate.i.g2) > 1) {
acts <- round(act.rate.i.g2[at - 1] *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
} else {
acts <- round(act.rate.i.g2 *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
}
}
}
## Edgelist
if (dat$param$groups == 1) {
p1 <- ssample(which(dat$attr$active == 1 & dat$attr$status != "f"), acts, replace = TRUE)
p2 <- ssample(which(dat$attr$active == 1 & dat$attr$status != "f"), acts, replace = TRUE)
} else {
p1 <- ssample(which(dat$attr$active == 1 & dat$attr$group == 1 & dat$attr$status != "f"),
acts, replace = TRUE)
p2 <- ssample(which(dat$attr$active == 1 & dat$attr$group == 2 & dat$attr$status != "f"),
acts, replace = TRUE)
}
del <- NULL
if (length(p1) > 0 & length(p2) > 0) {
del <- data.frame(p1, p2)
if (dat$param$groups == 1) {
while (any(del$p1 == del$p2)) {
del$p2 <- ifelse(del$p1 == del$p2,
ssample(which(dat$attr$active == 1 & dat$attr$status != "f"), 1), del$p2)
}
}
}
## Discordant edgelist (del)
del$p1.stat <- dat$attr$status[del$p1]
del$p2.stat <- dat$attr$status[del$p2]
serodis <- (del$p1.stat == "s" & del$p2.stat == "i") |
(del$p1.stat == "i" & del$p2.stat == "s")
del <- del[serodis == TRUE, ]
## Transmission on edgelist
if (nrow(del) > 0) {
if (dat$param$groups == 1) {
if (length(inf.prob.i) > 1) {
del$tprob <- inf.prob.i[at]
} else {
del$tprob <- inf.prob.i
}
} else {
if (length(inf.prob.i) > 1) {
del$tprob <- ifelse(del$p1.stat == "s", inf.prob.i[at],
inf.prob.i.g2[at])
} else {
del$tprob <- ifelse(del$p1.stat == "s", inf.prob.i,
inf.prob.i.g2)
}
}
if (!is.null(dat$param$inter.eff.i) && at >= dat$param$inter.start.i &&
at <= dat$param$inter.stop.i) {
del$tprob <- del$tprob * (1 - dat$param$inter.eff.i)
}
del$trans <- rbinom(nrow(del), 1, del$tprob)
del <- del[del$trans == TRUE, ]
if (nrow(del) > 0) {
if (dat$param$groups == 1) {
newIds <- unique(ifelse(del$p1.stat == "s", del$p1, del$p2))
nExp.i <- length(newIds)
}
if (dat$param$groups == 2) {
newIdsg1 <- unique(del$p1[del$p1.stat == "s"])
newIdsg2 <- unique(del$p2[del$p2.stat == "s"])
nExp.i <- length(newIdsg1)
nExpg2.i <- length(newIdsg2)
newIds <- c(newIdsg1, newIdsg2)
}
dat$attr$status[newIds] <- "e"
dat$attr$expTime[newIds] <- at
} else {
nExp.i <- nExpg2.i <- 0
}
} else {
nExp.i <- nExpg2.i <- 0
}
if (type %in% c("SEIQHRF")) {
# Transmission from exposed
## Expected acts
if (dat$param$groups == 1) {
if (length(act.rate.e) > 1) {
acts <- round(act.rate.e[at - 1] * dat$epi$num[at - 1] / 2)
} else {
acts <- round(act.rate.e * dat$epi$num[at - 1] / 2)
}
}
if (dat$param$groups == 2) {
if (dat$param$balance == "g1") {
if (length(act.rate.e.g2) > 1) {
acts <- round(act.rate.e.g2[at - 1] *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
} else {
acts <- round(act.rate.e.g2 *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
}
}
if (dat$param$balance == "g2") {
if (length(act.rate.e.g2) > 1) {
acts <- round(act.rate.e.g2[at - 1] *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
} else {
acts <- round(act.rate.e.g2 *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
}
}
}
## Edgelist
if (dat$param$groups == 1) {
p1 <- ssample(which(dat$attr$active == 1 & dat$attr$status != "f"), acts, replace = TRUE)
p2 <- ssample(which(dat$attr$active == 1 & dat$attr$status != "f"), acts, replace = TRUE)
} else {
p1 <- ssample(which(dat$attr$active == 1 & dat$attr$group == 1 & dat$attr$status != "f"),
acts, replace = TRUE)
p2 <- ssample(which(dat$attr$active == 1 & dat$attr$group == 2 & dat$attr$status != "f"),
acts, replace = TRUE)
}
del <- NULL
if (length(p1) > 0 & length(p2) > 0) {
del <- data.frame(p1, p2)
if (dat$param$groups == 1) {
while (any(del$p1 == del$p2)) {
del$p2 <- ifelse(del$p1 == del$p2,
ssample(which(dat$attr$active == 1 & dat$attr$status != "f"), 1), del$p2)
}
}
## Discordant edgelist (del)
del$p1.stat <- dat$attr$status[del$p1]
del$p2.stat <- dat$attr$status[del$p2]
# serodiscordance
serodis <- (del$p1.stat == "s" & del$p2.stat == "e") |
(del$p1.stat == "e" & del$p2.stat == "s")
del <- del[serodis == TRUE, ]
## Transmission on edgelist
if (nrow(del) > 0) {
if (dat$param$groups == 1) {
if (length(inf.prob.e) > 1) {
del$tprob <- inf.prob.e[at]
} else {
del$tprob <- inf.prob.e
}
} else {
if (length(inf.prob.e) > 1) {
del$tprob <- ifelse(del$p1.stat == "s", inf.prob.e[at],
inf.prob.e.g2[at])
} else {
del$tprob <- ifelse(del$p1.stat == "s", inf.prob.e,
inf.prob.e.g2)
}
}
if (!is.null(dat$param$inter.eff.e) && at >= dat$param$inter.start.e &&
at <= dat$param$inter.stop.e) {
del$tprob <- del$tprob * (1 - dat$param$inter.eff.e)
}
del$trans <- rbinom(nrow(del), 1, del$tprob)
del <- del[del$trans == TRUE, ]
if (nrow(del) > 0) {
if (dat$param$groups == 1) {
newIds <- unique(ifelse(del$p1.stat == "s", del$p1, del$p2))
nExp.e <- length(newIds)
}
if (dat$param$groups == 2) {
newIdsg1 <- unique(del$p1[del$p1.stat == "s"])
newIdsg2 <- unique(del$p2[del$p2.stat == "s"])
nExp.e <- length(newIdsg1)
nExpg2.e <- length(newIdsg2)
newIds <- c(newIdsg1, newIdsg2)
}
dat$attr$status[newIds] <- "e"
dat$attr$expTime[newIds] <- at
} else {
nExp.e <- nExpg2.e <- 0
}
} else {
nExp.e <- nExpg2.e <- 0
}
} else {
nExp.e <- nExpg2.e <- 0
}
# Transmission from quarantined
## Expected acts
if (dat$param$groups == 1) {
if (length(act.rate.q) > 1) {
acts <- round(act.rate.q[at - 1] * dat$epi$num[at - 1] / 2)
} else {
acts <- round(act.rate.q * dat$epi$num[at - 1] / 2)
}
}
if (dat$param$groups == 2) {
if (dat$param$balance == "g1") {
if (length(act.rate.q.g2) > 1) {
acts <- round(act.rate.q.g2[at - 1] *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
} else {
acts <- round(act.rate.q.g2 *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
}
}
if (dat$param$balance == "g2") {
if (length(act.rate.q.g2) > 1) {
acts <- round(act.rate.q.g2[at - 1] *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
} else {
acts <- round(act.rate.q.g2 *
(dat$epi$num[at - 1] + dat$epi$num.g2[at - 1]) / 2)
}
}
}
## Edgelist
if (dat$param$groups == 1) {
p1 <- ssample(which(dat$attr$active == 1 & dat$attr$status != "f"), acts, replace = TRUE)
p2 <- ssample(which(dat$attr$active == 1 & dat$attr$status != "f"), acts, replace = TRUE)
} else {
p1 <- ssample(which(dat$attr$active == 1 & dat$attr$group == 1 & dat$attr$status != "f"),
acts, replace = TRUE)
p2 <- ssample(which(dat$attr$active == 1 & dat$attr$group == 2 & dat$attr$status != "f"),
acts, replace = TRUE)
}
del <- NULL
if (length(p1) > 0 & length(p2) > 0) {
del <- data.frame(p1, p2)
if (dat$param$groups == 1) {
while (any(del$p1 == del$p2)) {
del$p2 <- ifelse(del$p1 == del$p2,
ssample(which(dat$attr$active == 1 & dat$attr$status != "f"), 1), del$p2)
}
}
## Discordant edgelist (del)
del$p1.stat <- dat$attr$status[del$p1]
del$p2.stat <- dat$attr$status[del$p2]
# serodiscordance
serodis <- (del$p1.stat == "s" & del$p2.stat == "q") |
(del$p1.stat == "q" & del$p2.stat == "s")
del <- del[serodis == TRUE, ]
## Transmission on edgelist
if (nrow(del) > 0) {
if (dat$param$groups == 1) {
if (length(inf.prob.q) > 1) {
del$tprob <- inf.prob.q[at]
} else {
del$tprob <- inf.prob.q
}
} else {
if (length(inf.prob.q) > 1) {
del$tprob <- ifelse(del$p1.stat == "s", inf.prob.q[at],
inf.prob.q.g2[at])
} else {
del$tprob <- ifelse(del$p1.stat == "s", inf.prob.q,
inf.prob.q.g2)
}
}
if (!is.null(dat$param$inter.eff.q) && at >= dat$param$inter.start.q &&
at <= dat$param$inter.stop.q) {
del$tprob <- del$tprob * (1 - dat$param$inter.eff.q)
}
del$trans <- rbinom(nrow(del), 1, del$tprob)
del <- del[del$trans == TRUE, ]
if (nrow(del) > 0) {
if (dat$param$groups == 1) {
newIds <- unique(ifelse(del$p1.stat == "s", del$p1, del$p2))
nExp.q <- length(newIds)
}
if (dat$param$groups == 2) {
newIdsg1 <- unique(del$p1[del$p1.stat == "s"])
newIdsg2 <- unique(del$p2[del$p2.stat == "s"])
nExp.q <- length(newIdsg1)
nExpg2.q <- length(newIdsg2)
newIds <- c(newIdsg1, newIdsg2)
}
dat$attr$status[newIds] <- "e"
dat$attr$expTime[newIds] <- at
} else {
nExp.q <- nExpg2.q <- 0
}
} else {
nExp.q <- nExpg2.q <- 0
}
} else {
nExp.q <- nExpg2.q <- 0
}
}
## Output
if (type %in% c("SEIQHR", "SEIQHRF")) {
if (at == 2) {
dat$epi$se.flow <- c(0, nExp.i + nExp.q)
} else {
dat$epi$se.flow[at] <- nExp.i + nExp.q
}
if (dat$param$groups == 2) {
if (at == 2) {
dat$epi$se.flow.g2 <- c(0, nExpg2.i + nExpg2.q )
} else {
dat$epi$se.flow.g2[at] <- nExpg2.i + nExpg2.q
}
}
} else {
if (at == 2) {
dat$epi$se.flow <- c(0, nExp.i)
} else {
dat$epi$se.flow[at] <- nExp.i
}
if (dat$param$groups == 2) {
if (at == 2) {
dat$epi$se.flow.g2 <- c(0, nExpg2.i)
} else {
dat$epi$se.flow.g2[at] <- nExpg2.i
}
}
}
return(dat)
}
# utility functions
cum_discr_si <- function(vecTimeSinceExp, scale, shape) {
vlen <- length(vecTimeSinceExp)
if (vlen > 0) {
probVec <- numeric(vlen)
for (p in 1:vlen) {
probVec[p] <- pweibull(vecTimeSinceExp[p], shape=shape, scale=scale)
}
} else {
probVec <- 0
}
return(probVec)
}
progress.seiqhrf.icm <- function(dat, at) {
#print(at)
#print(dat$control$type)
#print("-------")
# Conditions --------------------------------------------------------------
if (!(dat$control$type %in% c("SIR", "SIS", "SEIR", "SEIQHR", "SEIQHRF"))) {
return(dat)
}
# Variables ---------------------------------------------------------------
active <- dat$attr$active
status <- dat$attr$status
groups <- dat$param$groups
group <- dat$attr$group
type <- dat$control$type
recovState <- ifelse(type %in% c("SIR", "SEIR", "SEIQHR", "SEIQHRF"), "r", "s")
progState <- "i"
quarState <- "q"
hospState <- "h"
fatState <- "f"
# --- progress from exposed to infectious ----
prog.rand <- dat$control$prog.rand
prog.rate <- dat$param$prog.rate
prog.rate.g2 <- dat$param$prog.rate.g2
prog.dist.scale <- dat$param$prog.dist.scale
prog.dist.shape <- dat$param$prog.dist.shape
prog.dist.scale.g2 <- dat$param$prog.dist.scale.g2
prog.dist.shape.g2 <- dat$param$prog.dist.shape.g2
nProg <- nProgG2 <- 0
idsElig <- which(active == 1 & status == "e")
nElig <- length(idsElig)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(prog.rate, prog.rate.g2)
ratesElig <- rates[gElig]
if (prog.rand == TRUE) {
vecProg <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecProg) > 0) {
idsProg <- idsElig[vecProg]
nProg <- sum(group[idsProg] == 1)
nProgG2 <- sum(group[idsProg] == 2)
status[idsProg] <- progState
dat$attr$infTime[idsProg] <- at
}
} else {
vecTimeSinceExp <- at - dat$attr$expTime[idsElig]
gammaRatesElig <- pweibull(vecTimeSinceExp, prog.dist.shape, scale=prog.dist.scale)
nProg <- round(sum(gammaRatesElig[gElig == 1], na.rm=TRUE))
if (nProg > 0) {
ids2bProg <- ssample(idsElig[gElig == 1],
nProg, prob = gammaRatesElig[gElig == 1])
status[ids2bProg] <- progState
dat$attr$infTime[ids2bProg] <- at
# debug
if (FALSE & at <= 30) {
print(paste("at:", at))
print("idsElig:")
print(idsElig[gElig == 1])
print("vecTimeSinceExp:")
print(vecTimeSinceExp[gElig == 1])
print("gammaRatesElig:")
print(gammaRatesElig)
print(paste("nProg:",nProg))
print(paste("sum of elig rates:", round(sum(gammaRatesElig[gElig == 1]))))
print(paste("sum(gElig == 1):", sum(gElig == 1)))
print("ids progressed:")
print(ids2bProg)
print("probs of ids to be progressed:")
print(gammaRatesElig[which(idsElig %in% ids2bProg)])
print("days since exposed of ids to be progressed:")
print(vecTimeSinceExp[which(idsElig %in% ids2bProg)])
print("------")
}
}
if (groups == 2) {
nProgG2 <- round(sum(gammaRatesElig[gElig == 2], na.rm=TRUE))
if (nProgG2 > 0) {
ids2bProgG2 <- ssample(idsElig[gElig == 2],
nProgG2, prob = gammaRatesElig[gElig == 2])
status[ids2bProgG2] <- progState
dat$attr$infTime[ids2bProgG2] <- at
}
}
}
}
dat$attr$status <- status
if (type %in% c("SEIQHR", "SEIQHRF")) {
# ----- quarantine -------
quar.rand <- dat$control$quar.rand
quar.rate <- dat$param$quar.rate
quar.rate.g2 <- dat$param$quar.rate.g2
nQuar <- nQuarG2 <- 0
idsElig <- which(active == 1 & status == "i")
nElig <- length(idsElig)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(quar.rate, quar.rate.g2)
if (length(quar.rate) > 1) {
qrate <- quar.rate[at]
} else {
qrate <- quar.rate
}
if (length(quar.rate.g2) > 1) {
qrate.g2 <- quar.rate.g2[at]
} else {
qrate.g2 <- quar.rate.g2
}
rates <- c(qrate, qrate.g2)
ratesElig <- rates[gElig]
if (quar.rand == TRUE) {
vecQuar <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecQuar) > 0) {
idsQuar <- idsElig[vecQuar]
nQuar <- sum(group[idsQuar] == 1)
nQuarG2 <- sum(group[idsQuar] == 2)
status[idsQuar] <- quarState
dat$attr$quarTime[idsQuar] <- at
}
} else {
nQuar <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
idsQuar <- ssample(idsElig[gElig == 1], nQuar)
status[idsQuar] <- quarState
dat$attr$quarTime[idsQuar] <- at
if (groups == 2) {
nQuarG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
idsQuarG2 <- ssample(idsElig[gElig == 2], nQuarG2)
status[idsQuarG2] <- quarState
dat$attr$quarTime[idsQuarG2] <- at
}
}
}
dat$attr$status <- status
# ----- need to be hospitalised -------
hosp.rand <- dat$control$hosp.rand
hosp.rate <- dat$param$hosp.rate
hosp.rate.g2 <- dat$param$hosp.rate.g2
nHosp <- nHospG2 <- 0
idsElig <- which(active == 1 & (status == "i" | status == "q"))
nElig <- length(idsElig)
idsHosp <- numeric(0)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(hosp.rate, hosp.rate.g2)
ratesElig <- rates[gElig]
if (hosp.rand == TRUE) {
vecHosp <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecHosp) > 0) {
idsHosp <- idsElig[vecHosp]
nHosp <- sum(group[idsHosp] == 1)
nHospG2 <- sum(group[idsHosp] == 2)
status[idsHosp] <- hospState
}
} else {
nHosp <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
idsHosp <- ssample(idsElig[gElig == 1], nHosp)
status[idsHosp] <- hospState
if (groups == 2) {
nHospG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
idsHospG2 <- ssample(idsElig[gElig == 2], nHospG2)
status[idsHospG2] <- hospState
idsHosp <- c(idsHosp, idsHospG2)
}
}
}
dat$attr$status <- status
dat$attr$hospTime[idsHosp] <- at
# ----- discharge from need to be hospitalised -------
disch.rand <- dat$control$disch.rand
disch.rate <- dat$param$disch.rate
disch.rate.g2 <- dat$param$disch.rate.g2
nDisch <- nDischG2 <- 0
idsElig <- which(active == 1 & status == "h")
nElig <- length(idsElig)
idsDisch <- numeric(0)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(disch.rate, disch.rate.g2)
if (length(disch.rate) > 1) {
dcrate <- disch.rate[at]
} else {
dcrate <- disch.rate
}
if (length(disch.rate.g2) > 1) {
dcrate.g2 <- disch.rate.g2[at]
} else {
dcrate.g2 <- disch.rate.g2
}
rates <- c(dcrate, dcrate.g2)
ratesElig <- rates[gElig]
if (disch.rand == TRUE) {
vecDisch <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecDisch) > 0) {
idsDisch <- idsElig[vecDisch]
nDisch <- sum(group[idsDisch] == 1)
nDischG2 <- sum(group[idsDisch] == 2)
status[idsDisch] <- recovState
}
} else {
nDisch <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
idsDisch <- ssample(idsElig[gElig == 1], nDisch)
status[idsDisch] <- recovState
if (groups == 2) {
nDischG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
idsDischG2 <- ssample(idsElig[gElig == 2], nDischG2)
status[idsDischG2] <- recovState
idsDisch <- c(idsDisch, idsDischG2)
}
}
}
dat$attr$status <- status
dat$attr$dischTime[idsDisch] <- at
}
# ----- recover -------
rec.rand <- dat$control$rec.rand
rec.rate <- dat$param$rec.rate
rec.rate.g2 <- dat$param$rec.rate.g2
rec.dist.scale <- dat$param$rec.dist.scale
rec.dist.shape <- dat$param$rec.dist.shape
rec.dist.scale.g2 <- dat$param$rec.dist.scale.g2
rec.dist.shape.g2 <- dat$param$rec.dist.shape.g2
nRecov <- nRecovG2 <- 0
idsElig <- which(active == 1 & (status == "i" | status == "q" | status == "h"))
nElig <- length(idsElig)
idsRecov <- numeric(0)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(rec.rate, rec.rate.g2)
ratesElig <- rates[gElig]
if (rec.rand == TRUE) {
vecRecov <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecRecov) > 0) {
idsRecov <- idsElig[vecRecov]
nRecov <- sum(group[idsRecov] == 1)
nRecovG2 <- sum(group[idsRecov] == 2)
status[idsRecov] <- recovState
}
} else {
vecTimeSinceExp <- at - dat$attr$expTime[idsElig]
vecTimeSinceExp[is.na(vecTimeSinceExp)] <- 0
gammaRatesElig <- pweibull(vecTimeSinceExp, rec.dist.shape, scale=rec.dist.scale)
nRecov <- round(sum(gammaRatesElig[gElig == 1], na.rm=TRUE))
if (nRecov > 0) {
idsRecov <- ssample(idsElig[gElig == 1],
nRecov, prob = gammaRatesElig[gElig == 1])
status[idsRecov] <- recovState
# debug
if (FALSE & at <= 30) {
print(paste("at:", at))
print("idsElig:")
print(idsElig[gElig == 1])
print("vecTimeSinceExp:")
print(vecTimeSinceExp[gElig == 1])
print("gammaRatesElig:")
print(gammaRatesElig)
print(paste("nRecov:",nRecov))
print(paste("sum of elig rates:", round(sum(gammaRatesElig[gElig == 1]))))
print(paste("sum(gElig == 1):", sum(gElig == 1)))
print("ids recovered:")
print(idsRecov)
print("probs of ids to be progressed:")
print(gammaRatesElig[which(idsElig %in% idsRecov)])
print("days since exposed of ids to be Recovered:")
print(vecTimeSinceExp[which(idsElig %in% idsRecov)])
print("------")
}
}
if (groups == 2) {
nRecovG2 <- round(sum(gammaRatesElig[gElig == 2], na.rm=TRUE))
if (nRecovG2 > 0) {
idsRecovG2 <- ssample(idsElig[gElig == 2],
nRecovG2, prob = gammaRatesElig[gElig == 2])
status[idsRecovG2] <- recovState
idsRecov <- c(idsRecov, idsRecovG2)
}
}
}
}
dat$attr$status <- status
dat$attr$recovTime[idsRecov] <- at
fatEnable <- TRUE
if (fatEnable & type %in% c("SEIQHRF")) {
# ----- case fatality -------
fat.rand <- dat$control$fat.rand
fat.rate.base <- dat$param$fat.rate.base
fat.rate.base.g2 <- dat$param$fat.rate.base.g2
fat.rate.base.g2 <- ifelse(is.null(fat.rate.base.g2),
0, fat.rate.base.g2)
fat.rate.overcap <- dat$param$fat.rate.overcap
fat.rate.overcap.g2 <- dat$param$fat.rate.overcap.g2
fat.rate.overcap.g2 <- ifelse(is.null(fat.rate.overcap.g2),
0, fat.rate.overcap.g2)
hosp.cap <- dat$param$hosp.cap
fat.tcoeff <- dat$param$fat.tcoeff
nFat <- nFatG2 <- 0
idsElig <- which(active == 1 & status =="h")
nElig <- length(idsElig)
if (nElig > 0) {
gElig <- group[idsElig]
timeInHospElig <- at - dat$attr$hospTime[idsElig]
rates <- c(fat.rate.base, fat.rate.base.g2)
h.num.yesterday <- 0
if (!is.null(dat$epi$h.num[at - 1])) {
h.num.yesterday <- dat$epi$h.num[at - 1]
if (h.num.yesterday > hosp.cap) {
blended.rate <- ((hosp.cap * fat.rate.base) +
((h.num.yesterday - hosp.cap) * fat.rate.overcap)) /
h.num.yesterday
blended.rate.g2 <- ((hosp.cap * fat.rate.base.g2) +
((h.num.yesterday - hosp.cap) * fat.rate.overcap.g2)) /
h.num.yesterday
rates <- c(blended.rate, blended.rate.g2)
}
}
ratesElig <- rates[gElig]
ratesElig <- ratesElig + timeInHospElig*fat.tcoeff*ratesElig
if (fat.rand == TRUE) {
vecFat <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecFat) > 0) {
idsFat <- idsElig[vecFat]
nFat <- sum(group[idsFat] == 1)
nFatG2 <- sum(group[idsFat] == 2)
status[idsFat] <- fatState
dat$attr$fatTime[idsFat] <- at
}
} else {
nFat <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
idsFat <- ssample(idsElig[gElig == 1], nFat)
status[idsFat] <- fatState
dat$attr$fatTime[idsFat] <- at
if (groups == 2) {
nFatG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
idsFatG2 <- ssample(idsElig[gElig == 2], nFatG2)
status[idsFatG2] <- fatState
dat$attr$fatTime[idsFatG2] <- at
}
}
}
dat$attr$status <- status
}
# Output ------------------------------------------------------------------
outName_a <- ifelse(type %in% c("SIR", "SEIR"), "ir.flow", "is.flow")
outName_a[2] <- paste0(outName_a, ".g2")
if (type %in% c("SEIR", "SEIQHR", "SEIQHRF")) {
outName_b <- "ei.flow"
outName_b[2] <- paste0(outName_b, ".g2")
}
if (type %in% c("SEIQHR", "SEIQHRF")) {
outName_c <- "iq.flow"
outName_c[2] <- paste0(outName_c, ".g2")
outName_d <- "iq2h.flow"
outName_d[2] <- paste0(outName_d, ".g2")
}
if (type %in% c("SEIQHRF")) {
outName_e <- "hf.flow"
outName_e[2] <- paste0(outName_e, ".g2")
}
## Summary statistics
if (at == 2) {
dat$epi[[outName_a[1]]] <- c(0, nRecov)
if (type %in% c("SEIR", "SEIQHR")) {
dat$epi[[outName_b[1]]] <- c(0, nProg)
}
if (type %in% c("SEIQHR", "SEIQHRF")) {
dat$epi[[outName_c[1]]] <- c(0, nQuar)
dat$epi[[outName_d[1]]] <- c(0, nHosp)
}
if (fatEnable & type %in% c("SEIQHRF")) {
dat$epi[[outName_e[1]]] <- c(0, nFat)
}
} else {
dat$epi[[outName_a[1]]][at] <- nRecov
if (type %in% c("SEIR", "SEIQHR")) {
dat$epi[[outName_b[1]]][at] <- nProg
}
if (type %in% c("SEIQHR", "SEIQHRF")) {
dat$epi[[outName_c[1]]][at] <- nQuar
dat$epi[[outName_d[1]]][at] <- nHosp
}
if (fatEnable & type %in% c("SEIQHRF")) {
dat$epi[[outName_e[1]]][at] <- nFat
}
}
if (groups == 2) {
if (at == 2) {
dat$epi[[outName_a[2]]] <- c(0, nRecovG2)
if (type %in% c("SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi[[outName_b[2]]] <- c(0, nProgG2)
}
if (type %in% c("SEIQHR", "SEIQHRF")) {
dat$epi[[outName_c[2]]] <- c(0, nQuarG2)
dat$epi[[outName_d[2]]] <- c(0, nHospG2)
}
if (type %in% c("SEIQHRF")) {
dat$epi[[outName_e[2]]] <- c(0, nFatG2)
}
} else {
dat$epi[[outName_a[2]]][at] <- nRecovG2
if (type %in% c("SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi[[outName_b[2]]][at] <- nProgG2
}
if (type %in% c("SEIQHR", "SEIQHRF")) {
dat$epi[[outName_c[2]]][at] <- nQuarG2
dat$epi[[outName_d[2]]][at] <- nHospG2
}
if (type %in% c("SEIQHRF")) {
dat$epi[[outName_e[2]]][at] <- nFatG2
}
}
}
return(dat)
}
###############
Raw
_icm.mod.vital.seiqhrf.R
departures.seiqhrf.icm <- function(dat, at) {
# Conditions --------------------------------------------------------------
if (dat$param$vital == FALSE) {
return(dat)
}
# Variables ---------------------------------------------------------------
groups <- dat$param$groups
group <- dat$attr$group
# Susceptible departures ------------------------------------------------------
nDepartures <- nDeparturesG2 <- 0
idsElig <- which(dat$attr$active == 1 & dat$attr$status == "s")
nElig <- length(idsElig)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(dat$param$ds.rate, dat$param$ds.rate.g2)
ratesElig <- rates[gElig]
if (dat$control$d.rand == TRUE) {
vecDepartures <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecDepartures) > 0) {
idsDpt <- idsElig[vecDepartures]
nDepartures <- sum(group[idsDpt] == 1)
nDeparturesG2 <- sum(group[idsDpt] == 2)
dat$attr$active[idsDpt] <- 0
}
} else {
nDepartures <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
dat$attr$active[ssample(idsElig[gElig == 1], nDepartures)] <- 0
if (groups == 2) {
nDeparturesG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
dat$attr$active[ssample(idsElig[gElig == 2], nDeparturesG2)] <- 0
}
}
}
if (at == 2) {
dat$epi$ds.flow <- c(0, nDepartures)
if (groups == 2) {
dat$epi$ds.flow.g2 <- c(0, nDeparturesG2)
}
} else {
dat$epi$ds.flow[at] <- nDepartures
if (groups == 2) {
dat$epi$ds.flow.g2[at] <- nDeparturesG2
}
}
# Exposed Departures ---------------------------------------------------------
nDepartures <- nDeparturesG2 <- 0
idsElig <- which(dat$attr$active == 1 & dat$attr$status == "e")
nElig <- length(idsElig)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(dat$param$de.rate, dat$param$de.rate.g2)
ratesElig <- rates[gElig]
if (dat$control$d.rand == TRUE) {
vecDepartures <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecDepartures) > 0) {
idsDpt <- idsElig[vecDepartures]
nDepartures <- sum(group[idsDpt] == 1)
nDeparturesG2 <- sum(group[idsDpt] == 2)
dat$attr$active[idsDpt] <- 0
}
} else {
nDepartures <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
dat$attr$active[ssample(idsElig[gElig == 1], nDepartures)] <- 0
if (groups == 2) {
nDeparturesG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
dat$attr$active[ssample(idsElig[gElig == 2], nDeparturesG2)] <- 0
}
}
}
if (at == 2) {
dat$epi$de.flow <- c(0, nDepartures)
if (groups == 2) {
dat$epi$de.flow.g2 <- c(0, nDeparturesG2)
}
} else {
dat$epi$de.flow[at] <- nDepartures
if (groups == 2) {
dat$epi$de.flow.g2[at] <- nDeparturesG2
}
}
# Infected Departures ---------------------------------------------------------
nDepartures <- nDeparturesG2 <- 0
idsElig <- which(dat$attr$active == 1 & dat$attr$status == "i")
nElig <- length(idsElig)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(dat$param$di.rate, dat$param$di.rate.g2)
ratesElig <- rates[gElig]
if (dat$control$d.rand == TRUE) {
vecDepartures <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecDepartures) > 0) {
idsDpt <- idsElig[vecDepartures]
nDepartures <- sum(group[idsDpt] == 1)
nDeparturesG2 <- sum(group[idsDpt] == 2)
dat$attr$active[idsDpt] <- 0
}
} else {
nDepartures <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
dat$attr$active[ssample(idsElig[gElig == 1], nDepartures)] <- 0
if (groups == 2) {
nDeparturesG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
dat$attr$active[ssample(idsElig[gElig == 2], nDeparturesG2)] <- 0
}
}
}
if (at == 2) {
dat$epi$di.flow <- c(0, nDepartures)
if (groups == 2) {
dat$epi$di.flow.g2 <- c(0, nDeparturesG2)
}
} else {
dat$epi$di.flow[at] <- nDepartures
if (groups == 2) {
dat$epi$di.flow.g2[at] <- nDeparturesG2
}
}
# Quarantined Departures ---------------------------------------------------------
nDepartures <- nDeparturesG2 <- 0
idsElig <- which(dat$attr$active == 1 & dat$attr$status == "q")
nElig <- length(idsElig)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(dat$param$dq.rate, dat$param$dq.rate.g2)
ratesElig <- rates[gElig]
if (dat$control$d.rand == TRUE) {
vecDepartures <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecDepartures) > 0) {
idsDpt <- idsElig[vecDepartures]
nDepartures <- sum(group[idsDpt] == 1)
nDeparturesG2 <- sum(group[idsDpt] == 2)
dat$attr$active[idsDpt] <- 0
}
} else {
nDepartures <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
dat$attr$active[ssample(idsElig[gElig == 1], nDepartures)] <- 0
if (groups == 2) {
nDeparturesG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
dat$attr$active[ssample(idsElig[gElig == 2], nDeparturesG2)] <- 0
}
}
}
if (at == 2) {
dat$epi$dq.flow <- c(0, nDepartures)
if (groups == 2) {
dat$epi$dq.flow.g2 <- c(0, nDeparturesG2)
}
} else {
dat$epi$dq.flow[at] <- nDepartures
if (groups == 2) {
dat$epi$dq.flow.g2[at] <- nDeparturesG2
}
}
# Hospitalised Departures ---------------------------------------------------------
nDepartures <- nDeparturesG2 <- 0
idsElig <- which(dat$attr$active == 1 & dat$attr$status == "h")
nElig <- length(idsElig)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(dat$param$dh.rate, dat$param$dh.rate.g2)
ratesElig <- rates[gElig]
if (dat$control$d.rand == TRUE) {
vecDepartures <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecDepartures) > 0) {
idsDpt <- idsElig[vecDepartures]
nDepartures <- sum(group[idsDpt] == 1)
nDeparturesG2 <- sum(group[idsDpt] == 2)
dat$attr$active[idsDpt] <- 0
}
} else {
nDepartures <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
dat$attr$active[ssample(idsElig[gElig == 1], nDepartures)] <- 0
if (groups == 2) {
nDeparturesG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
dat$attr$active[ssample(idsElig[gElig == 2], nDeparturesG2)] <- 0
}
}
}
if (at == 2) {
dat$epi$dh.flow <- c(0, nDepartures)
if (groups == 2) {
dat$epi$dh.flow.g2 <- c(0, nDeparturesG2)
}
} else {
dat$epi$dh.flow[at] <- nDepartures
if (groups == 2) {
dat$epi$dh.flow.g2[at] <- nDeparturesG2
}
}
# Recovered Departures --------------------------------------------------------
nDepartures <- nDeparturesG2 <- 0
idsElig <- which(dat$attr$active == 1 & dat$attr$status == "r")
nElig <- length(idsElig)
if (nElig > 0) {
gElig <- group[idsElig]
rates <- c(dat$param$dr.rate, dat$param$dr.rate.g2)
ratesElig <- rates[gElig]
if (dat$control$d.rand == TRUE) {
vecDepartures <- which(rbinom(nElig, 1, ratesElig) == 1)
if (length(vecDepartures) > 0) {
idsDpt <- idsElig[vecDepartures]
nDepartures <- sum(group[idsDpt] == 1)
nDeparturesG2 <- sum(group[idsDpt] == 2)
dat$attr$active[idsDpt] <- 0
}
} else {
nDepartures <- min(round(sum(ratesElig[gElig == 1])), sum(gElig == 1))
dat$attr$active[ssample(idsElig[gElig == 1], nDepartures)] <- 0
if (groups == 2) {
nDeparturesG2 <- min(round(sum(ratesElig[gElig == 2])), sum(gElig == 2))
dat$attr$active[ssample(idsElig[gElig == 2], nDeparturesG2)] <- 0
}
}
}
if (at == 2) {
dat$epi$dr.flow <- c(0, nDepartures)
if (groups == 2) {
dat$epi$dr.flow.g2 <- c(0, nDeparturesG2)
}
} else {
dat$epi$dr.flow[at] <- nDepartures
if (groups == 2) {
dat$epi$dr.flow.g2[at] <- nDeparturesG2
}
}
return(dat)
}
arrivals.seiqhrf.icm <- function(dat, at) {
# Conditions --------------------------------------------------------------
if (dat$param$vital == FALSE) {
return(dat)
}
# Variables ---------------------------------------------------------------
a.rate <- dat$param$a.rate
a.rate.g2 <- dat$param$a.rate.g2
a.rand <- dat$control$a.rand
groups <- dat$param$groups
nOld <- dat$epi$num[at - 1]
# checking params, should be in control.icm or params.icm eventually
type <- dat$control$type
nsteps <- dat$control$nsteps
if (!(length(a.rate) == 1 || length(a.rate == nsteps))) {
stop("Length of a.rate must be 1 or the value of nsteps")
}
if (!is.null(a.rate.g2) &&
!(length(a.rate.g2) == 1 || length(a.rate.g2 == nsteps))) {
stop("Length of a.rate.g2 must be 1 or the value of nsteps")
}
a.prop.e <- dat$param$a.prop.e
if (!(length(a.prop.e) == 1 || length(a.prop.e == nsteps))) {
stop("Length of a.prop.e must be 1 or the value of nsteps")
}
a.prop.i <- dat$param$a.prop.i
if (!(length(a.prop.i) == 1 || length(a.prop.i == nsteps))) {
stop("Length of a.prop.i must be 1 or the value of nsteps")
}
a.prop.q <- dat$param$a.prop.q
if (!(length(a.prop.q) == 1 || length(a.prop.q == nsteps))) {
stop("Length of a.prop.q must be 1 or the value of nsteps")
}
a.prop.e.g2 <- dat$param$a.prop.e.g2
if (!is.null(a.prop.e.g2) &&
!(length(a.prop.e.g2) == 1 || length(a.prop.e.g2 == nsteps))) {
stop("Length of a.prop.e.g2 must be 1 or the value of nsteps")
}
a.prop.i.g2 <- dat$param$a.prop.i.g2
if (!is.null(a.prop.i.g2) &&
!(length(a.prop.i.g2) == 1 || length(a.prop.i.g2 == nsteps))) {
stop("Length of a.prop.i.g2 must be 1 or the value of nsteps")
}
a.prop.q.g2 <- dat$param$a.prop.q.g2
if (!is.null(a.prop.q.g2) &&
!(length(a.prop.q.g2) == 1 || length(a.prop.q.g2 == nsteps))) {
stop("Length of a.prop.q.g2 must be 1 or the value of nsteps")
}
# Process -----------------------------------------------------------------
nArrivals <- nArrivals.g2 <- 0
if (groups == 1) {
if (a.rand == TRUE) {
nArrivals <- sum(rbinom(nOld, 1, a.rate))
}
if (a.rand == FALSE) {
nArrivals <- round(nOld * a.rate)
}
}
if (groups == 2) {
nOldG2 <- dat$epi$num.g2[at - 1]
if (a.rand == TRUE) {
if (is.na(a.rate.g2)) {
nArrivals <- sum(rbinom(nOld, 1, a.rate))
nArrivals.g2 <- sum(rbinom(nOld, 1, a.rate))
} else {
nArrivals <- sum(rbinom(nOld, 1, a.rate))
nArrivals.g2 <- sum(rbinom(nOldG2, 1, a.rate.g2))
}
}
if (a.rand == FALSE) {
if (is.na(a.rate.g2)) {
nArrivals <- round(nOld * a.rate)
nArrivals.g2 <- round(nOld * a.rate)
} else {
nArrivals <- round(nOld * a.rate)
nArrivals.g2 <- round(nOldG2 * a.rate.g2)
}
}
}
## Set attributes
totArrivals <- 0
totArrivals.g2 <- 0
# partition arrivals into compartments
if (length(a.prop.e) > 1) {
nArrivals.e <- round(nArrivals*(a.prop.e[at]))
totArrivals <- totArrivals + nArrivals.e
if (!is.null(a.prop.e.g2)) {
nArrivals.e.g2 <- round(nArrivals.g2*(a.prop.e.g2[at]))
totArrivals.g2 <- totArrivals.g2 + nArrivals.e.g2
} else {
nArrivals.e.g2 <- round(nArrivals.g2*(a.prop.e.g2[at]))
totArrivals.g2 <- totArrivals.g2 + nArrivals.e.g2
}
} else {
nArrivals.e <- round(nArrivals*a.prop.e)
totArrivals <- totArrivals + nArrivals.e
if (!is.null(a.prop.e.g2)) {
nArrivals.e.g2 <- round(nArrivals.g2*(a.prop.e.g2))
totArrivals.g2 <- totArrivals.g2 + nArrivals.e.g2
} else {
nArrivals.e.g2 <- round(nArrivals.g2*(a.prop.e.g2))
totArrivals.g2 <- totArrivals.g2 + nArrivals.e.g2
}
}
if (length(a.prop.i) > 1) {
nArrivals.i <- round(nArrivals*(a.prop.i[at]))
totArrivals <- totArrivals + nArrivals.i
if (!is.null(a.prop.i.g2)) {
nArrivals.i.g2 <- round(nArrivals.g2*(a.prop.i.g2[at]))
totArrivals.g2 <- totArrivals.g2 + nArrivals.i.g2
} else {
nArrivals.i.g2 <- round(nArrivals.g2*(a.prop.i.g2[at]))
totArrivals.g2 <- totArrivals.g2 + nArrivals.i.g2
}
} else {
nArrivals.i <- round(nArrivals*a.prop.i)
totArrivals <- totArrivals + nArrivals.i
if (!is.null(a.prop.i.g2)) {
nArrivals.i.g2 <- round(nArrivals.g2*(a.prop.i.g2))
totArrivals.g2 <- totArrivals.g2 + nArrivals.i.g2
} else {
nArrivals.i.g2 <- round(nArrivals.g2*(a.prop.i.g2))
totArrivals.g2 <- totArrivals.g2 + nArrivals.i.g2
}
}
if (length(a.prop.q) > 1) {
nArrivals.q <- round(nArrivals*(a.prop.q[at]))
totArrivals <- totArrivals + nArrivals.q
if (!is.null(a.prop.q.g2)) {
nArrivals.q.g2 <- round(nArrivals.g2*(a.prop.q.g2[at]))
totArrivals.g2 <- totArrivals.g2 + nArrivals.q.g2
} else {
nArrivals.q.g2 <- round(nArrivals.g2*(a.prop.q.g2[at]))
totArrivals.g2 <- totArrivals.g2 + nArrivals.q.g2
}
} else {
nArrivals.q <- round(nArrivals*a.prop.q)
totArrivals <- totArrivals + nArrivals.q
if (!is.null(a.prop.q.g2)) {
nArrivals.q.g2 <- round(nArrivals.g2*(a.prop.q.g2))
totArrivals.g2 <- totArrivals.g2 + nArrivals.q.g2
} else {
nArrivals.q.g2 <- round(nArrivals.g2*(a.prop.q.g2))
totArrivals.g2 <- totArrivals.g2 + nArrivals.q.g2
}
}
# debug
print("totArrivals:")
print(totArrivals)
print("totArrivals.g2:")
print(totArrivals.g2)
print("----")
# group 1
dat$attr$active <- c(dat$attr$active, rep(1, totArrivals))
dat$attr$group <- c(dat$attr$group, rep(1, totArrivals))
dat$attr$status <- c(dat$attr$status,
rep("e", nArrivals.e),
rep("i", nArrivals.i),
rep("q", nArrivals.q),
rep("s", totArrivals - nArrivals.e - nArrivals.i - nArrivals.q))
dat$attr$expTime <- c(dat$attr$expTime, rep(NA, totArrivals))
dat$attr$infTime <- c(dat$attr$infTime, rep(NA, totArrivals))
dat$attr$quarTime <- c(dat$attr$quarTime, rep(NA, totArrivals))
dat$attr$hospTime <- c(dat$attr$ihospTime, rep(NA, totArrivals))
dat$attr$recovTime <- c(dat$attr$recovTime, rep(NA, totArrivals))
dat$attr$fatTime <- c(dat$attr$fatTime, rep(NA, totArrivals))
# group 2
if (length(totArrivals.g2) > 0) {
dat$attr$active <- c(dat$attr$active, rep(1, totArrivals.g2))
dat$attr$group <- c(dat$attr$group, rep(2, totArrivals.g2))
dat$attr$status <- c(dat$attr$status,
rep("e", nArrivals.e.g2),
rep("i", nArrivals.i.g2),
rep("q", nArrivals.q.g2),
rep("s", totArrivals.g2 - nArrivals.e.g2 -
nArrivals.i.g2 - nArrivals.q.g2))
dat$attr$expTime <- c(dat$attr$expTime, rep(NA, totArrivals.g2))
dat$attr$infTime <- c(dat$attr$infTime, rep(NA, totArrivals.g2))
dat$attr$quarTime <- c(dat$attr$quarTime, rep(NA, totArrivals.g2))
dat$attr$hospTime <- c(dat$attr$ihospTime, rep(NA, totArrivals.g2))
dat$attr$recovTime <- c(dat$attr$recovTime, rep(NA, totArrivals.g2))
dat$attr$fatTime <- c(dat$attr$fatTime, rep(NA, totArrivals.g2))
}
# Output ------------------------------------------------------------------
if (at == 2) {
dat$epi$a.flow <- c(0, totArrivals)
dat$epi$a.e.flow <- c(0, nArrivals.e)
dat$epi$a.i.flow <- c(0, nArrivals.i)
dat$epi$a.q.flow <- c(0, nArrivals.q)
} else {
dat$epi$a.flow[at] <- totArrivals
dat$epi$a.e.flow[at] <- c(0, nArrivals.e)
dat$epi$a.i.flow[at] <- c(0, nArrivals.i)
dat$epi$a.q.flow[at] <- c(0, nArrivals.q)
}
if (length(totArrivals.g2) > 0 && dat$param$groups == 2) {
if (at == 2) {
dat$epi$a.flow.g2 <- c(0, totArrivals.g2)
dat$epi$a.e.flow.g2 <- c(0, nArrivals.e.g2)
dat$epi$a.i.flow.g2 <- c(0, nArrivals.i.g2)
dat$epi$a.q.flow.g2 <- c(0, nArrivals.q.g2)
} else {
dat$epi$a.flow.g2[at] <- totArrivals.g2
dat$epi$a.e.flow.g2[at] <- c(0, nArrivals.e.g2)
dat$epi$a.i.flow.g2[at] <- c(0, nArrivals.i.g2)
dat$epi$a.q.flow.g2[at] <- c(0, nArrivals.q.g2)
}
}
return(dat)
}
Raw
_icm.saveout.seiqhrf.R
saveout.seiqhrf.icm <- function(dat, s, out = NULL) {
alist2df <- function(dat,s) {
alist <- list()
alist$expTime <- dat$attr$expTime
alist$infTime <- dat$attr$infTime
alist$quarTime <- dat$attr$quarTime
alist$recovTime <- dat$attr$recovTime
alist$hospTime <- dat$attr$hospTime
alist$dischTime <- dat$attr$dischTime
alist$fatTime <- dat$attr$fatTime
alist <- lapply(alist, `length<-`, max(lengths(alist)))
return(data.frame(alist))
}
if (s == 1) {
out <- list()
out$param <- dat$param
out$control <- dat$control
out$epi <- list()
for (j in 1:length(dat$epi)) {
out$epi[[names(dat$epi)[j]]] <- data.frame(dat$epi[j])
}
} else {
for (j in 1:length(dat$epi)) {
out$epi[[names(dat$epi)[j]]][, s] <- data.frame(dat$epi[j])
}
}
# capture transition times from attribs
if (dat$control$type %in% c("SEIQHR", "SEIQHRF")) {
out$times[[paste("sim",s,sep="")]] <- alist2df(dat,s)
}
## Processing for final run
if (s == dat$control$nsims) {
# Remove functions from control list
ftodel <- grep(".FUN", names(out$control), value = TRUE)
out$control[ftodel] <- NULL
# Set column names for varying list elements
for (i in as.vector(which(lapply(out$epi, class) == "data.frame"))) {
colnames(out$epi[[i]]) <- paste0("sim", 1:dat$control$nsims)
}
}
return(out)
}
Raw
_icm.utils.seiqhrf.R
get_prev.seiqhrf.icm <- function(dat, at) {
if (at == 1) {
dat$epi <- list()
dat$epi$s.num <- sum(dat$attr$active == 1 &
dat$attr$status == "s" &
dat$attr$group == 1)
dat$epi$i.num <- sum(dat$attr$active == 1 &
dat$attr$status == "i" &
dat$attr$group == 1)
if (dat$control$type %in% c("SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi$e.num <- sum(dat$attr$active == 1 &
dat$attr$status == "e" &
dat$attr$group == 1)
}
if (dat$control$type %in% c("SIR", "SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi$r.num <- sum(dat$attr$active == 1 &
dat$attr$status == "r" &
dat$attr$group == 1)
}
if (dat$control$type %in% c("SEIQHR", "SEIQHRF")) {
dat$epi$q.num <- sum(dat$attr$active == 1 &
dat$attr$status == "q" &
dat$attr$group == 1)
dat$epi$h.num <- sum(dat$attr$active == 1 &
dat$attr$status == "h" &
dat$attr$group == 1)
}
if (dat$control$type =="SEIQHRF") {
dat$epi$f.num <- sum(dat$attr$active == 1 &
dat$attr$status == "f" &
dat$attr$group == 1)
}
if (dat$control$type == "SIR") {
dat$epi$num <- dat$epi$s.num +
dat$epi$i.num +
dat$epi$r.num
} else if (dat$control$type == "SEIR") {
dat$epi$num <- dat$epi$s.num +
dat$epi$e.num +
dat$epi$i.num +
dat$epi$r.num
} else if (dat$control$type == "SEIQHR") {
dat$epi$num <- dat$epi$s.num +
dat$epi$e.num +
dat$epi$i.num +
dat$epi$q.num +
dat$epi$h.num +
dat$epi$r.num
} else if (dat$control$type == "SEIQHRF") {
dat$epi$num <- dat$epi$s.num +
dat$epi$e.num +
dat$epi$i.num +
dat$epi$q.num +
dat$epi$h.num +
dat$epi$r.num +
dat$epi$f.num
} else {
dat$epi$num <- dat$epi$s.num + dat$epi$i.num
}
if (dat$param$groups == 2) {
dat$epi$s.num.g2 <- sum(dat$attr$active == 1 &
dat$attr$status == "s" &
dat$attr$group == 2)
if (dat$control$type %in% c("SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi$e.num.g2 <- sum(dat$attr$active == 1 &
dat$attr$status == "e" &
dat$attr$group == 2)
}
if (dat$control$type %in% c("SEIQHR", "SEIQHRF")) {
dat$epi$q.num.g2 <- sum(dat$attr$active == 1 &
dat$attr$status == "q" &
dat$attr$group == 2)
dat$epi$h.num.g2 <- sum(dat$attr$active == 1 &
dat$attr$status == "h" &
dat$attr$group == 2)
}
if (dat$control$type %in% c("SEIQHRF")) {
dat$epi$f.num.g2 <- sum(dat$attr$active == 1 &
dat$attr$status == "f" &
dat$attr$group == 2)
}
dat$epi$i.num.g2 <- sum(dat$attr$active == 1 &
dat$attr$status == "i" &
dat$attr$group == 2)
dat$epi$num.g2 <- dat$epi$s.num.g2 + dat$epi$i.num.g2
if (dat$control$type %in% c("SIR", "SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi$r.num.g2 <- sum(dat$attr$active == 1 &
dat$attr$status == "r" &
dat$attr$group == 2)
}
if (dat$control$type == "SIR") {
dat$epi$num.g2 <- dat$epi$s.num.g2 +
dat$epi$i.num.g2 +
dat$epi$r.num.g2
} else if (dat$control$type == "SEIR") {
dat$epi$num.g2 <- dat$epi$s.num.g2 +
dat$epi$e.num.g2 +
dat$epi$i.num.g2 +
dat$epi$r.num.g2
} else if (dat$control$type == "SEIQHR") {
dat$epi$num.g2 <- dat$epi$s.num.g2 +
dat$epi$e.num.g2 +
dat$epi$i.num.g2 +
dat$epi$q.num.g2 +
dat$epi$h.num.g2 +
dat$epi$r.num.g2
} else if (dat$control$type == "SEIQHRF") {
dat$epi$num.g2 <- dat$epi$s.num.g2 +
dat$epi$e.num.g2 +
dat$epi$i.num.g2 +
dat$epi$q.num.g2 +
dat$epi$h.num.g2 +
dat$epi$r.num.g2 +
dat$epi$f.num.g2
} else {
dat$epi$num.g2 <- dat$epi$s.num.g2 + dat$epi$i.num.g2
}
}
} else {
dat$epi$s.num[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "s" &
dat$attr$group == 1)
if (dat$control$type %in% c("SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi$e.num[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "e" &
dat$attr$group == 1)
}
dat$epi$i.num[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "i" &
dat$attr$group == 1)
if (dat$control$type %in% c("SIR", "SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi$r.num[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "r" &
dat$attr$group == 1)
}
if (dat$control$type %in% c("SEIQHR", "SEIQHRF")) {
dat$epi$q.num[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "q" &
dat$attr$group == 1)
dat$epi$h.num[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "h" &
dat$attr$group == 1)
}
if (dat$control$type %in% c("SEIQHRF")) {
dat$epi$f.num[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "f" &
dat$attr$group == 1)
}
if (dat$control$type == "SIR") {
dat$epi$num[at] <- dat$epi$s.num[at] +
dat$epi$i.num[at] +
dat$epi$r.num[at]
} else if (dat$control$type == "SEIR") {
dat$epi$num[at] <- dat$epi$s.num[at] +
dat$epi$e.num[at] +
dat$epi$i.num[at] +
dat$epi$r.num[at]
} else if (dat$control$type == "SEIQHR") {
dat$epi$num[at] <- dat$epi$s.num[at] +
dat$epi$e.num[at] +
dat$epi$i.num[at] +
dat$epi$q.num[at] +
dat$epi$h.num[at] +
dat$epi$r.num[at]
} else if (dat$control$type == "SEIQHRF") {
dat$epi$num[at] <- dat$epi$s.num[at] +
dat$epi$e.num[at] +
dat$epi$i.num[at] +
dat$epi$q.num[at] +
dat$epi$h.num[at] +
dat$epi$r.num[at] +
dat$epi$f.num[at]
} else {
dat$epi$num[at] <- dat$epi$s.num[at] + dat$epi$i.num[at]
}
if (dat$param$groups == 2) {
dat$epi$s.num.g2[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "s" &
dat$attr$group == 2)
if (dat$control$type %in% c("SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi$i.num.g2[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "e" &
dat$attr$group == 2)
}
dat$epi$i.num.g2[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "i" &
dat$attr$group == 2)
if (dat$control$type %in% c("SIR", "SEIR", "SEIQHR", "SEIQHRF")) {
dat$epi$r.num.g2[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "r" &
dat$attr$group == 2)
}
if (dat$control$type %in% c("SEIQHR", "SEIQHRF")) {
dat$epi$q.num.g2[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "q" &
dat$attr$group == 2)
dat$epi$h.num.g2[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "h" &
dat$attr$group == 2)
}
if (dat$control$type %in% c("SEIQHRF")) {
dat$epi$f.num.g2[at] <- sum(dat$attr$active == 1 &
dat$attr$status == "f" &
dat$attr$group == 2)
}
if (dat$control$type == "SIR") {
dat$epi$num.g2[at] <- dat$epi$s.num.g2[at] +
dat$epi$i.num.g2[at] +
dat$epi$r.num.g2[at]
} else if (dat$control$type == "SEIR") {
dat$epi$num.g2[at] <- dat$epi$s.num.g2[at] +
dat$epi$e.num.g2[at] +
dat$epi$i.num.g2[at] +
dat$epi$r.num.g2[at]
} else if (dat$control$type == "SEIQHR") {
dat$epi$num.g2[at] <- dat$epi$s.num.g2[at] +
dat$epi$e.num.g2[at] +
dat$epi$i.num.g2[at] +
dat$epi$q.num.g2[at] +
dat$epi$h.num.g2[at] +
dat$epi$r.num.g2[at]
} else if (dat$control$type == "SEIQHRF") {
dat$epi$num.g2[at] <- dat$epi$s.num.g2[at] +
dat$epi$e.num.g2[at] +
dat$epi$i.num.g2[at] +
dat$epi$q.num.g2[at] +
dat$epi$h.num.g2[at] +
dat$epi$r.num.g2[at] +
dat$epi$f.num.g2[at]
} else {
dat$epi$num.g2[at] <- dat$epi$s.num.g2[at] +
dat$epi$i.num.g2[at]
}
}
}
return(dat)
}
@drnareshchauhan
Copy link

drnareshchauhan commented May 3, 2021

image
throwing this error sir kindly guide
for this code
baseline_sim <- simulate(ncores = 4)
even tried to change to 1
image

thanks !

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment