diff --git a/R/rmot_lm.R b/R/rmot.R similarity index 70% rename from R/rmot_lm.R rename to R/rmot.R index 2d6b6b9..16892a1 100644 --- a/R/rmot_lm.R +++ b/R/rmot.R @@ -10,8 +10,7 @@ #' @examples #' mtcars #' rmot_lm(mtcars$mpg, mtcars$disp) -rmot_lm <- function(x, y, ...) { - standata <- list(x = x, y = y, N = length(y)) - out <- rstan::sampling(stanmodels$lm, data = standata, ...) +rmot <- function(x, y, ...) { + out <- rstan::sampling(stanmodels$model, data = standata, ...) return(out) } diff --git a/R/rmot_models.R b/R/rmot_models.R new file mode 100644 index 0000000..aa94057 --- /dev/null +++ b/R/rmot_models.R @@ -0,0 +1,61 @@ +# Set list structures for different models +# An example for lm + +rmot_lm <- function(){ +list(X = NULL, + Y = NULL, + N = NULL, + model = "linear") +} + + +# Need a mechanism to select models +# rmot_config(model = "linear") + +rmot_config <- function(model=NULL){ + output <- switch(model, + linear = rmot_lm()) + + class(output) <- "rmot_object" + + return(output) +} + +# Need a mechanism to take user data and assign to slots in list +rmot_assign_data <- function(model_template, field, data){ + purrr::assign_in(model_template, field, data) +} + + +rmot_assign_data <- function(model_template, ...){ + # Grab user expressions + user_code <- rlang::enexprs(..., .check_assign = TRUE) + + # Evaluate the RHS of expressions (the values) + data <- purrr::map(user_code, + eval) + + # Grab the names + fields <- names(user_code) + + for(i in fields){ + model_template <- purrr::list_modify(model_template, !!!data[i]) + } + + return(model_template) +} + + + + +list_rename = function(data, ...) { + mapping = sapply( + rlang::enquos(...), + rlang::as_name + ) + new_names = stats::setNames(nm=names(data)) + # `new_name = old_name` for consistency with `dplyr::rename` + new_names[mapping] = names(mapping) + # for `old_name = new_name` use: `new_names[names(mapping)] = mapping` + stats::setNames(data, new_names) +}