R/to_overlay_data.R

In frhl/genoppi-v4: genoppi

#' @title to overlay data
#' @description Adds columns to a data.frame (e.g. one used for overlay)
#' that are required for downstream functionality in genoppi/
#' @param df a data.frame
#' @param dataset optional string, if dataset is not provided in df.
#' @export
#' @family interactive

to_overlay_data <- function(df, dataset=NULL) {
  cnames = colnames(df)
  # if the following columns are not specified in the reference
  # they are set to the default in this function.
  if ('dataset' %nin% cnames) df$dataset = dataset
  if ('label' %nin% cnames) df$label <- TRUE
  if ('stroke' %nin% cnames) df$stroke <- TRUE
  if ('col_significant' %nin% cnames) df$col_significant <- 'yellow'
  if ('col_other' %nin% cnames) df$col_other <- 'grey'
  if ('alt_label' %nin% cnames) df$alt_label <- NA
  if ('label_size' %nin% cnames) df$label_size = 3
  if ('pLI' %nin% cnames) df$pLI <- NA
  if ('shape' %nin% cnames) df$shape <- 21
  if ('symbol' %nin% cnames) df$symbol <- 'circle'
  if ('opacity' %nin% cnames) df$opacity <- 1
  if ('gg.size' %nin% cnames) df$gg.size <- 3.5
  if ('gene' %nin% cnames) df$gene <- NA
  if ('size' %nin% cnames) df$size <- 9
  if ('legend_order' %nin% cnames) df$legend_order <- NA 
  return(as.data.frame(df))
}