R/raincloud_1x1.R

In jorvlan/openvis: Create visualizations based on best open science practices

#' 1 x 1 comparison
#'
#'
#' This function visualizes a 1x1 repeated measures raincloud.
#' @name raincloud_1x1
#' @param data_1x1 <data.frame> the array of datapoints to be plotted
#' @param line_color <string> color lines
#' @param line_alpha <numeric> alpha lines
#' @param size <numeric> data size
#' @param alpha <numeric> data alpha
#' @importFrom ggplot2 ggplot aes geom_point geom_line position_nudge scale_x_continuous theme_classic ggsave
#' @importFrom gghalves geom_half_boxplot geom_half_violin
#' @importFrom magrittr %>%
#' @importFrom dplyr group_by mutate
#' @return figure_1x1_horizontal
#'
#'
#' @examples
#' \dontrun{
#' # Using an example dataset
#'raincloud_1x1 <- function(data_1x1,
#'                          line_color = 'lightgray',
#'                          line_alpha = .3,
#'                          size = 1.5,
#'                          alpha = .6,
#'                          ort = 'v')
#'
#'}
#'@export

globalVariables(c("ggplot", "aes", "geom_point", "geom_line",
                  "geom_half_boxplot", "geom_half_violin",
                  "position_nudge", "%>%", "position"))

raincloud_1x1 <- function(data_1x1,
                          line_color = 'lightgray',
                          line_alpha = .3,
                          size = 1.5,
                          alpha = .6,
                          ort = 'v') {

if (ort == 'v') {
figure_1x1_vertical <- ggplot(data = data_1x1) +

      #Add geom_() objects
      geom_point(data = data_1x1 %>% dplyr::filter(x_axis =="1"), aes(x = jit, y = y_axis, fill = x_axis, color = x_axis), alpha = alpha, show.legend = FALSE) +
      geom_point(data = data_1x1 %>% dplyr::filter(x_axis =="2"), aes(x = jit, y = y_axis, fill = x_axis, color = x_axis), alpha = alpha, show.legend = FALSE) +

      geom_half_violin(
        data = data_1x1 %>% dplyr::filter(x_axis=="1"),aes(x = x_axis, y = y_axis, color = x_axis, fill = x_axis), position = position_nudge(x = .35),
        side = "r", alpha = alpha, show.legend = FALSE) +

      geom_half_violin(
        data = data_1x1 %>% dplyr::filter(x_axis=="2"),aes(x = x_axis, y = y_axis, color = x_axis, fill = x_axis), position = position_nudge(x = .35),
        side = "r", alpha = alpha, show.legend = FALSE) +

      geom_half_boxplot(
        data = data_1x1 %>% dplyr::filter(x_axis=="1"), aes(x=x_axis, y = y_axis, color = x_axis, fill = x_axis), position = position_nudge(x = .2),
        side = "r",outlier.shape = NA, center = TRUE, errorbar.draw = FALSE, width = .2, alpha = alpha, show.legend = FALSE) +

      geom_half_boxplot(
        data = data_1x1 %>% dplyr::filter(x_axis=="2"), aes(x=x_axis, y = y_axis, color = x_axis, fill = x_axis), position = position_nudge(x = .2),
        side = "r",outlier.shape = NA, center = TRUE, errorbar.draw = FALSE, width = .2, alpha = alpha, show.legend = FALSE)

return(figure_1x1_vertical)

} else if (ort == 'h') {

  figure_1x1_horizontal <- ggplot(data = data_1x1) +

    #Add geom_() objects
    geom_point(data = data_1x1 %>% dplyr::filter(x_axis=="1"), aes(x = jit, y = y_axis, fill = x_axis, color = x_axis), alpha = alpha, show.legend = FALSE) +
    geom_point(data = data_1x1 %>% dplyr::filter(x_axis=="2"), aes(x = jit, y = y_axis, fill = x_axis, color = x_axis), alpha = alpha, show.legend = FALSE) +

    geom_flat_violin(data = data_1x1 %>% dplyr::filter(x_axis=="1"),aes(x = x_axis, y = y_axis, color = x_axis, fill = x_axis),
                     position = position_nudge(x = .35), alpha = alpha, show.legend = FALSE) +
    geom_flat_violin(data = data_1x1 %>% dplyr::filter(x_axis=="2"),aes(x = x_axis, y = y_axis, color = x_axis, fill = x_axis),
                     position = position_nudge(x = .35), alpha = alpha, show.legend = FALSE) +
    geom_half_boxplot(
      data = data_1x1 %>% dplyr::filter(x_axis=="1"), aes(x=x_axis, y = y_axis, color = x_axis, fill = x_axis), position = position_nudge(x = .2),
      side = "r",outlier.shape = NA, center = TRUE, errorbar.draw = FALSE, width = .2, alpha = alpha, show.legend = FALSE) +

    geom_half_boxplot(
      data = data_1x1 %>% dplyr::filter(x_axis=="2"), aes(x=x_axis, y = y_axis, color = x_axis, fill = x_axis), position = position_nudge(x = .2),
      side = "r",outlier.shape = NA, center = TRUE, errorbar.draw = FALSE, width = .2, alpha = alpha, show.legend = FALSE) +

    coord_flip()

  return(figure_1x1_horizontal)

}
}