R/icompress.R

In iterors: Fast, Compact Iterators and Tools

#' Iterator that filters elements where corresponding selector is false.
#'
#' Constructs an iterator that filters elements from iterable returning only
#' those for which the corresponding element from \code{selectors} is
#' \code{TRUE}.
#'
#' The iterator stops when either \code{object} or \code{selectors} has been
#' exhausted.
#'
#' @export
#' @param object an iterable object
#' @param selectors an iterable that determines whether the corresponding
#' element in \code{object} is returned.
#' @return iterator object
#'
#' @examples
#' # Filters out odd numbers and retains only even numbers
#' n <- 10
#' selectors <- rep(c(FALSE, TRUE), n)
#' it <- i_mask(seq_len(n), selectors)
#' as.list(it)
#'
#' # Similar idea here but anonymous function is used to filter out even
#' # numbers
#' n <- 10
#' it2 <- i_mask(seq_len(10), rep(c(TRUE, FALSE), n))
#' as.list(it2)
#'
#' it3 <- i_mask(letters, letters %in% c('a', 'e', 'i', 'o', 'u'))
#' as.list(it3)
i_mask <- function(object, selectors) {
  iter_i_zip <- i_zip(obj=object, select=selectors)

  nextOr_ <- function(or) {
    repeat {
      next_elem <- iter_i_zip(or = return(or))
      if (next_elem$select) {
        return(next_elem$obj)
      }
    }
  }

  iteror(nextOr_)
}