R/set_threads.R

In modsem: Latent Interaction (and Moderation) Analysis in Structural Equation Models (SEM)

ThreadEnv <- rlang::env(n.threads = NULL)


setThreadEnv <- function(n.threads) {
  ThreadEnv$n.threads <- n.threads
}


resetThreadEnv <- function() {
  setThreadEnv(NULL)
}


setThreads <- function(n) {
  if (is.null(n)) k <- getDefaultThreads()
  else if (is.numeric(n)) k <- getThreadsN(n)
  else if (is.character(n)) {
    k <- switch(n,
                "default" = getDefaultThreads(),
                "max" = getMaxThreads(),
                "min" = getMinThreads(),
                stop2("Invalid string specifying number of threads"))
  } else stop2("Invalid number of threads, must be integer, NULL, or character")

  ThreadEnv$n.threads <- k
}


resetThreads <- function() {
  resetThreadEnv()
}


getDefaultThreads <- function() {
  defaultCRAN <- 2
  getThreadsN(defaultCRAN)
}


getMaxThreads <- function() {
  parallel::detectCores()
}


getMinThreads <- function() {
  getThreadsN(n = 1)
}


getThreadsN <- function(n) {
  ncores <- parallel::detectCores()
  ifelse(n >= ncores, ncores, n)
}