max/ourMELONS
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Adjusted behavior to benchmark

Browse source
This commit is contained in:
Waldir Leoncio 2020-01-30 17:52:20 +01:00
parent c9c9b7cb26
commit f6768c67bc
3 changed files with 22 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
R/simulateAllFreqs.R
View file

@ -5,23 +5,27 @@
#' @param COUNTS COUNTS
#' @export
simulateAllFreqs <- function(noalle, COUNTS = matrix(0)) {
simulateAllFreqs <- function(noalle, COUNTS = matrix(NA, 0, 0)) {
max_noalle <- size(COUNTS, 1)
nloci <- size(COUNTS, 2)
npops <- size(COUNTS, 3)
prioriAlleelit <- zeros(max_noalle, nloci)
for (j in 1:nloci) {
prioriAlleelit[1:noalle[j], j] <- 1 / noalle[j]
if (nloci > 0) {
for (j in 1:nloci) {
prioriAlleelit[1:noalle[j], j] <- 1 / noalle[j]
}
}
prioriAlleelit <- repmat(prioriAlleelit, matrix(c(1, 1, npops), ncol = 1))
prioriAlleelit <- repmat(prioriAlleelit, matrix(c(1, 1, npops), 1))
counts <- COUNTS + prioriAlleelit
allfreqs <- zeros(size(counts))
for (i in 1:npops) {
for (j in 1:nloci) {
simuloidut <- randdir(counts[1:noalle[j], j, i] , noalle[j])
allfreqs[1:noalle[j], j, i] <- simuloidut
if (nloci > 0) {
for (j in 1:nloci) {
simuloidut <- randdir(counts[1:noalle[j], j, i] , noalle[j])
allfreqs[1:noalle[j], j, i] <- simuloidut
}
}
}
return(allfreqs)