R/riverbank.R

In cognopod/walter: Functions and Visualizations for Factor Analysis

#' riverbank
#'
#' A function to create edges and node for riverplot.
#'
#' @param factorlist A list of factors from \code{watershed()}
#' @return A list of components
#'
riverbank <- function(factorlist) {
  #palette <- paste0(RColorBrewer::brewer.pal(ncol(factorlist), "Set1"), "90")
  palette <- paste0(viridis::plasma(ncol(factorlist), alpha = 0.5625))
  for (this.col in 2:ncol(factorlist)){
    if (this.col == 2) {
      EDGES <- as.data.frame(table(factorlist[,(this.col-1)], factorlist[,this.col]))
    } else {
      EDGES <- rbind(EDGES, as.data.frame(table(factorlist[,(this.col-1)], factorlist[,this.col])))
    }
  }
  colnames(EDGES) <- c("N1", "N2", "Value")
  for (this.col in 1:ncol(factorlist)){
    if (this.col == 1) {
      NODES <- data.frame(ID = unique(factorlist[,this.col]), x = this.col)
    } else {
      NODES <- rbind(NODES, data.frame(ID = unique(factorlist[,this.col]), x = this.col))
    }
  }
  colnames(NODES) <- c("ID", "x")
  NODES <- na.omit(NODES)
  NODES <- NODES[order(NODES$ID),]
  NODES$col <- palette[NODES$x]
  if (ncol(factorlist) < 3) {
    textpalette <- c("white", "black")
  } else {
    textpalette <- c("white", "white", rep("black", ncol(factorlist) - 2))
  }
  NODES$textcol <- textpalette[NODES$x]
  #NODES$x <- as.numeric(levels(NODES$x))[NODES$x]
  return(list("NODES" = NODES, "EDGES" = EDGES))
}