Chris@49
|
1 // Copyright (C) 2009-2011 NICTA (www.nicta.com.au)
|
Chris@49
|
2 // Copyright (C) 2009-2011 Conrad Sanderson
|
Chris@49
|
3 //
|
Chris@49
|
4 // This Source Code Form is subject to the terms of the Mozilla Public
|
Chris@49
|
5 // License, v. 2.0. If a copy of the MPL was not distributed with this
|
Chris@49
|
6 // file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
Chris@49
|
7
|
Chris@49
|
8
|
Chris@49
|
9 //! \addtogroup fn_solve
|
Chris@49
|
10 //! @{
|
Chris@49
|
11
|
Chris@49
|
12
|
Chris@49
|
13
|
Chris@49
|
14 //! Solve a system of linear equations, i.e., A*X = B, where X is unknown.
|
Chris@49
|
15 //! For a square matrix A, this function is conceptually the same as X = inv(A)*B,
|
Chris@49
|
16 //! but is done more efficiently.
|
Chris@49
|
17 //! The number of rows in A and B must be the same.
|
Chris@49
|
18 //! B can be either a column vector or a matrix.
|
Chris@49
|
19 //! This function will also try to provide approximate solutions
|
Chris@49
|
20 //! to under-determined as well as over-determined systems (non-square A matrices).
|
Chris@49
|
21
|
Chris@49
|
22 template<typename T1, typename T2>
|
Chris@49
|
23 inline
|
Chris@49
|
24 const Glue<T1, T2, glue_solve>
|
Chris@49
|
25 solve
|
Chris@49
|
26 (
|
Chris@49
|
27 const Base<typename T1::elem_type,T1>& A,
|
Chris@49
|
28 const Base<typename T1::elem_type,T2>& B,
|
Chris@49
|
29 const bool slow = false,
|
Chris@49
|
30 const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
|
Chris@49
|
31 )
|
Chris@49
|
32 {
|
Chris@49
|
33 arma_extra_debug_sigprint();
|
Chris@49
|
34 arma_ignore(junk);
|
Chris@49
|
35
|
Chris@49
|
36 return Glue<T1, T2, glue_solve>(A.get_ref(), B.get_ref(), ((slow == false) ? 0 : 1) );
|
Chris@49
|
37 }
|
Chris@49
|
38
|
Chris@49
|
39
|
Chris@49
|
40
|
Chris@49
|
41 template<typename T1, typename T2>
|
Chris@49
|
42 inline
|
Chris@49
|
43 const Glue<T1, T2, glue_solve_tr>
|
Chris@49
|
44 solve
|
Chris@49
|
45 (
|
Chris@49
|
46 const Op<T1, op_trimat>& A,
|
Chris@49
|
47 const Base<typename T1::elem_type,T2>& B,
|
Chris@49
|
48 const bool slow = false,
|
Chris@49
|
49 const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
|
Chris@49
|
50 )
|
Chris@49
|
51 {
|
Chris@49
|
52 arma_extra_debug_sigprint();
|
Chris@49
|
53 arma_ignore(slow);
|
Chris@49
|
54 arma_ignore(junk);
|
Chris@49
|
55
|
Chris@49
|
56 return Glue<T1, T2, glue_solve_tr>(A.m, B.get_ref(), A.aux_uword_a);
|
Chris@49
|
57 }
|
Chris@49
|
58
|
Chris@49
|
59
|
Chris@49
|
60
|
Chris@49
|
61 template<typename T1, typename T2>
|
Chris@49
|
62 inline
|
Chris@49
|
63 bool
|
Chris@49
|
64 solve
|
Chris@49
|
65 (
|
Chris@49
|
66 Mat<typename T1::elem_type>& out,
|
Chris@49
|
67 const Base<typename T1::elem_type,T1>& A,
|
Chris@49
|
68 const Base<typename T1::elem_type,T2>& B,
|
Chris@49
|
69 const bool slow = false,
|
Chris@49
|
70 const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
|
Chris@49
|
71 )
|
Chris@49
|
72 {
|
Chris@49
|
73 arma_extra_debug_sigprint();
|
Chris@49
|
74 arma_ignore(junk);
|
Chris@49
|
75
|
Chris@49
|
76 try
|
Chris@49
|
77 {
|
Chris@49
|
78 out = solve( A.get_ref(), B.get_ref(), slow );
|
Chris@49
|
79 }
|
Chris@49
|
80 catch(std::runtime_error&)
|
Chris@49
|
81 {
|
Chris@49
|
82 return false;
|
Chris@49
|
83 }
|
Chris@49
|
84
|
Chris@49
|
85 return true;
|
Chris@49
|
86 }
|
Chris@49
|
87
|
Chris@49
|
88
|
Chris@49
|
89
|
Chris@49
|
90 //! @}
|