R/abstract-model-impl.R

In dirkschumacher/romp: Model and Solve Mixed Integer Linear Programs

#' @export
add_variable.abstract_model <- function(.model, .variable, ...,
                                        type = "continuous",
                                        lb = -Inf, ub = Inf) {
  add_variable_(
    .model = .model,
    .variable = lazyeval::as.lazy(
      substitute(.variable), parent.frame()
    ),
    type = type,
    lb = lb,
    ub = ub,
    .dots = lazyeval::lazy_dots(...)
  )
}

#' @export
set_objective.abstract_model <- function(model, expression,
                                         sense = c("max", "min")) {
  set_objective_(model,
    expression = lazyeval::as.lazy(
      substitute(expression),
      parent.frame()
    ),
    sense = sense
  )
}

#' @export
add_constraint.abstract_model <- function(.model, .constraint_expr, ...,
                                          .show_progress_bar = TRUE) {
  add_constraint_(.model, lazyeval::as.lazy(
    substitute(.constraint_expr),
    parent.frame()
  ),
  .dots = lazyeval::lazy_dots(...),
  .show_progress_bar = .show_progress_bar
  )
}

#' @export
set_bounds.abstract_model <- function(.model, .variable, ...,
                                      lb = NULL, ub = NULL) {
  set_bounds_(
    .model = .model,
    .variable = lazyeval::as.lazy(
      substitute(.variable), parent.frame()
    ),
    lb = lb,
    ub = ub,
    .dots = lazyeval::lazy_dots(...)
  )
}

#' @export
solve_model.abstract_model <- function(model, solver) {
  if (!is.function(solver)) {
    stop(paste0(
      "Solver is not a function Model -> Solution.\n",
      "Take a look at the examples on the website on how to call",
      " solve_model."
    ))
  }
  solver(model)
}