Skip to content

Instantly share code, notes, and snippets.

@RobertMyles
Last active February 8, 2017 14:32
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save RobertMyles/5288f15c217c402057ec4e6bd74ed9b1 to your computer and use it in GitHub Desktop.
Save RobertMyles/5288f15c217c402057ec4e6bd74ed9b1 to your computer and use it in GitHub Desktop.
Modify R package BEST's plotPost function to include other font family for Brazilian Political Science Review research note
# modified by R. McDonnell. Original at: https://github.com/cran/BEST/blob/master/R/plotPost.R
# Original code by John Kruschke, modified by Mike.
plotPost_2 <-
function( paramSampleVec, credMass=0.95, compVal=NULL, ROPE=NULL,
HDItextPlace=0.7, showMode=FALSE, showCurve=FALSE, ... ) {
# Does a plot for a single parameter. Called by plot.BEST but also exported.
# Returns a histogram object invisibly.
# This stuff should be in the ... argument:
# yaxt="n", ylab="", xlab="Parameter", main="", cex.lab=1.5, cex=1.4,
# xlim=range(compVal, paramSampleVec), col="skyblue", border="white",
# breaks=NULL
# Deal with ... argument:
dots <- list(...)
if(length(dots) == 1 && class(dots[[1]]) == "list")
dots <- dots[[1]]
defaultArgs <- list(xlab=deparse(substitute(paramSampleVec)),
yaxt="n", ylab="", main="", cex.lab=1.5,
cex=1.4, col="skyblue", border="white", bty="n", lwd=5, freq=FALSE,
family = "Cambria",
xlim=range(compVal, hdi(paramSampleVec, 0.99)))
useArgs <- modifyList(defaultArgs, dots)
# Get breaks argument
breaks <- dots$breaks
if (is.null(breaks)) {
if (all(paramSampleVec == round(paramSampleVec))) { # all integers
breaks <- seq(min(paramSampleVec), max(paramSampleVec) + 1) - 0.5
} else {
by <- diff(hdi(paramSampleVec))/18
breaks <- unique(c( seq( from=min(paramSampleVec), to=max(paramSampleVec),
by=by), max(paramSampleVec) ))
}
}
histinfo <- hist(paramSampleVec, breaks=breaks, plot=FALSE)
histinfo$xname <- useArgs$xlab
oldpar <- par(xpd=TRUE) ; on.exit(par(oldpar))
if (showCurve) {
densCurve <- density( paramSampleVec, adjust=2 )
selPlot <- names(useArgs) %in%
c(names(as.list(args(plot.default))), names(par(no.readonly=TRUE)))
plotArgs <- useArgs[selPlot]
plotArgs$x <- densCurve$x
plotArgs$y <- densCurve$y
plotArgs$type <- "l"
plotArgs$xpd <- FALSE
do.call(plot, plotArgs, quote=TRUE)
abline(h=0, col='grey', xpd=FALSE)
# Display the HDI.
if(!is.null(credMass)) {
HDI <- hdi(densCurve, credMass, allowSplit=TRUE)
ht <- attr(HDI, "height")
segments(HDI[, 1], ht, HDI[, 2], ht, lwd=4, lend='butt')
segments(HDI, 0, HDI, ht, lty=2)
text( mean(HDI), ht, bquote(.(100*credMass) * "% HDI" ),
adj=c(.5,-1.7), cex=useArgs$cex , family = "Cambria")
text( HDI, ht, bquote(.(signif(HDI, 3))),
pos=3, cex=useArgs$cex, family = "Cambria")
}
} else {
plot.histogram.args.names <- c("freq", "density", "angle", "border",
"main", "sub", "xlab", "ylab", "xlim", "ylim", "axes", "labels",
"add", "family") # plot.histogram not exported, so need to cheat!
selPlot <- names(useArgs) %in%
c(plot.histogram.args.names, names(par(no.readonly=TRUE)))
plotArgs <- useArgs[selPlot]
plotArgs$lwd <- 1
plotArgs$x <- histinfo
do.call(plot, plotArgs, quote=TRUE)
# Display the HDI.
if(!is.null(credMass)) {
HDI <- hdi( paramSampleVec, credMass )
lines(HDI, c(0,0), lwd=4, lend='butt')
text( mean(HDI), 0, bquote(.(100*credMass) * "% HDI" ),
adj=c(.5,-1.7), cex=useArgs$cex, family = "Cambria" )
text( HDI[1], 0, bquote(.(signif(HDI[1],3))),
adj=c(HDItextPlace,-0.5), cex=useArgs$cex, family = "Cambria" )
text( HDI[2], 0, bquote(.(signif(HDI[2],3))),
adj=c(1.0-HDItextPlace,-0.5), cex=useArgs$cex, family = "Cambria" )
}
}
# Display mean or mode:
cenTendHt <- 0.9 * max(histinfo$density)
if ( showMode==FALSE ) {
meanParam <- mean( paramSampleVec )
text( meanParam, cenTendHt, family = "Cambria",
bquote(mean==.(signif(meanParam,3))), adj=c(.5,0), cex=useArgs$cex )
} else {
dres <- density( paramSampleVec )
modeParam <- dres$x[which.max(dres$y)]
text( modeParam, cenTendHt, family = "Cambria",
bquote(mode==.(signif(modeParam,3))), adj=c(.5,0), cex=useArgs$cex )
}
# Display the comparison value.
if ( !is.null( compVal ) ) {
cvHt <- 0.7 * max(histinfo$density)
cvCol <- "darkgreen"
pcgtCompVal <- round( 100 * sum( paramSampleVec > compVal )
/ length( paramSampleVec ) , 1 )
pcltCompVal <- 100 - pcgtCompVal
lines( c(compVal,compVal), c(0.96*cvHt,0),
lty="dotted", lwd=1, col=cvCol )
text( compVal, cvHt, family = "Cambria",
bquote( .(pcltCompVal)*"% < " *
.(signif(compVal,3)) * " < "*.(pcgtCompVal)*"%" ),
adj=c(pcltCompVal/100,0), cex=0.8*useArgs$cex, col=cvCol )
}
# Display the ROPE.
if ( !is.null( ROPE ) ) {
ROPEtextHt <- 0.55 * max(histinfo$density)
ropeCol <- "darkred"
pcInROPE <- ( sum( paramSampleVec > ROPE[1] & paramSampleVec < ROPE[2] )
/ length( paramSampleVec ) )
lines( c(ROPE[1],ROPE[1]), c(0.96*ROPEtextHt,0), lty="dotted", lwd=2,
col=ropeCol )
lines( c(ROPE[2],ROPE[2]), c(0.96*ROPEtextHt,0), lty="dotted", lwd=2,
col=ropeCol)
text( mean(ROPE), ROPEtextHt, family = "Cambria",
bquote( .(round(100*pcInROPE))*"% in ROPE" ),
adj=c(.5,0), cex=1, col=ropeCol )
}
return(invisible(histinfo))
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment