R/plot.covfm.q

In robust: Port of the S+ "Robust Library"

plot.covfm <- function(x, which.plots = c(4, 3, 5), ...)
{
  n.models <- length(x)
  mod.names <- names(x)

  choices <- c("All",
               "Eigenvalues of Covariance Estimate", 
               "Mahalanobis Distances",
               "Ellipses Plot")

  if(n.models == 2)
    choices <- c(choices, "Distance - Distance Plot")
  else
    which.plots <- which.plots[which.plots != 5]

  all.plots <- 2:length(choices)

  tmenu <- paste("plot:", choices)

  if(is.numeric(which.plots)) {
    if(!all(which.plots %in% all.plots))
      stop(sQuote("which"), " must be in 2:", length(choices))

    if(length(which.plots) == 0)
      return(invisible(x))

    if(length(which.plots) > 1) {
      par.ask <- par(ask = TRUE)
      on.exit(par(ask = par.ask))
    }

    ask <- FALSE
    which.plots <- c(which.plots + 1, 1)
  }

  else if(which.plots == "all") {
    which.plots <- c(all.plots + 1, 1)
    ask <- FALSE
    par.ask <- par(ask = TRUE)
    on.exit(par(ask = par.ask))
  }

  else
    ask <- TRUE

  repeat {
    if(ask) {
      which.plots <- menu(tmenu,
                          title = "\nMake plot selections (or 0 to exit):\n")

      if(any(which.plots == 1)) {
        which.plots <- c(all.plots, 0)
        par.ask <- par(ask = TRUE)
        on.exit(par(ask = par.ask))
      }

    which.plots <- which.plots + 1
    }

    if(!length(which.plots))
      stop(paste("Invalid choice of plot in \'which.plots\'"))

    for(pick in which.plots) {
      switch(pick,
        return(invisible(x)),
        place.holder <- 1,

        screePlot.covfm(x,
                        xlab = "Principal Component",
                        ylab = "Variances",
                        main = "Screeplot",
                        ...),

        distancePlot.covfm(x,
                           xlab = "Index",
                           ylab = "Mahalanobis Distance",
                           pch = 16,
                           ...),

        ellipsesPlot.covfm(x, ...),
        
        {
          strip <- paste(mod.names[1], "Distance vs.", mod.names[2], "Distance")
          ddPlot.covfm(x,
                       strip = strip,
                       xlab = paste(mod.names[2], "Distance"),
                       ylab = paste(mod.names[1], "Distance"),
                       main = "Distance-Distance Plot",
                       pch = 16,
                       ...)
        }
      )
    }
  }
  invisible(x)
}