fix rTraitDisc

[ape.git] / R / rTrait.R

diff --git a/R/rTrait.R b/R/rTrait.R
index 3359df6f71da40c3b8a5d7a254844bafb50b79e3..642591e6e2523d2ac0da057b861f1831546f6621 100644 (file)
--- a/R/rTrait.R
+++ b/R/rTrait.R
@@ -1,4 +1,4 @@
-## rTrait.R (2010-01-11)
+## rTrait.R (2010-02-03)
 
 ##   Trait Evolution
 
@@ -59,6 +59,7 @@ rTraitDisc <-
         for (i in N:1) x[des[i]] <- model(x[anc[i]], el[i])
     } else {
         diag(Q) <- -rowSums(Q)
+        freq <- rep(freq, each = k)
         for (i in N:1) {
             p <- matexpo(Q * freq * el[i])[x[anc[i]], ]
             x[des[i]] <- .Internal(sample(k, size = 1, FALSE, prob = p))