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 GenCube
|
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 eT, typename gen_type>
|
Chris@49
|
15 class GenCube : public BaseCube<eT, GenCube<eT, gen_type> >
|
Chris@49
|
16 {
|
Chris@49
|
17 public:
|
Chris@49
|
18
|
Chris@49
|
19 typedef eT 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 = 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 arma_aligned const uword n_rows;
|
Chris@49
|
26 arma_aligned const uword n_cols;
|
Chris@49
|
27 arma_aligned const uword n_slices;
|
Chris@49
|
28
|
Chris@49
|
29 arma_inline GenCube(const uword in_n_rows, const uword in_n_cols, const uword in_n_slices);
|
Chris@49
|
30 arma_inline ~GenCube();
|
Chris@49
|
31
|
Chris@49
|
32 arma_inline static eT generate();
|
Chris@49
|
33
|
Chris@49
|
34 arma_inline eT operator[] (const uword i) const;
|
Chris@49
|
35 arma_inline eT at (const uword row, const uword col, const uword slice) const;
|
Chris@49
|
36 arma_inline eT at_alt (const uword i) const;
|
Chris@49
|
37
|
Chris@49
|
38 inline void apply (Cube<eT>& out) const;
|
Chris@49
|
39 inline void apply_inplace_plus (Cube<eT>& out) const;
|
Chris@49
|
40 inline void apply_inplace_minus(Cube<eT>& out) const;
|
Chris@49
|
41 inline void apply_inplace_schur(Cube<eT>& out) const;
|
Chris@49
|
42 inline void apply_inplace_div (Cube<eT>& out) const;
|
Chris@49
|
43 };
|
Chris@49
|
44
|
Chris@49
|
45
|
Chris@49
|
46
|
Chris@49
|
47 //! @}
|