R/sf-compat.R

In paleolimbot/geovctrs: Efficient Data Structures for Geometry Vectors

#' @rdname is_geovctr
#' @export
as_geovctr.sfc <- function(x, ...) {
  new_wk_wkb(unclass(sf::st_as_binary(x, EWKB = TRUE)))
}

#' @rdname is_geovctr
#' @export
as_geovctr.sf <- function(x, ...) {
  as_geovctr(x[[attr(x, "sf_column")]], ...)
}

#' @rdname is_geovctr
#' @export
restore_geovctr.sfc <- function(x, result, ...) {
  wkb <- unclass(as_wkb(result))
  wkb[vapply(wkb, is.null, logical(1))] <- as_wkb("GEOMETRYCOLLECTION EMPTY")
  sf::st_as_sfc(structure(wkb, class = "WKB"), EWKB = TRUE)
}

#' @rdname is_geovctr
#' @export
restore_geovctr.sf <- function(x, result, ...) {
  x <- vec_recycle(x, vec_size(result))
  x[[attr(x, "sf_column")]] <- restore_geovctr(x[[attr(x, "sf_column")]], result, ...)
  x
}