Mercurial > repos > public > sbplib

view diags.m @ 917:878652b22157 feature/timesteppers

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Make time.rk.rungekutta return stage approximations and stage rates in addition to the solution after one time step.
author Martin Almquist <malmquist@stanford.edu>
date Mon, 26 Nov 2018 16:12:27 -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