R/funnel-web-data.R

In animbook: Visualizing Changes in Performance Measures and Demographic Affiliations using Animation

#' Funnel web spider plot data
#'
#' This function performs data manipulation for facetting.
#'
#' @param data The categorized data.
#'
#' @return An object contained the modified data with additional data components.
#'
#' @details The function takes the categorized data and manipulates the data into the
#' format where it can be further facetting.
#'
#' @keywords internal

funnel_web_spider_data <- function(data) {


  # check -------------------------------------------------------------------

  stopifnot("Please use the prep function to convert the data into a categorized format" =
              any("categorized" %in% class(data)))

  class(data) <- c("tbl_df", "tbl", "data.frame")

  x <- unique(data$time)


  # Change the data format --------------------------------------------------

  start <- data |>
    dplyr::filter(time > min(time)) |>
    dplyr::group_by(id) |>
    dplyr::mutate(facet = dplyr::row_number())

  end <- data |>
    dplyr::filter(time < max(time)) |>
    dplyr::group_by(id) |>
    dplyr::mutate(facet = dplyr::row_number())

  full <- rbind(start, end) |>
    dplyr::arrange(id, time)

  object <- list(data = full,
                 xbreaks = sort(x))

  return(object)
}