Mercurial > repos > public > sbplib

view diags.m @ 718:71aa5828cbbf feature/utux2D

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add Schrödinger scheme for 2d single block. Will develop to multiblock with interpolation.
author Martin Almquist <malmquist@stanford.edu>
date Sat, 03 Mar 2018 16:18:33 -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