R/example_data.R

In CRI-iAtlas/iatlas.modules2: iAatlas Modules2

utils::globalVariables("iris")

#' Example Starwars Data
example_starwars_data <- function(){
  dplyr::starwars %>%
    dplyr::select(
      "sample" = "name",
      "group" = "species",
      "height",
      "mass"
    ) %>%
    tidyr::pivot_longer(
      -c("sample", "group"), names_to = "feature", values_to = "feature_value"
    )
}

example_starwars_data_func <- function(.feature_class){
  example_starwars_data()
}

#' Example Iris Data
#' @importFrom magrittr %>%
example_iris_data <- function(){
  iris %>%
    dplyr::as_tibble() %>%
    dplyr::mutate("sample" = as.character(1:dplyr::n())) %>%
    tidyr::pivot_longer(
      !c("Species", "sample"),
      names_to = "feature",
      values_to = "feature_value"
    ) %>%
    dplyr::rename("group" = "Species") %>%
    dplyr::inner_join(
      dplyr::tribble(
        ~group,       ~color,
        "setosa",     "#FF0000",
        "versicolor", "#0000FF",
        "virginica",  "#FFFF00"
      ),
      by = "group"
    ) %>%
    dplyr::mutate(
      "group_description" = stringr::str_c("Iris Species: ", .data$group),
    ) %>%
    dplyr::inner_join(
      dplyr::tribble(
        ~feature_class, ~feature,        ~feature_class2, ~feature_display,
        "Length",       "Sepal.Length",  "Sepal",         "Sepal Length",
        "Width",        "Sepal.Width",   "Sepal",         "Sepal Width",
        "Length",       "Petal.Length",  "Petal",         "Petal Length",
        "Width",        "Petal.Width",   "Petal",         "Petal Width"
      ),
      by = "feature"
    )
}

example_iris_data_func <- function(.feature_class = NULL, .feature = NULL){
  iris_data <- example_iris_data()
  if (!is.null(.feature_class)){
    iris_data <- dplyr::filter(iris_data, .data$feature_class == .feature_class)
  }
  if (!is.null(.feature)){
    iris_data <- dplyr::filter(iris_data, .data$feature == .feature)
  }
  dplyr::select(iris_data, -"feature_class")
}