R/all_words.R

In trinker/qdap: Bridging the Gap Between Qualitative Data and Quantitative Analysis

#' Searches Text Column for Words
#' 
#' A convenience function to find words that begin with or contain a letter 
#' chunk and returns the frequency counts of the number of occurrences of each 
#' word.
#' 
#' @param text.var The text variable.
#' @param begins.with This argument takes a word chunk.  Default is \code{NULL}. 
#' Use this if searching for a word beginning with the word chunk.
#' @param contains This argument takes a word chunk.  Default is \code{NULL}. 
#' Use this if searching for a word containing the word chunk.
#' @param alphabetical logical.  If \code{TRUE} orders rows alphabetically, if 
#' \code{FALSE} orders the rows by descending frequency.
#' @param apostrophe.remove logical.  If \code{TRUE} removes apostrophes from 
#' the text before examining.
#' @param char.keep A character vector of symbol character (i.e., punctuation) 
#' that strip should keep.  The default is to strip everything except 
#' apostrophes.  This enables the use of special characters to be turned into 
#' spaces or for characters to be retained.
#' @param char2space A vector of characters to be turned into spaces. 
#' @param \ldots Other argument supplied to \code{\link[qdap]{strip}}.
#' @return Returns a dataframe with frequency counts of words that begin with or 
#' contain the provided word chunk.
#' @note Cannot provide both \code{begins.with} and \code{contains} arguments 
#' at once.  If both begins.with and contains are \code{NULL}.
#' \code{\link[qdap]{all_words}} returns a 
#' frequency count for all words.
#' @seealso 
#' \code{\link[qdap]{term_match}}
#' @export
#' @examples
#' \dontrun{
#' x1 <- all_words(raj$dialogue, begins.with="re")
#' head(x1, 10)
#' x2 <- all_words(raj$dialogue, "q")
#' head(x2, 10)
#' all_words(raj$dialogue, contains="conc")
#' x3 <- all_words(raj$dialogue)
#' head(x3, 10)
#' x4 <- all_words(raj$dialogue, contains="the")
#' head(x4)
#' x5 <- all_words(raj$dialogue, contains="read")
#' head(x5)
#' 
#' ## Filter by nchar and stopwords
#' Filter(head(x3), min = 3)
#' 
#' ## Keep spaces
#' all_words(space_fill(DATA$state, c("are you", "can be")))
#' }
all_words <- 
    function(text.var, begins.with = NULL, contains = NULL, alphabetical = TRUE, 
             apostrophe.remove = FALSE,  char.keep = char2space, char2space = "~~", ...){
        
        if (!is.null(begins.with) & !is.null(contains)) {
            stop("Can not use both 'begins.with' & 'contains' arguments")
        }
        if(!is.null(begins.with)) begins.with <- tolower(begins.with)
        if(!is.null(contains)) contains <- tolower(contains)
        
        char.keep <- unique(c(char2space, char.keep))
        
        WORDS <- unlist(bag_o_words(text.var, apostrophe.remove = apostrophe.remove, 
                                    char.keep, char.keep, ...), use.names=FALSE)
        
        y <- data.frame(table(WORDS), stringsAsFactors = FALSE, row.names=NULL)
        names(y) <- c("WORD", "FREQ")
        y$WORD <- as.character(y$WORD)
        y[, "FREQ"] <- as.numeric(as.character(y[, "FREQ"]))
        if (!is.null(begins.with)) {
            y <- y[substring(y[, 1], 1, nchar(begins.with)) %in% begins.with, ]
            if(nrow(y)==0) stop("No words match")
        }
        if (!is.null(contains)) {
            y <- y[grep(contains, y[, 1]), ]
            if(nrow(y)==0) stop("No words match")
        }
        if (!alphabetical) {
            y <- y[order(-y$FREQ, y$WORD), ]
        }
        if (!is.null(char2space)) {
            y[["WORD"]] <- mgsub(char2space, " ", y[["WORD"]])
        }
        p <- class(y)
        class(y) <- c("all_words", p)
        y
    }

#' Prints an all_words Object
#' 
#' Prints an all_words object.
#' 
#' @param x The all_words object.
#' @param \ldots ignored
#' @method print all_words
#' @export
print.all_words <- function(x, ...) {
    class(x) <- "data.frame"
    print(left_just(x, "WORD"))
}

#' Filter
#' 
#' \code{Filter.all_words} - Filter words from a all_words 
#' that meet max/min word length criteria.
#' 
#' all_words Method for Filter
#' @rdname Filter
#' @export
#' @method Filter all_words
#' @return \code{Filter.all_words} - Returns a matrix of the class "all_words".
Filter.all_words <- function(x, min = 1, max = Inf, 
                             count.apostrophe = TRUE, stopwords = NULL, ignore.case = TRUE, ...) {
    
    word_list_filter_helper(x, min = min, max = max, 
                            count.apostrophe = count.apostrophe, stopwords = stopwords, 
                            ignore.case = ignore.case, ...)
}