R/propkc.r

In kcsamir/mdpop: What the Package Does (One Line, Title Case)

#' Proportion size in MSDem
#'
#' This function allows you to calculate population size by sex or by regions using the MSDem output
#' @param res1 data
#' @param agel lower limit - default is 0
#' @param ageu upper limit - default it 'max'
#' @param period specific periods - default is NULL
#' @param sex specific sex - default is NULL
#' @param bysex group by sex - default is FALSE
#' @param reg specific region(s)
#' @param byreg group by regions - default is FALSE
#' @param resi specific residence(s)
#' @param byresi group by residences - default is FALSE
#' @return R list with five data.table 1) var_def 2) state_space 3) mig_dom 4) mig_int 5) results
#' @keywords read
#' @export
#' @examples

propkc <- function(resX=res,
                  period = NULL,
                  agel=0,ageu="max",byage=FALSE,
                  sex=NULL, bysex=FALSE,
                  reg=NULL, byreg=FALSE,
                  resi=NULL,byresi=FALSE) {
          #Try this for with residence
          #need to try all possible combinations

            # #need res data
            # if(length(grep("res",ls()))==0) print("Run fread_all() first")

            var_def <- resX$var_def
            names(var_def)[1] <-c("varname")
            # head(var_def)
            # unique(var_def$varname)
            ages = var_def[varname=="age"][,values:=as.numeric(values)]
            if(ageu == "max") ageu = max(ages$values)

            periods = var_def[varname=="period"][,values:=as.numeric(values)]$values
            periods = c(periods,max(periods)+unique(diff(periods)))
            if(!is.null(period)) periods = period

            sexes = var_def[varname=="sex"]$values
            if(!is.null(sex)) sexes = sex

            regions = var_def[varname=="region"]$values
            if(!is.null(reg)) regions = reg

            group.var = c("age","period")
            if(bysex) group.var = c(group.var,"sex")
            if(byreg) group.var = c(group.var,"region")

            group.var.minus.age = setdiff(group.var,"age")
            resX <- resX$results

            if(any(grep("resi",unique(var_def$varname)))) {
              residences = var_def[varname=="resi"]$values
              if(!is.null(resi)) residences = resi
              if(byres) key.cols = c(group.var,"residence")
              res.pop = resX[period%in%periods&age%in%seq(agel,ageu)&sex%in%sexes&reg%in%regions&resi%in%residences
                                  ,.(pop=sum(pop)),by=group.var]
            } else {
              res.prop <- resX[period%in%periods & sex%in%sexes & region%in%regions,.(pop=sum(pop)),by=group.var][,.(age=age,prop=prop.table(pop)),by=group.var.minus.age]
              if(!byage) res.prop <- res.prop[age%in%seq(agel,ageu),.(age=paste(agel,ageu+4,sep="to"),prop=sum(prop)),group.var.minus.age]
            }
        return(res.prop)
  }