R/get_in_range.R

In carolyner/imabc: Incremental Mixture Approximate Bayesian Computation (IMABC)

get_in_range <- function(compare_list, check_dt, criteria = NULL, out = "logical") {
  UseMethod("get_in_range")
}

get_in_range.targets <- function(compare_list, check_dt, criteria, out = "logical") {
  # Determine which rows are between the selected bounds. In range == 0, out of range == -1
  check <- sapply(attr(compare_list, which = "target_names"), FUN = function(x, dt, compare_list) {
    if (criteria == "start") {
      in_range(dt[[x]], compare_list$current_lower_bounds[x], compare_list$current_upper_bounds[x]) - 1
    } else if (criteria == "update") {
      in_range(dt[[x]], compare_list$new_lower_bounds[x], compare_list$new_upper_bounds[x]) - 1
    } else if (criteria == "stop") {
      in_range(dt[[x]], compare_list$stopping_lower_bounds[x], compare_list$stopping_upper_bounds[x]) - 1
    }
  }, dt = check_dt, compare_list = compare_list)

  # If grouped targets exist, aggregate distances
  if (inherits(compare_list, "grouped")) {
    # Since in range values are 0 and out of range values are -1, any target group with at least one target out of range
    #  will have a negative check value
    check <- t(rowsum(t(check), attr(compare_list, which = "target_groups"), reorder = FALSE))
  }

  # Set output type
  if (out == "logical") {
    # Set 0 (in range) to TRUE, set anything else (out of range) to FALSE
    check <- check == 0
  } else {
    # Make sure groups with more than one out of range value are given a -1
    check <- pmax(check, -1)
    # Convert in range values to 1 while keeping out of range values as -1
    check <- (check + 1)*2 - 1
  }

  return(check)
}

get_in_range.priors <- function(compare_list, check_dt, out = "logical") {
  # All parms must be in range based on their prior information
  check <- Reduce(`&`, lapply(names(compare_list), FUN = function(y, check_dt, compare_list) {
    check_dt[[y]] >= attr(compare_list, which = "mins")[y] &
      check_dt[[y]] <= attr(compare_list, which = "maxs")[y]
  }, check_dt = check_dt, compare_list = compare_list))
  check <- as.numeric(check)
  check[is.na(check)] <- 0

  # Set output type
  if (out == "logical") {
    # Set 1 (in range) to TRUE, set anything else (out of range) to FALSE
    check <- check == 1
  } else {
    # Convert out of range values as -1 while keeping in range values as 1
    check <- (check)*2 - 1
  }

  return(check)
}