R/perm_lookup.R

In hnasif/movies: Pair of lookup functions given an incorrect permutation or spelling

#' Look up a title in a data set of titles given a permutation
#'
#' @param data Data frame of any kind
#' @param column Character of the name of the variable of interest
#' @param input Character of the full title of the variable of interest, in any permutation
#' @return The proper title of interest from the full data set
#' @examples
#' perm_lookup(movies, "title", "man the she's")
#' @export
perm_lookup <- function(data, column, input){
  tolower(input)
  perms <- unlist(lapply(permn(lapply(str_split(input, " "),
                        removePunctuation)[[1]]), str_c, collapse = " "))
  lowercase_column <- lapply(tolower(iconv(data[[column]])), removePunctuation)
  for (i in 1:length(perms)) {
    if (length(data[grepl(perms[[i]], lowercase_column),][[column]] %in% data) != 0) {
      print(data[grepl(perms[[i]], lowercase_column),][[column]])
      break
    }
  }
}

spell_lookup <- function(data, column, input){
  tolower(input)
  bad <- hunspell(input)
  correct <- lapply(str_split(input, " "), hunspell_check)
  suggest <- lapply(hunspell_suggest(bad[[1]]), '[[', 1)
  correct_list <- as.list(str_split(input, " ")[[1]])[correct[[1]]]
  perm_corrected_strings <- permn(unlist(lapply(append(suggest, correct_list), str_c)))
  corrected_strings <- unlist(lapply(perm_corrected_strings, str_c, collapse = " "))
  lowercase_column <- lapply(tolower(iconv(data[[column]])), removePunctuation)
  for (i in 1:length(corrected_strings)) {
    if (length(data[grepl(corrected_strings[[i]], lowercase_column),][[column]] %in% movies) != 0) {
      print(data[grepl(corrected_strings[[i]], lowercase_column),][[column]])
      break
    }
  }
}