summaryrefslogtreecommitdiff
path: root/modules/compatibility_functions/macros/mtlb_diag.sci
blob: 1f6aae9e7bc262f6942c46418dba42aead0c6ae4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
// Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
// Copyright (C) 2002-2004 - INRIA - Vincent COUVERT
//
// This file must be used under the terms of the CeCILL.
// This source file is licensed as described in the file COPYING, which
// you should have received as part of this distribution.  The terms
// are also available at
// http://www.cecill.info/licences/Licence_CeCILL_V2.1-en.txt

function y=mtlb_diag(x,dim)
    // Emulation function for diag() Matlab function


    rhs=argn(2)

    // B = diag(A)
    if rhs==1 then
        if type(x)==10 then
            x=mstr2sci(x);
            tmp=diag(x)
            tmp(find(tmp==" "))=ascii(0)
            y=[]
            for k=1:size(tmp,1)
                y=[y;strcat(tmp(k,:))]
            end
        elseif or(type(x)==[4,6]) then
            y=diag(bool2s(x))
        else
            y=diag(x)
        end
        // B = diag(A,k)
    else
        if type(x)==10 then
            x=asciimat(x);
            x=diag(x,dim)
            y=[]
            for k=1:size(x,1)
                tmp=mstr2sci(ascii(x(k,:)))
                tmp(find(tmp==ascii(0)))=ascii(0)
                y=[y;strcat(tmp)]
            end
        elseif or(type(x)==[4,6]) then
            y=diag(bool2s(x),dim)
        else
            y=diag(x,dim)
        end
    end
endfunction