R/intarface-kintoki.R

In paithiov909/krikinton: rJava Wrapper of Sudachi and Kintoki

#' Rebuild Kintoki Parser
#'
#' @return a binding to the instance of
#' `com.worksap.nlp.kintoki.cabocha.Cabocha` is returned invisibly.
#'
#' @import rJava
#' @export
rebuild_parser <- function() {
  Parser(rJava::.jnew("com.worksap.nlp.kintoki.cabocha.Cabocha", Param()))
  return(invisible(Parser()))
}

#' Call Kintoki Parser
#'
#' @param chr character vector
#' @return tibble
#'
#' @export
kintoki <- function(chr) {
  if (!is.character(chr) || is.na(chr)) {
    message("Invalid characters provided. Chr must be a character vector, not NA_character_.")
    return(invisible(tibble()))
  } else {
    texts <- stri_enc_toutf8(chr)
    if (is.null(Parser())) rebuild_parser()

    results <- imap(texts, function(sent, idx) {
      if (!is.na(sent)) {
        tree <- rJava::J(Parser(), "parse", sent)

        chunk_size <- tree$getChunkSize()

        if (!rJava::is.jnull(chunk_size)) {
          df <- lapply(seq_len(chunk_size), function(i) {
            chunk <- tree$chunk(i - 1L)
            token_size <- chunk$getTokenSize()
            tokens <- lapply(seq_len(token_size), function(j) {
              token <- chunk$token(j - 1L)
              res <- list(
                Surface = token$getSurface(),
                Feature = token$getFeature()
              )
              res <- lapply(res, function(elem) {
                Encoding(elem) <- "UTF-8"
                return(elem)
              })
              res <- list(
                sentence_id = idx,
                chunk_id = i - 1L,
                Link = chunk$getLink(),
                Score = chunk$getScore(),
                Head = chunk$getHeadPos(),
                Func = chunk$getFuncPos(),
                token_id = j - 1L,
                res
              )
              return(as.data.frame(res, stringsAsFactors = FALSE))
            })
            tokens <- map_dfr(
              tokens,
              ~ separate(
                .,
                col = "Feature",
                into = c(
                  "POS1",
                  "POS2",
                  "POS3",
                  "POS4",
                  "X5StageUse1",
                  "X5StageUse2"
                ),
                sep = ","
              )
            )
            return(mutate(
              as_tibble(tokens),
              across(
                where(is.character),
                ~ if_else(
                  . == "*",
                  NA_character_,
                  .
                )
              )
            ))
          })
          return(map_dfr(df, ~.))
        } else {
          return(tibble())
        }
      } else {
        return(tibble())
      }
    })

    return(map_dfr(
      discard(results, ~ is_empty(.)),
      ~.
    ))
  }
}