R/casting.R

In hacksaw: Additional Tools for Splitting and Cleaning Data

#' Cast columns to a specified data type
#'
#' @param .data a table of data.
#' @param ... A selection of columns.
#'
#' @export
#' @rdname casting
#' @return a data frame.
#'
#' @examples
#' library(dplyr)
#' df <- tibble(x = 1:3, y = as.character(1:3), z = c(0, 0, 1))
#' df %>% cast_character(x)
#' df %>% cast_numeric(y)
#' df %>% cast_logical(z)
cast_character <- function(.data, ...) {
  .data <- assert_df(.data)
  vars <- tidyselect::eval_select(rlang::expr(c(...)), .data)
  if (rlang::is_empty(vars)) return(dplyr::mutate_all(.data, as.character))
  dplyr::mutate_at(.data, dplyr::vars(...), as.character)
}

#' @rdname casting
#' @export
cast_numeric <- function(.data, ...) {
  .data <- assert_df(.data)
  vars <- tidyselect::eval_select(rlang::expr(c(...)), .data)
  if (rlang::is_empty(vars)) return(dplyr::mutate_all(.data, as.numeric))
  dplyr::mutate_at(.data, dplyr::vars(...), as.numeric)
}

#' @rdname casting
#' @export
cast_logical <- function(.data, ...) {
  .data <- assert_df(.data)
  vars <- tidyselect::eval_select(rlang::expr(c(...)), .data)
  if (rlang::is_empty(vars)) return(dplyr::mutate_all(.data, as.logical))
  dplyr::mutate_at(.data, dplyr::vars(...), as.logical)
}