plot.tree.EJN <- function (x, y = NULL, type = c("proportional", "uniform"),
                       scale = 1,...) {
   if (inherits(x, "singlenode"))
      stop("cannot plot singlenode tree")
   if (!inherits(x, "tree"))
      stop("not legitimate tree")
   type <- match.arg(type)
   uniform <- type == "uniform"
   dev <- dev.cur()
   if (dev == 1)
      dev <- 2
   assign(paste(".Tree.unif", dev, sep = "."), uniform, envir = .GlobalEnv)
#EJN>>> Tell the environment the current scale
   assign(paste(".Tree.scale", dev, sep = "."), scale, envir=.GlobalEnv)
#<<<EJN
   invisible(treepl.EJN(treeco(x), node = as.numeric(row.names(x$frame)), ...))
}

treepl.EJN <- function (xy, node, erase = FALSE,
                        scale = paste(".Tree.scale", dev.cur(), sep = "."), ...) {
   x <- xy$x
   y <- xy$y
#EJN>>>  Get the current scale
   if(exists(scale)) scale <- get(scale)
   else scale <- 1
#<<<EJN
   parent <- match((node%/%2), node)
   sibling <- match(ifelse(node%%2, node - 1, node + 1), node)
   xx <- rbind(x, x, x[sibling], x[sibling], NA)
   yy <- rbind(y, y[parent], y[parent], y[sibling], NA)
   if (any(erase)) {
      lines(c(xx[, erase]), c(yy[, erase]), col = par("bg"))
      return(x = x[!erase], y = y[!erase])
   }
#EJN>>> Apply the scale
   x.range <- range(x)/scale
   y.range <- range(y)/scale
   plot(x.range - diff(x.range)*(1 - scale)/2, y.range + diff(y.range)*(1 - scale)/2,
         type = "n", axes = FALSE, xlab = "", ylab = "")
#<<<EJN
#Orig   plot(range(x), range(y), type = "n", axes = FALSE, xlab = "",
#Orig        ylab = "")

   text(x[1], y[1], "|", ...)
   lines(c(xx[, -1]), c(yy[, -1]), ...)
   list(x = x, y = y)
}

treeco <- function (tree, uniform = paste(".Tree.unif", dev.cur(), sep = ".")) {
   frame <- tree$frame
   node <- as.numeric(row.names(frame))
   depth <- tree.depth(node)
   x <- -depth
   if (exists(uniform))
      uniform <- get(uniform)
   else uniform <- 0
   if (uniform)
      y <- x
   else {
      y <- dev <- frame$dev
      depth <- split(seq(node), depth)
      parent <- match(node%/%2, node)
      sibling <- match(ifelse(node%%2, node - 1, node + 1), node)
      for (i in depth[-1])
         y[i] <- y[parent[i]] - dev[parent[i]] + dev[i] + dev[sibling[i]]
   }
   depth <- -x
   leaves <- frame$var == "<leaf>"
   x[leaves] <- seq(sum(leaves))
   depth <- split(seq(node)[!leaves], depth[!leaves])
   left.child <- match(node * 2, node)
   right.child <- match(node * 2 + 1, node)
   for (i in rev(depth)) x[i] <- 0.5 * (x[left.child[i]] + x[right.child[i]])
   list(x = x, y = y)
}

tree.depth <- function (nodes) {
   depth <- floor(log(nodes, base = 2) + 1e-07)
   as.vector(depth - min(depth))
}