R/browse.R

In dataview: Data and Workspace Browser for Terminals

interrupt <- function(){
    signalCondition(structure(simpleCondition("User interrupt"),
                              class=c("interrupt", "condition")))
}

#' Browse the contents of a nested data structure
#' 
#' Manually step in and out of the elements of complex data structure with
#' \code{\link{whos}}. To move around enter the name or number of the element
#' you want to inspect next. Partial names will automatically be matched against
#' the possible element names.
#' 
#' @param x Object.
#' @param name Name of the object.
#' @return Nothing.
#' @examples
#' \dontrun{
#' require(ggplot2)
#' p <- ggplot(iris, aes(x=Sepal.Length, y=Sepal.Width)) +
#'     geom_point()
#' browse(p)
#' }
#' @seealso whos
#' @author Christofer \enc{Bäcklin}{Backlin}
#' @export
browse <- function(x=.GlobalEnv, name){
    if(missing(name)) name <- deparse(substitute(x))
    elem <- NA
    w <- whos(x)

    # I haven't yet figured out why browse cannot find the [.whos if it is not
    # specified here too... weird
    `[.whos` <- function(x, ...){
        structure(as.data.table(x)[...], class=class(x))
    }

    if(!identical(x, .GlobalEnv)){
        try({
            suppressMessages(attach(x)) # To get tab completion
            on.exit(detach(x))
        }, silent=TRUE)
    }
    invisible(tryCatch({
        while(is.na(elem) || elem != ""){
            cat("\nBrowsing ", name, "\n", sep="")
            print(w)
            elem <- readline("Select an element to inspect: ")
            if(elem != ""){
                if(grepl("^\\d+$", elem) && !elem %in% w$name){
                    i <- as.integer(elem)
                    if(is.na(w$name[i])){
                        elem <- if(i > nrow(w)) NA else i
                    } else {
                        elem <- w$name[i]
                    }
                } else if(elem == "Q" && !"Q" %in% w$name){
                    interrupt()
                } else if(!elem %in% w$name){
                    i <- grep(sprintf("^%s", elem), w$name)
                    while(length(i) > 1){
                        cat("\n")
                        print(w[i])
                        elem <- readline("Please specify: ")
                        if(grepl("^\\d+$", elem) && !elem %in% w$name[i]){
                            i <- i[as.integer(elem)]
                        } else if(elem == "Q" && !"Q" %in% w$name){
                            interrupt()
                        } else {
                            i <- grep(sprintf("^%s", elem), w$name)
                        }
                    }
                    elem <- if(is.na(i) || length(i) == 0) NA else w$name[i]
                }
                if(is.na(elem)){
                    cat("No matches, please try again. Enter blank or press ctrl+c to exit.\n")
                } else {
                    # Get the object
                    if(isS4(x)){
                        obj <- slot(x, elem)
                        obj.name <- sprintf("%s@%s", name, elem)
                    } else if(is.integer(elem)){
                        obj <- x[[elem]]
                        obj.name <- sprintf("%s[[%i]]", name, elem)
                    } else {
                        obj <- get(elem, x)
                        obj.name <- sprintf("%s$%s", name,
                            if(elem == make.names(elem)) elem else sprintf("`%s`", elem))
                    }
                    # Browse or show the object
                    if(is.data.frame(obj)){
                        entry.view(obj)
                    } else if(is.list(obj)){
                        browse(obj, obj.name)
                    } else if(is.function(obj)){
                        page(obj)
                    } else {
                        print(summary(obj))
                        readline("Press enter to continue.")
                    }
                }
            }
        }
    }, interrupt = function(cond){
        signalCondition(cond)
    }))
}