Mercurial > hg > octave-lyh
diff liboctave/fCNDArray.h @ 7789:82be108cc558
First attempt at single precision tyeps
* * *
corrections to qrupdate single precision routines
* * *
prefer demotion to single over promotion to double
* * *
Add single precision support to log2 function
* * *
Trivial PROJECT file update
* * *
Cache optimized hermitian/transpose methods
* * *
Add tests for tranpose/hermitian and ChangeLog entry for new transpose code
author | David Bateman <dbateman@free.fr> |
---|---|
date | Sun, 27 Apr 2008 22:34:17 +0200 |
parents | |
children | 935be827eaf8 |
line wrap: on
line diff
new file mode 100644 --- /dev/null +++ b/liboctave/fCNDArray.h @@ -0,0 +1,164 @@ +/* + +Copyright (C) 2003, 2004, 2005, 2006, 2007 John W. Eaton + +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_FloatComplexNDArray_h) +#define octave_FloatComplexNDArray_h 1 + +#include "MArrayN.h" +#include "fCMatrix.h" + +#include "mx-defs.h" +#include "mx-op-defs.h" + +class +OCTAVE_API +FloatComplexNDArray : public MArrayN<FloatComplex> +{ +public: + + FloatComplexNDArray (void) : MArrayN<FloatComplex> () { } + + FloatComplexNDArray (const dim_vector& dv) : MArrayN<FloatComplex> (dv) { } + + FloatComplexNDArray (const dim_vector& dv, const FloatComplex& val) + : MArrayN<FloatComplex> (dv, val) { } + + FloatComplexNDArray (const FloatComplexNDArray& a) : MArrayN<FloatComplex> (a) { } + + FloatComplexNDArray (const FloatComplexMatrix& a) : MArrayN<FloatComplex> (a) { } + + template <class U> + FloatComplexNDArray (const MArrayN<U>& a) : MArrayN<FloatComplex> (a) { } + + template <class U> + FloatComplexNDArray (const ArrayN<U>& a) : MArrayN<FloatComplex> (a) { } + + FloatComplexNDArray& operator = (const FloatComplexNDArray& a) + { + MArrayN<FloatComplex>::operator = (a); + return *this; + } + + // unary operations + + boolNDArray operator ! (void) const; + + // FIXME -- this is not quite the right thing. + + bool any_element_is_inf_or_nan (void) const; + bool all_elements_are_real (void) const; + bool all_integers (float& max_val, float& min_val) const; + bool too_large_for_float (void) const; + + boolNDArray all (int dim = -1) const; + boolNDArray any (int dim = -1) const; + + FloatComplexNDArray cumprod (int dim = -1) const; + FloatComplexNDArray cumsum (int dim = -1) const; + FloatComplexNDArray prod (int dim = -1) const; + FloatComplexNDArray sum (int dim = -1) const; + FloatComplexNDArray sumsq (int dim = -1) const; + FloatComplexNDArray concat (const FloatComplexNDArray& rb, const Array<octave_idx_type>& ra_idx); + FloatComplexNDArray concat (const FloatNDArray& rb, const Array<octave_idx_type>& ra_idx); + + FloatComplexNDArray max (int dim = 0) const; + FloatComplexNDArray max (ArrayN<octave_idx_type>& index, int dim = 0) const; + FloatComplexNDArray min (int dim = 0) const; + FloatComplexNDArray min (ArrayN<octave_idx_type>& index, int dim = 0) const; + FloatComplexNDArray& insert (const NDArray& a, octave_idx_type r, octave_idx_type c); + FloatComplexNDArray& insert (const FloatComplexNDArray& a, octave_idx_type r, octave_idx_type c); + FloatComplexNDArray& insert (const FloatComplexNDArray& a, const Array<octave_idx_type>& ra_idx); + + FloatNDArray abs (void) const; + + FloatComplexNDArray fourier (int dim = 1) const; + FloatComplexNDArray ifourier (int dim = 1) const; + + FloatComplexNDArray fourier2d (void) const; + FloatComplexNDArray ifourier2d (void) const; + + FloatComplexNDArray fourierNd (void) const; + FloatComplexNDArray ifourierNd (void) const; + + FloatComplexMatrix matrix_value (void) const; + + FloatComplexNDArray squeeze (void) const { return MArrayN<FloatComplex>::squeeze (); } + + static void increment_index (Array<octave_idx_type>& ra_idx, + const dim_vector& dimensions, + int start_dimension = 0); + + static octave_idx_type compute_index (Array<octave_idx_type>& ra_idx, + const dim_vector& dimensions); + + // i/o + + friend OCTAVE_API std::ostream& operator << (std::ostream& os, const FloatComplexNDArray& a); + friend OCTAVE_API std::istream& operator >> (std::istream& is, FloatComplexNDArray& a); + + static FloatComplex resize_fill_value (void) { return FloatComplex (0.0, 0.0); } + + // bool all_elements_are_real (void) const; + // bool all_integers (float& max_val, float& min_val) const; + + FloatComplexNDArray diag (octave_idx_type k = 0) const; + + typedef float (*dmapper) (const FloatComplex&); + typedef FloatComplex (*cmapper) (const FloatComplex&); + typedef bool (*bmapper) (const FloatComplex&); + + FloatNDArray map (dmapper fcn) const; + FloatComplexNDArray map (cmapper fcn) const; + boolNDArray map (bmapper fcn) const; + +private: + + FloatComplexNDArray (FloatComplex *d, const dim_vector& dv) + : MArrayN<FloatComplex> (d, dv) { } +}; + +extern OCTAVE_API FloatComplexNDArray min (const FloatComplex& c, const FloatComplexNDArray& m); +extern OCTAVE_API FloatComplexNDArray min (const FloatComplexNDArray& m, const FloatComplex& c); +extern OCTAVE_API FloatComplexNDArray min (const FloatComplexNDArray& a, const FloatComplexNDArray& b); + +extern OCTAVE_API FloatComplexNDArray max (const FloatComplex& c, const FloatComplexNDArray& m); +extern OCTAVE_API FloatComplexNDArray max (const FloatComplexNDArray& m, const FloatComplex& c); +extern OCTAVE_API FloatComplexNDArray max (const FloatComplexNDArray& a, const FloatComplexNDArray& b); + +NDS_CMP_OP_DECLS (FloatComplexNDArray, FloatComplex, OCTAVE_API) +NDS_BOOL_OP_DECLS (FloatComplexNDArray, FloatComplex, OCTAVE_API) + +SND_CMP_OP_DECLS (FloatComplex, FloatComplexNDArray, OCTAVE_API) +SND_BOOL_OP_DECLS (FloatComplex, FloatComplexNDArray, OCTAVE_API) + +NDND_CMP_OP_DECLS (FloatComplexNDArray, FloatComplexNDArray, OCTAVE_API) +NDND_BOOL_OP_DECLS (FloatComplexNDArray, FloatComplexNDArray, OCTAVE_API) + +MARRAY_FORWARD_DEFS (MArrayN, FloatComplexNDArray, FloatComplex) + +#endif + +/* +;;; Local Variables: *** +;;; mode: C++ *** +;;; End: *** +*/