R/zdd_or.R

In jordagaman/zddr: A zero-suppressed binary decision diagram implementation for R

#' zdd_or
#'
#' @param ... a list of objects that can be coerced to zdds
#'
#' @return a zdd that is the union of all inputs
#' @export
#'
#' @examples
#' zdd_or(1,2,3,4)
#' zdd_or( zdd_and(1,2), zdd_and(3,4) ) # { {1,2}, {3,4} }
#' zdd_or( zdd_and(1,2,3,4,5), 4)       # { {4} }
zdd_or <- function(...) {
  input_list <- list(...)
  return(lzdd_or(input_list))
}

lzdd_or <- function(lst) {
  input_list <- purrr::map_chr(lst, as_zdd)  # future optimization opportunity: only as_zdd everything if there are no zdd1's in here
  if(any(purrr::map_lgl(input_list, is_one)))
    return(as_zdd(T))
  return(
    purrr::reduce(input_list, zdd_union)  # this would probably be faster if the list was sorted first
  )
}