load_stein.R

# Load all package sources from R/ into the global environment (script workflow).
# For package installs use: devtools::load_all() or library(steingof)
# Set STEIN_SRC_ROOT to the repository root if getwd() is not the project root
# (e.g. parallel workers).
(function() {
  root <- Sys.getenv("STEIN_SRC_ROOT", unset = "")
  if (!nzchar(root)) {
    root <- normalizePath(".", mustWork = FALSE)
    Sys.setenv(STEIN_SRC_ROOT = root)
  }
  rd <- file.path(root, "R")
  if (!dir.exists(rd)) {
    stop("Cannot find R/ under ", root, "; set STEIN_SRC_ROOT to the repo root.", call. = FALSE)
  }
  owd <- getwd()
  setwd(rd)
  on.exit(setwd(owd), add = TRUE)
  collate <- c(
    "stein_helpers.R",
    "kernel_classes.R",
    "bootstrap.R",
    "gmm_model.R",
    "ksd_u_test.R",
    "ksd_v_test.R",
    "fssd_test.R",
    "stein_discrepancy_gof.R",
    "svgd.R",
    "stein_thinning.R",
    "stein_points.R",
    "stein_point_mcmc.R"
  )
  for (fn in collate) {
    if (!file.exists(fn)) {
      stop("Missing required file: ", file.path(rd, fn), call. = FALSE)
    }
    sys.source(fn, envir = .GlobalEnv, keep.source = FALSE)
  }
  invisible(TRUE)
})()