Revision 9e5b9bb8a0564f8bf629df4ca5e96d3078f50363 authored by jmc on 14 July 2008, 00:00:00 UTC, committed by Gabor Csardi on 14 July 2008, 00:00:00 UTC
1 parent a3b36b1
Raw File
triDiag2.R
triDiag2 <- function(diagonal, upper, lower,
                    nrow = length(diagonal), ncol = nrow) {
    value <- diag(diagonal, nrow = nrow, ncol = ncol)
    n <- min(nrow, ncol-1)
    if(n>0) {
        rseq <- 1:n
        value[cbind(rseq, rseq+1)] <- upper
    }
    n <- min(nrow-1, ncol)
    if(n > 0) {
        rseq <- 1:n
        value[cbind(rseq+1, rseq)] <- lower
    }
    value
}
back to top