annotate armadillo-3.900.4/include/armadillo_bits/op_toeplitz_meat.hpp @ 84:55a047986812 tip

Update library URI so as not to be document-local
author Chris Cannam
date Wed, 22 Apr 2020 14:21:57 +0100
parents 1ec0e2823891
children
rev   line source
Chris@49 1 // Copyright (C) 2013 Conrad Sanderson
Chris@49 2 // Copyright (C) 2013 NICTA (www.nicta.com.au)
Chris@49 3 //
Chris@49 4 // This Source Code Form is subject to the terms of the Mozilla Public
Chris@49 5 // License, v. 2.0. If a copy of the MPL was not distributed with this
Chris@49 6 // file, You can obtain one at http://mozilla.org/MPL/2.0/.
Chris@49 7
Chris@49 8
Chris@49 9 //! \addtogroup op_toeplitz
Chris@49 10 //! @{
Chris@49 11
Chris@49 12
Chris@49 13
Chris@49 14 template<typename T1>
Chris@49 15 inline
Chris@49 16 void
Chris@49 17 op_toeplitz::apply(Mat<typename T1::elem_type>& out, const Op<T1,op_toeplitz>& in)
Chris@49 18 {
Chris@49 19 arma_extra_debug_sigprint();
Chris@49 20
Chris@49 21 typedef typename T1::elem_type eT;
Chris@49 22
Chris@49 23 const unwrap_check<T1> tmp(in.m, out);
Chris@49 24 const Mat<eT>& X = tmp.M;
Chris@49 25
Chris@49 26 arma_debug_check( ((X.is_vec() == false) && (X.is_empty() == false)), "toeplitz(): given object is not a vector" );
Chris@49 27
Chris@49 28 const uword N = X.n_elem;
Chris@49 29 const eT* X_mem = X.memptr();
Chris@49 30
Chris@49 31 out.set_size(N,N);
Chris@49 32
Chris@49 33 for(uword col=0; col < N; ++col)
Chris@49 34 {
Chris@49 35 eT* col_mem = out.colptr(col);
Chris@49 36
Chris@49 37 uword i;
Chris@49 38
Chris@49 39 i = col;
Chris@49 40 for(uword row=0; row < col; ++row, --i) { col_mem[row] = X_mem[i]; }
Chris@49 41
Chris@49 42 i = 0;
Chris@49 43 for(uword row=col; row < N; ++row, ++i) { col_mem[row] = X_mem[i]; }
Chris@49 44 }
Chris@49 45 }
Chris@49 46
Chris@49 47
Chris@49 48
Chris@49 49 template<typename T1>
Chris@49 50 inline
Chris@49 51 void
Chris@49 52 op_toeplitz_c::apply(Mat<typename T1::elem_type>& out, const Op<T1,op_toeplitz_c>& in)
Chris@49 53 {
Chris@49 54 arma_extra_debug_sigprint();
Chris@49 55
Chris@49 56 typedef typename T1::elem_type eT;
Chris@49 57
Chris@49 58 const unwrap_check<T1> tmp(in.m, out);
Chris@49 59 const Mat<eT>& X = tmp.M;
Chris@49 60
Chris@49 61 arma_debug_check( ((X.is_vec() == false) && (X.is_empty() == false)), "circ_toeplitz(): given object is not a vector" );
Chris@49 62
Chris@49 63 const uword N = X.n_elem;
Chris@49 64 const eT* X_mem = X.memptr();
Chris@49 65
Chris@49 66 out.set_size(N,N);
Chris@49 67
Chris@49 68 if(X.is_rowvec() == true)
Chris@49 69 {
Chris@49 70 for(uword row=0; row < N; ++row)
Chris@49 71 {
Chris@49 72 uword i;
Chris@49 73
Chris@49 74 i = row;
Chris@49 75 for(uword col=0; col < row; ++col, --i) { out.at(row,col) = X_mem[N-i]; }
Chris@49 76
Chris@49 77 i = 0;
Chris@49 78 for(uword col=row; col < N; ++col, ++i) { out.at(row,col) = X_mem[i]; }
Chris@49 79 }
Chris@49 80 }
Chris@49 81 else
Chris@49 82 {
Chris@49 83 for(uword col=0; col < N; ++col)
Chris@49 84 {
Chris@49 85 eT* col_mem = out.colptr(col);
Chris@49 86
Chris@49 87 uword i;
Chris@49 88
Chris@49 89 i = col;
Chris@49 90 for(uword row=0; row < col; ++row, --i) { col_mem[row] = X_mem[N-i]; }
Chris@49 91
Chris@49 92 i = 0;
Chris@49 93 for(uword row=col; row < N; ++row, ++i) { col_mem[row] = X_mem[i]; }
Chris@49 94 }
Chris@49 95 }
Chris@49 96 }
Chris@49 97
Chris@49 98
Chris@49 99
Chris@49 100 //! @}