Add support for ZCTAs and 2020 age and sex data

.Rbuildignore

@@ -12,3 +12,5 @@ ChangeLog
 
 ^cran-comments\.md$
 ^CRAN-SUBMISSION$
+^README\.Rmd$
+^data-raw$

.gitignore

@@ -1,3 +1,5 @@
+.DS_Store
+
 # History files
 .Rhistory
 .Rapp.history
@@ -8,6 +10,7 @@
 # RStudio files
 .Rproj.user/
 .Rproj
+.lazytest
 
 # produced vignettes
 vignettes/*.html
@@ -21,4 +24,4 @@ vignettes/*.pdf
 src/RcppExports.o
 src/aux_funs.o
 src/sample_me.o
-src/wru.so
+src/wru.so

DESCRIPTION

@@ -1,47 +1,56 @@
 Package: wru
-Version: 2.0.0
-Date: 2023-07-12
-Title: Who are You? Bayesian Prediction of Racial Category Using Surname, First Name, Middle Name, and
-    Geolocation
+Title: Who are You? Bayesian Prediction of Racial Category Using Surname,
+    First Name, Middle Name, and Geolocation
+Version: 3.0.0
+Date: 2023-12-09
 Authors@R: c(
-  person("Kabir", "Khanna", email = "[email protected]", role = c("aut")),
-  person("Brandon", "Bertelsen", email = "[email protected]", role = c("aut","cre")),
-  person("Santiago", "Olivella", email = "[email protected]", role = c("aut")),
-  person("Evan", "Rosenman", email = "[email protected]", role = c("aut")),
-  person("Kosuke", "Imai", email = "[email protected]", role = c("aut"))
+    person("Kabir", "Khanna", , "[email protected]", role = "aut"),
+    person("Brandon", "Bertelsen", , "[email protected]", role = c("aut", "cre")),
+    person("Santiago", "Olivella", , "[email protected]", role = "aut"),
+    person("Evan", "Rosenman", , "[email protected]", role = "aut"),
+    person("Kosuke", "Imai", , "[email protected]", role = "aut")
   )
-Description: Predicts individual race/ethnicity using surname, first name, middle name, geolocation,
-    and other attributes, such as gender and age. The method utilizes Bayes'
-    Rule (with optional measurement error correction) to compute the posterior probability of each racial category for any given
-    individual. The package implements methods described in Imai and Khanna (2016)
-    "Improving Ecological Inference by Predicting Individual Ethnicity from Voter
-    Registration Records" Political Analysis <DOI:10.1093/pan/mpw001> and Imai, Olivella, and Rosenman (2022) 
-    "Addressing census data problems in race imputation via fully Bayesian Improved Surname Geocoding and name supplements" 
-    <DOI:10.1126/sciadv.adc9824>.  The package also incorporates the data described in Rosenman, Olivella, and Imai (2023) 
-    "Race and ethnicity data for first, middle, and surnames" <DOI:10.1038/s41597-023-02202-2>.
+Description: Predicts individual race/ethnicity using surname, first name,
+    middle name, geolocation, and other attributes, such as gender and
+    age. The method utilizes Bayes' Rule (with optional measurement error
+    correction) to compute the posterior probability of each racial
+    category for any given individual. The package implements methods
+    described in Imai and Khanna (2016) "Improving Ecological Inference by
+    Predicting Individual Ethnicity from Voter Registration Records"
+    Political Analysis <DOI:10.1093/pan/mpw001> and Imai, Olivella, and
+    Rosenman (2022) "Addressing census data problems in race imputation
+    via fully Bayesian Improved Surname Geocoding and name supplements"
+    <DOI:10.1126/sciadv.adc9824>.  The package also incorporates the data
+    described in Rosenman, Olivella, and Imai (2023) "Race and ethnicity
+    data for first, middle, and surnames"
+    <DOI:10.1038/s41597-023-02202-2>.
+License: GPL (>= 3)
 URL: https://github.com/kosukeimai/wru
 BugReports: https://github.com/kosukeimai/wru/issues
 Depends:
     R (>= 4.1.0),
     utils
 Imports:
+    cli,
     dplyr,
     furrr,
     future,
+    piggyback (>= 0.1.4),
+    PL94171,
     purrr,
     Rcpp,
-    piggyback (>= 0.1.4), 
-    PL94171
+    rlang
 Suggests:
+    covr,
     testthat (>= 3.0.0),
-    covr
+    tidycensus
 LinkingTo:
     Rcpp,
     RcppArmadillo
-LazyLoad: yes
+Config/testthat/edition: 3
+Encoding: UTF-8
 LazyData: yes
 LazyDataCompression: xz
-License: GPL (>= 3)
+LazyLoad: yes
+Roxygen: list(markdown = TRUE)
 RoxygenNote: 7.2.3
-Encoding: UTF-8
-Config/testthat/edition: 3

NAMESPACE

@@ -6,9 +6,11 @@ export(predict_race)
 import(PL94171)
 importFrom(Rcpp,evalCpp)
 importFrom(dplyr,coalesce)
+importFrom(dplyr,pull)
 importFrom(furrr,future_map_dfr)
 importFrom(piggyback,pb_download)
 importFrom(purrr,map_dfr)
+importFrom(rlang,"%||%")
 importFrom(stats,rmultinom)
 importFrom(utils,setTxtProgressBar)
 importFrom(utils,txtProgressBar)

R/census_data_preflight.R

@@ -1,31 +1,15 @@
 #' Preflight census data
 #' 
-#' @param census.data See documentation in \code{race_predict}.
-#' @param census.geo See documentation in \code{race_predict}.
-#' @param year See documentation in \code{race_predict}.
+#' @inheritParams predict_race
 #' @keywords internal
 
 census_data_preflight <- function(census.data, census.geo, year) {
-
-  if (year != "2020"){
-    vars_ <- c(
-      pop_white = 'P005003', pop_black = 'P005004',
-      pop_aian = 'P005005', pop_asian = 'P005006',
-      pop_nhpi = 'P005007', pop_other = 'P005008', 
-      pop_two = 'P005009', pop_hisp = 'P005010'
-    )
-  } else {
-    vars_ <- c(
-      pop_white = 'P2_005N', pop_black = 'P2_006N',
-      pop_aian = 'P2_007N', pop_asian = 'P2_008N', 
-      pop_nhpi = 'P2_009N', pop_other = 'P2_010N', 
-      pop_two = 'P2_011N', pop_hisp = 'P2_002N'
-    )
-  }
+  vars_ <- unlist(census_geo_api_names(year = year))
+  legacy_vars <- unlist(census_geo_api_names_legacy(year = year))
 
   test <- lapply(census.data, function(x) {
     nms_to_test <- names(x[[census.geo]])
-    all(vars_ %in% nms_to_test)
+    all(vars_ %in% nms_to_test) || all(legacy_vars %in% nms_to_test)
   }) 
   missings <- names(test)[!unlist(test)]