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

Improved function behavior on empty matrices

Browse source
This commit is contained in:
Waldir Leoncio 2020-02-25 14:28:39 +01:00
parent 8323d54c01
commit 37822b2ec8
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
R/computeAllFreqs2.R
View file

@ -6,7 +6,7 @@
#' @param SUMCOUNTS sumcounts #' @param SUMCOUNTS sumcounts
#' @export #' @export
computeAllFreqs2 <- function (noalle, COUNTS = matrix(NA, 0, 0), computeAllFreqs2 <- function (noalle, COUNTS = matrix(NA, 0, 0),
SUMCOUNTS = sum(COUNTS)) { SUMCOUNTS = NA) {
max_noalle <- size(COUNTS, 1) max_noalle <- size(COUNTS, 1)
nloci <- size(COUNTS,2) nloci <- size(COUNTS,2)
@ -17,8 +17,10 @@ computeAllFreqs2 <- function (noalle, COUNTS = matrix(NA, 0, 0),
sumCounts <- repmat(sumCounts, c(max_noalle, 1, 1)) sumCounts <- repmat(sumCounts, c(max_noalle, 1, 1))
prioriAlleelit <- zeros(max_noalle, nloci) prioriAlleelit <- zeros(max_noalle, nloci)
for (j in 1:nloci) { if (nloci > 0) {
prioriAlleelit[1:noalle[j], j] <- 1 / noalle[j] for (j in 1:nloci) {
prioriAlleelit[1:noalle[j], j] <- 1 / noalle[j]
}
} }
prioriAlleelit <- repmat(prioriAlleelit, c(1, 1, npops)) prioriAlleelit <- repmat(prioriAlleelit, c(1, 1, npops))
counts <- COUNTS + prioriAlleelit counts <- COUNTS + prioriAlleelit