max@0: // Copyright (C) 2010-2011 NICTA (www.nicta.com.au)
max@0: // Copyright (C) 2010-2011 Conrad Sanderson
max@0: // Copyright (C) 2011 Alcatel Lucent
max@0: // Copyright (C) 2011 Gerhard Schreiber
max@0: // 
max@0: // This file is part of the Armadillo C++ library.
max@0: // It is provided without any warranty of fitness
max@0: // for any purpose. You can redistribute this file
max@0: // and/or modify it under the terms of the GNU
max@0: // Lesser General Public License (LGPL) as published
max@0: // by the Free Software Foundation, either version 3
max@0: // of the License or (at your option) any later version.
max@0: // (see http://www.opensource.org/licenses for more info)
max@0: 
max@0: 
max@0: 
max@0: //! \addtogroup glue_toeplitz
max@0: //! @{
max@0: 
max@0: 
max@0: 
max@0: template<typename T1, typename T2>
max@0: inline
max@0: void
max@0: glue_toeplitz::apply(Mat<typename T1::elem_type>& out, const Glue<T1, T2, glue_toeplitz>& in)
max@0:   {
max@0:   arma_extra_debug_sigprint();
max@0:   
max@0:   typedef typename T1::elem_type eT;
max@0:   
max@0:   if( ((void*)(&in.A)) == ((void*)(&in.B)) )
max@0:     {
max@0:     arma_extra_debug_print("glue_toeplitz::apply(): one argument version");
max@0:     
max@0:     const unwrap_check<T1>  tmp(in.A, out);
max@0:     const Mat<eT>& A      = tmp.M;
max@0:     
max@0:     arma_debug_check( (A.is_vec() == false), "toeplitz(): input argument must be a vector" );
max@0:     
max@0:     const uword N     = A.n_elem;
max@0:     const eT* A_mem = A.memptr();
max@0:     
max@0:     out.set_size(N,N);
max@0:     
max@0:     for(uword col=0; col<N; ++col)
max@0:       {
max@0:       eT* col_mem = out.colptr(col);
max@0:       
max@0:       uword i;
max@0:       
max@0:       i = col;
max@0:       for(uword row=0; row<col; ++row, --i)
max@0:         {
max@0:         col_mem[row] = A_mem[i];
max@0:         }
max@0:       
max@0:       i = 0;
max@0:       for(uword row=col; row<N; ++row, ++i)
max@0:         {
max@0:         col_mem[row] = A_mem[i];
max@0:         }      
max@0:       }
max@0:     }
max@0:   else
max@0:     {
max@0:     arma_extra_debug_print("glue_toeplitz::apply(): two argument version");
max@0:     
max@0:     const unwrap_check<T1> tmp1(in.A, out);
max@0:     const unwrap_check<T2> tmp2(in.B, out);
max@0:     
max@0:     const Mat<eT>& A = tmp1.M;
max@0:     const Mat<eT>& B = tmp2.M;
max@0:     
max@0:     arma_debug_check( ( (A.is_vec() == false) || (B.is_vec() == false) ), "toeplitz(): input arguments must be vectors" );
max@0:     
max@0:     const uword A_N = A.n_elem;
max@0:     const uword B_N = B.n_elem;
max@0:     
max@0:     const eT* A_mem = A.memptr();
max@0:     const eT* B_mem = B.memptr();
max@0:     
max@0:     out.set_size(A_N, B_N);
max@0:     
max@0:     if( out.is_empty() )
max@0:       {
max@0:       return;
max@0:       }
max@0:     
max@0:     for(uword col=0; col<B_N; ++col)
max@0:       {
max@0:       eT* col_mem = out.colptr(col);
max@0:       
max@0:       uword i = 0;
max@0:       for(uword row=col; row<A_N; ++row, ++i)
max@0:         {
max@0:         col_mem[row] = A_mem[i];
max@0:         }
max@0:       }
max@0:     
max@0:     for(uword row=0; row<A_N; ++row)
max@0:       {
max@0:       uword i = 1;
max@0:       for(uword col=(row+1); col<B_N; ++col, ++i)
max@0:         {
max@0:         out.at(row,col) = B_mem[i];
max@0:         }
max@0:       }
max@0:     
max@0:     }
max@0:   
max@0:   
max@0:   }
max@0: 
max@0: 
max@0: 
max@0: template<typename T1, typename T2>
max@0: inline
max@0: void
max@0: glue_toeplitz_circ::apply(Mat<typename T1::elem_type>& out, const Glue<T1, T2, glue_toeplitz_circ>& in)
max@0:   {
max@0:   arma_extra_debug_sigprint();
max@0:   
max@0:   typedef typename T1::elem_type eT;
max@0:   
max@0:   if( ((void*)(&in.A)) == ((void*)(&in.B)) )
max@0:     {
max@0:     arma_extra_debug_print("glue_toeplitz_circ::apply(): one argument version");
max@0:     
max@0:     const unwrap_check<T1>  tmp(in.A, out);
max@0:     const Mat<eT>& A      = tmp.M;
max@0:     
max@0:     arma_debug_check( (A.is_vec() == false), "toeplitz(): input argument must be a vector" );
max@0:     
max@0:     const uword N     = A.n_elem;
max@0:     const eT* A_mem = A.memptr();
max@0:     
max@0:     out.set_size(N,N);
max@0:     
max@0:     
max@0:     if(A.is_colvec())
max@0:       {
max@0:       // A is interpreted as colvec
max@0:       
max@0:       for(uword col=0; col<N; ++col)
max@0:         {
max@0:         eT* col_mem = out.colptr(col);
max@0:         
max@0:         
max@0:         uword i = col;
max@0:         
max@0:         for(uword row=0; row<col; ++row, --i)
max@0:           {
max@0:           col_mem[row] = A_mem[N-i];
max@0:           }
max@0:         
max@0:         
max@0:         i = 0;
max@0:         
max@0:         for(uword row=col; row<N; ++row, ++i)
max@0:           {
max@0:           col_mem[row] = A_mem[i];
max@0:           }      
max@0:         }
max@0:       
max@0:       }
max@0:     else
max@0:       {
max@0:       // A is interpreted as rowvec - probably not the computationally most efficient way to do this ;-)
max@0:       
max@0:       for(uword row=0; row<N; ++row)
max@0:         {
max@0:         uword i = row;
max@0:          
max@0:         for(uword col=0; col<row; ++col, --i)
max@0:           {
max@0:           out.at(row,col) = A_mem[N-i];
max@0:           }
max@0:          
max@0:         i = 0;
max@0:          
max@0:         for(uword col=row; col<N; ++col, ++i)
max@0:           {
max@0:           out.at(row,col) = A_mem[i];
max@0:           }
max@0:         }
max@0:       }
max@0:     }
max@0:   }
max@0: 
max@0: 
max@0: 
max@0: //! @}