|
max@0
|
1 // Copyright (C) 2011 NICTA (www.nicta.com.au)
|
|
max@0
|
2 // Copyright (C) 2011 Conrad Sanderson
|
|
max@0
|
3 //
|
|
max@0
|
4 // This file is part of the Armadillo C++ library.
|
|
max@0
|
5 // It is provided without any warranty of fitness
|
|
max@0
|
6 // for any purpose. You can redistribute this file
|
|
max@0
|
7 // and/or modify it under the terms of the GNU
|
|
max@0
|
8 // Lesser General Public License (LGPL) as published
|
|
max@0
|
9 // by the Free Software Foundation, either version 3
|
|
max@0
|
10 // of the License or (at your option) any later version.
|
|
max@0
|
11 // (see http://www.opensource.org/licenses for more info)
|
|
max@0
|
12
|
|
max@0
|
13
|
|
max@0
|
14 //! \addtogroup GenCube
|
|
max@0
|
15 //! @{
|
|
max@0
|
16
|
|
max@0
|
17
|
|
max@0
|
18 //! support class for generator functions (eg. zeros, randu, randn, ...)
|
|
max@0
|
19 template<typename eT, typename gen_type>
|
|
max@0
|
20 class GenCube : public BaseCube<eT, GenCube<eT, gen_type> >
|
|
max@0
|
21 {
|
|
max@0
|
22 public:
|
|
max@0
|
23
|
|
max@0
|
24 typedef eT elem_type;
|
|
max@0
|
25 typedef typename get_pod_type<elem_type>::result pod_type;
|
|
max@0
|
26
|
|
max@0
|
27 static const bool prefer_at_accessor = false;
|
|
max@0
|
28
|
|
max@0
|
29 arma_aligned const uword n_rows;
|
|
max@0
|
30 arma_aligned const uword n_cols;
|
|
max@0
|
31 arma_aligned const uword n_slices;
|
|
max@0
|
32
|
|
max@0
|
33 arma_inline GenCube(const uword in_n_rows, const uword in_n_cols, const uword in_n_slices);
|
|
max@0
|
34 arma_inline ~GenCube();
|
|
max@0
|
35
|
|
max@0
|
36 arma_inline static eT generate();
|
|
max@0
|
37
|
|
max@0
|
38 arma_inline eT operator[] (const uword i) const;
|
|
max@0
|
39 arma_inline eT at (const uword row, const uword col, const uword slice) const;
|
|
max@0
|
40
|
|
max@0
|
41 inline void apply (Cube<eT>& out) const;
|
|
max@0
|
42 inline void apply_inplace_plus (Cube<eT>& out) const;
|
|
max@0
|
43 inline void apply_inplace_minus(Cube<eT>& out) const;
|
|
max@0
|
44 inline void apply_inplace_schur(Cube<eT>& out) const;
|
|
max@0
|
45 inline void apply_inplace_div (Cube<eT>& out) const;
|
|
max@0
|
46 };
|
|
max@0
|
47
|
|
max@0
|
48
|
|
max@0
|
49
|
|
max@0
|
50 //! @}
|
