R/Heatmap_Legend.R

In James-Thorson/spatial_condition_factor: Package for estimating spatial and spatiotemporal variation in individual condition.

Heatmap_Legend <-
function( colvec, heatrange, margintext=NULL ){
  par( xaxs="i", yaxs="i", mar=c(1,0,1,2+ifelse(is.null(margintext),0,1.5)), mgp=c(1.5,0.25,0), tck=-0.02 )
  N = length(colvec)
  Y = seq(heatrange[1], heatrange[2], length=N+1)
  plot( 1, type="n", xlim=c(0,1), ylim=heatrange, xlab="", ylab="", main="", xaxt="n", yaxt="n", cex.main=1.5)
  for( i in 1:N) polygon( x=c(0,1,1,0), y=Y[c(i,i,i+1,i+1)], col=colvec[i], border=NA)
  axis(side=4, at=pretty(heatrange) )
  if(!is.null(margintext)) mtext(side=4, text=margintext, line=2, cex=1.5)
  #mtext(side=1, outer=TRUE, line=1, "Legend")
}