Use vectorized randomNames() call

R/add_cols.R

@@ -189,30 +189,14 @@ NULL
 #' @name .add_info
 .add_names <- function(.data) {
   # create sample of names so there are no duplicates
-  cases_m <- sum(.data$gender == "m")
-  cases_f <- sum(.data$gender == "f")
-  names <- list(
-    names_masc = randomNames::randomNames(
-      n = cases_m,
-      which.names = "both",
-      name.sep = " ",
-      name.order = "first.last",
-      gender = 0,
-      sample.with.replacement = FALSE
-    ),
-    names_fem = randomNames::randomNames(
-      n = cases_f,
-      which.names = "both",
-      name.sep = " ",
-      name.order = "first.last",
-      gender = 1,
-      sample.with.replacement = FALSE
-    )
+  .data$case_name <- randomNames::randomNames(
+    which.names = "both",
+    name.sep = " ",
+    name.order = "first.last",
+    gender = .data$gender,
+    sample.with.replacement = FALSE
   )
 
-  .data$case_name[.data$gender == "m"] <- names$names_masc # move to start of df
-  .data$case_name[.data$gender == "f"] <- names$names_fem
-
   # left join corresponding names to infectors preserving column and row order
   infector_names <- data.frame(id = .data$id, infector_name = .data$case_name)
   col_order <- c(colnames(.data), "infector_name")