max@0
|
1 // Copyright (C) 2008-2010 NICTA (www.nicta.com.au)
|
max@0
|
2 // Copyright (C) 2008-2010 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
|
max@0
|
15 //! \addtogroup operator_times
|
max@0
|
16 //! @{
|
max@0
|
17
|
max@0
|
18
|
max@0
|
19
|
max@0
|
20 //! Base * scalar
|
max@0
|
21 template<typename T1>
|
max@0
|
22 arma_inline
|
max@0
|
23 const eOp<T1, eop_scalar_times>
|
max@0
|
24 operator*
|
max@0
|
25 (const Base<typename T1::elem_type,T1>& X, const typename T1::elem_type k)
|
max@0
|
26 {
|
max@0
|
27 arma_extra_debug_sigprint();
|
max@0
|
28
|
max@0
|
29 return eOp<T1, eop_scalar_times>(X.get_ref(),k);
|
max@0
|
30 }
|
max@0
|
31
|
max@0
|
32
|
max@0
|
33
|
max@0
|
34 //! scalar * Base
|
max@0
|
35 template<typename T1>
|
max@0
|
36 arma_inline
|
max@0
|
37 const eOp<T1, eop_scalar_times>
|
max@0
|
38 operator*
|
max@0
|
39 (const typename T1::elem_type k, const Base<typename T1::elem_type,T1>& X)
|
max@0
|
40 {
|
max@0
|
41 arma_extra_debug_sigprint();
|
max@0
|
42
|
max@0
|
43 return eOp<T1, eop_scalar_times>(X.get_ref(),k); // NOTE: order is swapped
|
max@0
|
44 }
|
max@0
|
45
|
max@0
|
46
|
max@0
|
47
|
max@0
|
48 //! non-complex Base * complex scalar (experimental)
|
max@0
|
49 template<typename T1>
|
max@0
|
50 arma_inline
|
max@0
|
51 const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_times>
|
max@0
|
52 operator*
|
max@0
|
53 (
|
max@0
|
54 const Base<typename T1::pod_type, T1>& X,
|
max@0
|
55 const std::complex<typename T1::pod_type>& k
|
max@0
|
56 )
|
max@0
|
57 {
|
max@0
|
58 arma_extra_debug_sigprint();
|
max@0
|
59
|
max@0
|
60 return mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_times>('j', X.get_ref(), k);
|
max@0
|
61 }
|
max@0
|
62
|
max@0
|
63
|
max@0
|
64
|
max@0
|
65 //! complex scalar * non-complex Base (experimental)
|
max@0
|
66 template<typename T1>
|
max@0
|
67 arma_inline
|
max@0
|
68 const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_times>
|
max@0
|
69 operator*
|
max@0
|
70 (
|
max@0
|
71 const std::complex<typename T1::pod_type>& k,
|
max@0
|
72 const Base<typename T1::pod_type, T1>& X
|
max@0
|
73 )
|
max@0
|
74 {
|
max@0
|
75 arma_extra_debug_sigprint();
|
max@0
|
76
|
max@0
|
77 return mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_times>('j', X.get_ref(), k);
|
max@0
|
78 }
|
max@0
|
79
|
max@0
|
80
|
max@0
|
81
|
max@0
|
82 //! scalar * trans(T1)
|
max@0
|
83 template<typename T1>
|
max@0
|
84 arma_inline
|
max@0
|
85 const Op<T1, op_htrans2>
|
max@0
|
86 operator*
|
max@0
|
87 (const typename T1::elem_type k, const Op<T1, op_htrans>& X)
|
max@0
|
88 {
|
max@0
|
89 arma_extra_debug_sigprint();
|
max@0
|
90
|
max@0
|
91 return Op<T1, op_htrans2>(X.m, k);
|
max@0
|
92 }
|
max@0
|
93
|
max@0
|
94
|
max@0
|
95
|
max@0
|
96 //! trans(T1) * scalar
|
max@0
|
97 template<typename T1>
|
max@0
|
98 arma_inline
|
max@0
|
99 const Op<T1, op_htrans2>
|
max@0
|
100 operator*
|
max@0
|
101 (const Op<T1, op_htrans>& X, const typename T1::elem_type k)
|
max@0
|
102 {
|
max@0
|
103 arma_extra_debug_sigprint();
|
max@0
|
104
|
max@0
|
105 return Op<T1, op_htrans2>(X.m, k);
|
max@0
|
106 }
|
max@0
|
107
|
max@0
|
108
|
max@0
|
109
|
max@0
|
110 //! Base * diagmat
|
max@0
|
111 template<typename T1, typename T2>
|
max@0
|
112 arma_inline
|
max@0
|
113 const Glue<T1, Op<T2, op_diagmat>, glue_times_diag>
|
max@0
|
114 operator*
|
max@0
|
115 (const Base<typename T2::elem_type,T1>& X, const Op<T2, op_diagmat>& Y)
|
max@0
|
116 {
|
max@0
|
117 arma_extra_debug_sigprint();
|
max@0
|
118
|
max@0
|
119 return Glue<T1, Op<T2, op_diagmat>, glue_times_diag>(X.get_ref(), Y);
|
max@0
|
120 }
|
max@0
|
121
|
max@0
|
122
|
max@0
|
123
|
max@0
|
124 //! diagmat * Base
|
max@0
|
125 template<typename T1, typename T2>
|
max@0
|
126 arma_inline
|
max@0
|
127 const Glue<Op<T1, op_diagmat>, T2, glue_times_diag>
|
max@0
|
128 operator*
|
max@0
|
129 (const Op<T1, op_diagmat>& X, const Base<typename T1::elem_type,T2>& Y)
|
max@0
|
130 {
|
max@0
|
131 arma_extra_debug_sigprint();
|
max@0
|
132
|
max@0
|
133 return Glue<Op<T1, op_diagmat>, T2, glue_times_diag>(X, Y.get_ref());
|
max@0
|
134 }
|
max@0
|
135
|
max@0
|
136
|
max@0
|
137
|
max@0
|
138 //! diagmat * diagmat
|
max@0
|
139 template<typename T1, typename T2>
|
max@0
|
140 arma_inline
|
max@0
|
141 Mat< typename promote_type<typename T1::elem_type, typename T2::elem_type>::result >
|
max@0
|
142 operator*
|
max@0
|
143 (const Op<T1, op_diagmat>& X, const Op<T2, op_diagmat>& Y)
|
max@0
|
144 {
|
max@0
|
145 arma_extra_debug_sigprint();
|
max@0
|
146
|
max@0
|
147 typedef typename T1::elem_type eT1;
|
max@0
|
148 typedef typename T2::elem_type eT2;
|
max@0
|
149
|
max@0
|
150 typedef typename promote_type<eT1,eT2>::result out_eT;
|
max@0
|
151
|
max@0
|
152 promote_type<eT1,eT2>::check();
|
max@0
|
153
|
max@0
|
154 const diagmat_proxy<T1> A(X.m);
|
max@0
|
155 const diagmat_proxy<T2> B(Y.m);
|
max@0
|
156
|
max@0
|
157 arma_debug_assert_mul_size(A.n_elem, A.n_elem, B.n_elem, B.n_elem, "matrix multiply");
|
max@0
|
158
|
max@0
|
159 const uword N = A.n_elem;
|
max@0
|
160
|
max@0
|
161 Mat<out_eT> out(N,N);
|
max@0
|
162
|
max@0
|
163 out.zeros();
|
max@0
|
164
|
max@0
|
165 for(uword i=0; i<N; ++i)
|
max@0
|
166 {
|
max@0
|
167 out.at(i,i) = upgrade_val<eT1,eT2>::apply( A[i] ) * upgrade_val<eT1,eT2>::apply( B[i] );
|
max@0
|
168 }
|
max@0
|
169
|
max@0
|
170 return out;
|
max@0
|
171 }
|
max@0
|
172
|
max@0
|
173
|
max@0
|
174
|
max@0
|
175 //! multiplication of Base objects with same element type
|
max@0
|
176 template<typename T1, typename T2>
|
max@0
|
177 arma_inline
|
max@0
|
178 const Glue<T1, T2, glue_times>
|
max@0
|
179 operator*
|
max@0
|
180 (const Base<typename T1::elem_type,T1>& X, const Base<typename T1::elem_type,T2>& Y)
|
max@0
|
181 {
|
max@0
|
182 arma_extra_debug_sigprint();
|
max@0
|
183
|
max@0
|
184 return Glue<T1, T2, glue_times>(X.get_ref(), Y.get_ref());
|
max@0
|
185 }
|
max@0
|
186
|
max@0
|
187
|
max@0
|
188
|
max@0
|
189 //! multiplication of Base objects with different element types
|
max@0
|
190 template<typename T1, typename T2>
|
max@0
|
191 inline
|
max@0
|
192 const mtGlue< typename promote_type<typename T1::elem_type, typename T2::elem_type>::result, T1, T2, glue_mixed_times >
|
max@0
|
193 operator*
|
max@0
|
194 (
|
max@0
|
195 const Base< typename force_different_type<typename T1::elem_type, typename T2::elem_type>::T1_result, T1>& X,
|
max@0
|
196 const Base< typename force_different_type<typename T1::elem_type, typename T2::elem_type>::T2_result, T2>& Y
|
max@0
|
197 )
|
max@0
|
198 {
|
max@0
|
199 arma_extra_debug_sigprint();
|
max@0
|
200
|
max@0
|
201 typedef typename T1::elem_type eT1;
|
max@0
|
202 typedef typename T2::elem_type eT2;
|
max@0
|
203
|
max@0
|
204 typedef typename promote_type<eT1,eT2>::result out_eT;
|
max@0
|
205
|
max@0
|
206 promote_type<eT1,eT2>::check();
|
max@0
|
207
|
max@0
|
208 return mtGlue<out_eT, T1, T2, glue_mixed_times>( X.get_ref(), Y.get_ref() );
|
max@0
|
209 }
|
max@0
|
210
|
max@0
|
211
|
max@0
|
212
|
max@0
|
213 //! @}
|