R/function_sumryList.R

In mcmcOutput: Functions to Store, Manipulate and Display Markov Chain Monte Carlo (MCMC) Output

# Function to create a summary in list-of-lists format so that foo$mean$theta
#   works and gives output with appropriate structure if theta is an array.

sumryList <- function(object, median=TRUE,  mode=FALSE,
    CRItype=c("hdi", "symmetrical", "none"),
    CRImass=0.95, Rhat=TRUE, MCEpc = TRUE, n.eff=FALSE,
    overlap0=FALSE, f=FALSE, ...) {

  if(!inherits(object, "mcmcOutput"))
    stop("The first argument must an object of class 'mcmcOutput'.", call.=FALSE)

  CRItype <- match.arg(CRItype)
  sumtab <- summarise(object, median=median, mode=mode, CRItype=CRItype,
    CRImass=CRImass, Rhat=Rhat, MCEpc = MCEpc, n.eff=n.eff,
    overlap0=overlap0, f=f, ...)

  sl <- attr(object, "simsList")
  D <- lapply(sl, dim)
  # function to do 1 column of sumtable
  do1stat <- function(x1) {
    t1 <- lapply(sl, function(x) x1[x])
    for(i in seq_along(t1))
      dim(t1[[i]]) <- D[[i]]
    return(t1)
  }
  return(lapply(sumtab, do1stat))
}