Mercurial > repos > public > sbplib
diff spdiagVariable.m @ 756:f891758ad7a4 feature/d1_staggered
Merge with feature/utux2d.
| author | Martin Almquist <malmquist@stanford.edu> |
|---|---|
| date | Sat, 16 Jun 2018 14:30:45 -0700 |
| parents | 7368affc8f78 |
| children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/spdiagVariable.m Sat Jun 16 14:30:45 2018 -0700 @@ -0,0 +1,17 @@ +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 \ No newline at end of file