1 ////////////////////////////////////////////////////////////////////////
2 //
3 // Copyright (C) 1996-2021 The Octave Project Developers
4 //
5 // See the file COPYRIGHT.md in the top-level directory of this
6 // distribution or <https://octave.org/copyright/>.
7 //
8 // This file is part of Octave.
9 //
10 // Octave is free software: you can redistribute it and/or modify it
11 // under the terms of the GNU General Public License as published by
12 // the Free Software Foundation, either version 3 of the License, or
13 // (at your option) any later version.
14 //
15 // Octave is distributed in the hope that it will be useful, but
16 // WITHOUT ANY WARRANTY; without even the implied warranty of
17 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18 // GNU General Public License for more details.
19 //
20 // You should have received a copy of the GNU General Public License
21 // along with Octave; see the file COPYING.  If not, see
22 // <https://www.gnu.org/licenses/>.
23 //
24 ////////////////////////////////////////////////////////////////////////
25 
26 #if defined (HAVE_CONFIG_H)
27 #  include "config.h"
28 #endif
29 
30 #include "ovl.h"
31 #include "ov.h"
32 #include "ov-cx-mat.h"
33 #include "ov-flt-cx-mat.h"
34 #include "ov-flt-complex.h"
35 #include "ov-complex.h"
36 #include "ov-typeinfo.h"
37 #include "ops.h"
38 #include "xdiv.h"
39 #include "xpow.h"
40 
41 // complex matrix by complex scalar ops.
42 
43 DEFNDBINOP_OP (add, float_complex_matrix, float_complex,
44                float_complex_array, float_complex, +)
45 DEFNDBINOP_OP (sub, float_complex_matrix, float_complex,
46                float_complex_array, float_complex, -)
47 DEFNDBINOP_OP (mul, float_complex_matrix, float_complex,
48                float_complex_array, float_complex, *)
49 
DEFBINOP(div,float_complex_matrix,float_complex)50 DEFBINOP (div, float_complex_matrix, float_complex)
51 {
52   const octave_float_complex_matrix& v1
53     = dynamic_cast<const octave_float_complex_matrix&> (a1);
54   const octave_float_complex& v2
55     = dynamic_cast<const octave_float_complex&> (a2);
56 
57   return octave_value (v1.float_complex_array_value () / v2.float_complex_value ());
58 }
59 
DEFBINOP_FN(pow,float_complex_matrix,float_complex,xpow)60 DEFBINOP_FN (pow, float_complex_matrix, float_complex, xpow)
61 
62 DEFBINOP (ldiv, float_complex_matrix, float_complex)
63 {
64   const octave_float_complex_matrix& v1
65     = dynamic_cast<const octave_float_complex_matrix&> (a1);
66   const octave_float_complex& v2
67     = dynamic_cast<const octave_float_complex&> (a2);
68 
69   FloatComplexMatrix m1 = v1.float_complex_matrix_value ();
70   FloatComplexMatrix m2 = v2.float_complex_matrix_value ();
71   MatrixType typ = v1.matrix_type ();
72 
73   FloatComplexMatrix ret = xleftdiv (m1, m2, typ);
74   v1.matrix_type (typ);
75   return ret;
76 }
77 
DEFNDCMPLXCMPOP_FN(lt,float_complex_matrix,float_complex,float_complex_array,float_complex,mx_el_lt)78 DEFNDCMPLXCMPOP_FN (lt, float_complex_matrix, float_complex,
79                     float_complex_array, float_complex, mx_el_lt)
80 DEFNDCMPLXCMPOP_FN (le, float_complex_matrix, float_complex,
81                     float_complex_array, float_complex, mx_el_le)
82 DEFNDCMPLXCMPOP_FN (eq, float_complex_matrix, float_complex,
83                     float_complex_array, float_complex, mx_el_eq)
84 DEFNDCMPLXCMPOP_FN (ge, float_complex_matrix, float_complex,
85                     float_complex_array, float_complex, mx_el_ge)
86 DEFNDCMPLXCMPOP_FN (gt, float_complex_matrix, float_complex,
87                     float_complex_array, float_complex, mx_el_gt)
88 DEFNDCMPLXCMPOP_FN (ne, float_complex_matrix, float_complex,
89                     float_complex_array, float_complex, mx_el_ne)
90 
91 DEFNDBINOP_OP (el_mul, float_complex_matrix, float_complex,
92                float_complex_array, float_complex, *)
93 
94 DEFBINOP (el_div, float_complex_matrix, float_complex)
95 {
96   const octave_float_complex_matrix& v1
97     = dynamic_cast<const octave_float_complex_matrix&> (a1);
98   const octave_float_complex& v2
99     = dynamic_cast<const octave_float_complex&> (a2);
100 
101   return octave_value (v1.float_complex_array_value () / v2.float_complex_value ());
102 }
103 
DEFNDBINOP_FN(el_pow,float_complex_matrix,float_complex,float_complex_array,float_complex,elem_xpow)104 DEFNDBINOP_FN (el_pow, float_complex_matrix, float_complex,
105                float_complex_array, float_complex, elem_xpow)
106 
107 DEFBINOP (el_ldiv, float_complex_matrix, float_complex)
108 {
109   const octave_float_complex_matrix& v1
110     = dynamic_cast<const octave_float_complex_matrix&> (a1);
111   const octave_float_complex& v2
112     = dynamic_cast<const octave_float_complex&> (a2);
113 
114   return x_el_div (v2.float_complex_value (), v1.float_complex_array_value ());
115 }
116 
DEFNDBINOP_FN(el_and,float_complex_matrix,float_complex,float_complex_array,float_complex,mx_el_and)117 DEFNDBINOP_FN (el_and, float_complex_matrix, float_complex,
118                float_complex_array, float_complex, mx_el_and)
119 DEFNDBINOP_FN (el_or,  float_complex_matrix, float_complex,
120                float_complex_array, float_complex, mx_el_or)
121 
122 DEFNDCATOP_FN (fcm_fcs, float_complex_matrix, float_complex,
123                float_complex_array, float_complex_array, concat)
124 
125 DEFNDCATOP_FN (cm_fcs, complex_matrix, float_complex,
126                float_complex_array, float_complex_array, concat)
127 
128 DEFNDCATOP_FN (fcm_cs, float_complex_matrix, complex,
129                float_complex_array, float_complex_array, concat)
130 
131 DEFNDASSIGNOP_FN (assign, float_complex_matrix, float_complex,
132                   float_complex, assign)
133 DEFNDASSIGNOP_FN (dbl_assign, complex_matrix, float_complex,
134                   complex, assign)
135 
136 DEFNDASSIGNOP_OP (assign_add, float_complex_matrix, float_complex_scalar,
137                   float_complex, +=)
138 DEFNDASSIGNOP_OP (assign_sub, float_complex_matrix, float_complex_scalar,
139                   float_complex, -=)
140 DEFNDASSIGNOP_OP (assign_mul, float_complex_matrix, float_complex_scalar,
141                   float_complex, *=)
142 DEFNDASSIGNOP_OP (assign_div, float_complex_matrix, float_complex_scalar,
143                   float_complex, /=)
144 
145 void
146 install_fcm_fcs_ops (octave::type_info& ti)
147 {
148   INSTALL_BINOP_TI (ti, op_add, octave_float_complex_matrix,
149                     octave_float_complex, add);
150   INSTALL_BINOP_TI (ti, op_sub, octave_float_complex_matrix,
151                     octave_float_complex, sub);
152   INSTALL_BINOP_TI (ti, op_mul, octave_float_complex_matrix,
153                     octave_float_complex, mul);
154   INSTALL_BINOP_TI (ti, op_div, octave_float_complex_matrix,
155                     octave_float_complex, div);
156   INSTALL_BINOP_TI (ti, op_pow, octave_float_complex_matrix,
157                     octave_float_complex, pow);
158   INSTALL_BINOP_TI (ti, op_ldiv, octave_float_complex_matrix,
159                     octave_float_complex, ldiv);
160   INSTALL_BINOP_TI (ti, op_lt, octave_float_complex_matrix, octave_float_complex,
161                     lt);
162   INSTALL_BINOP_TI (ti, op_le, octave_float_complex_matrix, octave_float_complex,
163                     le);
164   INSTALL_BINOP_TI (ti, op_eq, octave_float_complex_matrix, octave_float_complex,
165                     eq);
166   INSTALL_BINOP_TI (ti, op_ge, octave_float_complex_matrix, octave_float_complex,
167                     ge);
168   INSTALL_BINOP_TI (ti, op_gt, octave_float_complex_matrix, octave_float_complex,
169                     gt);
170   INSTALL_BINOP_TI (ti, op_ne, octave_float_complex_matrix, octave_float_complex,
171                     ne);
172   INSTALL_BINOP_TI (ti, op_el_mul, octave_float_complex_matrix,
173                     octave_float_complex, el_mul);
174   INSTALL_BINOP_TI (ti, op_el_div, octave_float_complex_matrix,
175                     octave_float_complex, el_div);
176   INSTALL_BINOP_TI (ti, op_el_pow, octave_float_complex_matrix,
177                     octave_float_complex, el_pow);
178   INSTALL_BINOP_TI (ti, op_el_ldiv, octave_float_complex_matrix,
179                     octave_float_complex, el_ldiv);
180   INSTALL_BINOP_TI (ti, op_el_and, octave_float_complex_matrix,
181                     octave_float_complex, el_and);
182   INSTALL_BINOP_TI (ti, op_el_or, octave_float_complex_matrix,
183                     octave_float_complex, el_or);
184 
185   INSTALL_CATOP_TI (ti, octave_float_complex_matrix, octave_float_complex,
186                     fcm_fcs);
187   INSTALL_CATOP_TI (ti, octave_complex_matrix, octave_float_complex, cm_fcs);
188   INSTALL_CATOP_TI (ti, octave_float_complex_matrix, octave_complex, fcm_cs);
189 
190   INSTALL_ASSIGNOP_TI (ti, op_asn_eq, octave_float_complex_matrix,
191                        octave_float_complex, assign);
192   INSTALL_ASSIGNOP_TI (ti, op_asn_eq, octave_complex_matrix,
193                        octave_float_complex, dbl_assign);
194 
195   INSTALL_ASSIGNOP_TI (ti, op_add_eq, octave_float_complex_matrix,
196                        octave_float_complex_scalar, assign_add);
197   INSTALL_ASSIGNOP_TI (ti, op_sub_eq, octave_float_complex_matrix,
198                        octave_float_complex_scalar, assign_sub);
199   INSTALL_ASSIGNOP_TI (ti, op_mul_eq, octave_float_complex_matrix,
200                        octave_float_complex_scalar, assign_mul);
201   INSTALL_ASSIGNOP_TI (ti, op_div_eq, octave_float_complex_matrix,
202                        octave_float_complex_scalar, assign_div);
203 }
204