view +sbp/diagInd.m @ 912:1cdf5ead2a16 feature/utux2D

Refactor interface method for Schrodinger2d
author Martin Almquist <malmquist@stanford.edu>
date Sun, 25 Nov 2018 19:46:39 -0800
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