Chris@49: // Copyright (C) 2009-2012 NICTA (www.nicta.com.au)
Chris@49: // Copyright (C) 2009-2012 Conrad Sanderson
Chris@49: // 
Chris@49: // This Source Code Form is subject to the terms of the Mozilla Public
Chris@49: // License, v. 2.0. If a copy of the MPL was not distributed with this
Chris@49: // file, You can obtain one at http://mozilla.org/MPL/2.0/.
Chris@49: 
Chris@49: 
Chris@49: //! \addtogroup glue_solve
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT, typename T2>
Chris@49: inline
Chris@49: void
Chris@49: glue_solve::solve_direct(Mat<eT>& out, Mat<eT>& A, const Base<eT,T2>& X, const bool slow)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const uword A_n_rows = A.n_rows;
Chris@49:   const uword A_n_cols = A.n_cols;
Chris@49:   
Chris@49:   bool status = false;
Chris@49:   
Chris@49:   if(A_n_rows == A_n_cols)
Chris@49:     {
Chris@49:     status = auxlib::solve(out, A, X, slow);
Chris@49:     }
Chris@49:   else
Chris@49:   if(A_n_rows > A_n_cols)
Chris@49:     {
Chris@49:     arma_extra_debug_print("solve(): detected over-determined system");
Chris@49:     status = auxlib::solve_od(out, A, X);
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     arma_extra_debug_print("solve(): detected under-determined system");
Chris@49:     status = auxlib::solve_ud(out, A, X);
Chris@49:     }
Chris@49:   
Chris@49:   if(status == false)
Chris@49:     {
Chris@49:     out.reset();
Chris@49:     arma_bad("solve(): solution not found");
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1, typename T2>
Chris@49: inline
Chris@49: void
Chris@49: glue_solve::apply(Mat<typename T1::elem_type>& out, const Glue<T1,T2,glue_solve>& X)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename T1::elem_type eT;
Chris@49:   
Chris@49:   Mat<eT> A = X.A.get_ref();
Chris@49:   
Chris@49:   glue_solve::solve_direct( out, A, X.B, (X.aux_uword == 1) );
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1, typename T2>
Chris@49: inline
Chris@49: void
Chris@49: glue_solve_tr::apply(Mat<typename T1::elem_type>& out, const Glue<T1,T2,glue_solve_tr>& X)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename T1::elem_type eT;
Chris@49:   
Chris@49:   const unwrap_check<T1> A_tmp(X.A, out);
Chris@49:   const unwrap_check<T2> B_tmp(X.B, out);
Chris@49:   
Chris@49:   const Mat<eT>& A = A_tmp.M;
Chris@49:   const Mat<eT>& B = B_tmp.M;
Chris@49:   
Chris@49:   bool  err_state = false;
Chris@49:   char* err_msg   = 0;
Chris@49:   
Chris@49:   arma_debug_set_error( err_state, err_msg, ((&A) == (&B)),           "solve(): A is an alias of B" );
Chris@49:   arma_debug_set_error( err_state, err_msg, (A.n_rows != B.n_rows),   "solve(): number of rows in A and B must be the same" );
Chris@49:   arma_debug_set_error( err_state, err_msg, (A.is_square() == false), "solve(): A is not a square matrix" );
Chris@49:   
Chris@49:   arma_debug_check(err_state, err_msg);
Chris@49:   
Chris@49:   const bool status = auxlib::solve_tr(out, A, B, X.aux_uword);
Chris@49:   
Chris@49:   if(status == false)
Chris@49:     {
Chris@49:     out.reset();
Chris@49:     arma_bad("solve(): solution not found");
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! @}