32a08784bf72b9a6c0826af799b94b15836d91de
1 // Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
3 //
4 // Copyright (C) 2012 - 2016 - Scilab Enterprises
5 //
6 // This file is hereby licensed under the terms of the GNU GPL v2.0,
7 // pursuant to article 5.3.4 of the CeCILL v.2.1.
8 // This file was originally licensed under the terms of the CeCILL v2.1,
9 // and continues to be available under such terms.
10 // For more information, see the COPYING file which you should have received
11 // along with this program.
13 function d=%sp_diag(a,k)
14     // %sp_diag - implement diag function for sparse matrix, rational matrix ,..
16     [lhs,rhs]=argn(0)
17     if rhs==1 then k=0,end
19     [ij,v,sz]=spget(a)
20     m=sz(1);n=sz(2)
21     if m>1&n>1 then
22         l=find(ij(:,1)==(ij(:,2)-k))
23         if k<=0 then
24             mn=min(m+k,n)
25             i0=-k
26         else
27             mn=min(m,n-k)
28             i0=0
29         end
30         kk=abs(k)
31         if l==[] then d=sparse([],[],[mn,1]);return;end
32         d=sparse([ij(l,1)-i0,ones(ij(l,1))],v(l),[mn,1])
33     else
34         if m>1 then ij=ij(:,1);else ij=ij(:,2);end
35         nn = max(m,n)+abs(k)
36         if ij==[] then
37             d=sparse([],[],[nn,nn])
38         else
39             if k>0 then
40                 d=sparse([ij,ij+k],v,[nn,nn])
41             else
42                 d=sparse([ij-k,ij],v,[nn,nn])
43             end
44         end
45     end
46 endfunction