Mercurial > octave-nkf
view liboctave/operators/Sparse-diag-op-defs.h @ 17822:ebb3ef964372
maint: Use common #define syntax "octave_filename_h" in h_files.
* libgui/src/dialog.h, libgui/src/find-files-dialog.h,
libgui/src/find-files-model.h, libgui/src/m-editor/file-editor-tab.h,
libgui/src/m-editor/octave-qscintilla.h, libgui/src/main-window.h,
libgui/src/octave-dock-widget.h, libgui/src/workspace-model.h,
libgui/src/workspace-view.h, libinterp/corefcn/Cell.h,
libinterp/corefcn/gl-render.h, libinterp/corefcn/gl2ps-renderer.h,
libinterp/corefcn/graphics.in.h, libinterp/corefcn/oct-errno.h,
libinterp/corefcn/oct-fstrm.h, libinterp/corefcn/oct-handle.h,
libinterp/corefcn/oct-hdf5.h, libinterp/corefcn/oct-hist.h,
libinterp/corefcn/oct-iostrm.h, libinterp/corefcn/oct-lvalue.h,
libinterp/corefcn/oct-prcstrm.h, libinterp/corefcn/oct-procbuf.h,
libinterp/corefcn/oct-stdstrm.h, libinterp/corefcn/oct-stream.h,
libinterp/corefcn/oct-strstrm.h, libinterp/corefcn/octave-link.h,
libinterp/corefcn/pt-jit.h, libinterp/octave-value/ov-base-diag.h,
libinterp/octave-value/ov-base-int.h, libinterp/octave-value/ov-base-mat.h,
libinterp/octave-value/ov-base-scalar.h,
libinterp/octave-value/ov-base-sparse.h, libinterp/octave-value/ov-base.h,
libinterp/octave-value/ov-bool-mat.h, libinterp/octave-value/ov-bool-sparse.h,
libinterp/octave-value/ov-bool.h, libinterp/octave-value/ov-builtin.h,
libinterp/octave-value/ov-cell.h, libinterp/octave-value/ov-ch-mat.h,
libinterp/octave-value/ov-class.h, libinterp/octave-value/ov-colon.h,
libinterp/octave-value/ov-complex.h, libinterp/octave-value/ov-cs-list.h,
libinterp/octave-value/ov-cx-diag.h, libinterp/octave-value/ov-cx-mat.h,
libinterp/octave-value/ov-cx-sparse.h, libinterp/octave-value/ov-dld-fcn.h,
libinterp/octave-value/ov-fcn-handle.h, libinterp/octave-value/ov-fcn-inline.h,
libinterp/octave-value/ov-fcn.h, libinterp/octave-value/ov-float.h,
libinterp/octave-value/ov-flt-complex.h,
libinterp/octave-value/ov-flt-cx-diag.h,
libinterp/octave-value/ov-flt-cx-mat.h,
libinterp/octave-value/ov-flt-re-diag.h,
libinterp/octave-value/ov-flt-re-mat.h, libinterp/octave-value/ov-int-traits.h,
libinterp/octave-value/ov-int16.h, libinterp/octave-value/ov-int32.h,
libinterp/octave-value/ov-int64.h, libinterp/octave-value/ov-int8.h,
libinterp/octave-value/ov-java.h, libinterp/octave-value/ov-lazy-idx.h,
libinterp/octave-value/ov-mex-fcn.h, libinterp/octave-value/ov-null-mat.h,
libinterp/octave-value/ov-perm.h, libinterp/octave-value/ov-range.h,
libinterp/octave-value/ov-re-diag.h, libinterp/octave-value/ov-re-mat.h,
libinterp/octave-value/ov-re-sparse.h, libinterp/octave-value/ov-scalar.h,
libinterp/octave-value/ov-str-mat.h, libinterp/octave-value/ov-struct.h,
libinterp/octave-value/ov-typeinfo.h, libinterp/octave-value/ov-uint16.h,
libinterp/octave-value/ov-uint32.h, libinterp/octave-value/ov-uint64.h,
libinterp/octave-value/ov-uint8.h, libinterp/octave-value/ov-usr-fcn.h,
libinterp/octave-value/ov.h, libinterp/parse-tree/pt-all.h,
libinterp/parse-tree/pt-arg-list.h, libinterp/parse-tree/pt-array-list.h,
libinterp/parse-tree/pt-assign.h, libinterp/parse-tree/pt-binop.h,
libinterp/parse-tree/pt-bp.h, libinterp/parse-tree/pt-cbinop.h,
libinterp/parse-tree/pt-cell.h, libinterp/parse-tree/pt-check.h,
libinterp/parse-tree/pt-cmd.h, libinterp/parse-tree/pt-colon.h,
libinterp/parse-tree/pt-const.h, libinterp/parse-tree/pt-decl.h,
libinterp/parse-tree/pt-eval.h, libinterp/parse-tree/pt-except.h,
libinterp/parse-tree/pt-exp.h, libinterp/parse-tree/pt-fcn-handle.h,
libinterp/parse-tree/pt-id.h, libinterp/parse-tree/pt-idx.h,
libinterp/parse-tree/pt-jump.h, libinterp/parse-tree/pt-loop.h,
libinterp/parse-tree/pt-mat.h, libinterp/parse-tree/pt-misc.h,
libinterp/parse-tree/pt-pr-code.h, libinterp/parse-tree/pt-select.h,
libinterp/parse-tree/pt-stmt.h, libinterp/parse-tree/pt-unop.h,
libinterp/parse-tree/pt-walk.h, libinterp/parse-tree/pt.h,
liboctave/array/CColVector.h, liboctave/array/CDiagMatrix.h,
liboctave/array/CMatrix.h, liboctave/array/CNDArray.h,
liboctave/array/CRowVector.h, liboctave/array/boolMatrix.h,
liboctave/array/chMatrix.h, liboctave/array/chNDArray.h,
liboctave/array/dColVector.h, liboctave/array/dDiagMatrix.h,
liboctave/array/dMatrix.h, liboctave/array/dNDArray.h,
liboctave/array/dRowVector.h, liboctave/array/fCColVector.h,
liboctave/array/fCDiagMatrix.h, liboctave/array/fCMatrix.h,
liboctave/array/fCNDArray.h, liboctave/array/fCRowVector.h,
liboctave/array/fColVector.h, liboctave/array/fDiagMatrix.h,
liboctave/array/fMatrix.h, liboctave/array/fNDArray.h,
liboctave/array/fRowVector.h, liboctave/cruft/misc/lo-error.h,
liboctave/numeric/CmplxAEPBAL.h, liboctave/numeric/CmplxCHOL.h,
liboctave/numeric/CmplxGEPBAL.h, liboctave/numeric/CmplxHESS.h,
liboctave/numeric/CmplxLU.h, liboctave/numeric/CmplxQR.h,
liboctave/numeric/CmplxQRP.h, liboctave/numeric/CmplxSCHUR.h,
liboctave/numeric/CmplxSVD.h, liboctave/numeric/Quad.h,
liboctave/numeric/SparseCmplxCHOL.h, liboctave/numeric/SparseCmplxLU.h,
liboctave/numeric/SparseCmplxQR.h, liboctave/numeric/SparseQR.h,
liboctave/numeric/SparsedbleCHOL.h, liboctave/numeric/SparsedbleLU.h,
liboctave/numeric/bsxfun.h, liboctave/numeric/dbleAEPBAL.h,
liboctave/numeric/dbleCHOL.h, liboctave/numeric/dbleGEPBAL.h,
liboctave/numeric/dbleHESS.h, liboctave/numeric/dbleLU.h,
liboctave/numeric/dbleQR.h, liboctave/numeric/dbleQRP.h,
liboctave/numeric/dbleSCHUR.h, liboctave/numeric/dbleSVD.h,
liboctave/numeric/fCmplxAEPBAL.h, liboctave/numeric/fCmplxCHOL.h,
liboctave/numeric/fCmplxGEPBAL.h, liboctave/numeric/fCmplxHESS.h,
liboctave/numeric/fCmplxLU.h, liboctave/numeric/fCmplxQR.h,
liboctave/numeric/fCmplxQRP.h, liboctave/numeric/fCmplxSCHUR.h,
liboctave/numeric/fCmplxSVD.h, liboctave/numeric/fEIG.h,
liboctave/numeric/floatAEPBAL.h, liboctave/numeric/floatCHOL.h,
liboctave/numeric/floatGEPBAL.h, liboctave/numeric/floatHESS.h,
liboctave/numeric/floatLU.h, liboctave/numeric/floatQR.h,
liboctave/numeric/floatQRP.h, liboctave/numeric/floatSCHUR.h,
liboctave/numeric/floatSVD.h, liboctave/numeric/lo-mappers.h,
liboctave/numeric/lo-specfun.h, liboctave/numeric/oct-convn.h,
liboctave/numeric/oct-norm.h, liboctave/numeric/oct-rand.h,
liboctave/operators/Sparse-diag-op-defs.h,
liboctave/operators/Sparse-op-defs.h,
liboctave/operators/Sparse-perm-op-defs.h, liboctave/system/lo-sysdep.h,
liboctave/system/oct-env.h, liboctave/system/oct-group.h,
liboctave/system/oct-openmp.h, liboctave/system/oct-passwd.h,
liboctave/system/oct-syscalls.h, liboctave/system/oct-time.h,
liboctave/system/oct-uname.h, liboctave/util/lo-array-gripes.h,
liboctave/util/lo-cutils.h, liboctave/util/lo-ieee.h,
liboctave/util/lo-macros.h, liboctave/util/lo-math.h,
liboctave/util/lo-regexp.h, liboctave/util/lo-traits.h,
liboctave/util/lo-utils.h, liboctave/util/oct-base64.h,
liboctave/util/oct-binmap.h, liboctave/util/oct-glob.h,
liboctave/util/oct-inttypes.h, liboctave/util/oct-locbuf.h,
liboctave/util/oct-mem.h, liboctave/util/oct-mutex.h,
liboctave/util/oct-refcount.h, liboctave/util/oct-rl-edit.h,
liboctave/util/oct-rl-hist.h, liboctave/util/oct-shlib.h,
liboctave/util/oct-sort.h, liboctave/util/oct-sparse.h:
Use common #define syntax "octave_filename_h" in h_files.
author | Rik <rik@octave.org> |
---|---|
date | Thu, 31 Oct 2013 22:19:07 -0700 |
parents | 49a5a4be04a1 |
children | 4197fc428c7d |
line wrap: on
line source
/* -*- C++ -*- Copyright (C) 2009-2013 Jason Riedy, Jaroslav Hajek 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/>. */ #if !defined (octave_Sparse_diag_op_defs_h) #define octave_Sparse_diag_op_defs_h 1 // Matrix multiplication template <typename RT, typename DM, typename SM> RT do_mul_dm_sm (const DM& d, const SM& a) { const octave_idx_type nr = d.rows (); const octave_idx_type nc = d.cols (); const octave_idx_type a_nr = a.rows (); const octave_idx_type a_nc = a.cols (); if (nc != a_nr) { gripe_nonconformant ("operator *", nr, nc, a_nr, a_nc); return RT (); } else { RT r (nr, a_nc, a.nnz ()); octave_idx_type l = 0; for (octave_idx_type j = 0; j < a_nc; j++) { r.xcidx (j) = l; const octave_idx_type colend = a.cidx (j+1); for (octave_idx_type k = a.cidx (j); k < colend; k++) { const octave_idx_type i = a.ridx (k); if (i >= nr) break; r.xdata (l) = d.dgelem (i) * a.data (k); r.xridx (l) = i; l++; } } r.xcidx (a_nc) = l; r.maybe_compress (true); return r; } } template <typename RT, typename SM, typename DM> RT do_mul_sm_dm (const SM& a, const DM& d) { const octave_idx_type nr = d.rows (); const octave_idx_type nc = d.cols (); const octave_idx_type a_nr = a.rows (); const octave_idx_type a_nc = a.cols (); if (nr != a_nc) { gripe_nonconformant ("operator *", a_nr, a_nc, nr, nc); return RT (); } else { const octave_idx_type mnc = nc < a_nc ? nc: a_nc; RT r (a_nr, nc, a.cidx (mnc)); for (octave_idx_type j = 0; j < mnc; ++j) { const typename DM::element_type s = d.dgelem (j); const octave_idx_type colend = a.cidx (j+1); r.xcidx (j) = a.cidx (j); for (octave_idx_type k = a.cidx (j); k < colend; ++k) { r.xdata (k) = s * a.data (k); r.xridx (k) = a.ridx (k); } } for (octave_idx_type j = mnc; j <= nc; ++j) r.xcidx (j) = a.cidx (mnc); r.maybe_compress (true); return r; } } // FIXME: functors such as this should be gathered somewhere template <typename T> struct identity_val : public std::unary_function <T, T> { T operator () (const T x) { return x; } }; // Matrix addition template <typename RT, typename SM, typename DM, typename OpA, typename OpD> RT inner_do_add_sm_dm (const SM& a, const DM& d, OpA opa, OpD opd) { using std::min; const octave_idx_type nr = d.rows (); const octave_idx_type nc = d.cols (); const octave_idx_type n = min (nr, nc); const octave_idx_type a_nr = a.rows (); const octave_idx_type a_nc = a.cols (); const octave_idx_type nz = a.nnz (); RT r (a_nr, a_nc, nz + n); octave_idx_type k = 0; for (octave_idx_type j = 0; j < nc; ++j) { octave_quit (); const octave_idx_type colend = a.cidx (j+1); r.xcidx (j) = k; octave_idx_type k_src = a.cidx (j), k_split; for (k_split = k_src; k_split < colend; k_split++) if (a.ridx (k_split) >= j) break; for (; k_src < k_split; k_src++, k++) { r.xridx (k) = a.ridx (k_src); r.xdata (k) = opa (a.data (k_src)); } if (k_src < colend && a.ridx (k_src) == j) { r.xridx (k) = j; r.xdata (k) = opa (a.data (k_src)) + opd (d.dgelem (j)); k++; k_src++; } else { r.xridx (k) = j; r.xdata (k) = opd (d.dgelem (j)); k++; } for (; k_src < colend; k_src++, k++) { r.xridx (k) = a.ridx (k_src); r.xdata (k) = opa (a.data (k_src)); } } r.xcidx (nc) = k; r.maybe_compress (true); return r; } template <typename RT, typename DM, typename SM> RT do_commutative_add_dm_sm (const DM& d, const SM& a) { // Extra function to ensure this is only emitted once. return inner_do_add_sm_dm<RT> (a, d, identity_val<typename SM::element_type> (), identity_val<typename DM::element_type> ()); } template <typename RT, typename DM, typename SM> RT do_add_dm_sm (const DM& d, const SM& a) { if (a.rows () != d.rows () || a.cols () != d.cols ()) { gripe_nonconformant ("operator +", d.rows (), d.cols (), a.rows (), a.cols ()); return RT (); } else return do_commutative_add_dm_sm<RT> (d, a); } template <typename RT, typename DM, typename SM> RT do_sub_dm_sm (const DM& d, const SM& a) { if (a.rows () != d.rows () || a.cols () != d.cols ()) { gripe_nonconformant ("operator -", d.rows (), d.cols (), a.rows (), a.cols ()); return RT (); } else return inner_do_add_sm_dm<RT> (a, d, std::negate<typename SM::element_type> (), identity_val<typename DM::element_type> ()); } template <typename RT, typename SM, typename DM> RT do_add_sm_dm (const SM& a, const DM& d) { if (a.rows () != d.rows () || a.cols () != d.cols ()) { gripe_nonconformant ("operator +", a.rows (), a.cols (), d.rows (), d.cols ()); return RT (); } else return do_commutative_add_dm_sm<RT> (d, a); } template <typename RT, typename SM, typename DM> RT do_sub_sm_dm (const SM& a, const DM& d) { if (a.rows () != d.rows () || a.cols () != d.cols ()) { gripe_nonconformant ("operator -", a.rows (), a.cols (), d.rows (), d.cols ()); return RT (); } else return inner_do_add_sm_dm<RT> (a, d, identity_val<typename SM::element_type> (), std::negate<typename DM::element_type> ()); } #endif // octave_Sparse_diag_op_defs_h