R/be_1.R

In omstuhler/semgram: Extracting Semantic Motifs from Textual Data

#' @importFrom rsyntax annotate_tqueries cast_text children NOT OR not_children parents tquery
NULL

###############################################################################################
##### Rule: Being adjective
##### Example: "ENTITY is nice but dumb." (nice, dumb)
##### Example: "ENTITY is a winner and nice." (winner, nice)
##### Example: "ENTITY looks nice." (nice)
##### Example: "ENTITY is nice and a cool person." (nice, humble, cool, person)


be_1 = function(tokens, entities, verb_pos, agent_patient_pos, extract){
  rule = tquery(OR(token = entities, appos_child = "appos_child"), relation = "nsubj",
                label = "Entity", fill = F,
                parents(pos = c("AUX", "VERB"),
                        children(pos = c("ADJ", "NOUN", "PROPN"), relation = c("acomp", "attr"), phrase_replacement = NA,
                                 label = "characterization",
                                 fill = F,
                                 children(pos = c("ADJ", "NOUN", "PROPN"), relation = c("conj", "appos", "amod"), req = F, phrase_replacement = NA,
                                          label = "characterization",
                                          fill = F, depth = 3
                                 )
                        )
                )
  )

  tokens = annotate_tqueries(tokens, "query", rule, overwrite = T, copy = F)
  if(all(is.na(tokens$query))){
    casted = data.table(doc_id = character(), ann_id = factor(), Entity = character(), characterization = character())
  } else {
    casted = cast_text(tokens, 'query', text_col = extract)
  }
  return(casted)
}