R/list-object-sizes.R

In csgillespie/rprofile: My Rprofile in a Handy R Package

# SO: http://stackoverflow.com/q/1358003/203420
# improved list of objects
lsos = function(order.by = c("PrettySize", "Type", "Size", "Rows", "Columns"), #nolint
                pos = 1) {
  napply = function(names, fn) sapply(names, function(x) fn(get(x, pos = pos)))
  names = ls(pos = pos)

  obj.class = napply(names, function(x) as.character(class(x))[1]) #nolint
  obj.mode = napply(names, mode) #nolint
  obj.type = ifelse(is.na(obj.class), obj.mode, obj.class) #nolint
  obj.prettysize = napply(names, function(x) { #nolint
    utils::capture.output(print(utils::object.size(x), units = "auto"))
  })
  obj.size = napply(names, utils::object.size) #nolint
  obj.dim = t(napply(names, function(x) as.numeric(dim(x))[1:2])) #nolint

  if (length(names) > 0) {
    vec = is.na(obj.dim)[, 1] & (obj.type != "function")
    obj.dim[vec, 1] = napply(names, length)[vec] #nolint
    out = data.frame(obj.type, obj.size, obj.prettysize, obj.dim)
  } else {
    out = tibble::tibble("a", "b", "c", "d", "e")
    out = out[FALSE, ]
  }
  names(out) = c("Type", "Size", "PrettySize", "Rows", "Columns")
  order.by = match.arg(order.by) #nolint
  out = out[order(out[[order.by]], decreasing = TRUE), ]
  tibble::tibble(out)
}