add argument to avoid forking

stemangiola · Nov 3, 2023 · d3b7438 · d3b7438
1 parent 0444538
commit d3b7438
Showing 1 changed file with 20 additions and 5 deletions.
diff --git a/R/glmmSeq.R b/R/glmmSeq.R
@@ -453,7 +453,7 @@ glmmSeq = function (modelFormula, countdata, metadata, id = NULL, dispersion = N
                     sizeFactors = NULL, reduced = NULL, modelData = NULL, designMatrix = NULL,
                     method = c("lme4", "glmmTMB"), control = NULL, family = glmmTMB::nbinom2,
                     cores = 1, removeSingles = FALSE, zeroCount = 0.125, verbose = TRUE,
-                    returnList = FALSE, progress = FALSE, max_rows_for_matrix_multiplication = Inf, ...)
+                    returnList = FALSE, progress = FALSE, max_rows_for_matrix_multiplication = Inf, avoid_forking = FALSE, ...)
 {
   glmmcall <- match.call(expand.dots = TRUE)
   method <- match.arg(method)
@@ -617,23 +617,38 @@ glmmSeq = function (modelFormula, countdata, metadata, id = NULL, dispersion = N
       }
     }
     else {
-      if (progress) {
-
+      if(avoid_forking){
+        library(parallel)
+        cl = parallel::makeCluster(cores, type = "PSOCK")
+        #parallel::clusterEvalQ(cl,c(library(dplyr),library(glmmSeq)))
+        #clusterExport(cl, list("varname1", "varname2"),envir=environment())
+        resultList <- parallel::clusterApply(
+          cl,
+          fullList,
+          function(geneList) {
+            glmerCore(geneList, fullFormula, reduced,
+                      subsetMetadata, control, offset, modelData,
+                      designMatrix, hyp.matrix,, max_rows_for_matrix_multiplication = max_rows_for_matrix_multiplication, ...)
+          }
+        )
+      } 
+      else if (progress) {
+
         # Check if package is installed, otherwise install
         if (find.package("pbmcapply", quiet = TRUE) %>% length %>% equals(0)) {
           message("tidybulk says: Installing pbmcapply needed for differential transcript abundance analyses")
           if (!requireNamespace("BiocManager", quietly = TRUE))
             install.packages("BiocManager", repos = "https://cloud.r-project.org")
           BiocManager::install("pbmcapply", ask = FALSE)
         }
-
+        
         resultList <- pbmcapply::pbmclapply(fullList, function(geneList) {
           glmerCore(geneList, fullFormula, reduced,
                     subsetMetadata, control, offset, modelData,
                     designMatrix, hyp.matrix, , max_rows_for_matrix_multiplication = max_rows_for_matrix_multiplication, ...)
         }, mc.cores = cores)
         if ("value" %in% names(resultList)) resultList <- resultList$value
-
+        
       }
       else {
         resultList <- mclapply(fullList, function(geneList) {