R/PrepMap.R

In TommiKarki/SurvMaps: Get geo data from ECDC GIS server, prepare geodata for ggplot2, EU/EEA survmapper

#' Prepare your data and spatial data for SurvMapper (ggplot)
#'
#' @param data Your data that you want to include in the map, including the variable GEO_ID with the correct codes
#' @param geo Your geometry data, either SpatialPolygonsDataFrame or SpatialPointsDataFrame with variable GEO_ID

#' @keywords map
#' @export
#' @examples
#' 
#' # load the included dummy data
#' load(system.file("extdata", "dummy_data.rds", package = "SurvMaps"))
#' # load the included EU/EEA SpatialPolygonsDataframe (includes Asia and Africa for background)
#' load(system.file("extdata", "EU_AFR_AS_plgs.rds", package = "SurvMaps"))
#' 
#' mymap <- PrepMap(data = dummy_data , geo = plg_map)

PrepMap <- function(data, geo){
  if(is(geo, "SpatialPolygonsDataFrame")){
  geo@data[["id"]] <- rownames(geo@data)
  geo@data   <- merge(geo@data, data, by="GEO_ID", all.x=TRUE)
  z     <- broom::tidy(geo)
  z    <- merge(z,geo@data, by="id", all.x=TRUE)
  }else if(is(geo, "SpatialPointsDataFrame")){
    z <- data.frame(geo)
    z <- merge(z, data, by="GEO_ID", all.x=TRUE)
  }else
    stop("Only SpatialPolygonsDataFrame or SpatialPointsDataFrame accepted! Please check your geodata!")
  return(z)}