Mercurial > repos > public > sbplib

view diags.m @ 885:18e10217dca9 feature/d1_staggered

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Generalize Staggered1DAcoustics to handle variable coefficients defined by both function handles and vectors.
author Martin Almquist <malmquist@stanford.edu>
date Sun, 04 Nov 2018 12:36:30 -0800
parents 0b87863cb29c
children
line wrap: on
line source

function A = diags(B,d,m,n)
    assert(size(B,1) == m);

    A = repmat(B(:,1)*0, [1, n]);

    for i = 1:size(B,2)
        A(:,d(i)+ (1:m)) = A(:,d(i)+ (1:m)) + diag(B(:,i));
    end
end