Chris@49
|
1 // Copyright (C) 2011-2013 NICTA (www.nicta.com.au)
|
Chris@49
|
2 // Copyright (C) 2011-2013 Conrad Sanderson
|
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 Gen
|
Chris@49
|
10 //! @{
|
Chris@49
|
11
|
Chris@49
|
12
|
Chris@49
|
13 //! support class for generator functions (eg. zeros, randu, randn, ...)
|
Chris@49
|
14 template<typename T1, typename gen_type>
|
Chris@49
|
15 class Gen : public Base<typename T1::elem_type, Gen<T1, gen_type> >
|
Chris@49
|
16 {
|
Chris@49
|
17 public:
|
Chris@49
|
18
|
Chris@49
|
19 typedef typename T1::elem_type elem_type;
|
Chris@49
|
20 typedef typename get_pod_type<elem_type>::result pod_type;
|
Chris@49
|
21
|
Chris@49
|
22 static const bool prefer_at_accessor = (is_same_type<gen_type, gen_ones_diag>::value) ? true : false;
|
Chris@49
|
23 static const bool is_simple = (is_same_type<gen_type, gen_ones_full>::value) || (is_same_type<gen_type, gen_zeros>::value);
|
Chris@49
|
24
|
Chris@49
|
25 static const bool is_row = T1::is_row;
|
Chris@49
|
26 static const bool is_col = T1::is_col;
|
Chris@49
|
27
|
Chris@49
|
28 arma_aligned const uword n_rows;
|
Chris@49
|
29 arma_aligned const uword n_cols;
|
Chris@49
|
30
|
Chris@49
|
31 arma_inline Gen(const uword in_n_rows, const uword in_n_cols);
|
Chris@49
|
32 arma_inline ~Gen();
|
Chris@49
|
33
|
Chris@49
|
34 arma_inline static elem_type generate();
|
Chris@49
|
35
|
Chris@49
|
36 arma_inline elem_type operator[] (const uword ii) const;
|
Chris@49
|
37 arma_inline elem_type at (const uword row, const uword col) const;
|
Chris@49
|
38 arma_inline elem_type at_alt (const uword ii) const;
|
Chris@49
|
39
|
Chris@49
|
40 inline void apply (Mat<elem_type>& out) const;
|
Chris@49
|
41 inline void apply_inplace_plus (Mat<elem_type>& out) const;
|
Chris@49
|
42 inline void apply_inplace_minus(Mat<elem_type>& out) const;
|
Chris@49
|
43 inline void apply_inplace_schur(Mat<elem_type>& out) const;
|
Chris@49
|
44 inline void apply_inplace_div (Mat<elem_type>& out) const;
|
Chris@49
|
45 };
|
Chris@49
|
46
|
Chris@49
|
47
|
Chris@49
|
48
|
Chris@49
|
49 //! @}
|