R/recontactReports.R

In Crunch-io/crunchtabs: Custom Report Generation for Crunch Datasets

#' Recontact Toplines
#'
#' Allows the user to create a simple report that shows recontact question.
#' @param dataset A crunch dataset
#' @param questions A character vector of aliases that should be included in the
#' report. If your recontact has been named using a suffix such as _pre, _post
#' leave that out.
#' @param suffixes The suffixes of recontact questions, for example _pre, _post
#' @param labels Formal labels for
#' the election", "After the election".
#' @param weights A character vector of equal to the length of suffixes. You may
#' specify a unique weight per recontact period. The default would return
#' all variables with the default survey weighting `weight(ds)`. Your weights
#' should be in the same order as your suffixies.
#' @param default_weight The default weight of the dataset, if any.
recontact_toplines <- function(dataset, questions, suffixes, labels,
                               weights = crunch::weight(dataset),
                               default_weight = crunch::alias(crunch::weight(dataset))) {
  stopifnot(is.dataset(dataset))
  stopifnot(is.character(questions))
  stopifnot(is.character(suffixes))
  stopifnot(is.character(labels))

  groupings <- lapply(questions, function(x) paste0(x, suffixes))
  names(groupings) <- questions
  vars <- unlist(groupings)

  if (length(weights) > 1) {
    weight_spec <- lapply(suffixes, function(x) vars[grepl(x, vars)])
    names(weight_spec) <- weights
  }

  ct <- crosstabs(
    dataset,
    vars = c(unlist(groupings), names(weight_spec)),
    weight = weight_spec,
    include_original_weighted = FALSE
  )

  for (question in questions) {
    if (!is.null(weights)) {
      if (weights[1] == default_weight) {
        p1 <- groupings[[question]][1]
      } else { # nocov start
        p1 <- paste0(groupings[[question]][1], "_", weights[1])
      }

      if (weights[2] == default_weight) {
        p2 <- groupings[[question]][2]
      } else {
        p2 <- paste0(groupings[[question]][2], "_", weights[2])
      }
    } else {
      p1 <- groupings[[question]][1]
      p2 <- groupings[[question]][2]
    }
    # nocov end

    ct$results[[question]] <- as.ToplineCategoricalArray(
      questions = list(ct$results[[p1]], ct$results[[p2]]),
      question_alias = question,
      labels = labels
    )

    ct$results[[p1]] <- NULL
    ct$results[[p2]] <- NULL
  }

  ct
}