Mercurial > repos > public > sbplib
view spdiagVariable.m @ 885:18e10217dca9 feature/d1_staggered
Generalize Staggered1DAcoustics to handle variable coefficients defined by both function handles and vectors.
author | Martin Almquist <malmquist@stanford.edu> |
---|---|
date | Sun, 04 Nov 2018 12:36:30 -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