view +sbp/diagInd.m @ 1330:855871e0b852 feature/D2_boundary_opt

Add size checks to the operators
author Vidar Stiernström <vidar.stiernstrom@it.uu.se>
date Thu, 17 Feb 2022 18:55:11 +0100
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