R/barplot.R

In jcai849/iNZightTA: iNZight Text Analytics

#' output a ggplot column graph of the top texts from some insight function
#'
#' @param .data a dataframe containing "text" and insight columns as
#'     per the output of the get_(term|aggregate)_insight wrapper
#'     function
#' 
#' @param y symbol name of the column insight was
#'     outputted to
#' 
#' @param x symbol name of column for insight labels
#'
#' @param n number of bars to display
#'
#' @param desc bool: show bars in descending order
#'
#' @export
score_barplot <- function(.data, y, n = 15,
                                x = text, desc = FALSE){
    wrap <- 50
    text <- dplyr::enquo(x)
    insight_col <- dplyr::enquo(y)
    if (desc == FALSE){
        .data %>%
            dplyr::distinct(!! text, .keep_all=TRUE) %>%
            dplyr::arrange(dplyr::desc(!! insight_col)) %>%
            dplyr::group_modify(~{.x %>% head(n)}) %>%
            dplyr::ungroup() %>%
            dplyr::mutate(text = forcats::fct_reorder(shorten(!! text, wrap),
                                                      !! insight_col,
                                                      #.desc = desc
                                                      )) %>%
            ggplot2::ggplot(ggplot2::aes(x = text)) +
            ggplot2::geom_col(ggplot2::aes(y = !! insight_col)) +
            ggplot2::coord_flip() 
    }
    else {
        .data %>%
            dplyr::distinct(!! text, .keep_all=TRUE) %>%
            dplyr::arrange(!! insight_col) %>%
            dplyr::group_modify(~{.x %>% head(n)}) %>%
            dplyr::ungroup() %>%
            dplyr::mutate(text = forcats::fct_reorder(shorten(!! text, wrap),
                                                      !! insight_col,
                                                      #.desc = desc
            )) %>%
            ggplot2::ggplot(ggplot2::aes(x = text)) +
            ggplot2::geom_col(ggplot2::aes(y = !! insight_col)) +
            ggplot2::coord_flip() 
    }
}

#' Shorten some text up to n characters
#'
#' @param .data character vector
#' 
#' @param n wrap length of text
#'
#' @return shortened form of .data
#'
shorten <- function(.data, n){
    ifelse(nchar(.data) > n,
           paste(substr(.data, 1, n), "...", sep = ""),
           .data)
}