Chris@49: // Copyright (C) 2008-2010 NICTA (www.nicta.com.au)
Chris@49: // Copyright (C) 2008-2010 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 op_dotext
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT
Chris@49: op_dotext::direct_rowvec_mat_colvec
Chris@49:   (
Chris@49:   const eT*      A_mem,
Chris@49:   const Mat<eT>& B,
Chris@49:   const eT*      C_mem
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const uword cost_AB = B.n_cols;
Chris@49:   const uword cost_BC = B.n_rows;
Chris@49:   
Chris@49:   if(cost_AB <= cost_BC)
Chris@49:     {
Chris@49:     podarray<eT> tmp(B.n_cols);
Chris@49:     
Chris@49:     for(uword col=0; col<B.n_cols; ++col)
Chris@49:       {
Chris@49:       const eT* B_coldata = B.colptr(col);
Chris@49:       
Chris@49:       eT val = eT(0);
Chris@49:       for(uword i=0; i<B.n_rows; ++i)
Chris@49:         {
Chris@49:         val += A_mem[i] * B_coldata[i];
Chris@49:         }
Chris@49:         
Chris@49:       tmp[col] = val;
Chris@49:       }
Chris@49:     
Chris@49:     return op_dot::direct_dot(B.n_cols, tmp.mem, C_mem);
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     podarray<eT> tmp(B.n_rows);
Chris@49:     
Chris@49:     for(uword row=0; row<B.n_rows; ++row)
Chris@49:       {
Chris@49:       eT val = eT(0);
Chris@49:       for(uword col=0; col<B.n_cols; ++col)
Chris@49:         {
Chris@49:         val += B.at(row,col) * C_mem[col];
Chris@49:         }
Chris@49:       
Chris@49:       tmp[row] = val;
Chris@49:       }
Chris@49:     
Chris@49:     return op_dot::direct_dot(B.n_rows, A_mem, tmp.mem);
Chris@49:     }
Chris@49:   
Chris@49:   
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT
Chris@49: op_dotext::direct_rowvec_transmat_colvec
Chris@49:   (
Chris@49:   const eT*      A_mem,
Chris@49:   const Mat<eT>& B,
Chris@49:   const eT*      C_mem
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const uword cost_AB = B.n_rows;
Chris@49:   const uword cost_BC = B.n_cols;
Chris@49:   
Chris@49:   if(cost_AB <= cost_BC)
Chris@49:     {
Chris@49:     podarray<eT> tmp(B.n_rows);
Chris@49:     
Chris@49:     for(uword row=0; row<B.n_rows; ++row)
Chris@49:       {
Chris@49:       eT val = eT(0);
Chris@49:       
Chris@49:       for(uword i=0; i<B.n_cols; ++i)
Chris@49:         {
Chris@49:         val += A_mem[i] * B.at(row,i);
Chris@49:         }
Chris@49:         
Chris@49:       tmp[row] = val;
Chris@49:       }
Chris@49:     
Chris@49:     return op_dot::direct_dot(B.n_rows, tmp.mem, C_mem);
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     podarray<eT> tmp(B.n_cols);
Chris@49:     
Chris@49:     for(uword col=0; col<B.n_cols; ++col)
Chris@49:       {
Chris@49:       const eT* B_coldata = B.colptr(col);
Chris@49:       
Chris@49:       eT val = eT(0);
Chris@49:       
Chris@49:       for(uword i=0; i<B.n_rows; ++i)
Chris@49:         {
Chris@49:         val += B_coldata[i] * C_mem[i];
Chris@49:         }
Chris@49:       
Chris@49:       tmp[col] = val;
Chris@49:       }
Chris@49:     
Chris@49:     return op_dot::direct_dot(B.n_cols, A_mem, tmp.mem);
Chris@49:     }
Chris@49:   
Chris@49:   
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT
Chris@49: op_dotext::direct_rowvec_diagmat_colvec
Chris@49:   (
Chris@49:   const eT*      A_mem,
Chris@49:   const Mat<eT>& B,
Chris@49:   const eT*      C_mem
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   eT val = eT(0);
Chris@49: 
Chris@49:   for(uword i=0; i<B.n_rows; ++i)
Chris@49:     {
Chris@49:     val += A_mem[i] * B.at(i,i) * C_mem[i];
Chris@49:     }
Chris@49: 
Chris@49:   return val;
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT
Chris@49: op_dotext::direct_rowvec_invdiagmat_colvec
Chris@49:   (
Chris@49:   const eT*      A_mem,
Chris@49:   const Mat<eT>& B,
Chris@49:   const eT*      C_mem
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   eT val = eT(0);
Chris@49: 
Chris@49:   for(uword i=0; i<B.n_rows; ++i)
Chris@49:     {
Chris@49:     val += (A_mem[i] * C_mem[i]) / B.at(i,i);
Chris@49:     }
Chris@49: 
Chris@49:   return val;
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT
Chris@49: op_dotext::direct_rowvec_invdiagvec_colvec
Chris@49:   (
Chris@49:   const eT*      A_mem,
Chris@49:   const Mat<eT>& B,
Chris@49:   const eT*      C_mem
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const eT* B_mem = B.mem;
Chris@49:   
Chris@49:   eT val = eT(0);
Chris@49: 
Chris@49:   for(uword i=0; i<B.n_elem; ++i)
Chris@49:     {
Chris@49:     val += (A_mem[i] * C_mem[i]) / B_mem[i];
Chris@49:     }
Chris@49: 
Chris@49:   return val;
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! @}