tests/sim/20210701-sim/setup-marPack.R

In jlivsey/Dvar: What the Package Does (One Line, Title Case)

# ---- County recodes ----

# Load VA tract level data.frame
# source('~/github/Dvar/tests/survey/20210410-raceHisp-tarWts.R')

# Count the number of tracts in each county
# x_recode <- temp_x %>%
  # count(County_name)
# save(x_recode, file = "tests/sim/20210429-sim/x_recode.Rdata")
load("tests/sim/20210429-sim/x_recode.Rdata")

countyRecode <- matrix(-99, 7, 2)
countyRecode[1, ] <- c(1, x_recode$n[1])
idx = x_recode$n[1] + 1
for(i in 2:7){
  countyRecode[i, ] <- c(idx, idx + x_recode$n[i] - 1)
  idx <- idx + x_recode$n[i]
}

# ---- State recodes ----
stateRecode <- matrix(-99, 2, 2)
stateRecode[1, ] <- c(1, 28)
stateRecode[2, ] <- c(29, 43)

# setup marPack
# marPack = vector("list", 1872)
marPack = list()

idx <- 1

# ---- Own/rent --------------------------------------------------------------------

# Add Own/rent margins at tract level
for(i in 1:2){     # own/rent
  for(j in 1:43){  # tract
    marPack[[idx]] <- list(i, 0, 0, 0, 0, j)
    idx <- idx + 1
  }
}
# # Add Own/rent margins at county level
# for(i in 1:2){    # own/rent
#   for(j in 1:7){  # county
#     marPack[[idx]] <- list(i, 0, 0, 0, 0, countyRecode[j, 1]:countyRecode[j, 2])
#     idx <- idx + 1
#   }
# }
# # Add Own/rent margins at state level
# for(i in 1:2){   # own/rent
#   for(j in 1:2){ # state
#     marPack[[idx]] <- list(i, 0, 0, 0, 0, stateRecode[j, 1]:stateRecode[j, 2])
#     idx <- idx + 1
#   }
# }

# ---- votingAge x hisp x cenRace ---------------------------------------------

votingAgeRecode <- rbind(c(1, 1),
                         c(2, 3))

# Add margins at tract level
for(i in 1:43){ # tract levels
  for(j in 1:2){ # voting age levels
    for(k in 1:2){ # hisp levels
      for(el in 1:7){ # cenRace levels
        marPack[[idx]] <- list(0, 0, el,
                               votingAgeRecode[j, 1]:votingAgeRecode[j, 2],
                               k, i)
        idx <- idx + 1
      }
    }
  }
}
# # Add margins at county level
# for(i in 1:7){ # county levels
#   for(j in 1:2){ # voting age levels
#     for(k in 1:2){ # hisp levels
#       for(el in 1:7){ # cenRace levels
#         marPack[[idx]] <- list(0, 0, el,
#                                votingAgeRecode[j, 1]:votingAgeRecode[j, 2],
#                                k,
#                                countyRecode[i, 1]:countyRecode[i, 2])
#         idx <- idx + 1
#       }
#     }
#   }
# }
# # Add margins at state level
# for(i in 1:2){ # state levels
#   for(j in 1:2){ # voting age levels
#     for(k in 1:2){ # hisp levels
#       for(el in 1:7){ # cenRace levels
#         marPack[[idx]] <- list(0, 0, el,
#                                votingAgeRecode[j, 1]:votingAgeRecode[j, 2],
#                                k,
#                                stateRecode[i, 1]:stateRecode[i, 2])
#         idx <- idx + 1
#       }
#     }
#   }
# }

# ---- Age x Sex ---------------------------------------------------------------

# Add margins at tract level
for(i in 1:43){ # tract levels
  for(j in 1:3){ # age levels
    for(k in 1:2){ # sex levels
      marPack[[idx]] <- list(0, k, 0, j, 0, i)
      idx <- idx + 1
    }
  }
}
# # Add margins at county level
# for(i in 1:7){ # county levels
#   for(j in 1:3){ # age levels
#     for(k in 1:2){ # sex levels
#       marPack[[idx]] <- list(0, k, 0, j, 0,
#                              countyRecode[i,1]:countyRecode[i, 2])
#       idx <- idx + 1
#     }
#   }
# }
# # Add margins at state level
# for(i in 1:2){ # state levels
#   for(j in 1:3){ # age levels
#     for(k in 1:2){ # sex levels
#       marPack[[idx]] <- list(0, k, 0, j, 0,
#                              stateRecode[i,1]:stateRecode[i, 2])
#       idx <- idx + 1
#     }
#   }
# }

# -----------------------------------------------------------------------------
# ---- SAVE -------------------------------------------------------------------

save(marPack, file = "marPack-tr8.RData")