R/IO.R

In Deleetdk/megadataset: Worldwide data on countries and intra-country units

#' Abbreviate country and regional names to ISO-3X.
#'
#' To enable easier merging of datasets of international data. You need to download the countrylist.csv file yourself.
#' @param x (character vector) The full names of countries and subcountry units.
#' @param abbrev (data.framea) Optional: A data.frame of abbreviations to use.
#' @param georgia (character scalar) Whether "georgia" is a country or a US state ("state"). Defaults to "country".
#' @export
#' @examples
#' as_abbrev(c("Denmark", "USA", "India", "Thailand", "Russia", "Korea, South"))
as_abbrev2 = function(x, georgia = "country", abbrev = NA){
  library(stringr)
  library(magrittr)

  #use built in?
  if (is.na(abbrev)) {
    #get dictionary
    abbreviations = read.csv("inst/extdata/abbreviations.csv", sep=";", encoding = "UTF-8", stringsAsFactors = F)
  } else {
    abbreviations = abbrev
  }

  #Georgia as state or country?
  if (substr(georgia, 1, 1) == "s") {
    abbreviations["Georgia", ] = "USA_GA"
  }

  #get abbrevs
  sapply(x, function(i) {
    indice = (abbreviations$Names == i) %>% #find matches
      which %>% #their indices
      `[`(1) #get the first
    if(is.na(indice)) message(str_c(i, " could not be found!"))

    return(abbreviations$ISO3X[indice])
  })
}


#' Get full country names from ISO-3X.
#'
#' To enable easier merging of datasets of international data. You need to download the countrylist.csv file yourself.
#' @param x (character vector) The ISO-3X codes.
#' @export
#' @examples
#' as_long(c("DNK", "USA", "IND", "THA", "RUS", "KOR"))
as_long2 = function(x, abbrev = NA) {
  library(stringr)
  library(magrittr)

  #use built in?
  if (is.na(abbrev)) {
    #get dictionary
    abbreviations = read.csv("inst/extdata/abbreviations.csv", sep=";", encoding = "UTF-8", stringsAsFactors = F)
  } else {
    abbreviations = abbrev
  }

  sapply(x, function(i) {
    # browser()
    indice = (abbreviations$ISO3X == i) %>% #find matches
      which %>% #their indices
      `[`(1) #get the first
    if(is.na(indice)) message(str_c(i, " could not be found!"))

    return(abbreviations$Names[indice])
  })
}


#' Get the abbreviations data.frame.
#'
#' Fetches the abbreviation dictionary and returns it to the user. Useful if one wants to alter it and pass the modified version to as_abbrev() or as_long().
#' @param x (character vector) The ISO-3X codes.
#' @export
#' @examples
#' get_abbrev()
get_abbrev2 = function() {
  abbreviations = read.csv("inst/extdata/abbreviations.csv", sep = ";", header = T, stringsAsFactors = F, encoding = "UTF-8")
  abbreviations
}