|
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 Gen
|
|
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 Gen : public Base<eT, Gen<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 = (is_same_type<gen_type, gen_ones_diag>::value) ? true : 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
|
|
max@0
|
32 arma_inline Gen(const uword in_n_rows, const uword in_n_cols);
|
|
max@0
|
33 arma_inline ~Gen();
|
|
max@0
|
34
|
|
max@0
|
35 arma_inline static eT generate();
|
|
max@0
|
36
|
|
max@0
|
37 arma_inline eT operator[] (const uword i) const;
|
|
max@0
|
38 arma_inline eT at (const uword row, const uword col) const;
|
|
max@0
|
39
|
|
max@0
|
40 inline void apply (Mat<eT>& out) const;
|
|
max@0
|
41 inline void apply_inplace_plus (Mat<eT>& out) const;
|
|
max@0
|
42 inline void apply_inplace_minus(Mat<eT>& out) const;
|
|
max@0
|
43 inline void apply_inplace_schur(Mat<eT>& out) const;
|
|
max@0
|
44 inline void apply_inplace_div (Mat<eT>& out) const;
|
|
max@0
|
45 };
|
|
max@0
|
46
|
|
max@0
|
47
|
|
max@0
|
48
|
|
max@0
|
49 //! @}
|
