R/postprocess.R

In datasketch/dsvizprep: Data processing

# sort and slice
#'@export
postprocess <- function(d, col, sort = NULL, slice_n = NULL) {

  if (is.null(sort)) {
    d <- d
  } else if (sort == "asc") {
    d <- d[order(d[col],na.last = TRUE),]
  } else if (sort == "desc") {
    d <- d[order(-as.numeric(d[[col]]), na.last = TRUE),]
  }

  if (!is.null(slice_n)) {
    d <- d %>%
      dplyr::slice(1:slice_n)
  }

  d
}

#' Format data
#'
#' function to format the strings or number of the data from scratch.
#'
#' @param data A data.frame
#' @param dic Dic of data to plot
#' @param formats list with sample_num, sample_cat, suffix and prefix
#'
#' @export
format_prep <- function(data, dic, formats) {

  if (is.null(data)) return()

  var_nums <- grep("Num", dic$hdType)

  if (!identical(var_nums, integer())) {
    var_nums <- dic$id[var_nums]

    l_nums <- purrr::map(var_nums, function(f_nums){
      data[[paste0(f_nums, "_label")]] <<- ifelse(is.na(data[[f_nums]]), NA,
                                                  paste0(formats$prefix,
                                                         makeup::makeup_num(as.numeric(data[[f_nums]]),
                                                                            formats$sample_num),
                                                         formats$suffix))
    })}

  var_coor <- grep("Glt|Gln", dic$hdType)

  if (!identical(var_coor, integer())) {
    var_coor <- dic$id[var_coor]

    l_nums <- purrr::map(var_coor, function(f_coor){
      data[[paste0(f_coor, "_label")]] <<- ifelse(is.na(data[[f_coor]]), NA, round(as.numeric(data[[f_coor]]),2))
    })}

  var_cats <- grep("^Cat$|Gnm|Gcd", dic$hdType)
  if (!identical(var_cats, integer())) {
    var_cats <- dic$id[var_cats]
    l_nums <- purrr::map(var_cats, function(f_cats){
      data[[paste0(f_cats, "_label")]] <<- ifelse(is.na(data[[f_cats]]), NA,
                                                  makeup::makeup_chr(data[[f_cats]],
                                                                     formats$sample_cat))
    })}

  var_cats_extra <- grep("Cat..", dic$hdType)

  if (!identical(var_cats_extra, integer())) {
    var_cats_extra <- dic$id[var_cats_extra]

    l_nums <- purrr::map(var_cats_extra, function(f_cats..){
      data[[paste0(f_cats.., "_label")]] <<- makeup::makeup_chr(as.character(data[[f_cats..]]), sample = NULL)
    })}


  data

}