R/foreign_map.R

In yiluheihei/ncovmap:

#' Foreign ncov map
#'
#' @param country country name
#' @export
#' @rdname plot_province_map
plot_foreign_map <- function(ncov,
                            country,
                            key = c("confirmedCount", "suspectedCount",
                                    "curedCount", "deadCount"),
                            bins = c(0, 10, 100, 500, 1000),
                            legend_title ='Confirmed',
                            legend_position = c("bottomright", "topright",
                                                "bottomleft", "topleft"),
                            color = "Reds"
) {
  key <- match.arg(key)
  legend_position <- match.arg(legend_position)

  country <- tolower(country)

  country_states <- system.file(
    "foreign_cities.csv",
    package = "ncovmap"
  ) %>%
    readr::read_csv() %>%
    filter(name_zh == country | name == country)

  if (grepl(country, "korea")) {
    country <- "south korea"
  }

  bins <- setdiff(bins, c(0, 1)) %>%
    c(0, 1, .)

  if (country == "italy") {
    json_file <- system.file(
      paste0("json/", country, ".json"),
      package = "ncovmap"
    )
    country_map <- leafletCN::read.geoShape(json_file)
  } else {
    country_map <- rnaturalearth::ne_states(country)
    # country_states <- system.file(
    #   paste0(sub(" ", "_", country), "_states.csv"),
    #   package = "ncovmap"
    # ) %>%
    #   readr::read_csv()
  }

  if (country == "south korea") {
    country_map$name <- country_map$name_de
  }

  country_map$provinceName <- country_states$provinceName[
    match(country_map$name, country_states$provinceEnglishName)
  ]
  # country_map$name_zh <- country_states$name_zh[
  #  match(country_map$name, country_states$name)
  #  ]

  ncov <- right_join(
    ncov,
    country_states,
    by = c("provinceName" = "provinceName")
  )
  ncov <-  mutate_if(ncov, is.numeric, ~ ifelse(is.na(.x), 0, .x))

  ncov$key <- ncov[[key]]
  ncov$key_level <-  cut(
    ncov$key,
    breaks = c(bins, Inf),
    labels = format_labels(bins),
    include.lowest = TRUE,
    right = FALSE
  )

  index <- match(country_map$provinceName, ncov$provinceName)
  country_map$value <- ncov[["key_level"]][index]
  ncov <- ncov[index, ]

  pal <- leaflet::colorFactor(
    palette = "Reds",
    domain = country_map$value
  )

  colors <- pal(country_map$value)
  # if the count is 0, manual set the color as white
  colors[country_map$value == 0] <-  "#FFFFFF"
  map_colors <- colors
  names(colors) <- country_map$value
  legend_colors <- colors[!duplicated(colors)] %>%
    sort(decreasing = TRUE)


  leaflet::leaflet(country_map) %>%
    leaflet::addPolygons(
      smoothFactor = 1,
      fillOpacity = 0.7,
      weight = 1,
      color = map_colors,
      popup =  paste(
        ncov$provinceName,
        ncov$key)
    ) %>%
    leaflet::addLegend(
      "bottomleft",
      colors = legend_colors,
      labels = names(legend_colors),
      labFormat = leaflet::labelFormat(prefix = ""),
      opacity = 1
    )
}