|
max@0
|
1 // Copyright (C) 2009-2011 NICTA (www.nicta.com.au)
|
|
max@0
|
2 // Copyright (C) 2009-2011 Conrad Sanderson
|
|
max@0
|
3 // Copyright (C) 2009-2010 Dimitrios Bouzas
|
|
max@0
|
4 //
|
|
max@0
|
5 // This file is part of the Armadillo C++ library.
|
|
max@0
|
6 // It is provided without any warranty of fitness
|
|
max@0
|
7 // for any purpose. You can redistribute this file
|
|
max@0
|
8 // and/or modify it under the terms of the GNU
|
|
max@0
|
9 // Lesser General Public License (LGPL) as published
|
|
max@0
|
10 // by the Free Software Foundation, either version 3
|
|
max@0
|
11 // of the License or (at your option) any later version.
|
|
max@0
|
12 // (see http://www.opensource.org/licenses for more info)
|
|
max@0
|
13
|
|
max@0
|
14
|
|
max@0
|
15
|
|
max@0
|
16 //! \addtogroup op_repmat
|
|
max@0
|
17 //! @{
|
|
max@0
|
18
|
|
max@0
|
19
|
|
max@0
|
20
|
|
max@0
|
21 //! \brief
|
|
max@0
|
22 //! implementation of the 'repeat matrix' operation, used for constructing matrices
|
|
max@0
|
23 template<typename T1>
|
|
max@0
|
24 inline
|
|
max@0
|
25 void
|
|
max@0
|
26 op_repmat::apply(Mat<typename T1::elem_type>& out, const Op<T1,op_repmat>& in)
|
|
max@0
|
27 {
|
|
max@0
|
28 arma_extra_debug_sigprint();
|
|
max@0
|
29
|
|
max@0
|
30 typedef typename T1::elem_type eT;
|
|
max@0
|
31
|
|
max@0
|
32 const unwrap_check<T1> tmp(in.m, out);
|
|
max@0
|
33 const Mat<eT>& X = tmp.M;
|
|
max@0
|
34
|
|
max@0
|
35 const uword copies_per_row = in.aux_uword_a;
|
|
max@0
|
36 const uword copies_per_col = in.aux_uword_b;
|
|
max@0
|
37
|
|
max@0
|
38 const uword X_n_rows = X.n_rows;
|
|
max@0
|
39 const uword X_n_cols = X.n_cols;
|
|
max@0
|
40
|
|
max@0
|
41 out.set_size(X_n_rows * copies_per_row, X_n_cols * copies_per_col);
|
|
max@0
|
42
|
|
max@0
|
43 const uword out_n_rows = out.n_rows;
|
|
max@0
|
44 const uword out_n_cols = out.n_cols;
|
|
max@0
|
45
|
|
max@0
|
46 if( (out_n_rows > 0) && (out_n_cols > 0) )
|
|
max@0
|
47 {
|
|
max@0
|
48 for(uword col = 0; col < out_n_cols; col += X_n_cols)
|
|
max@0
|
49 {
|
|
max@0
|
50 for(uword row = 0; row < out_n_rows; row += X_n_rows)
|
|
max@0
|
51 {
|
|
max@0
|
52 out.submat(row, col, row+X_n_rows-1, col+X_n_cols-1) = X;
|
|
max@0
|
53 }
|
|
max@0
|
54 }
|
|
max@0
|
55 }
|
|
max@0
|
56 }
|
|
max@0
|
57
|
|
max@0
|
58
|
|
max@0
|
59
|
|
max@0
|
60 //! @}
|
