FreesearchR/R/plot_violin.R

#' Beatiful violin plot
#'
#' @returns ggplot2 object
#' @export
#'
#' @name data-plots
#'
#' @examples
#' mtcars |> plot_violin(pri = "mpg", sec = "cyl", ter = "gear")
plot_violin <- function(data, pri, sec, ter = NULL) {
  if (!is.null(ter)) {
    ds <- split(data, data[ter])
  } else {
    ds <- list(data)
  }

  out <- lapply(ds, \(.ds){
    rempsyc::nice_violin(
      data = .ds,
      group = sec,
      response = pri,
      xtitle = get_label(data, var = sec),
      ytitle = get_label(data, var = pri)
    )
  })

  wrap_plot_list(out)
  # patchwork::wrap_plots(out,guides = "collect")
}