R/cas_show_ts.R

In giocomai/castarter: Content Analysis Starter Toolkit

#' Create dygraphs based on a data frame typically generated with cas_count()
#'
#' @param count_df
#'
#' @return
#' @export
#'
#' @examples
#'
#' count_df <- castarter::cas_count(
#'   corpus = castarter::cas_demo_corpus,
#'   words = c("russia", "moscow")
#' ) %>%
#'   cas_summarise(before = 15, after = 15)
#' cas_show_ts_dygraph(count_df)
cas_show_ts_dygraph <- function(count_df,
                                date_column_name = date,
                                n_column_name = n,
                                pattern_column_name = pattern,
                                range_selector = TRUE) {
  if (is.data.frame(count_df) == FALSE) {
    return(NULL)
  }
  xts_df <- count_df %>%
    tidyr::pivot_wider(
      id_cols = {{ date_column_name }},
      names_from = {{ pattern_column_name }},
      values_from = {{ n_column_name }}
    ) %>%
    tbl2xts::tbl_xts()

  dy <- xts_df %>%
    dygraphs::dygraph()

  if (isTRUE(range_selector)) {
    dy <- dy %>%
      dygraphs::dyRangeSelector()
  }
  dy
}