Mercurial > repos > public > sbplib

annotate diags.m @ 602:2dee6c9e827d feature/grids

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add assert function for number of arguments
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 28 Sep 2017 11:15:29 +0200
parents 0b87863cb29c
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
412
5eb312dfe2ab Add full matrix equivalent of spdiags
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
1 function A = diags(B,d,m,n)
5eb312dfe2ab Add full matrix equivalent of spdiags
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
2 assert(size(B,1) == m);
5eb312dfe2ab Add full matrix equivalent of spdiags
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
3
5eb312dfe2ab Add full matrix equivalent of spdiags
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
4 A = repmat(B(:,1)*0, [1, n]);
5eb312dfe2ab Add full matrix equivalent of spdiags
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
5
5eb312dfe2ab Add full matrix equivalent of spdiags
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
6 for i = 1:size(B,2)
5eb312dfe2ab Add full matrix equivalent of spdiags
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
7 A(:,d(i)+ (1:m)) = A(:,d(i)+ (1:m)) + diag(B(:,i));
5eb312dfe2ab Add full matrix equivalent of spdiags
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8 end
413
0b87863cb29c Add missing eol
Jonatan Werpers <jonatan@werpers.com>
parents: 412
diff changeset
9 end