R/visregPlot.R

In visreg: Visualization of Regression Models

visregPlot <- function(v, partial, rug, band, whitespace, top, line.par, fill.par, points.par, ...) {
  ## Setup
  x <- v$res[, v$meta$x]
  y <- v$res$visregRes
  xx <- v$fit[, v$meta$x]
  yy <- v$fit$visregFit
  lwr <- v$fit$visregLwr
  upr <- v$fit$visregUpr
  xlim <- if (is.factor(xx)) c(0, 1) else range(xx)
  ylab <- if (is.null(v$meta$yName)) paste("f(", v$meta$x, ")", sep="") else v$meta$yName

  if (partial && sum(!is.na(y))>0) {
    ylim <- range(c(y, lwr, upr), na.rm=TRUE)
  } else if (band) {
    ylim <- range(c(yy, lwr, upr), na.rm=TRUE)
  } else {
    ylim <- range(yy)
  }
  plot.args <- list(x=1, y=1, ylim=ylim, xlab=v$meta$x, ylab=ylab, type="n", xlim=xlim, xaxt=ifelse(is.factor(xx),'n','s'), las=1)
  new.args <- list(...)
  if (length(new.args)) plot.args[names(new.args)] <- new.args
  do.call("plot", plot.args)

  if (is.factor(xx)) {
    factorPlot(v, partial, band, rug, whitespace, top, line.par, fill.par, points.par, ...)
    if (!("xaxt" %in% names(new.args) && new.args$xaxt=="n")) factorAxis(x, whitespace, new.args)
  } else {
    if (band) {
      fill.args <- list(x=c(xx, rev(xx)), y=c(lwr, rev(upr)), col="gray85", border=F)
      if (length(fill.par)) fill.args[names(fill.par)] <- fill.par
      do.call("polygon", fill.args)
    }
    line.args <- list(x=xx, y=yy, lwd=3, lty=1, col="#008DFFFF")
    if (length(line.par)) line.args[names(line.par)] <- line.par
    if (!partial) {
      do.call("lines", line.args)
    } else {
      points.args <- list(x=x, y=y, pch=19, cex=0.4, col="gray50")
      if (length(points.par)) points.args[names(points.par)] <- points.par
      if (top == 'line') {
        do.call("points", points.args)
        do.call("lines", line.args)
      } else {
        do.call("lines", line.args)
        do.call("points", points.args)
      }
    }
    if (rug==1) rug(x, side=1)
    if (rug==2) {
      rug(x[!v$res$visregPos], side=1)
      rug(x[v$res$visregPos], side=3)
    }
  }
}