comparison armadillo-2.4.4/include/armadillo_bits/fn_solve.hpp @ 0:8b6102e2a9b0

Armadillo Library

+// Copyright (C) 2009-2011 NICTA (www.nicta.com.au)
+// Copyright (C) 2009-2011 Conrad Sanderson
+// 
+// This file is part of the Armadillo C++ library.
+// It is provided without any warranty of fitness
+// for any purpose. You can redistribute this file
+// and/or modify it under the terms of the GNU
+// Lesser General Public License (LGPL) as published
+// by the Free Software Foundation, either version 3
+// of the License or (at your option) any later version.
+// (see http://www.opensource.org/licenses for more info)
+
+
+//! \addtogroup fn_solve
+//! @{
+
+
+
+//! Solve a system of linear equations, i.e., A*X = B, where X is unknown.
+//! For a square matrix A, this function is conceptually the same as X = inv(A)*B,
+//! but is done more efficiently.
+//! The number of rows in A and B must be the same.
+//! B can be either a column vector or a matrix.
+//! This function will also try to provide approximate solutions
+//! to under-determined as well as over-determined systems (non-square A matrices).
+
+template<typename T1, typename T2>
+inline
+const Glue<T1, T2, glue_solve>
+solve
+  (
+  const Base<typename T1::elem_type,T1>& A,
+  const Base<typename T1::elem_type,T2>& B,
+  const bool slow = false,
+  const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
+  )
+  {
+  arma_extra_debug_sigprint();
+  arma_ignore(junk);
+  
+  return Glue<T1, T2, glue_solve>(A.get_ref(), B.get_ref(), ((slow == false) ? 0 : 1) );
+  }
+
+
+
+template<typename T1, typename T2>
+inline
+const Glue<T1, T2, glue_solve_tr>
+solve
+  (
+  const Op<T1, op_trimat>& A,
+  const Base<typename T1::elem_type,T2>& B,
+  const bool slow = false,
+  const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
+  )
+  {
+  arma_extra_debug_sigprint();
+  arma_ignore(slow);
+  arma_ignore(junk);
+  
+  return Glue<T1, T2, glue_solve_tr>(A.m, B.get_ref(), A.aux_uword_a);
+  }
+
+
+
+template<typename T1, typename T2>
+inline
+bool
+solve
+  (
+         Mat<typename T1::elem_type>&    out,
+  const Base<typename T1::elem_type,T1>& A,
+  const Base<typename T1::elem_type,T2>& B,
+  const bool slow = false,
+  const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
+  )
+  {
+  arma_extra_debug_sigprint();
+  arma_ignore(junk);
+  
+  try
+    {
+    out = solve( A.get_ref(), B.get_ref(), slow );
+    }
+  catch(std::runtime_error&)
+    {
+    return false;
+    }
+  
+  return true;
+  }
+
+
+
+//! @}