R/percolate.R

In melody789/percolate: What the Package Does (One Line, Title Case)

#' percoalte function
#'
#' @param x board
#' @param ...
#'
#' @return a borad
#' @export
#'
#' @examples percolate(myboard())
percolate <- function(x, ...) {
  UseMethod("percolate")
}
#'Check if there is more possible open in the baords
#'
#' @param x board object
#'
#' @return boolean true or false
#' @export
#'
#' @examples check_possible(board(matrix(1, 10, 10)))
check_possible <- function(x) {
  n <- ncol(x)
  dirs <- list(list(0, -1), list(0, 1), list(1, 0), list(-1, 0))
  for (i in 1:n) {
    for (j in 1:n) {
      if (x[i, j] == 1) {
        for (d in 1:4) {
          d1 <- dirs[[d]][[1]]
          d2 <- dirs[[d]][[2]]
          if ((1<=i+d1 && i+d1<=n && 1<=j+d2 && j+d2<=n) && (x[i+d1, j+d2] == 2)) {
            return (TRUE)
          }
        }
      }
    }
  }
  return (FALSE)
}
#' percolate the boards
#'
#' @param x board object
#'
#' @return a percolate matrix
#' @export
#'
#' @examples percolate(board(matrix(1, 10, 10)))
percolate.board <- function(x) {
  is_valid(x)
  n <- ncol(x)
  res <- FALSE
  assert_that(all(x == 0 | x == 1), msg = "not valid matrix")
  #fill first row
  for (j in 1:n) {
    if (x[1, j] == 1) {
      x[1, j] = 2
      res <- TRUE
    }
  }
  #fill rest
  dirs <- list(list(0, -1), list(0, 1), list(1, 0), list(-1, 0))
  while (check_possible(x)) {
    for (i in 1:n) {
      for (j in 1:n) {
        if (x[i, j] == 2) {
          for (d in 1:4) {
            d1 <- dirs[[d]][[1]]
            d2 <- dirs[[d]][[2]]
            if ((1<=i+d1 && i+d1<=n && 1<=j+d2 && j+d2<=n) && (x[i+d1, j+d2] == 1)) {
              x[i+d1, j+d2] = 2
              res <- TRUE
            }
          }
        }
      }
    }
  }
  return (list(result_board = x, result = res))
}