R/slimSummary.R

In jmcurran/s20xHelpR:

#' To be documented
#'
#' @export
slimSummary = function(lm.obj,
                       showCall = !showAOVsummary,
                       showResiduals = FALSE,
                       showSignif = FALSE,
                       showAOVsummary = ifelse(class(lm.obj)[1] == "lm", TRUE, FALSE),
                       showGLMsummary = !showAOVsummary,
                       ...){

  Lines = capture.output(summary(lm.obj, ...))

  callBlock = grep("^(Call|lm|glm).*$", Lines)
  resBlock = grep("^Residuals.*$", Lines) + 0:2
  coefTable = list(start = grep("^Coefficients:.*$", Lines),
                   end = grep("^---$", Lines))

  ## special case for constant mean model, e.g. y ~ 1
  if(length(coefTable$end) == 0){
    blanks = which(nchar(Lines) == 0)
    coefTable$end = (blanks[blanks > coefTable$start])[1] ## first blank line after start
  }

  signifLine = grep("^Signif.*$", Lines)
  anovaSummaryBlock = list(start = grep("^Residual standard.*$", Lines),
                           end = grep("^F-statistic.*$", Lines))
  glmSummaryBlock = list(start = grep("^\\(Dispersion parameter.*$", Lines),
                           end = grep("^Residual deviance.*$", Lines))

  ## special case for constant mean model, e.g. y ~ 1
  if(length(anovaSummaryBlock$end) == 0){
    blanks = which(nchar(Lines) == 0)
    anovaSummaryBlock$end = (blanks[blanks > anovaSummaryBlock$start])[1] ## first blank line after start
  }

  if(showCall){
    cat(paste0(Lines[callBlock], collapse = "\n"))
    cat("\n\n")
  }

  if(showResiduals){
    cat(paste0(Lines[resBlock], "\n"))
    cat("\n")
  }

  cat(paste0(Lines[coefTable$start:coefTable$end], collapse = "\n"))
  cat("\n")

  if(showSignif){
    cat(paste0(Lines[signifLine], "\n"))
    cat("\n")
  }

  if(showAOVsummary){
    cat(paste0(Lines[anovaSummaryBlock$start:anovaSummaryBlock$end],
               collapse = "\n"))
  }

  if(showGLMsummary){
    cat(paste0(Lines[glmSummaryBlock$start:glmSummaryBlock$end],
               collapse = "\n"))
  }
}