Replace stats::ave() (#6698)

* use vctrs-variant of `stats::ave()`

* replace other instances of `stats::ave()` too

Author: teunbrand

R/facet-.R

@@ -1421,14 +1421,14 @@ censor_labels <- function(ranges, layout, labels) {
   )
 
   if (!labels$x) {
-    xmax <- stats::ave(layout$ROW, layout$COL, FUN = max)
-    xmin <- stats::ave(layout$ROW, layout$COL, FUN = min)
+    xmax <- vec_ave(layout$ROW, layout$COL, max)
+    xmin <- vec_ave(layout$ROW, layout$COL, min)
     draw[which(layout$ROW != xmax), "bottom"] <- FALSE
     draw[which(layout$ROW != xmin), "top"] <- FALSE
   }
   if (!labels$y) {
-    ymax <- stats::ave(layout$COL, layout$ROW, FUN = max)
-    ymin <- stats::ave(layout$COL, layout$ROW, FUN = min)
+    ymax <- vec_ave(layout$COL, layout$ROW, max)
+    ymin <- vec_ave(layout$COL, layout$ROW, min)
     draw[which(layout$COL != ymax), "right"] <- FALSE
     draw[which(layout$COL != ymin), "left"] <- FALSE
   }

R/geom-path.R

@@ -19,7 +19,7 @@ GeomPath <- ggproto("GeomPath", Geom,
     # middle since you expect those to be shown by a break in the line
     aesthetics <- c(self$required_aes, self$non_missing_aes)
     complete <- stats::complete.cases(data[names(data) %in% aesthetics])
-    kept <- stats::ave(complete, data$group, data$PANEL, FUN = keep_mid_true)
+    kept <- vec_ave(complete, interaction(data$group, data$PANEL, drop = TRUE), keep_mid_true)
     data <- data[kept, ]
 
     if (!all(kept) && !params$na.rm) {
@@ -48,7 +48,7 @@ GeomPath <- ggproto("GeomPath", Geom,
     munched <- coord_munch(coord, data, panel_params)
 
     # Silently drop lines with less than two points, preserving order
-    rows <- stats::ave(seq_len(nrow(munched)), munched$group, FUN = length)
+    rows <- vec_ave(seq_len(nrow(munched)), munched$group, length)
     munched <- munched[rows >= 2, ]
     if (nrow(munched) < 2) return(zeroGrob())
 

R/stat-sum.R

@@ -15,7 +15,7 @@ StatSum <- ggproto(
 
     counts <- count(data, group_by, wt_var = "weight")
     counts <- rename(counts, c(freq = "n"))
-    counts$prop <- stats::ave(counts$n, counts$group, FUN = prop.table)
+    counts$prop <- vec_ave(counts$n, counts$group, prop.table)
     counts
   }
 )

R/utilities.R

@@ -931,7 +931,7 @@ compute_data_size <- function(data, size, default = 0.9,
     res <- vapply(res, resolution, FUN.VALUE = numeric(1), ...)
     res <- min(res, na.rm = TRUE)
   } else if (panels == "by") {
-    res <- stats::ave(data[[var]], data$PANEL, FUN = function(x) resolution(x, ...))
+    res <- vec_ave(data[[var]], data$PANEL, function(x) resolution(x, ...))
   } else {
     res <- resolution(data[[var]], ...)
   }
@@ -951,3 +951,14 @@ try_prop <- function(object, name, default = NULL) {
   }
   S7::prop(object, name)
 }
+
+vec_ave <- function(x, by, fn, ...) {
+  idx <- vec_group_loc(by)$loc
+  list_unchop(
+    lapply(
+      vec_chop(x, indices = idx),
+      FUN = fn, ...
+    ),
+    indices = idx
+  )
+}