R/post.rpart.R

In henckr/distRforest: Distribution-based Random Forest

##
post.rpart <- function(tree, title.,
		       filename = paste(deparse(substitute(tree)), ".ps", sep = ""),
		       digits = getOption("digits") - 2, pretty = TRUE,
		       use.n = TRUE, horizontal = TRUE, ...)
{
    if (filename != "") {
	postscript(file = filename, horizontal = horizontal, ...)
	par(mar = c(2,2,4,2) + 0.1)
	on.exit(dev.off())
    } else {
	oldpar <- par(mar = c(2,2,4,2) + 0.1)
	on.exit(invisible(par(oldpar)))
    }

    plot(tree, uniform = TRUE, branch = 0.2, compress = TRUE, margin = 0.1)
    text(tree, all = TRUE, use.n = use.n, fancy = TRUE, digits = digits,
         pretty = pretty)

    if (missing(title.)) {
        temp <- attr(tree$terms, "variables")[2L]
        title(paste("Endpoint =", temp), cex = 0.8)
    } else if (nzchar(title.)) title(title., cex = 0.8)
}