R/plot_projection.r

In pop.wolf: Models for Simulating Wolf Populations

#########################################################################
# PLOT POPULATION COUNTS
#########################################################################

plot_projection <- function(projection, title) {

	if (title == "Packs") varname <- "packs"
	if (title == "Pairs") varname <- "pairs"
	if (title == "Reproductions") varname <- "repros"
	if (title == "Population") varname <- "pop_size"

	years <- projection$parameters$years
	runs <- projection$runs

	dmed <- apply(runs[,varname,], 1, median)
	dlci <- apply(runs[,varname,], 1, quantile, 0.025)
	duci <- apply(runs[,varname,], 1, quantile, 0.975)
	dmonths <- 1:nrow(runs)
	dyears <- c(1, (1:projection$parameters$years)*12 + 1)

	plot(0,0, type="l", xaxt="n", xlab="Years", ylab="", xlim=c(1,nrow(runs)), ylim=c(0, max(duci)), main=title)
	polygon(x=c(dmonths, rev(dmonths)), y=c(dlci, rev(duci)), col="gray95", border=NA)
	lines(dmonths, dlci, lty="dashed")
	lines(dmonths, duci, lty="dashed")
	lines(dmonths, dmed, lty="solid")
	points(dyears, dmed[dyears], pch=21, bg="white")
	axis(1, at=2+seq(0,years*12,12), labels=0:years, las=2)
	abline(v=2+seq(0,years*12,12), lty="dotted", col="lightgray")

}