summaryrefslogtreecommitdiff
path: root/modules/cacsd/macros/trisolve.sci
blob: a1d039646032103a2f525ae8a86b1afa6d54ae58 (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
// Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
// Copyright (C) INRIA - F. Delebecque , Serge STEER
//
// 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 [b,sexp]=trisolve(a,b,sexp)
    //[x [,sexp]] = trisolve(A,b [,sexp])  symbolically solves A*x =b
    // when A and b are matrices of character strings, A being assumed to be
    // upper triangular.
    //sexp : vector of common subexpressions in A, b and x.
    //!

    [lhs,rhs]=argn(0)
    [n0,m]=size(b)
    if rhs==2 then
        for n=n0:-1:1
            pivot=a(n,n)
            for k=1:m,b(n,k)=ldivf(pivot,"("+b(n,k)+")"),end
            if n==1 then return,end
            for l=1:n-1
                for k=1:m,
                    b(l,k)=addf(b(l,k),mulf(mulf("-1",a(l,n)),b(n,k)))
                end
            end
        end
    else
        ns=prod(size(sexp))
        for n=n0:-1:1
            pivot=a(n,n)
            for k=1:m,
                ns=ns+1
                sexp(ns)=ldivf(pivot,"("+b(n,k)+")")
                b(n,k)="%("+string(ns)+")";
            end
            if n==1 then return,end
            for l=1:n-1
                for k=1:m,
                    b(l,k)=addf(b(l,k),mulf(mulf("-1",a(l,n)),b(n,k)))
                end
            end
        end
    end
endfunction