GPL + CeCILL Header change
[scilab.git] / scilab / modules / elementary_functions / macros / %sp_diag.sci
1 // Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
2 // Copyright (C) INRIA
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.
12 // === LICENSE_END ===
13
14 function d=%sp_diag(a,k)
15     // %sp_diag - implement diag function for sparse matrix, rational matrix ,..
16
17     [lhs,rhs]=argn(0)
18     if rhs==1 then k=0,end
19
20     [ij,v,sz]=spget(a)
21     m=sz(1);n=sz(2)
22     if m>1&n>1 then
23         l=find(ij(:,1)==(ij(:,2)-k))
24         if k<=0 then
25             mn=min(m+k,n)
26             i0=-k
27         else
28             mn=min(m,n-k)
29             i0=0
30         end
31         kk=abs(k)
32         if l==[] then d=sparse([],[],[mn,1]);return;end
33         d=sparse([ij(l,1)-i0,ones(ij(l,1))],v(l),[mn,1])
34     else
35         if m>1 then ij=ij(:,1);else ij=ij(:,2);end
36         nn = max(m,n)+abs(k)
37         if ij==[] then
38             d=sparse([],[],[nn,nn])
39         else
40             if k>0 then
41                 d=sparse([ij,ij+k],v,[nn,nn])
42             else
43                 d=sparse([ij-k,ij],v,[nn,nn])
44             end
45         end
46     end
47 endfunction