R/map.R

In connorrothschild/tpltheme: Creates custom ggplot themes in the style of Texas Policy Lab

config <- yaml::read_yaml("https://raw.githubusercontent.com/connorrothschild/tpltheme/master/config.yaml", eval.expr = TRUE)

#'@importFrom magrittr "%>%"

cmp <- config$maps$base

#' @importFrom magrittr "%>%"
cntys <- ggplot2::map_data("county") %>%
  subset(region == "texas")

cities <- maps::'us.cities'
cities <- cities[cities$country.etc == "TX", ]

state <- ggplot2::map_data("state")

ditch_the_axes <- ggplot2::theme(
  axis.text = ggplot2::element_blank(),
  axis.line = ggplot2::element_blank(),
  axis.ticks = ggplot2::element_blank(),
  panel.border = ggplot2::element_blank(),
  panel.grid = ggplot2::element_blank(),
  axis.title = ggplot2::element_blank()
)

#' @title base map
#' @param data the data to use
base_map <- function(data) {

  ggplot2::ggplot() +
    ggplot2::geom_polygon(data = data,
                          ggplot2::aes(x = state$long, y = state$lat, fill = state$region, group = state$group)) +#, color = cmp$color, fill = cmp$fill) +
    ggplot2::coord_fixed(1.3) +
    ggplot2::guides(fill = FALSE) +
    ggplot2::theme_bw() +
    ditch_the_axes
}

#' @title United States base map
usa_base_map <- function() {

  base_map(data = state) + theme_tpl_texas()

}

#' @title Texas base map
#' @param data Geographic data passed to the function
tx_base_map <- function(data = "cntys") {

  texas <- state[state$region == "texas", ]

  base_map <- function(data) {

    ggplot2::ggplot() +
      ggplot2::geom_polygon(data = data,
                            ggplot2::aes(x = data$long, y = data$lat, fill = data$region, group = data$group)) +#, color = cmp$color, fill = cmp$fill) +
      ggplot2::coord_fixed(1.3) +
      ggplot2::guides(fill = FALSE) +
      ggplot2::theme_bw() +
      ditch_the_axes
  }

  if (data == "none") {
  base_map(data = texas)
  } else if (data == "cntys") {
  base_map(data = cntys) + theme_tpl_texas()
  } else if (data == "cities") {
  base_map(data = texas)
  maps::map("state", "Texas")
  maps::map.cities(maps::us.cities, country = "TX", label = TRUE)
  } else {
    stop("Geographic object does not exist. Try 'cntys' or 'cities', or 'none' to return a blank Texas map.",
    .call = FALSE)
  }

}

#' @title Add Texas cities
#'
#' @param label Whether the plotted cities should also have labels
#' @export
add_tx_cities <- function(label = FALSE) {

    maps::map("state", "Texas")
    maps::map.cities(maps::us.cities, country = "TX", label = label)

  }