view spdiagVariable.m @ 718:71aa5828cbbf feature/utux2D

Add Schrödinger scheme for 2d single block. Will develop to multiblock with interpolation.
author Martin Almquist <malmquist@stanford.edu>
date Sat, 03 Mar 2018 16:18:33 -0800
parents 7368affc8f78
children
line wrap: on
line source

function A = spdiagVariable(a,i)
    default_arg('i',0);

    if isrow(a)
        a = a';
    end

    n = length(a)+abs(i);

    if i > 0
    	a = [sparse(i,1); a];
    elseif i < 0
    	a = [a; sparse(abs(i),1)];
    end

    A = spdiags(a,i,n,n);
end