Chris@49: // Copyright (C) 2013 Conrad Sanderson
Chris@49: // Copyright (C) 2013 NICTA (www.nicta.com.au)
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_toeplitz
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_toeplitz::apply(Mat<typename T1::elem_type>& out, const Op<T1,op_toeplitz>& in)
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>  tmp(in.m, out);
Chris@49:   const Mat<eT>& X      = tmp.M;
Chris@49:   
Chris@49:   arma_debug_check( ((X.is_vec() == false) && (X.is_empty() == false)), "toeplitz(): given object is not a vector" );
Chris@49:   
Chris@49:   const uword N     = X.n_elem;
Chris@49:   const eT*   X_mem = X.memptr();
Chris@49:   
Chris@49:   out.set_size(N,N);
Chris@49:   
Chris@49:   for(uword col=0; col < N; ++col)
Chris@49:     {
Chris@49:     eT* col_mem = out.colptr(col);
Chris@49:     
Chris@49:     uword i;
Chris@49:     
Chris@49:     i = col;
Chris@49:     for(uword row=0; row < col; ++row, --i) { col_mem[row] = X_mem[i]; }
Chris@49:     
Chris@49:     i = 0;
Chris@49:     for(uword row=col; row < N; ++row, ++i) { col_mem[row] = X_mem[i]; }      
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_toeplitz_c::apply(Mat<typename T1::elem_type>& out, const Op<T1,op_toeplitz_c>& in)
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>  tmp(in.m, out);
Chris@49:   const Mat<eT>& X      = tmp.M;
Chris@49:   
Chris@49:   arma_debug_check( ((X.is_vec() == false) && (X.is_empty() == false)), "circ_toeplitz(): given object is not a vector" );
Chris@49:   
Chris@49:   const uword N     = X.n_elem;
Chris@49:   const eT*   X_mem = X.memptr();
Chris@49:   
Chris@49:   out.set_size(N,N);
Chris@49:   
Chris@49:   if(X.is_rowvec() == true)
Chris@49:     {
Chris@49:     for(uword row=0; row < N; ++row)
Chris@49:       {
Chris@49:       uword i;
Chris@49:       
Chris@49:       i = row;
Chris@49:       for(uword col=0; col < row; ++col, --i)  { out.at(row,col) = X_mem[N-i]; }
Chris@49:       
Chris@49:       i = 0;
Chris@49:       for(uword col=row; col < N; ++col, ++i)  { out.at(row,col) = X_mem[i];   }
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     for(uword col=0; col < N; ++col)
Chris@49:       {
Chris@49:       eT* col_mem = out.colptr(col);
Chris@49:       
Chris@49:       uword i;
Chris@49:       
Chris@49:       i = col;
Chris@49:       for(uword row=0; row < col; ++row, --i)  { col_mem[row] = X_mem[N-i]; }
Chris@49:       
Chris@49:       i = 0;
Chris@49:       for(uword row=col; row < N; ++row, ++i)  { col_mem[row] = X_mem[i];   }
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! @}