Mercurial > octave
view liboctave/numeric/oct-fftw.h @ 21202:f7121e111991
maint: indent #ifdef blocks in liboctave and src directories.
* Array-C.cc, Array-b.cc, Array-ch.cc, Array-d.cc, Array-f.cc, Array-fC.cc,
Array-i.cc, Array-idx-vec.cc, Array-s.cc, Array-str.cc, Array-util.cc,
Array-voidp.cc, Array.cc, CColVector.cc, CDiagMatrix.cc, CMatrix.cc,
CNDArray.cc, CRowVector.cc, CSparse.cc, CSparse.h, DiagArray2.cc, MArray-C.cc,
MArray-d.cc, MArray-f.cc, MArray-fC.cc, MArray-i.cc, MArray-s.cc, MArray.cc,
MDiagArray2.cc, MSparse-C.cc, MSparse-d.cc, MSparse.h, MatrixType.cc,
PermMatrix.cc, Range.cc, Sparse-C.cc, Sparse-b.cc, Sparse-d.cc, Sparse.cc,
boolMatrix.cc, boolNDArray.cc, boolSparse.cc, chMatrix.cc, chNDArray.cc,
dColVector.cc, dDiagMatrix.cc, dMatrix.cc, dNDArray.cc, dRowVector.cc,
dSparse.cc, dSparse.h, dim-vector.cc, fCColVector.cc, fCDiagMatrix.cc,
fCMatrix.cc, fCNDArray.cc, fCRowVector.cc, fColVector.cc, fDiagMatrix.cc,
fMatrix.cc, fNDArray.cc, fRowVector.cc, idx-vector.cc, int16NDArray.cc,
int32NDArray.cc, int64NDArray.cc, int8NDArray.cc, intNDArray.cc,
uint16NDArray.cc, uint32NDArray.cc, uint64NDArray.cc, uint8NDArray.cc,
blaswrap.c, cquit.c, f77-extern.cc, f77-fcn.c, f77-fcn.h, lo-error.c, quit.cc,
quit.h, CmplxAEPBAL.cc, CmplxCHOL.cc, CmplxGEPBAL.cc, CmplxHESS.cc, CmplxLU.cc,
CmplxQR.cc, CmplxQRP.cc, CmplxSCHUR.cc, CmplxSVD.cc, CollocWt.cc, DASPK.cc,
DASRT.cc, DASSL.cc, EIG.cc, LSODE.cc, ODES.cc, Quad.cc, base-lu.cc, base-qr.cc,
dbleAEPBAL.cc, dbleCHOL.cc, dbleGEPBAL.cc, dbleHESS.cc, dbleLU.cc, dbleQR.cc,
dbleQRP.cc, dbleSCHUR.cc, dbleSVD.cc, eigs-base.cc, fCmplxAEPBAL.cc,
fCmplxCHOL.cc, fCmplxGEPBAL.cc, fCmplxHESS.cc, fCmplxLU.cc, fCmplxQR.cc,
fCmplxQRP.cc, fCmplxSCHUR.cc, fCmplxSVD.cc, fEIG.cc, floatAEPBAL.cc,
floatCHOL.cc, floatGEPBAL.cc, floatHESS.cc, floatLU.cc, floatQR.cc,
floatQRP.cc, floatSCHUR.cc, floatSVD.cc, lo-mappers.cc, lo-specfun.cc,
oct-convn.cc, oct-fftw.cc, oct-fftw.h, oct-norm.cc, oct-rand.cc,
oct-spparms.cc, randgamma.c, randmtzig.c, randpoisson.c, sparse-chol.cc,
sparse-dmsolve.cc, sparse-lu.cc, sparse-qr.cc, mx-defs.h, dir-ops.cc,
file-ops.cc, file-stat.cc, lo-sysdep.cc, mach-info.cc, oct-env.cc,
oct-group.cc, oct-openmp.h, oct-passwd.cc, oct-syscalls.cc, oct-time.cc,
oct-uname.cc, pathlen.h, sysdir.h, syswait.h, cmd-edit.cc, cmd-hist.cc,
data-conv.cc, f2c-main.c, glob-match.cc, lo-array-errwarn.cc,
lo-array-gripes.cc, lo-cutils.c, lo-cutils.h, lo-ieee.cc, lo-math.h,
lo-regexp.cc, lo-utils.cc, oct-base64.cc, oct-glob.cc, oct-inttypes.cc,
oct-inttypes.h, oct-locbuf.cc, oct-mutex.cc, oct-refcount.h, oct-rl-edit.c,
oct-rl-hist.c, oct-shlib.cc, oct-sort.cc, pathsearch.cc, singleton-cleanup.cc,
sparse-sort.cc, sparse-util.cc, statdefs.h, str-vec.cc, unwind-prot.cc,
url-transfer.cc, display-available.h, main-cli.cc, main-gui.cc, main.in.cc,
mkoctfile.in.cc, octave-config.in.cc, shared-fcns.h:
indent #ifdef blocks in liboctave and src directories.
author | Rik <rik@octave.org> |
---|---|
date | Sat, 06 Feb 2016 06:40:13 -0800 |
parents | f7084eae3318 |
children | 1473547f50f5 |
line wrap: on
line source
/* Copyright (C) 2001-2015 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_oct_fftw_h) #define octave_oct_fftw_h 1 #include <cstddef> #if defined (HAVE_FFTW3_H) # include <fftw3.h> #endif #include "oct-cmplx.h" #include "dim-vector.h" #if defined (HAVE_FFTW) class OCTAVE_API octave_fftw_planner { protected: octave_fftw_planner (void); public: ~octave_fftw_planner (void); enum FftwMethod { UNKNOWN = -1, ESTIMATE, MEASURE, PATIENT, EXHAUSTIVE, HYBRID }; static bool instance_ok (void); static fftw_plan create_plan (int dir, const int rank, const dim_vector dims, octave_idx_type howmany, octave_idx_type stride, octave_idx_type dist, const Complex *in, Complex *out) { static fftw_plan dummy; return instance_ok () ? instance->do_create_plan (dir, rank, dims, howmany, stride, dist, in, out) : dummy; } static fftw_plan create_plan (const int rank, const dim_vector dims, octave_idx_type howmany, octave_idx_type stride, octave_idx_type dist, const double *in, Complex *out) { static fftw_plan dummy; return instance_ok () ? instance->do_create_plan (rank, dims, howmany, stride, dist, in, out) : dummy; } static FftwMethod method (void) { static FftwMethod dummy; return instance_ok () ? instance->do_method () : dummy; } static FftwMethod method (FftwMethod _meth) { static FftwMethod dummy; return instance_ok () ? instance->do_method (_meth) : dummy; } #if defined (HAVE_FFTW3F_THREADS) static void threads (int _nthreads) { if (instance_ok () && _nthreads != threads ()) { instance->nthreads = _nthreads; fftw_plan_with_nthreads (_nthreads); //Clear the current plans instance->rplan = instance->plan[0] = instance->plan[1] = 0; } } static int threads () { return instance_ok () ? instance->nthreads : 0; } #endif private: // No copying! octave_fftw_planner (const octave_fftw_planner&); octave_fftw_planner& operator = (const octave_fftw_planner&); static octave_fftw_planner *instance; static void cleanup_instance (void) { delete instance; instance = 0; } fftw_plan do_create_plan (int dir, const int rank, const dim_vector dims, octave_idx_type howmany, octave_idx_type stride, octave_idx_type dist, const Complex *in, Complex *out); fftw_plan do_create_plan (const int rank, const dim_vector dims, octave_idx_type howmany, octave_idx_type stride, octave_idx_type dist, const double *in, Complex *out); FftwMethod do_method (void); FftwMethod do_method (FftwMethod _meth); FftwMethod meth; // FIXME: perhaps this should be split into two classes? // Plan for fft and ifft of complex values fftw_plan plan[2]; // dist octave_idx_type d[2]; // stride octave_idx_type s[2]; // rank int r[2]; // howmany octave_idx_type h[2]; // dims dim_vector n[2]; bool simd_align[2]; bool inplace[2]; // Plan for fft of real values fftw_plan rplan; // dist octave_idx_type rd; // stride octave_idx_type rs; // rank int rr; // howmany octave_idx_type rh; // dims dim_vector rn; bool rsimd_align; #if defined (HAVE_FFTW3_THREADS) //number of threads when compiled with Multi-threading support int nthreads; #endif }; class OCTAVE_API octave_float_fftw_planner { protected: octave_float_fftw_planner (void); public: ~octave_float_fftw_planner (void); enum FftwMethod { UNKNOWN = -1, ESTIMATE, MEASURE, PATIENT, EXHAUSTIVE, HYBRID }; static bool instance_ok (void); static fftwf_plan create_plan (int dir, const int rank, const dim_vector dims, octave_idx_type howmany, octave_idx_type stride, octave_idx_type dist, const FloatComplex *in, FloatComplex *out) { static fftwf_plan dummy; return instance_ok () ? instance->do_create_plan (dir, rank, dims, howmany, stride, dist, in, out) : dummy; } static fftwf_plan create_plan (const int rank, const dim_vector dims, octave_idx_type howmany, octave_idx_type stride, octave_idx_type dist, const float *in, FloatComplex *out) { static fftwf_plan dummy; return instance_ok () ? instance->do_create_plan (rank, dims, howmany, stride, dist, in, out) : dummy; } static FftwMethod method (void) { static FftwMethod dummy; return instance_ok () ? instance->do_method () : dummy; } static FftwMethod method (FftwMethod _meth) { static FftwMethod dummy; return instance_ok () ? instance->do_method (_meth) : dummy; } #if defined (HAVE_FFTW3F_THREADS) static void threads (int _nthreads) { if (instance_ok () && _nthreads != threads ()) { instance->nthreads = _nthreads; fftwf_plan_with_nthreads (_nthreads); //Clear the current plans instance->rplan = instance->plan[0] = instance->plan[1] = 0; } } static int threads () { return instance_ok () ? instance->nthreads : 0; } #endif private: // No copying! octave_float_fftw_planner (const octave_float_fftw_planner&); octave_float_fftw_planner& operator = (const octave_float_fftw_planner&); static octave_float_fftw_planner *instance; static void cleanup_instance (void) { delete instance; instance = 0; } fftwf_plan do_create_plan (int dir, const int rank, const dim_vector dims, octave_idx_type howmany, octave_idx_type stride, octave_idx_type dist, const FloatComplex *in, FloatComplex *out); fftwf_plan do_create_plan (const int rank, const dim_vector dims, octave_idx_type howmany, octave_idx_type stride, octave_idx_type dist, const float *in, FloatComplex *out); FftwMethod do_method (void); FftwMethod do_method (FftwMethod _meth); FftwMethod meth; // FIXME: perhaps this should be split into two classes? // Plan for fft and ifft of complex values fftwf_plan plan[2]; // dist octave_idx_type d[2]; // stride octave_idx_type s[2]; // rank int r[2]; // howmany octave_idx_type h[2]; // dims dim_vector n[2]; bool simd_align[2]; bool inplace[2]; // Plan for fft of real values fftwf_plan rplan; // dist octave_idx_type rd; // stride octave_idx_type rs; // rank int rr; // howmany octave_idx_type rh; // dims dim_vector rn; bool rsimd_align; #if defined (HAVE_FFTW3F_THREADS) //number of threads when compiled with Multi-threading support int nthreads; #endif }; class OCTAVE_API octave_fftw { public: static int fft (const double *in, Complex *out, size_t npts, size_t nsamples = 1, octave_idx_type stride = 1, octave_idx_type dist = -1); static int fft (const Complex *in, Complex *out, size_t npts, size_t nsamples = 1, octave_idx_type stride = 1, octave_idx_type dist = -1); static int ifft (const Complex *in, Complex *out, size_t npts, size_t nsamples = 1, octave_idx_type stride = 1, octave_idx_type dist = -1); static int fftNd (const double*, Complex*, const int, const dim_vector &); static int fftNd (const Complex*, Complex*, const int, const dim_vector &); static int ifftNd (const Complex*, Complex*, const int, const dim_vector &); static int fft (const float *in, FloatComplex *out, size_t npts, size_t nsamples = 1, octave_idx_type stride = 1, octave_idx_type dist = -1); static int fft (const FloatComplex *in, FloatComplex *out, size_t npts, size_t nsamples = 1, octave_idx_type stride = 1, octave_idx_type dist = -1); static int ifft (const FloatComplex *in, FloatComplex *out, size_t npts, size_t nsamples = 1, octave_idx_type stride = 1, octave_idx_type dist = -1); static int fftNd (const float*, FloatComplex*, const int, const dim_vector &); static int fftNd (const FloatComplex*, FloatComplex*, const int, const dim_vector &); static int ifftNd (const FloatComplex*, FloatComplex*, const int, const dim_vector &); private: octave_fftw (void); octave_fftw (const octave_fftw&); octave_fftw& operator = (const octave_fftw&); }; #endif #endif