R/return_mrconso_net.R

In patelm9/cartographR:

#' Returns MRCONSO NETS that match the phrase
#' @param phrase character vector of length 1. 
#' @param rm_forward_slash forward slashes throws errors with the write_csv and read_csv proceses downstream and are removed from the phrase by default
#' @param cast_wide_net TRUE if any NET with a grep logical match is to be included
#' @param list_by_cui selects only CUI and STR variables and splits output by CUI into a list
#' @import dplyr
#' @import readr
#' @import mirroR
#' @import typewriteR
#' @import stringr
#' @export


return_mrconso_net <-
        function(phrase, rm_forward_slash = TRUE, cast_wide_net = TRUE, list_by_cui = TRUE) {
                glossary_fn <- "/Users/meerapatel/GitHub/MSK_KMI_Enterprise/TerminologyBiblioTech/CATALOGUE/UMLS_MT_MRCONSO/GLOSSARY_UMLS_MT_MRCONSO_QUERY_TERMS.csv"
                files_list <- list.files("/Users/meerapatel/GitHub/MSK_KMI_Enterprise/TerminologyBiblioTech/CATALOGUE/UMLS_MT_MRCONSO/NETS", full.names = TRUE)
                glossary <- readr::read_csv(glossary_fn, col_types = cols(.default = "c"))
                
                phrase_00 <- phrase
                if (rm_forward_slash == TRUE) {
                        phrase_01 <- stringr::str_remove_all(phrase_00, "[/]")
                } else {
                        phrase_01 <- phrase
                }
                
                exact_phrase_file_ext <- paste0("[0-9]{12}[_]{1}", phrase_01, "[.]{1}csv$")
                
                if (cast_wide_net == TRUE) {
                        phrase_02 <- phrase_01
                } else {
                        phrase_02 <- paste0("^", phrase_01, "$")
                }
                
                if (any(grepl(exact_phrase_file_ext, files_list)) == TRUE) {
                                fns <- grep(phrase_02, files_list, value = TRUE, ignore.case = TRUE)
                                if (length(fns) > 0) {
                                        x <- lapply(fns, readr::read_csv, col_types = cols(.default = "c"))
                                        x <- dplyr::bind_rows(x)
                                        
                                        if (nrow(x) > 0) {
                                                
                                                if (list_by_cui == TRUE) {
                                                        x <- split(x, x$CUI)
                                                        cuis <- names(x)
                                                        x <- lapply(1:length(x), function(i) x[[i]] %>% dplyr::select(STR) %>% dplyr::distinct() %>% unlist() %>% unname())
                                                        names(x) <- cuis
                                                        return(x)
                                                } else {
                                                        return(x)
                                                }
                                                
                                        } else {
                                                typewriteR::tell_me("\tInput phrase:", phrase)
                                                cat("\n")
                                                typewriteR::tell_me("\tPhrase 01:", phrase_01)
                                                cat("\n")
                                                typewriteR::tell_me("\tPhrase 02:", phrase_02)
                                                cat("\n")
                                                typewriteR::tell_me("\t\tNo match to CUIs in MRCONSO.")
                                                cat("\n")
                                                return(x)
                                                }
                                }
                } else {
                        write_mrconso_net(phrase, rm_forward_slash = rm_forward_slash)
                        
                        glossary_fn <- "/Users/meerapatel/GitHub/MSK_KMI_Enterprise/TerminologyBiblioTech/CATALOGUE/UMLS_MT_MRCONSO/GLOSSARY_UMLS_MT_MRCONSO_QUERY_TERMS.csv"
                        files_list <- list.files("/Users/meerapatel/GitHub/MSK_KMI_Enterprise/TerminologyBiblioTech/CATALOGUE/UMLS_MT_MRCONSO/NETS", full.names = TRUE)
                        glossary <- readr::read_csv(glossary_fn, col_types = cols(.default = "c"))
                        phrase_00 <- phrase
                        if (rm_forward_slash == TRUE) {
                                phrase_01 <- stringr::str_remove_all(phrase_00, "[/]")
                        } else {
                                phrase_01 <- phrase
                        }
                        
                        exact_phrase_file_ext <- paste0("[0-9]{12}[_]{1}", phrase_01, "[.]{1}csv$")
                        
                        if (cast_wide_net == TRUE) {
                                phrase_02 <- phrase_01
                        } else {
                                phrase_02 <- paste0("^", phrase_01, "$")
                        }
                        
                        fns <- grep(phrase_02, files_list, value = TRUE, ignore.case = TRUE)
                        x <- lapply(fns, readr::read_csv, col_types = cols(.default = "c"))
                        x <- dplyr::bind_rows(x)
                        
                        if (nrow(x) > 0) {
                                
                                if (list_by_cui == TRUE) {
                                        x <- split(x, x$CUI)
                                        cuis <- names(x)
                                        x <- lapply(1:length(x), function(i) x[[i]] %>% dplyr::select(STR) %>% dplyr::distinct() %>% unlist() %>% unname())
                                        names(x) <- cuis
                                        return(x)
                                } else {
                                        return(x)
                                }
                                
                        } else {
                                typewriteR::tell_me("\tInput phrase:", phrase)
                                cat("\n")
                                typewriteR::tell_me("\tPhrase 01:", phrase_01)
                                cat("\n")
                                typewriteR::tell_me("\tPhrase 02:", phrase_02)
                                cat("\n")
                                typewriteR::tell_me("\t\tNo match to CUIs in MRCONSO.")
                                cat("\n")
                                return(x)
                        }
                        
                }
                
        }