100 lines
2.3 KiB
R
100 lines
2.3 KiB
R
parse_args <- function(x, mc) {
|
|
fmls <- lapply(x, eval)
|
|
if (!all(names(mc) %in% names(fmls))) {
|
|
stop(
|
|
"Invalid argument(s): ",
|
|
paste(setdiff(names(mc), names(fmls)), collapse = ", ")
|
|
)
|
|
}
|
|
args <- lapply(names(fmls), function(n) {
|
|
tryCatch(
|
|
{
|
|
fmls[[n]]$check(eval(mc[[n]]))
|
|
},
|
|
error = function(e) {
|
|
stop("Invalid argument '", n, "': ", e$message, call. = FALSE)
|
|
}
|
|
)
|
|
})
|
|
names(args) <- names(fmls)
|
|
args
|
|
}
|
|
|
|
check_args <- function(args, fmls) {
|
|
args <- as.list(args)[-1]
|
|
fmls <- lapply(fmls, eval)
|
|
lapply(names(fmls), function(n) {
|
|
tryCatch(
|
|
{
|
|
fmls[[n]]$check(args[[n]])
|
|
},
|
|
error = function(e) {
|
|
stop("Invalid argument '", n, "': ", e$message, call. = FALSE)
|
|
}
|
|
)
|
|
})
|
|
}
|
|
|
|
ts_result <- function(type, value) {
|
|
if (type$check(value)) {
|
|
return(value)
|
|
}
|
|
stop("Expected a value of type ", type$type)
|
|
}
|
|
|
|
#' TS function definition
|
|
#'
|
|
#' @param f an R function
|
|
#' @param ... argument definitions (only required if f does not specify these in its formals)
|
|
#' @param result return type (ignored if overloads are provided)
|
|
#' @export
|
|
ts_function <- function(f, ..., result = ts_void()) {
|
|
args <- list(...)
|
|
if (!is.null(result) && !is_ts_object(result)) {
|
|
stop("Invalid return type")
|
|
}
|
|
|
|
if (length(args) == 0) {
|
|
args <- lapply(formals(f), eval)
|
|
}
|
|
|
|
e <- new.env()
|
|
e$f <- f
|
|
# e$env <- env
|
|
e$args <- args
|
|
e$result <- result
|
|
|
|
e$call <- function(...) {
|
|
mc <- match.call(e$f)
|
|
.args <- parse_args(args, mc[-1])
|
|
.res <- do.call(e$f, .args)
|
|
check_type(result, .res)
|
|
}
|
|
|
|
e$copy <- function(env = parent.frame()) {
|
|
e2 <- e
|
|
environment(e2$f) <- rlang::env_clone(environment(e$f), env)
|
|
e2
|
|
}
|
|
|
|
class(e) <- "ts_function"
|
|
e
|
|
}
|
|
|
|
|
|
|
|
#' @export
|
|
print.ts_function <- function(x, ...) {
|
|
cli::cli_h3("Ocap function")
|
|
|
|
cli::cli_text("Arguments:")
|
|
args <- lapply(x$args, \(z) z$input_type)
|
|
lapply(names(args), \(n) {
|
|
cat("- ", n, ": ", args[[n]], "\n", sep = "")
|
|
})
|
|
cli::cli_text("\n\n")
|
|
|
|
cli::cli_text("Return type:")
|
|
cat(x$result$return_type)
|
|
}
|