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 "mx-cm-m.h"
31 #include "mx-m-cm.h"
32 #include "mx-cnda-nda.h"
33 #include "mx-nda-cnda.h"
34
35 #include "errwarn.h"
36 #include "ovl.h"
37 #include "ov.h"
38 #include "ov-cx-mat.h"
39 #include "ov-re-mat.h"
40 #include "ov-typeinfo.h"
41 #include "ops.h"
42 #include "xdiv.h"
43 #include "xpow.h"
44
45 // complex matrix by matrix ops.
46
47 DEFNDBINOP_OP (add, complex_matrix, matrix, complex_array, array, +)
48 DEFNDBINOP_OP (sub, complex_matrix, matrix, complex_array, array, -)
49
50 DEFBINOP_OP (mul, complex_matrix, matrix, *)
51
DEFBINOP(mul_trans,complex_matrix,matrix)52 DEFBINOP (mul_trans, complex_matrix, matrix)
53 {
54 const octave_complex_matrix& v1
55 = dynamic_cast<const octave_complex_matrix&> (a1);
56 const octave_matrix& v2 = dynamic_cast<const octave_matrix&> (a2);
57
58 ComplexMatrix m1 = v1.complex_matrix_value ();
59 Matrix m2 = v2.matrix_value ();
60
61 return ComplexMatrix (xgemm (real (m1), m2, blas_no_trans, blas_trans),
62 xgemm (imag (m1), m2, blas_no_trans, blas_trans));
63 }
64
DEFBINOP(div,complex_matrix,matrix)65 DEFBINOP (div, complex_matrix, matrix)
66 {
67 const octave_complex_matrix& v1
68 = dynamic_cast<const octave_complex_matrix&> (a1);
69 const octave_matrix& v2 = dynamic_cast<const octave_matrix&> (a2);
70 MatrixType typ = v2.matrix_type ();
71
72 ComplexMatrix ret = xdiv (v1.complex_matrix_value (),
73 v2.matrix_value (), typ);
74
75 v2.matrix_type (typ);
76 return ret;
77 }
78
DEFBINOPX(pow,complex_matrix,matrix)79 DEFBINOPX (pow, complex_matrix, matrix)
80 {
81 error ("can't do A ^ B for A and B both matrices");
82 }
83
DEFBINOP(ldiv,complex_matrix,matrix)84 DEFBINOP (ldiv, complex_matrix, matrix)
85 {
86 const octave_complex_matrix& v1
87 = dynamic_cast<const octave_complex_matrix&> (a1);
88 const octave_matrix& v2 = dynamic_cast<const octave_matrix&> (a2);
89 MatrixType typ = v1.matrix_type ();
90
91 ComplexMatrix ret = xleftdiv (v1.complex_matrix_value (),
92 v2.matrix_value (), typ);
93
94 v1.matrix_type (typ);
95 return ret;
96 }
97
DEFNDCMPLXCMPOP_FN(lt,complex_matrix,matrix,complex_array,array,mx_el_lt)98 DEFNDCMPLXCMPOP_FN (lt, complex_matrix, matrix, complex_array, array, mx_el_lt)
99 DEFNDCMPLXCMPOP_FN (le, complex_matrix, matrix, complex_array, array, mx_el_le)
100 DEFNDCMPLXCMPOP_FN (eq, complex_matrix, matrix, complex_array, array, mx_el_eq)
101 DEFNDCMPLXCMPOP_FN (ge, complex_matrix, matrix, complex_array, array, mx_el_ge)
102 DEFNDCMPLXCMPOP_FN (gt, complex_matrix, matrix, complex_array, array, mx_el_gt)
103 DEFNDCMPLXCMPOP_FN (ne, complex_matrix, matrix, complex_array, array, mx_el_ne)
104
105 DEFNDBINOP_FN (el_mul, complex_matrix, matrix, complex_array, array, product)
106 DEFNDBINOP_FN (el_div, complex_matrix, matrix, complex_array, array, quotient)
107 DEFNDBINOP_FN (el_pow, complex_matrix, matrix, complex_array, array, elem_xpow)
108
109 DEFBINOP (el_ldiv, complex_matrix, matrix)
110 {
111 const octave_complex_matrix& v1
112 = dynamic_cast<const octave_complex_matrix&> (a1);
113 const octave_matrix& v2 = dynamic_cast<const octave_matrix&> (a2);
114
115 return quotient (v2.array_value (), v1.complex_array_value ());
116 }
117
DEFNDBINOP_FN(el_and,complex_matrix,matrix,complex_array,array,mx_el_and)118 DEFNDBINOP_FN (el_and, complex_matrix, matrix, complex_array, array, mx_el_and)
119 DEFNDBINOP_FN (el_or, complex_matrix, matrix, complex_array, array, mx_el_or)
120
121 DEFNDCATOP_FN (cm_m, complex_matrix, matrix, complex_array, array, concat)
122
123 DEFNDASSIGNOP_FN (assign, complex_matrix, matrix, complex_array, assign)
124
125 void
126 install_cm_m_ops (octave::type_info& ti)
127 {
128 INSTALL_BINOP_TI (ti, op_add, octave_complex_matrix, octave_matrix, add);
129 INSTALL_BINOP_TI (ti, op_sub, octave_complex_matrix, octave_matrix, sub);
130 INSTALL_BINOP_TI (ti, op_mul, octave_complex_matrix, octave_matrix, mul);
131 INSTALL_BINOP_TI (ti, op_div, octave_complex_matrix, octave_matrix, div);
132 INSTALL_BINOP_TI (ti, op_pow, octave_complex_matrix, octave_matrix, pow);
133 INSTALL_BINOP_TI (ti, op_ldiv, octave_complex_matrix, octave_matrix, ldiv);
134 INSTALL_BINOP_TI (ti, op_lt, octave_complex_matrix, octave_matrix, lt);
135 INSTALL_BINOP_TI (ti, op_le, octave_complex_matrix, octave_matrix, le);
136 INSTALL_BINOP_TI (ti, op_eq, octave_complex_matrix, octave_matrix, eq);
137 INSTALL_BINOP_TI (ti, op_ge, octave_complex_matrix, octave_matrix, ge);
138 INSTALL_BINOP_TI (ti, op_gt, octave_complex_matrix, octave_matrix, gt);
139 INSTALL_BINOP_TI (ti, op_ne, octave_complex_matrix, octave_matrix, ne);
140 INSTALL_BINOP_TI (ti, op_el_mul, octave_complex_matrix, octave_matrix, el_mul);
141 INSTALL_BINOP_TI (ti, op_el_div, octave_complex_matrix, octave_matrix, el_div);
142 INSTALL_BINOP_TI (ti, op_el_pow, octave_complex_matrix, octave_matrix, el_pow);
143 INSTALL_BINOP_TI (ti, op_el_ldiv, octave_complex_matrix, octave_matrix,
144 el_ldiv);
145 INSTALL_BINOP_TI (ti, op_el_and, octave_complex_matrix, octave_matrix, el_and);
146 INSTALL_BINOP_TI (ti, op_el_or, octave_complex_matrix, octave_matrix, el_or);
147 INSTALL_BINOP_TI (ti, op_mul_trans, octave_complex_matrix, octave_matrix,
148 mul_trans);
149 INSTALL_BINOP_TI (ti, op_mul_herm, octave_complex_matrix, octave_matrix,
150 mul_trans);
151
152 INSTALL_CATOP_TI (ti, octave_complex_matrix, octave_matrix, cm_m);
153
154 INSTALL_ASSIGNOP_TI (ti, op_asn_eq, octave_complex_matrix, octave_matrix,
155 assign);
156 }
157