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 //! @}
|