R/vital_births_module.R

In EvoNetHIV/EvoNetVaccine: Evonet: Integrated Bio-Social Models for HIV Epidemiology

#' @export
vital_births_module <- function(dat, at) {
  
  #Description:
  #Calculates number of births (additions) to population for timestep
  #Expands network and population data structures based on number of births
  #Calls: vital_births_calculate_new(), vital_births_bookkeeping_pop(), vital_births_bookkeeping_misc() 
  #Inputs: param$birth_model==
              #"annual_growth_rate"
              #"births=deaths"
              #"poisson_birth_numbers"
              #"constant_rate"
              #"constant_number"
  #Outputs: dat$nw, dat$pop, dat$popsumm, dat$attr$status
    
  nBirths <- vital_births_calculate_new(dat,at)
  if(nBirths==0){return(dat)}
  
  #expand "pop" lists and fill in default values as needed
  dat$pop <- vital_births_bookkeeping_pop(no_births = nBirths, 
                                          dat=dat, timestep = at)
  
  #modify network and "status" vectors as needed
  dat <- vital_births_bookkeeping_misc(dat,at)  
  
  return(dat)
}