R/animalType.R

In UBC-MDS/Ranimalsgonewild: Animal themed text filters

#' Analyzes text based on average word length and returns an animal picture representing language complexity.
#'
#' @param text A string of characters to be analyzed
#' @param species A string representing species, must be one of "Duck", "Monkey", "Giraffe", "Whale"
#'
#' @return A jpeg of an animal
#' @export
#'
#' @examples
#' text <- "I like pie"
#' animal_rep <- animalType(animalClassifier(text), text)
#'
#' text <- "Pie's abundantly awesome"
#' species <- "Duck"
#' animal_rep <- animalType(species, text)
animalType <- function(species, text){
  if (!is.character(species)) {
    stop("Species is not a string")
  }

  if (!is.character(text)) {
    stop("Text is not a string")
  }
  words <- gsub('[[:punct:]]+','',text)

  iq = "smart"
  avg <- 0

  for (w in words) {avg <- avg + nchar(w)}
  avg <- avg/length(strsplit(words, " ")[[1]])

  if (avg < 5) {iq = "dumb"}

  if (endsWith(getwd(), "Ranimalsgonewild/tests/testthat")){
    dir <- dirname(getwd())
    dir <- dirname(dir)
    filename <- paste(dir, "/imgs/",iq, "_", species, ".jpeg", sep = "")
    plot(imager::load.image(filename))
  }
  else if(endsWith(getwd(), "Ranimalsgonewild")){
    filename <- paste0("./imgs/",iq, "_", species, ".jpeg")
  } else {
    return (paste0(iq,"_", species))
  }
  plot(imager::load.image(filename))
  paste0(iq,"_", species)
}