1 ////////////////////////////////////////////////////////////////////////
2 //
3 // Copyright (C) 1998-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-typeinfo.h"
33 #include "ov-cx-mat.h"
34 #include "ops.h"
35 #include "xdiv.h"
36
37 #include "sparse-xpow.h"
38 #include "sparse-xdiv.h"
39 #include "smx-sm-cm.h"
40 #include "smx-cm-sm.h"
41 #include "ov-re-sparse.h"
42
43 // sparse matrix by complex matrix ops.
44
45 DEFBINOP_OP (add, sparse_matrix, complex_matrix, +)
46 DEFBINOP_OP (sub, sparse_matrix, complex_matrix, -)
47
48 DEFBINOP_OP (mul, sparse_matrix, complex_matrix, *)
49
DEFBINOP(div,sparse_matrix,complex_matrix)50 DEFBINOP (div, sparse_matrix, complex_matrix)
51 {
52 const octave_sparse_matrix& v1
53 = dynamic_cast<const octave_sparse_matrix&> (a1);
54 const octave_complex_matrix& v2
55 = dynamic_cast<const octave_complex_matrix&> (a2);
56 MatrixType typ = v2.matrix_type ();
57
58 ComplexMatrix ret = xdiv (v1.matrix_value (),
59 v2.complex_matrix_value (), typ);
60
61 v2.matrix_type (typ);
62 return ret;
63 }
64
DEFBINOPX(pow,sparse_matrix,complex_matrix)65 DEFBINOPX (pow, sparse_matrix, complex_matrix)
66 {
67 error ("can't do A ^ B for A and B both matrices");
68 }
69
DEFBINOP(ldiv,sparse_matrix,complex_matrix)70 DEFBINOP (ldiv, sparse_matrix, complex_matrix)
71 {
72 const octave_sparse_matrix& v1
73 = dynamic_cast<const octave_sparse_matrix&> (a1);
74 const octave_complex_matrix& v2
75 = dynamic_cast<const octave_complex_matrix&> (a2);
76
77 if (v1.rows () == 1 && v1.columns () == 1)
78 return octave_value (v2.complex_array_value () / v1.scalar_value ());
79 else
80 {
81 MatrixType typ = v1.matrix_type ();
82
83 ComplexMatrix ret = xleftdiv (v1.sparse_matrix_value (),
84 v2.complex_matrix_value (), typ);
85
86 v1.matrix_type (typ);
87 return ret;
88 }
89 }
90
DEFBINOP_FN(lt,sparse_matrix,complex_matrix,mx_el_lt)91 DEFBINOP_FN (lt, sparse_matrix, complex_matrix, mx_el_lt)
92 DEFBINOP_FN (le, sparse_matrix, complex_matrix, mx_el_le)
93 DEFBINOP_FN (eq, sparse_matrix, complex_matrix, mx_el_eq)
94 DEFBINOP_FN (ge, sparse_matrix, complex_matrix, mx_el_ge)
95 DEFBINOP_FN (gt, sparse_matrix, complex_matrix, mx_el_gt)
96 DEFBINOP_FN (ne, sparse_matrix, complex_matrix, mx_el_ne)
97
98 DEFBINOP_FN (el_mul, sparse_matrix, complex_matrix, product)
99 DEFBINOP_FN (el_div, sparse_matrix, complex_matrix, quotient)
100
101 DEFBINOP (el_pow, sparse_matrix, complex_matrix)
102 {
103 const octave_sparse_matrix& v1
104 = dynamic_cast<const octave_sparse_matrix&> (a1);
105 const octave_complex_matrix& v2
106 = dynamic_cast<const octave_complex_matrix&> (a2);
107
108 return octave_value
109 (elem_xpow (v1.sparse_matrix_value (), SparseComplexMatrix
110 (v2.complex_matrix_value ())));
111 }
112
DEFBINOP(el_ldiv,sparse_matrix,complex_matrix)113 DEFBINOP (el_ldiv, sparse_matrix, complex_matrix)
114 {
115 const octave_sparse_matrix& v1
116 = dynamic_cast<const octave_sparse_matrix&> (a1);
117 const octave_complex_matrix& v2
118 = dynamic_cast<const octave_complex_matrix&> (a2);
119
120 return octave_value
121 (quotient (v2.complex_matrix_value (), v1.sparse_matrix_value ()));
122 }
123
DEFBINOP_FN(el_and,sparse_matrix,complex_matrix,mx_el_and)124 DEFBINOP_FN (el_and, sparse_matrix, complex_matrix, mx_el_and)
125 DEFBINOP_FN (el_or, sparse_matrix, complex_matrix, mx_el_or)
126
127 DEFCATOP (sm_cm, sparse_matrix, complex_matrix)
128 {
129 octave_sparse_matrix& v1 = dynamic_cast<octave_sparse_matrix&> (a1);
130 const octave_complex_matrix& v2
131 = dynamic_cast<const octave_complex_matrix&> (a2);
132 SparseComplexMatrix tmp (v2.complex_matrix_value ());
133 return octave_value (v1.sparse_matrix_value (). concat (tmp, ra_idx));
134 }
135
DEFCONV(sparse_complex_matrix_conv,sparse_matrix,sparse_complex_matrix)136 DEFCONV (sparse_complex_matrix_conv, sparse_matrix, sparse_complex_matrix)
137 {
138 const octave_sparse_matrix& v
139 = dynamic_cast<const octave_sparse_matrix&> (a);
140 return new octave_complex_matrix (v.complex_matrix_value ());
141 }
142
143 void
install_sm_cm_ops(octave::type_info & ti)144 install_sm_cm_ops (octave::type_info& ti)
145 {
146 INSTALL_BINOP_TI (ti, op_add, octave_sparse_matrix, octave_complex_matrix, add);
147 INSTALL_BINOP_TI (ti, op_sub, octave_sparse_matrix, octave_complex_matrix, sub);
148 INSTALL_BINOP_TI (ti, op_mul, octave_sparse_matrix, octave_complex_matrix, mul);
149 INSTALL_BINOP_TI (ti, op_div, octave_sparse_matrix, octave_complex_matrix, div);
150 INSTALL_BINOP_TI (ti, op_pow, octave_sparse_matrix, octave_complex_matrix, pow);
151 INSTALL_BINOP_TI (ti, op_ldiv, octave_sparse_matrix, octave_complex_matrix,
152 ldiv);
153 INSTALL_BINOP_TI (ti, op_lt, octave_sparse_matrix, octave_complex_matrix, lt);
154 INSTALL_BINOP_TI (ti, op_le, octave_sparse_matrix, octave_complex_matrix, le);
155 INSTALL_BINOP_TI (ti, op_eq, octave_sparse_matrix, octave_complex_matrix, eq);
156 INSTALL_BINOP_TI (ti, op_ge, octave_sparse_matrix, octave_complex_matrix, ge);
157 INSTALL_BINOP_TI (ti, op_gt, octave_sparse_matrix, octave_complex_matrix, gt);
158 INSTALL_BINOP_TI (ti, op_ne, octave_sparse_matrix, octave_complex_matrix, ne);
159 INSTALL_BINOP_TI (ti, op_el_mul, octave_sparse_matrix, octave_complex_matrix,
160 el_mul);
161 INSTALL_BINOP_TI (ti, op_el_div, octave_sparse_matrix, octave_complex_matrix,
162 el_div);
163 INSTALL_BINOP_TI (ti, op_el_pow, octave_sparse_matrix, octave_complex_matrix,
164 el_pow);
165 INSTALL_BINOP_TI (ti, op_el_ldiv, octave_sparse_matrix, octave_complex_matrix,
166 el_ldiv);
167 INSTALL_BINOP_TI (ti, op_el_and, octave_sparse_matrix, octave_complex_matrix,
168 el_and);
169 INSTALL_BINOP_TI (ti, op_el_or, octave_sparse_matrix, octave_complex_matrix,
170 el_or);
171
172 INSTALL_CATOP_TI (ti, octave_sparse_matrix, octave_complex_matrix, sm_cm);
173
174 INSTALL_ASSIGNCONV_TI (ti, octave_sparse_matrix, octave_complex_matrix,
175 octave_sparse_complex_matrix);
176
177 INSTALL_WIDENOP_TI (ti, octave_sparse_matrix, octave_complex_matrix,
178 sparse_complex_matrix_conv);
179 }
180