view diags.m @ 814:3a5e635a93fd feature/burgers1d

Add scheme for 1D Burgers equation - Add scheme for discretizing the 1D burgers equation, with spatially variable coefficients.
author Vidar Stiernstrom <vidar.stiernstrom@it.uu.se>
date Mon, 03 Sep 2018 14:50:27 +0200
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