R/add_macron.R

In joevalalipin/reohelpers: Making te reo easier to use in R

#' @import dplyr
#' @import stringr
#' @import magrittr


library(dplyr)
library(stringr)
library(magrittr)

load("data/glossary.rdata")

#' @export
add_macron <- function(string, sep = " ") {
  data.frame(str_split(string, sep), stringsAsFactors = FALSE) %>%
    rename(word_ascii_input = 1) %>%
    mutate(word_ascii_lower = tolower(word_ascii_input)) %>%
    left_join(glossary, by = "word_ascii_lower") %>%
    mutate(word_macron = ifelse(meaning != "Place" & str_detect(str_sub(word_ascii_input, 1, 1), "[A-Z]"), str_to_sentence(word_macron), word_macron)) %>%
    mutate(word_macron = ifelse(is.na(word_macron), word_ascii_input, word_macron)) %>%
    pull(word_macron) %>%
    paste0(collapse = sep)
}
add_macron <- Vectorize(add_macron)