Mercurial > octave-nkf
view src/OPERATORS/op-sm-scm.cc @ 8968:91d53dc37f79
Add perm * sparse, perm \ sparse, sparse * perm, and sparse / perm operations.
Nothing terribly fancy in any of this. There probably is some
mechanism for using the permutation vectors and some assign or index
method in the sparse classes, but I've never understood all the
intricacies. I'm opting for a simple implementation at the cost of
possibly duplicating some functionality.
author | Jason Riedy <jason@acm.org> |
---|---|
date | Tue, 10 Mar 2009 21:54:44 -0400 |
parents | a1dbe9d80eee |
children | fb6b6fcafa62 |
line wrap: on
line source
/* Copyright (C) 2004, 2005, 2006, 2007 David Bateman Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003, 2004 Andy Adler This file is part of Octave. Octave is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 3 of the License, or (at your option) any later version. Octave is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with Octave; see the file COPYING. If not, see <http://www.gnu.org/licenses/>. */ #ifdef HAVE_CONFIG_H #include <config.h> #endif #include "gripes.h" #include "oct-obj.h" #include "ov.h" #include "ov-typeinfo.h" #include "ops.h" #include "sparse-xdiv.h" #include "sparse-xpow.h" #include "smx-sm-scm.h" #include "smx-scm-sm.h" #include "ov-re-sparse.h" #include "ov-cx-sparse.h" // sparse matrix by sparse complex matrix ops. DEFBINOP_OP (add, sparse_matrix, sparse_complex_matrix, +) DEFBINOP_OP (sub, sparse_matrix, sparse_complex_matrix, -) DEFBINOP_OP (mul, sparse_matrix, sparse_complex_matrix, *) DEFBINOP (div, sparse_matrix, sparse_complex_matrix) { CAST_BINOP_ARGS (const octave_sparse_matrix&, const octave_sparse_complex_matrix&); if (v2.rows() == 1 && v2.columns() == 1) { Complex d = v2.complex_value (); if (d == 0.0) gripe_divide_by_zero (); return octave_value (v1.sparse_matrix_value () / d); } else { MatrixType typ = v2.matrix_type (); SparseComplexMatrix ret = xdiv (v1.sparse_matrix_value (), v2.sparse_complex_matrix_value (), typ); v2.matrix_type (typ); return ret; } } DEFBINOPX (pow, sparse_matrix, sparse_complex_matrix) { error ("can't do A ^ B for A and B both matrices"); return octave_value (); } DEFBINOP (ldiv, sparse_matrix, sparse_complex_matrix) { CAST_BINOP_ARGS (const octave_sparse_matrix&, const octave_sparse_complex_matrix&); if (v1.rows() == 1 && v1.columns() == 1) { double d = v1.scalar_value (); if (d == 0.0) gripe_divide_by_zero (); return octave_value (v2.sparse_complex_matrix_value () / d); } else { MatrixType typ = v1.matrix_type (); SparseComplexMatrix ret = xleftdiv (v1.sparse_matrix_value (), v2.sparse_complex_matrix_value (), typ); v1.matrix_type (typ); return ret; } } DEFBINOP_FN (lt, sparse_matrix, sparse_complex_matrix, mx_el_lt) DEFBINOP_FN (le, sparse_matrix, sparse_complex_matrix, mx_el_le) DEFBINOP_FN (eq, sparse_matrix, sparse_complex_matrix, mx_el_eq) DEFBINOP_FN (ge, sparse_matrix, sparse_complex_matrix, mx_el_ge) DEFBINOP_FN (gt, sparse_matrix, sparse_complex_matrix, mx_el_gt) DEFBINOP_FN (ne, sparse_matrix, sparse_complex_matrix, mx_el_ne) DEFBINOP_FN (el_mul, sparse_matrix, sparse_complex_matrix, product) DEFBINOP_FN (el_div, sparse_matrix, sparse_complex_matrix, quotient) DEFBINOP_FN (el_pow, sparse_matrix, sparse_complex_matrix, elem_xpow) DEFBINOP (el_ldiv, sparse_matrix, sparse_complex_matrix) { CAST_BINOP_ARGS (const octave_sparse_matrix&, const octave_sparse_complex_matrix&); return octave_value (quotient (v2.sparse_complex_matrix_value (), v1.sparse_matrix_value ())); } DEFBINOP_FN (el_and, sparse_matrix, sparse_complex_matrix, mx_el_and) DEFBINOP_FN (el_or, sparse_matrix, sparse_complex_matrix, mx_el_or) DEFCATOP_FN (sm_scm, sparse_matrix, sparse_complex_matrix, concat) DEFCONV (sparse_complex_matrix_conv, sparse_matrix, sparse_complex_matrix) { CAST_CONV_ARG (const octave_sparse_matrix&); return new octave_sparse_complex_matrix (v.sparse_complex_matrix_value ()); } void install_sm_scm_ops (void) { INSTALL_BINOP (op_add, octave_sparse_matrix, octave_sparse_complex_matrix, add); INSTALL_BINOP (op_sub, octave_sparse_matrix, octave_sparse_complex_matrix, sub); INSTALL_BINOP (op_mul, octave_sparse_matrix, octave_sparse_complex_matrix, mul); INSTALL_BINOP (op_div, octave_sparse_matrix, octave_sparse_complex_matrix, div); INSTALL_BINOP (op_pow, octave_sparse_matrix, octave_sparse_complex_matrix, pow); INSTALL_BINOP (op_ldiv, octave_sparse_matrix, octave_sparse_complex_matrix, ldiv); INSTALL_BINOP (op_lt, octave_sparse_matrix, octave_sparse_complex_matrix, lt); INSTALL_BINOP (op_le, octave_sparse_matrix, octave_sparse_complex_matrix, le); INSTALL_BINOP (op_eq, octave_sparse_matrix, octave_sparse_complex_matrix, eq); INSTALL_BINOP (op_ge, octave_sparse_matrix, octave_sparse_complex_matrix, ge); INSTALL_BINOP (op_gt, octave_sparse_matrix, octave_sparse_complex_matrix, gt); INSTALL_BINOP (op_ne, octave_sparse_matrix, octave_sparse_complex_matrix, ne); INSTALL_BINOP (op_el_mul, octave_sparse_matrix, octave_sparse_complex_matrix, el_mul); INSTALL_BINOP (op_el_div, octave_sparse_matrix, octave_sparse_complex_matrix, el_div); INSTALL_BINOP (op_el_pow, octave_sparse_matrix, octave_sparse_complex_matrix, el_pow); INSTALL_BINOP (op_el_ldiv, octave_sparse_matrix, octave_sparse_complex_matrix, el_ldiv); INSTALL_BINOP (op_el_and, octave_sparse_matrix, octave_sparse_complex_matrix, el_and); INSTALL_BINOP (op_el_or, octave_sparse_matrix, octave_sparse_complex_matrix, el_or); INSTALL_CATOP (octave_sparse_matrix, octave_sparse_complex_matrix, sm_scm); INSTALL_ASSIGNCONV (octave_sparse_matrix, octave_sparse_complex_matrix, octave_sparse_complex_matrix); INSTALL_WIDENOP (octave_sparse_matrix, octave_sparse_complex_matrix, sparse_complex_matrix_conv); } /* ;;; Local Variables: *** ;;; mode: C++ *** ;;; End: *** */