Mercurial > octave
view liboctave/array/CNDArray.cc @ 21136:7cac4e7458f2
maint: clean up code around calls to current_liboctave_error_handler.
Remove statements after call to handler that are no longer reachable.
Place input validation first and immediately call handler if necessary.
Change if/error_handler/else to if/error_handler and re-indent code.
* Array-util.cc, Array.cc, CColVector.cc, CDiagMatrix.cc, CMatrix.cc,
CNDArray.cc, CRowVector.cc, CSparse.cc, DiagArray2.cc, MArray.cc,
PermMatrix.cc, Sparse.cc, Sparse.h, chMatrix.cc, chNDArray.cc, dColVector.cc,
dDiagMatrix.cc, dMatrix.cc, dNDArray.cc, dRowVector.cc, dSparse.cc,
fCColVector.cc, fCDiagMatrix.cc, fCMatrix.cc, fCNDArray.cc, fCRowVector.cc,
fColVector.cc, fDiagMatrix.cc, fMatrix.cc, fNDArray.cc, fRowVector.cc,
idx-vector.cc, CmplxAEPBAL.cc, CmplxCHOL.cc, CmplxGEPBAL.cc, CmplxHESS.cc,
CmplxLU.cc, CmplxQR.cc, CmplxSCHUR.cc, CmplxSVD.cc, DASPK.cc, EIG.cc, LSODE.cc,
Quad.cc, SparseCmplxCHOL.cc, SparseCmplxLU.cc, SparseCmplxQR.cc, SparseQR.cc,
SparsedbleCHOL.cc, SparsedbleLU.cc, base-lu.cc, bsxfun-defs.cc, dbleAEPBAL.cc,
dbleCHOL.cc, dbleGEPBAL.cc, dbleHESS.cc, dbleLU.cc, dbleQR.cc, dbleSCHUR.cc,
dbleSVD.cc, eigs-base.cc, fCmplxAEPBAL.cc, fCmplxCHOL.cc, fCmplxLU.cc,
fCmplxQR.cc, fCmplxSCHUR.cc, fEIG.cc, floatAEPBAL.cc, floatCHOL.cc,
floatGEPBAL.cc, floatHESS.cc, floatLU.cc, floatQR.cc, floatSCHUR.cc,
floatSVD.cc, lo-specfun.cc, oct-fftw.cc, oct-rand.cc, oct-spparms.cc,
sparse-base-chol.cc, sparse-dmsolve.cc, file-ops.cc, lo-sysdep.cc,
mach-info.cc, oct-env.cc, oct-syscalls.cc, cmd-edit.cc, cmd-hist.cc,
data-conv.cc, lo-ieee.cc, lo-regexp.cc, oct-base64.cc, oct-shlib.cc,
pathsearch.cc, singleton-cleanup.cc, sparse-util.cc, unwind-prot.cc:
Remove statements after call to handler that are no longer reachable.
Place input validation first and immediately call handler if necessary.
Change if/error_handler/else to if/error_handler and re-indent code.
author | Rik <rik@octave.org> |
---|---|
date | Sat, 23 Jan 2016 13:52:03 -0800 |
parents | e39e05d90788 |
children | 26f85aa072de |
line wrap: on
line source
// N-D Array manipulations. /* Copyright (C) 1996-2015 John W. Eaton Copyright (C) 2009 VZLU Prague, a.s. 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 <cfloat> #include <vector> #include "Array-util.h" #include "CNDArray.h" #include "f77-fcn.h" #include "functor.h" #include "lo-ieee.h" #include "lo-mappers.h" #include "mx-base.h" #include "mx-op-defs.h" #include "mx-cnda-s.h" #include "oct-fftw.h" #include "oct-locbuf.h" #include "bsxfun-defs.cc" ComplexNDArray::ComplexNDArray (const charNDArray& a) : MArray<Complex> (a.dims ()) { octave_idx_type n = a.numel (); for (octave_idx_type i = 0; i < n; i++) xelem (i) = static_cast<unsigned char> (a(i)); } #if defined (HAVE_FFTW) ComplexNDArray ComplexNDArray::fourier (int dim) const { dim_vector dv = dims (); if (dim > dv.length () || dim < 0) return ComplexNDArray (); octave_idx_type stride = 1; octave_idx_type n = dv(dim); for (int i = 0; i < dim; i++) stride *= dv(i); octave_idx_type howmany = numel () / dv(dim); howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); octave_idx_type nloop = (stride == 1 ? 1 : numel () / dv(dim) / stride); octave_idx_type dist = (stride == 1 ? n : 1); const Complex *in (fortran_vec ()); ComplexNDArray retval (dv); Complex *out (retval.fortran_vec ()); // Need to be careful here about the distance between fft's for (octave_idx_type k = 0; k < nloop; k++) octave_fftw::fft (in + k * stride * n, out + k * stride * n, n, howmany, stride, dist); return retval; } ComplexNDArray ComplexNDArray::ifourier (int dim) const { dim_vector dv = dims (); if (dim > dv.length () || dim < 0) return ComplexNDArray (); octave_idx_type stride = 1; octave_idx_type n = dv(dim); for (int i = 0; i < dim; i++) stride *= dv(i); octave_idx_type howmany = numel () / dv(dim); howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); octave_idx_type nloop = (stride == 1 ? 1 : numel () / dv(dim) / stride); octave_idx_type dist = (stride == 1 ? n : 1); const Complex *in (fortran_vec ()); ComplexNDArray retval (dv); Complex *out (retval.fortran_vec ()); // Need to be careful here about the distance between fft's for (octave_idx_type k = 0; k < nloop; k++) octave_fftw::ifft (in + k * stride * n, out + k * stride * n, n, howmany, stride, dist); return retval; } ComplexNDArray ComplexNDArray::fourier2d (void) const { dim_vector dv = dims (); if (dv.length () < 2) return ComplexNDArray (); dim_vector dv2 (dv(0), dv(1)); const Complex *in = fortran_vec (); ComplexNDArray retval (dv); Complex *out = retval.fortran_vec (); octave_idx_type howmany = numel () / dv(0) / dv(1); octave_idx_type dist = dv(0) * dv(1); for (octave_idx_type i=0; i < howmany; i++) octave_fftw::fftNd (in + i*dist, out + i*dist, 2, dv2); return retval; } ComplexNDArray ComplexNDArray::ifourier2d (void) const { dim_vector dv = dims (); if (dv.length () < 2) return ComplexNDArray (); dim_vector dv2 (dv(0), dv(1)); const Complex *in = fortran_vec (); ComplexNDArray retval (dv); Complex *out = retval.fortran_vec (); octave_idx_type howmany = numel () / dv(0) / dv(1); octave_idx_type dist = dv(0) * dv(1); for (octave_idx_type i=0; i < howmany; i++) octave_fftw::ifftNd (in + i*dist, out + i*dist, 2, dv2); return retval; } ComplexNDArray ComplexNDArray::fourierNd (void) const { dim_vector dv = dims (); int rank = dv.length (); const Complex *in (fortran_vec ()); ComplexNDArray retval (dv); Complex *out (retval.fortran_vec ()); octave_fftw::fftNd (in, out, rank, dv); return retval; } ComplexNDArray ComplexNDArray::ifourierNd (void) const { dim_vector dv = dims (); int rank = dv.length (); const Complex *in (fortran_vec ()); ComplexNDArray retval (dv); Complex *out (retval.fortran_vec ()); octave_fftw::ifftNd (in, out, rank, dv); return retval; } #else extern "C" { // Note that the original complex fft routines were not written for // double complex arguments. They have been modified by adding an // implicit double precision (a-h,o-z) statement at the beginning of // each subroutine. F77_RET_T F77_FUNC (zffti, ZFFTI) (const octave_idx_type&, Complex*); F77_RET_T F77_FUNC (zfftf, ZFFTF) (const octave_idx_type&, Complex*, Complex*); F77_RET_T F77_FUNC (zfftb, ZFFTB) (const octave_idx_type&, Complex*, Complex*); } ComplexNDArray ComplexNDArray::fourier (int dim) const { dim_vector dv = dims (); if (dim > dv.length () || dim < 0) return ComplexNDArray (); ComplexNDArray retval (dv); octave_idx_type npts = dv(dim); octave_idx_type nn = 4*npts+15; Array<Complex> wsave (dim_vector (nn, 1)); Complex *pwsave = wsave.fortran_vec (); OCTAVE_LOCAL_BUFFER (Complex, tmp, npts); octave_idx_type stride = 1; for (int i = 0; i < dim; i++) stride *= dv(i); octave_idx_type howmany = numel () / npts; howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); octave_idx_type dist = (stride == 1 ? npts : 1); F77_FUNC (zffti, ZFFTI) (npts, pwsave); for (octave_idx_type k = 0; k < nloop; k++) { for (octave_idx_type j = 0; j < howmany; j++) { octave_quit (); for (octave_idx_type i = 0; i < npts; i++) tmp[i] = elem ((i + k*npts)*stride + j*dist); F77_FUNC (zfftf, ZFFTF) (npts, tmp, pwsave); for (octave_idx_type i = 0; i < npts; i++) retval((i + k*npts)*stride + j*dist) = tmp[i]; } } return retval; } ComplexNDArray ComplexNDArray::ifourier (int dim) const { dim_vector dv = dims (); if (dim > dv.length () || dim < 0) return ComplexNDArray (); ComplexNDArray retval (dv); octave_idx_type npts = dv(dim); octave_idx_type nn = 4*npts+15; Array<Complex> wsave (dim_vector (nn, 1)); Complex *pwsave = wsave.fortran_vec (); OCTAVE_LOCAL_BUFFER (Complex, tmp, npts); octave_idx_type stride = 1; for (int i = 0; i < dim; i++) stride *= dv(i); octave_idx_type howmany = numel () / npts; howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); octave_idx_type dist = (stride == 1 ? npts : 1); F77_FUNC (zffti, ZFFTI) (npts, pwsave); for (octave_idx_type k = 0; k < nloop; k++) { for (octave_idx_type j = 0; j < howmany; j++) { octave_quit (); for (octave_idx_type i = 0; i < npts; i++) tmp[i] = elem ((i + k*npts)*stride + j*dist); F77_FUNC (zfftb, ZFFTB) (npts, tmp, pwsave); for (octave_idx_type i = 0; i < npts; i++) retval((i + k*npts)*stride + j*dist) = tmp[i] / static_cast<double> (npts); } } return retval; } ComplexNDArray ComplexNDArray::fourier2d (void) const { dim_vector dv = dims (); dim_vector dv2 (dv(0), dv(1)); int rank = 2; ComplexNDArray retval (*this); octave_idx_type stride = 1; for (int i = 0; i < rank; i++) { octave_idx_type npts = dv2(i); octave_idx_type nn = 4*npts+15; Array<Complex> wsave (dim_vector (nn, 1)); Complex *pwsave = wsave.fortran_vec (); Array<Complex> row (dim_vector (npts, 1)); Complex *prow = row.fortran_vec (); octave_idx_type howmany = numel () / npts; howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); octave_idx_type dist = (stride == 1 ? npts : 1); F77_FUNC (zffti, ZFFTI) (npts, pwsave); for (octave_idx_type k = 0; k < nloop; k++) { for (octave_idx_type j = 0; j < howmany; j++) { octave_quit (); for (octave_idx_type l = 0; l < npts; l++) prow[l] = retval((l + k*npts)*stride + j*dist); F77_FUNC (zfftf, ZFFTF) (npts, prow, pwsave); for (octave_idx_type l = 0; l < npts; l++) retval((l + k*npts)*stride + j*dist) = prow[l]; } } stride *= dv2(i); } return retval; } ComplexNDArray ComplexNDArray::ifourier2d (void) const { dim_vector dv = dims (); dim_vector dv2 (dv(0), dv(1)); int rank = 2; ComplexNDArray retval (*this); octave_idx_type stride = 1; for (int i = 0; i < rank; i++) { octave_idx_type npts = dv2(i); octave_idx_type nn = 4*npts+15; Array<Complex> wsave (dim_vector (nn, 1)); Complex *pwsave = wsave.fortran_vec (); Array<Complex> row (dim_vector (npts, 1)); Complex *prow = row.fortran_vec (); octave_idx_type howmany = numel () / npts; howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); octave_idx_type dist = (stride == 1 ? npts : 1); F77_FUNC (zffti, ZFFTI) (npts, pwsave); for (octave_idx_type k = 0; k < nloop; k++) { for (octave_idx_type j = 0; j < howmany; j++) { octave_quit (); for (octave_idx_type l = 0; l < npts; l++) prow[l] = retval((l + k*npts)*stride + j*dist); F77_FUNC (zfftb, ZFFTB) (npts, prow, pwsave); for (octave_idx_type l = 0; l < npts; l++) retval((l + k*npts)*stride + j*dist) = prow[l] / static_cast<double> (npts); } } stride *= dv2(i); } return retval; } ComplexNDArray ComplexNDArray::fourierNd (void) const { dim_vector dv = dims (); int rank = dv.length (); ComplexNDArray retval (*this); octave_idx_type stride = 1; for (int i = 0; i < rank; i++) { octave_idx_type npts = dv(i); octave_idx_type nn = 4*npts+15; Array<Complex> wsave (dim_vector (nn, 1)); Complex *pwsave = wsave.fortran_vec (); Array<Complex> row (dim_vector (npts, 1)); Complex *prow = row.fortran_vec (); octave_idx_type howmany = numel () / npts; howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); octave_idx_type dist = (stride == 1 ? npts : 1); F77_FUNC (zffti, ZFFTI) (npts, pwsave); for (octave_idx_type k = 0; k < nloop; k++) { for (octave_idx_type j = 0; j < howmany; j++) { octave_quit (); for (octave_idx_type l = 0; l < npts; l++) prow[l] = retval((l + k*npts)*stride + j*dist); F77_FUNC (zfftf, ZFFTF) (npts, prow, pwsave); for (octave_idx_type l = 0; l < npts; l++) retval((l + k*npts)*stride + j*dist) = prow[l]; } } stride *= dv(i); } return retval; } ComplexNDArray ComplexNDArray::ifourierNd (void) const { dim_vector dv = dims (); int rank = dv.length (); ComplexNDArray retval (*this); octave_idx_type stride = 1; for (int i = 0; i < rank; i++) { octave_idx_type npts = dv(i); octave_idx_type nn = 4*npts+15; Array<Complex> wsave (dim_vector (nn, 1)); Complex *pwsave = wsave.fortran_vec (); Array<Complex> row (dim_vector (npts, 1)); Complex *prow = row.fortran_vec (); octave_idx_type howmany = numel () / npts; howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); octave_idx_type dist = (stride == 1 ? npts : 1); F77_FUNC (zffti, ZFFTI) (npts, pwsave); for (octave_idx_type k = 0; k < nloop; k++) { for (octave_idx_type j = 0; j < howmany; j++) { octave_quit (); for (octave_idx_type l = 0; l < npts; l++) prow[l] = retval((l + k*npts)*stride + j*dist); F77_FUNC (zfftb, ZFFTB) (npts, prow, pwsave); for (octave_idx_type l = 0; l < npts; l++) retval((l + k*npts)*stride + j*dist) = prow[l] / static_cast<double> (npts); } } stride *= dv(i); } return retval; } #endif // unary operations boolNDArray ComplexNDArray::operator ! (void) const { if (any_element_is_nan ()) err_nan_to_logical_conversion (); return do_mx_unary_op<bool, Complex> (*this, mx_inline_not); } // FIXME: this is not quite the right thing. bool ComplexNDArray::any_element_is_nan (void) const { return do_mx_check<Complex> (*this, mx_inline_any_nan); } bool ComplexNDArray::any_element_is_inf_or_nan (void) const { return ! do_mx_check<Complex> (*this, mx_inline_all_finite); } // Return true if no elements have imaginary components. bool ComplexNDArray::all_elements_are_real (void) const { return do_mx_check<Complex> (*this, mx_inline_all_real); } // Return nonzero if any element of CM has a non-integer real or // imaginary part. Also extract the largest and smallest (real or // imaginary) values and return them in MAX_VAL and MIN_VAL. bool ComplexNDArray::all_integers (double& max_val, double& min_val) const { octave_idx_type nel = numel (); if (nel > 0) { Complex val = elem (0); double r_val = std::real (val); double i_val = std::imag (val); max_val = r_val; min_val = r_val; if (i_val > max_val) max_val = i_val; if (i_val < max_val) min_val = i_val; } else return false; for (octave_idx_type i = 0; i < nel; i++) { Complex val = elem (i); double r_val = std::real (val); double i_val = std::imag (val); if (r_val > max_val) max_val = r_val; if (i_val > max_val) max_val = i_val; if (r_val < min_val) min_val = r_val; if (i_val < min_val) min_val = i_val; if (D_NINT (r_val) != r_val || D_NINT (i_val) != i_val) return false; } return true; } bool ComplexNDArray::too_large_for_float (void) const { return test_any (xtoo_large_for_float); } boolNDArray ComplexNDArray::all (int dim) const { return do_mx_red_op<bool, Complex> (*this, dim, mx_inline_all); } boolNDArray ComplexNDArray::any (int dim) const { return do_mx_red_op<bool, Complex> (*this, dim, mx_inline_any); } ComplexNDArray ComplexNDArray::cumprod (int dim) const { return do_mx_cum_op<Complex, Complex> (*this, dim, mx_inline_cumprod); } ComplexNDArray ComplexNDArray::cumsum (int dim) const { return do_mx_cum_op<Complex, Complex> (*this, dim, mx_inline_cumsum); } ComplexNDArray ComplexNDArray::prod (int dim) const { return do_mx_red_op<Complex, Complex> (*this, dim, mx_inline_prod); } ComplexNDArray ComplexNDArray::sum (int dim) const { return do_mx_red_op<Complex, Complex> (*this, dim, mx_inline_sum); } ComplexNDArray ComplexNDArray::xsum (int dim) const { return do_mx_red_op<Complex, Complex> (*this, dim, mx_inline_xsum); } ComplexNDArray ComplexNDArray::sumsq (int dim) const { return do_mx_red_op<double, Complex> (*this, dim, mx_inline_sumsq); } ComplexNDArray ComplexNDArray::diff (octave_idx_type order, int dim) const { return do_mx_diff_op<Complex> (*this, dim, order, mx_inline_diff); } ComplexNDArray ComplexNDArray::concat (const ComplexNDArray& rb, const Array<octave_idx_type>& ra_idx) { if (rb.numel () > 0) insert (rb, ra_idx); return *this; } ComplexNDArray ComplexNDArray::concat (const NDArray& rb, const Array<octave_idx_type>& ra_idx) { ComplexNDArray tmp (rb); if (rb.numel () > 0) insert (tmp, ra_idx); return *this; } ComplexNDArray concat (NDArray& ra, ComplexNDArray& rb, const Array<octave_idx_type>& ra_idx) { ComplexNDArray retval (ra); if (rb.numel () > 0) retval.insert (rb, ra_idx); return retval; } static const Complex Complex_NaN_result (octave_NaN, octave_NaN); ComplexNDArray ComplexNDArray::max (int dim) const { return do_mx_minmax_op<Complex> (*this, dim, mx_inline_max); } ComplexNDArray ComplexNDArray::max (Array<octave_idx_type>& idx_arg, int dim) const { return do_mx_minmax_op<Complex> (*this, idx_arg, dim, mx_inline_max); } ComplexNDArray ComplexNDArray::min (int dim) const { return do_mx_minmax_op<Complex> (*this, dim, mx_inline_min); } ComplexNDArray ComplexNDArray::min (Array<octave_idx_type>& idx_arg, int dim) const { return do_mx_minmax_op<Complex> (*this, idx_arg, dim, mx_inline_min); } ComplexNDArray ComplexNDArray::cummax (int dim) const { return do_mx_cumminmax_op<Complex> (*this, dim, mx_inline_cummax); } ComplexNDArray ComplexNDArray::cummax (Array<octave_idx_type>& idx_arg, int dim) const { return do_mx_cumminmax_op<Complex> (*this, idx_arg, dim, mx_inline_cummax); } ComplexNDArray ComplexNDArray::cummin (int dim) const { return do_mx_cumminmax_op<Complex> (*this, dim, mx_inline_cummin); } ComplexNDArray ComplexNDArray::cummin (Array<octave_idx_type>& idx_arg, int dim) const { return do_mx_cumminmax_op<Complex> (*this, idx_arg, dim, mx_inline_cummin); } NDArray ComplexNDArray::abs (void) const { return do_mx_unary_map<double, Complex, std::abs> (*this); } boolNDArray ComplexNDArray::isnan (void) const { return do_mx_unary_map<bool, Complex, xisnan> (*this); } boolNDArray ComplexNDArray::isinf (void) const { return do_mx_unary_map<bool, Complex, xisinf> (*this); } boolNDArray ComplexNDArray::isfinite (void) const { return do_mx_unary_map<bool, Complex, xfinite> (*this); } ComplexNDArray conj (const ComplexNDArray& a) { return do_mx_unary_map<Complex, Complex, std::conj<double> > (a); } ComplexNDArray& ComplexNDArray::insert (const NDArray& a, octave_idx_type r, octave_idx_type c) { dim_vector a_dv = a.dims (); int n = a_dv.length (); if (n != dimensions.length ()) (*current_liboctave_error_handler) ("Array<T>::insert: invalid indexing operation"); Array<octave_idx_type> a_ra_idx (dim_vector (a_dv.length (), 1), 0); a_ra_idx.elem (0) = r; a_ra_idx.elem (1) = c; for (int i = 0; i < n; i++) { if (a_ra_idx(i) < 0 || (a_ra_idx(i) + a_dv(i)) > dimensions(i)) (*current_liboctave_error_handler) ("Array<T>::insert: range error for insert"); } a_ra_idx.elem (0) = 0; a_ra_idx.elem (1) = 0; octave_idx_type n_elt = a.numel (); // IS make_unique () NECESSARY HERE? for (octave_idx_type i = 0; i < n_elt; i++) { Array<octave_idx_type> ra_idx = a_ra_idx; ra_idx.elem (0) = a_ra_idx(0) + r; ra_idx.elem (1) = a_ra_idx(1) + c; elem (ra_idx) = a.elem (a_ra_idx); increment_index (a_ra_idx, a_dv); } return *this; } ComplexNDArray& ComplexNDArray::insert (const ComplexNDArray& a, octave_idx_type r, octave_idx_type c) { Array<Complex>::insert (a, r, c); return *this; } ComplexNDArray& ComplexNDArray::insert (const ComplexNDArray& a, const Array<octave_idx_type>& ra_idx) { Array<Complex>::insert (a, ra_idx); return *this; } void ComplexNDArray::increment_index (Array<octave_idx_type>& ra_idx, const dim_vector& dimensions, int start_dimension) { ::increment_index (ra_idx, dimensions, start_dimension); } octave_idx_type ComplexNDArray::compute_index (Array<octave_idx_type>& ra_idx, const dim_vector& dimensions) { return ::compute_index (ra_idx, dimensions); } ComplexNDArray ComplexNDArray::diag (octave_idx_type k) const { return MArray<Complex>::diag (k); } ComplexNDArray ComplexNDArray::diag (octave_idx_type m, octave_idx_type n) const { return MArray<Complex>::diag (m, n); } // This contains no information on the array structure !!! std::ostream& operator << (std::ostream& os, const ComplexNDArray& a) { octave_idx_type nel = a.numel (); for (octave_idx_type i = 0; i < nel; i++) { os << " "; octave_write_complex (os, a.elem (i)); os << "\n"; } return os; } std::istream& operator >> (std::istream& is, ComplexNDArray& a) { octave_idx_type nel = a.numel (); if (nel > 0) { Complex tmp; for (octave_idx_type i = 0; i < nel; i++) { tmp = octave_read_value<Complex> (is); if (is) a.elem (i) = tmp; else goto done; } } done: return is; } MINMAX_FCNS (ComplexNDArray, Complex) NDS_CMP_OPS (ComplexNDArray, Complex) NDS_BOOL_OPS (ComplexNDArray, Complex) SND_CMP_OPS (Complex, ComplexNDArray) SND_BOOL_OPS (Complex, ComplexNDArray) NDND_CMP_OPS (ComplexNDArray, ComplexNDArray) NDND_BOOL_OPS (ComplexNDArray, ComplexNDArray) ComplexNDArray& operator *= (ComplexNDArray& a, double s) { if (a.is_shared ()) a = a * s; else do_ms_inplace_op<Complex, double> (a, s, mx_inline_mul2); return a; } ComplexNDArray& operator /= (ComplexNDArray& a, double s) { if (a.is_shared ()) return a = a / s; else do_ms_inplace_op<Complex, double> (a, s, mx_inline_div2); return a; } BSXFUN_STDOP_DEFS_MXLOOP (ComplexNDArray) BSXFUN_STDREL_DEFS_MXLOOP (ComplexNDArray) BSXFUN_OP_DEF_MXLOOP (pow, ComplexNDArray, mx_inline_pow)