R/aa_caching.R

Defines functions gemma_memoise in_memory_cache gemmaCache

Documented in gemmaCache gemma_memoise

#' Gemma Cache
#' @return A memoise filesystem
#' @keywords internal
gemmaCache <- function(){
  dir <- getOption(x = "gemma.cache",
                   rappdirs::user_cache_dir(appname ='gemmaR'))

  if(dir == "cache_in_memory"){
      return("cache_in_memory")
  }
  dir.create(dir,recursive = TRUE,showWarnings = FALSE)
  memoise::cache_filesystem(dir, algo = "xxhash64", compress = FALSE)
}

in_memory_cache <- function(...){
    args <- list(...)
    do.call(get(args[[1]]), args[-1])
}

#' Enable and disable memoisation of gemma.R functions
#' @param memoised boolean. If TRUE memoisation will be enabled
#' @param cache File path or "cache_in_memory". File path will chose a location
#' to save the cache files for memoisation. "cache_in_memory" will store the cache
#' in the current R session
#' @return NULL
#' @keywords misc
#' @export
gemma_memoise <- function(memoised = FALSE,
                          cache = rappdirs::user_cache_dir(appname ='gemmaR')){
    options('gemma.memoised' = memoised)
    options('gemma.cache' = cache)
    NULL
}

mem_in_memory_cache <- memoise::memoise(in_memory_cache, cache = memoise::cache_memory())
jsicherman/Gemma-API documentation built on Oct. 25, 2024, 5:18 p.m.