R/print.R

In pierucci/rgho: Access WHO Global Health Observatory Data from R

to_data_frame <- function(x, n = length(x)) {
  data.frame(
    Label = head(attr(x, "labels"), n = n),
    ID = head(as.vector(x), n = n),
    stringsAsFactors = FALSE
  )
}

#' @rdname man-gho
#' @export
print.gho <- function(x, n = options()$rgho.n, ...) {
  if (n == Inf) {
    n <- length(x)
  }

  cat(sprintf("A 'GHO' object of %i elements.\n\n", length(x)))

  print(to_data_frame(x, n = n), ...)

  if (n < length(x)) {
    cat(sprintf("...\n\n(Printing %i first elements.)\n", n))
  }

  if (! is.null(attr(x, "attrs"))) {
    names_attr <- names(attr(x, "attrs"))
    names_attr <- names_attr[names_attr != "code"]
    cat("\nAttributes:\n\n")
    cat(names_attr, sep = "\n")
  }
}