R/persp.multi_tpfit.R

In spMC: Continuous-Lag Spatial Markov Chains

persp.multi_tpfit <-
function(x, mpoints, which.dire, max.dist, main, mar, ask = TRUE, col = "white", ...) {
  # 3D-Plot transition probabilities matrices 2D
  # through ellispoidal interpolation
  #
  #          x multi_tpfit object
  #    mpoints number of points per axes
  # which.dire two choosen 1D directions
  #   max.dist vector of maximum distances
  #       main title string
  #        mar vector to pass to par()  
  #        ask boolean to pass to par()
  #        col vector of colors ordered from 1 to 0
  #        ... other args to pass to persp()

  if(!is.multi_tpfit(x)) stop("argument \"x\" must be a 'multi_tpfit' object.")

  nc <- length(x$coefficients)
  nk <- dim(x$coefficients[[1]]$coefficients)[1]

  if (missing(main) || !is.character(main)) main <- "Multidimensional transiogram"
  if (missing(mpoints)) stop("\"mpoints\" is missing")
  if (length(mpoints) != 1) mpoints <- mpoints[1]
  all <- FALSE
  if (missing(which.dire)) all <- TRUE
  if (missing(max.dist)) max.dist <- 1
  if (length(max.dist) == 1) max.dist <- rep(max.dist, nc)
  if (length(max.dist) != nc) 
    stop("\"max.dist\" must be a scalar or vector with the same dimension of coordinates")
  max.dist <- abs(max.dist)

  rawLags <- sapply(max.dist, function(h)
    seq(-h, h, length = mpoints)
  )

  which.dire <- if (all) combn(1:nc, 2) else matrix(which.dire)
  nimg <- dim(which.dire)
  if (nimg[1] != 2) stop("wrong length of \"which.dire\"")

  ly <- matrix(c(rep(1, nk), nk^2+(1:nk)+2, 2:(nk^2+1), rep(0, nk), rep(nk^2+2, nk)), ncol = nk, byrow = TRUE)
  yl <- c(rep(0, nk + 4))
  yl1 <- c(rep(0, 2), (nk+1)*nk+2+(1:nk), rep(0, 2))
  ly <- cbind(yl, yl1, ly, yl, yl)
  widths <- c(rep(4 - 0.5 * nk, 2), rep(30/nk, nk), rep(4 - 0.5 * nk, 2)) / 4
  heights <- c(0.75, 1/3.5, rep(7.5/nk, nk), 2/3.5, 1)
  ly <- layout(ly, widths = widths, heights = heights, respect = TRUE)

  nomi <- names(x$prop)
  if (missing(mar) || !is.numeric(mar)) {
    mar <- sum(c(0, 1, 0, -0.5, -0.2, rep(0, 4), 0.2, rep(c(-0.1, rep(0, 3)), 2), 0)[1:nk])
    mar <- rep(mar, 4)
  }
  else {
    if (length(mar) < 4) {
      mar <- rep(mar[1], 4)
    }
    else {
      mar <- mar[1:4]
    }
  }

  for (i in 1:nimg[2]) {
    lagsMat <- as.list(rep(0, nc))
    lagsMat[[which.dire[1, i]]] <- rawLags[, which.dire[1, i]]
    lagsMat[[which.dire[2, i]]] <- rawLags[, which.dire[2, i]]
    lagsMat <- as.matrix(expand.grid(lagsMat))
    Tprobs <- predict(x, lagsMat)
    if (ask && all && (nimg[2] > 1)) {
      devAskNewPage(TRUE)
      on.exit(devAskNewPage(FALSE))
    }

    par(mar = c(0.3, 0.1, 0.3, 0.1))
    plot.new()
    title <- paste(main, " (", x$coordsnames[which.dire[1, i]], ", ", x$coordsnames[which.dire[2, i]], ")", sep = "")
    text(0.5, 0.5, labels = title, cex = 2)

    par(mar = mar)
    for (j in 1:nk) {
      for (k in 1:nk) {
        zzz <- t(matrix(rev(Tprobs$Tmat[j, k, ]), mpoints, mpoints))
        colid <- 1
        if (length(col) > 1) {
          colth <- seq(1 / length(col), 1, length.out = length(col))
          colid <- sapply(zzz, function(xx) sum(xx < colth))
          dim(colid) <- dim(zzz)
          colid <- colid[-1, -1]
          zlim <- range(zzz, na.rm = TRUE)
        }
        else {
          zlim <- 0:1
        }

        persp(rawLags[, which.dire[1, i]], rawLags[, which.dire[2, i]],
              zzz, col = rev(col)[colid], zlab = "", zlim = zlim,
              xlab = x$coordsnames[which.dire[1, i]], 
              ylab = x$coordsnames[which.dire[2, i]], ...) #sub = paste(nomi[j], "to", nomi[k])

#        box()
#        if (k == 1) axis(2)
#        if (k == nk) axis(4, 0, labels = nomi[j], tick = FALSE, font = 3)
#        if (j == nk) axis(1)
#        if (j == 1) axis(3, 0, labels = nomi[k], tick = FALSE, font = 3)
      }
    }

    par(mar = c(2.2, 0.1, 1.2, 0.1))
    if (length(col) > 1) {
      image(matrix(colth, ncol = 1), col = col, main="Legend", axes = FALSE)
      box()
      axis(1)
    }
    else {
      plot.new()
    }
    par(mar = c(0.1, 0.1, 0.1, 0.1))
    for (j in c(0, 90)) {
      for (lbls in nomi) {
        plot.new()
        text(0.5, 0.5, labels = lbls, srt = j, font = 3)
      }
    }
  }

}