R/averageLayout.R

In qgraph: Graph Plotting Methods, Psychometric Data Visualization and Graphical Model Estimation

# Function averages the layout of multiple graphs:
averageWmat <- function(...)
{
  dotList <- list(...)
  
  # Get W mats:
  Wmats <- lapply(dotList,getWmat)
  
  # Replace list with averaged Wmats, and rescale:
  for (i in seq_along(Wmats))
  {

    if (is.list(Wmats[[i]]))
    {
      Wmats[[i]] <- do.call(averageWmat,Wmats[[i]])
    }
    if (!all(Wmats[[i]]==0)){
      Wmats[[i]] <- abs(Wmats[[i]]/max(abs(Wmats[[i]])))
    } else  Wmats[[i]][] <- 0
  }
  
  if (!(length(unique(sapply(Wmats,nrow))) == 1 | length(unique(sapply(Wmats,ncol))) == 1 )) stop("Graphs of different dimensions")
  avgWmat <- Reduce('+',Wmats)/length(Wmats)
  return(avgWmat)
  
}

averageLayout <- function(..., layout = "spring", repulsion = 1, layout.par)
{
  avgWmat <- averageWmat(...)
  
  if (missing(layout.par)){
    layout.par <- list(repulse.rad = ncol(avgWmat)^(repulsion * 3))
  }
  
  Q <- qgraph(avgWmat, DoNotPlot = TRUE, layout = layout, layout.par = layout.par)
  
  return(Q$layout)
}