R/utilities.R

In erocoar/vetoboxr: Spatial Voting in R

create_index <- function(idx, dimension, sort = TRUE) {
  if (dimension > 1) {
    idx <- c(idx, idx + seq(dimension - 1))
    if (isTRUE(sort)) {
      idx <- sort(idx)
    }
  }
  idx
}

get_range_x <- function(vote) {
  x_idx <- seq(1, ncol(vote$voter_position), vote$dimension)
  xmin <- min(
    vote$voter_position[, x_idx] - vote$voter_radii,
    vote$voter_position[, x_idx] + vote$voter_radii,
    vote$status_quo[, 1],
    vote$outcome[, 1]
  ) * 1
  xmax <- max(
    vote$voter_position[, x_idx] - vote$voter_radii,
    vote$voter_position[, x_idx] + vote$voter_radii,
    vote$status_quo[, 1],
    vote$outcome[, 1]
  ) * 1
  c(xmin, xmax)
}

get_range_y <- function(vote) {
  if (vote$dimension == 2) {
    y_idx <- seq(1, ncol(vote$voter_position), vote$dimension) + 1
    ymin <- min(
      vote$voter_position[, y_idx] - vote$voter_radii,
      vote$voter_position[, y_idx] + vote$voter_radii,
      vote$status_quo[, 2],
      vote$outcome[, 2]
    ) * 1
    ymax <- max(
      vote$voter_position[, y_idx] - vote$voter_radii,
      vote$voter_position[, y_idx] + vote$voter_radii,
      vote$status_quo[, 2],
      vote$outcome[, 2]
    ) * 1
  } else {
    ymin <- 0
    ymax <- max(
      vote$voter_position + vote$voter_radii / 2
    ) * 1
  }
  c(ymin, ymax)
}

seq_v <- function(x) seq(nrow(x))

seq_h <- function(x) seq(ncol(x))