R/StatMeanSDText.R

In EnvStats: Package for Environmental Statistics, Including US EPA Guidance

StatMeanSDText <- ggplot2::ggproto("StatMeanSDText", ggplot2::Stat,

  required_aes = c("x", "y"),

  setup_params = function(data, params) {
    if(!is.null(params$y.pos))
      return(params)

    range.y <- range(data$y, na.rm = TRUE)
    pos <- range.y[2] + diff(range.y) * params$y.expand.factor
    params$y.pos <- pos
    params
  },

  compute_panel = function(data, scales, y.pos, y.expand.factor, digits, digit.type, nsmall, family) {
    Mean.orig <- as.vector(by(data$y, data$x, mean, na.rm = TRUE))
    SD.orig <- as.vector(by(data$y, data$x, sd, na.rm = TRUE))

    Mean <- do.call(digit.type, list(x = Mean.orig, digits = digits))
    SD   <- do.call(digit.type, list(x = SD.orig,   digits = digits))

    dum.mat <- format(rbind(Mean, SD), nsmall = nsmall)
    lab <- paste0("Mean=", dum.mat[1, ],
        ifelse(family == "mono", "\nSD  =", "\nSD   ="),
        dum.mat[2, ])

    data.frame(x = as.vector(by(data$x, data$x, function(x)x[1])), y = y.pos, label = lab)
  }

)