Mercurial > repos > public > sbplib

view diags.m @ 641:070d578997f6 feature/d1_staggered

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add helper function for primal and dual 1D staggered grid, and a test for that function.
author Martin Almquist <malmquist@stanford.edu>
date Sat, 11 Nov 2017 20:26:37 -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