view +sbp/diagInd.m @ 737:521076a6ffa5 feature/d1_staggered

Update Runge Kutta for discrete data to return stage rates if needed.
author Martin Almquist <malmquist@stanford.edu>
date Wed, 25 Apr 2018 16:58:45 -0700
parents 32b39dc44474
children
line wrap: on
line source

function A = diagInd(d,n,m)
    A = zeros(n,length(d));
    for i = 1:length(d)
        i0 = 1;
        j0 = d(i)+1;

        I = i0 + (0:(n-1))';
        J = j0 + (0:(n-1))';

        A(:,i) = matInd2VecInd(I,J,n);

    end
end

function I = matInd2VecInd(i,j,n)
    I = i + (j-1)*n;
end