Added code and tests for simuloiAlleeli

R/admix1.R

@@ -478,12 +478,4 @@ admix1 <- function(tietue) {
 #         end
 #     end
 #     counter = counter+rowsFromInd;
-# end
-
-# function all = simuloiAlleeli(allfreqs,pop,loc)
-# % Simuloi populaation pop lokukseen loc alleelin.
-# freqs = allfreqs(:,loc,pop);
-# cumsumma = cumsum(freqs);
-# arvo = rand;
-# isommat = find(cumsumma>arvo);
-# all = min(isommat);
+# end

R/simuloiAlleeli.R

@@ -0,0 +1,17 @@
+#' @title simuloiAlleeli
+#' @description Simuloi populaation pop lokukseen loc alleelin.
+#' @note This function is (only?) called by `simulateIndividuals()`. Therefore, exporting it is probably unnecessary.
+#' @export
+
+simuloiAlleeli <- function(allfreqs, pop, loc) {
+    if (length(dim(allfreqs)) == 3) { # distinguish between arrays and matrices
+        freqs <- allfreqs[, loc, pop]
+    } else {
+        freqs <- allfreqs[, loc]
+    }
+    cumsumma <- cumsum(freqs)
+    arvo <- runif(1)
+    isommat <- which(cumsumma > arvo)
+    all <- min(isommat)
+    return(all)
+}