Implement vec_pany() and vec_pall()

NAMESPACE

@@ -569,6 +569,8 @@ export(vec_math_base)
 export(vec_names)
 export(vec_names2)
 export(vec_order)
+export(vec_pall)
+export(vec_pany)
 export(vec_proxy)
 export(vec_proxy_compare)
 export(vec_proxy_equal)

NEWS.md

@@ -1,5 +1,9 @@
 # vctrs (development version)
 
+* New `vec_pany()` and `vec_pall()` parallel variants of `any()` and `all()`
+  (in the same way that `pmin()` and `pmax()` are parallel variants of `min()`
+  and `max()`) (#1675).
+
 * `vec_c(outer = c(inner = 1))` now produces correct error messages (#522).
 
 * If a data frame is returned as the proxy from `vec_proxy_equal()`,

R/parallel.R

@@ -0,0 +1,57 @@
+#' Parallel `any()` and `all()`
+#'
+#' @description
+#' These functions are variants of [any()] and [all()] that work in parallel on
+#' multiple inputs at once. They work similarly to how [pmin()] and [pmax()] are
+#' parallel variants of [min()] and [max()].
+#'
+#' @details
+#' `vec_pany()` and `vec_pall()` are consistent with [any()] and [all()] when
+#' there are no inputs to process in parallel:
+#'
+#' - `any()` returns `FALSE` with no inputs. Similarly, `vec_pany(.size = 1)`
+#'   and `vec_pany(NA, .na_rm = TRUE)` both return `FALSE`.
+#'
+#' - `all()` returns `TRUE` with no inputs. Similarly, `vec_pall(.size = 1)`
+#'   and `vec_pall(NA, .na_rm = TRUE)` both return `TRUE`.
+#'
+#' @param ... Logical vectors. These will be [recycled][vector_recycling_rules]
+#'   to their common size.
+#'
+#' @param .na_rm Should missing values be removed?
+#'
+#' @param .size An optional output size that overrides the common size of the
+#'   inputs in `...`.
+#'
+#' @name parallel-operators
+#'
+#' @examples
+#' x <- c(TRUE, FALSE, NA, TRUE, NA)
+#' y <- c(FALSE, FALSE, TRUE, TRUE, NA)
+#'
+#' vec_pany(x, y)
+#' vec_pall(x, y)
+#'
+#' # Missing values can be removed from the computation
+#' vec_pany(x, y, .na_rm = TRUE)
+#' vec_pall(x, y, .na_rm = TRUE)
+#'
+#' # Same empty behavior as `any()` and `all()`
+#' vec_pany(.size = 1)
+#' any()
+#'
+#' vec_pall(.size = 1)
+#' all()
+NULL
+
+#' @rdname parallel-operators
+#' @export
+vec_pany <- function(..., .na_rm = FALSE, .size = NULL) {
+  .Call(ffi_vec_pany, list2(...), .na_rm, .size, environment())
+}
+
+#' @rdname parallel-operators
+#' @export
+vec_pall <- function(..., .na_rm = FALSE, .size = NULL) {
+  .Call(ffi_vec_pall, list2(...), .na_rm, .size, environment())
+}

_pkgdown.yml

@@ -9,7 +9,7 @@ template:
 
   includes:
     in_header: |
-      <script defer data-domain="vctrs.r-lib.org,all.tidyverse.org" src="https://plausible.io/js/plausible.js"></script>  
+      <script defer data-domain="vctrs.r-lib.org,all.tidyverse.org" src="https://plausible.io/js/plausible.js"></script>
 
 reference:
 - title: User FAQ
@@ -93,6 +93,10 @@ reference:
   - vec_seq_along
   - vec_identify_runs
 
+- title: Reducers
+  contents:
+  - vec_pall
+
 - title: New classes
   contents:
   - list_of

src/decl/parallel-decl.h

@@ -0,0 +1,43 @@
+static inline
+r_obj* ffi_vec_p(r_obj* xs,
+                 r_obj* ffi_na_rm,
+                 r_obj* ffi_size,
+                 r_obj* frame,
+                 enum vctrs_parallel parallel);
+
+static
+r_obj* vec_p(r_obj* xs,
+             bool na_rm,
+             r_ssize size,
+             enum vctrs_parallel parallel,
+             struct r_lazy call);
+
+static inline
+void vec_pall_init(const int* v_x, bool na_rm, r_ssize size, int* v_out);
+static inline
+void vec_pany_init(const int* v_x, bool na_rm, r_ssize size, int* v_out);
+
+static inline
+void vec_pall_fill(const int* v_x, bool na_rm, r_ssize size, int* v_out);
+static inline
+void vec_pany_fill(const int* v_x, bool na_rm, r_ssize size, int* v_out);
+
+static inline
+void vec_pall_init_na_rm(const int* v_x, r_ssize size, int* v_out);
+static inline
+void vec_pall_fill_na_rm(const int* v_x, r_ssize size, int* v_out);
+
+static inline
+void vec_pany_init_na_rm(const int* v_x, r_ssize size, int* v_out);
+static inline
+void vec_pany_fill_na_rm(const int* v_x, r_ssize size, int* v_out);
+
+static inline
+void vec_pall_init_na_keep(const int* v_x, r_ssize size, int* v_out);
+static inline
+void vec_pall_fill_na_keep(const int* v_x, r_ssize size, int* v_out);
+
+static inline
+void vec_pany_init_na_keep(const int* v_x, r_ssize size, int* v_out);
+static inline
+void vec_pany_fill_na_keep(const int* v_x, r_ssize size, int* v_out);

src/init.c

@@ -150,6 +150,8 @@ extern r_obj* ffi_list_all_vectors(r_obj*, r_obj*);
 extern r_obj* ffi_list_check_all_vectors(r_obj*, r_obj*);
 extern r_obj* ffi_as_short_length(r_obj*, r_obj*);
 extern r_obj* ffi_s3_get_method(r_obj*, r_obj*, r_obj*);
+extern r_obj* ffi_vec_pall(r_obj*, r_obj*, r_obj*, r_obj*);
+extern r_obj* ffi_vec_pany(r_obj*, r_obj*, r_obj*, r_obj*);
 
 
 // Maturing
@@ -321,6 +323,8 @@ static const R_CallMethodDef CallEntries[] = {
   {"ffi_list_check_all_vectors",            (DL_FUNC) &ffi_list_check_all_vectors, 2},
   {"ffi_as_short_length",                   (DL_FUNC) &ffi_as_short_length, 2},
   {"ffi_s3_get_method",                     (DL_FUNC) &ffi_s3_get_method, 3},
+  {"ffi_vec_pall",                          (DL_FUNC) &ffi_vec_pall, 4},
+  {"ffi_vec_pany",                          (DL_FUNC) &ffi_vec_pany, 4},
   {"ffi_exp_vec_cast",                      (DL_FUNC) &exp_vec_cast, 2},
   {NULL, NULL, 0}
 };