R/magic_result.R

In hoxo-m/magicfor: Magic Functions to Obtain Results from for Loops

#' Get values stored by magicalized for loops as a list
#'
#' @return list
#'
#' @export
magic_result <- function() {
  .result_env$result
}

#' Get values stored by magicalized for loops as a vector
#'
#' @return vector
#'
#' @export
magic_result_as_vector <- function() {
  res <- .result_env$result
  if (is.null(res)) {
    NULL
  } else if (length(res) == 1) {
    unlist(res[[1]])
  } else {
    Map(unlist, .result_env$result)
  }
}

#' Get values stored by magicalized for loops as a data.frame
#'
#' @param iter logical. Include iterator into the result.
#'
#' @return data.frame
#'
#' @export
magic_result_as_dataframe <- function(iter = TRUE) {
  result <- .result_env$result
  if(iter && exists("input", envir = .result_env)) {
    result <- c(list(.result_env$input), result)
    names(result)[1] <- .result_env$input_name
  }
  data.frame(Map(function(res) if(is.atomic(res[[1]])) unlist(res) else I(res), result),
             stringsAsFactors = FALSE, check.names = FALSE)
}