R/annotate.R

In vega-ai/vietr: R Wrapper for VnCoreNLP Word tokenizer (ver 1.1.1)

#' @export
vncorenlp <- R6::R6Class('vncorenlp',
  public = list(
    annotations = NULL,
    initialize = function(annotations=c("wseg", "pos", "ner", "parse")) {
      ann <- match.arg(annotations)
      anns <- NULL
      switch (ann,
        wseg = {
          anns <- c("wseg")
        },
        pos = {
          anns <- c("wseg", "pos")
        },
        ner = {
          anns <- c("wseg", "ner")
        },
        parse = {
          anns <- c("wseg", "pos", "ner", "parse")
        },
        stop("unrecognized annotation '", ann, "'", call.=FALSE)
      )
      self$annotations = annotations
      ann_args <- .jarray(anns)
      private$pipe = .jnew("vn.pipeline.VnCoreNLP", ann_args)
      self
    },

    tokenize = function(sen) {
      ann <- .jnew("vn.pipeline.Annotation", sen)
      private$pipe$annotate(ann)
      words <- ann$getWords()
      ws <- words$toArray()
      vapply(.jevalArray(ws),
         function(x) {
           .jcall(x, returnSig="S", method="getForm")},
         character(1))
    },

    postag = function(sen) {
      ann <- .jnew("vn.pipeline.Annotation", sen)
      private$pipe$annotate(ann)
      words <- ann$getWords() #
      ws <- words$toArray()
      vapply(.jevalArray(ws),
         function(x) {
           list(word=.jcall(x, returnSig="S", method="getForm"),
                tag=.jcall(x, returnSig="S", method="getPosTag"))},
         list(word="", tag=""))
    },

    ner = function(sen) {
      ann <- .jnew("vn.pipeline.Annotation", sen)
      private$pipe$annotate(ann)
      words <- ann$getWords() #
      ws <- words$toArray()
      vapply(.jevalArray(ws),
         function(x) {
           list(word=.jcall(x, returnSig="S", method="getForm"),
                ner=.jcall(x, returnSig="S", method="getNerLabel"))},
         list(word="", ner=""))
  },

  depparse = function(sen) {
    ann <- .jnew("vn.pipeline.Annotation", sen)
    private$pipe$annotate(ann)
    words <- ann$getWords() #
    ws <- words$toArray()
    vapply(.jevalArray(ws),
           function(x) {
             list(word=.jcall(x, returnSig="S", method="getForm"),
                  head=.jcall(x, returnSig="I", method="getHead"),
                  dep=.jcall(x, returnSig="S", method="getDepLabel"))},
           list(word="", head="", dep=""))
  }),

  private = list(
    pipe = NULL
  )
)