R/grf_pair_adv.R

In daltoolbox: Leveraging Experiment Lines to Data Analytics

#'@title Plot advanced scatter matrix
#'@description Scatter matrix with class coloring and manual palette application.
#'@param data data.frame
#'@param cnames column names to include
#'@param title optional title
#'@param clabel optional class label column name
#'@param colors optional vector of colors for classes
#'@return returns a ggplot2::ggplot graphic
#'@examples
#'data(iris)
#'grf <- plot_pair_adv(iris, cnames = colnames(iris)[1:4], title = "Iris")
#'print(grf)
#'@export
plot_pair_adv <- function(data, cnames, title = NULL, clabel = NULL, colors = NULL) {
  if (!requireNamespace("GGally", quietly = TRUE)) {
    stop("plot_pair_adv requires the 'GGally' package. Install with install.packages('GGally').")
  }

  plot_data <- as.data.frame(data)
  if (!is.null(clabel)) {
    if (!clabel %in% colnames(plot_data)) {
      stop("plot_pair_adv: 'clabel' is not a valid column in 'data'.")
    }
    plot_data$clabel <- plot_data[[clabel]]
    cnames <- c(cnames, "clabel")
  }
  icol <- match(cnames, colnames(plot_data))
  icol <- icol[!is.na(icol)]
  if (length(icol) == 0) {
    stop("plot_pair_adv: no valid columns in 'cnames'.")
  }

  if (!is.null(clabel)) {
    grf <- GGally::ggpairs(
      plot_data,
      columns = icol,
      mapping = ggplot2::aes(colour = clabel, alpha = 0.4),
      progress = FALSE
    ) + ggplot2::theme_bw(base_size = 10)
    if (!is.null(colors)) {
      for (i in 1:grf$nrow) {
        for (j in 1:grf$ncol) {
          grf[i, j] <- grf[i, j] + ggplot2::scale_fill_manual(values = colors) + ggplot2::scale_color_manual(values = colors)
        }
      }
    }
  } else {
    grf <- GGally::ggpairs(plot_data, columns = icol, progress = FALSE) + ggplot2::theme_bw(base_size = 10)
  }
  if (!is.null(title)) grf <- grf + ggplot2::ggtitle(title)
  grf
}