octave: liboctave/numeric/oct-fftw.cc annotate

annotate liboctave/numeric/oct-fftw.cc @ 27919:1891570abac8

update Octave Project Developers copyright for the new year In files that have the "Octave Project Developers" copyright notice, update for 2020.

author	John W. Eaton <jwe@octave.org>
date	Mon, 06 Jan 2020 22:29:51 -0500
parents	b442ec6dda5c
children	bd51beb6205e

rev	line source
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	1 /*
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	2
27919 1891570abac8 update Octave Project Developers copyright for the new year John W. Eaton <jwe@octave.org> parents: 27918 diff changeset	3 Copyright (C) 2001-2020 The Octave Project Developers
27918 b442ec6dda5c use centralized file for copyright info for individual contributors John W. Eaton <jwe@octave.org> parents: 27277 diff changeset	4
b442ec6dda5c use centralized file for copyright info for individual contributors John W. Eaton <jwe@octave.org> parents: 27277 diff changeset	5 See the file COPYRIGHT.md in the top-level directory of this distribution
b442ec6dda5c use centralized file for copyright info for individual contributors John W. Eaton <jwe@octave.org> parents: 27277 diff changeset	6 or <https://octave.org/COPYRIGHT.html/>.
b442ec6dda5c use centralized file for copyright info for individual contributors John W. Eaton <jwe@octave.org> parents: 27277 diff changeset	7
7017 a1dbe9d80eee [project @ 2007-10-12 21:27:11 by jwe] jwe parents: 7016 diff changeset	8
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	9 This file is part of Octave.
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	10
24534 194eb4bd202b maint: Update punctuation for GPL v3 license text. Rik <rik@octave.org> parents: 23795 diff changeset	11 Octave is free software: you can redistribute it and/or modify it
22755 3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	12 under the terms of the GNU General Public License as published by
24534 194eb4bd202b maint: Update punctuation for GPL v3 license text. Rik <rik@octave.org> parents: 23795 diff changeset	13 the Free Software Foundation, either version 3 of the License, or
22755 3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	14 (at your option) any later version.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	15
22755 3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	16 Octave is distributed in the hope that it will be useful, but
3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	17 WITHOUT ANY WARRANTY; without even the implied warranty of
3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	19 GNU General Public License for more details.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	20
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	21 You should have received a copy of the GNU General Public License
7016 93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 6482 diff changeset	22 along with Octave; see the file COPYING. If not, see
24534 194eb4bd202b maint: Update punctuation for GPL v3 license text. Rik <rik@octave.org> parents: 23795 diff changeset	23 <https://www.gnu.org/licenses/>.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	24
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	25 */
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	26
21724 aba2e6293dd8 use "#if ..." consistently instead of "#ifdef" and "#ifndef" John W. Eaton <jwe@octave.org> parents: 21688 diff changeset	27 #if defined (HAVE_CONFIG_H)
21301 40de9f8f23a6 Use '#include "config.h"' rather than <config.h>. Rik <rik@octave.org> parents: 21202 diff changeset	28 # include "config.h"
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	29 #endif
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	30
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	31 #if defined (HAVE_FFTW3_H)
763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	32 # include <fftw3.h>
763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	33 #endif
763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	34
4786 fc316bde0053 [project @ 2004-02-18 12:52:20 by jwe] jwe parents: 4785 diff changeset	35 #include "lo-error.h"
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	36 #include "oct-fftw.h"
23475 d691ed308237 maint: Clean up #includes in liboctave/numeric directory. Rik <rik@octave.org> parents: 23457 diff changeset	37 #include "oct-locbuf.h"
4786 fc316bde0053 [project @ 2004-02-18 12:52:20 by jwe] jwe parents: 4785 diff changeset	38 #include "quit.h"
13983 7dd7cccf0757 clean up memory allocated for singletons before exit John W. Eaton <jwe@octave.org> parents: 11586 diff changeset	39 #include "singleton-cleanup.h"
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	40
21520 e1c921322ba2 Fix configuration conditions in oct-fftw.cc Mike Miller <mtmiller@octave.org> parents: 21451 diff changeset	41 #if defined (HAVE_FFTW3_THREADS) \|\| defined (HAVE_FFTW3F_THREADS)
21893 0a6f387d18d3 * nproc.cc, oct-fftw.cc: Use octave_num_processors_wrapper. John W. Eaton <jwe@octave.org> parents: 21751 diff changeset	42 # include "nproc-wrapper.h"
15960 bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	43 #endif
bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	44
23062 9ca0829b160e fix editing error in previous change to oct-fftw.cc (bug #50070) John W. Eaton <jwe@octave.org> parents: 23058 diff changeset	45 namespace octave
9ca0829b160e fix editing error in previous change to oct-fftw.cc (bug #50070) John W. Eaton <jwe@octave.org> parents: 23058 diff changeset	46 {
21688 68ab693d16e4 * oct-fftw.cc: Allow compilation without FFTW. John W. Eaton <jwe@octave.org> parents: 21660 diff changeset	47 #if defined (HAVE_FFTW)
68ab693d16e4 * oct-fftw.cc: Allow compilation without FFTW. John W. Eaton <jwe@octave.org> parents: 21660 diff changeset	48
23457 21baad6b35c4 maint: Use C++11 nullptr rather than 0 or NULL when possible. Rik <rik@octave.org> parents: 23440 diff changeset	49 fftw_planner *fftw_planner::instance = nullptr;
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	50
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	51 // Helper class to create and cache FFTW plans for both 1D and
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	52 // 2D. This implementation defaults to using FFTW_ESTIMATE to create
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	53 // the plans, which in theory is suboptimal, but provides quite
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	54 // reasonable performance in practice.
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	55
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	56 // Also note that if FFTW_ESTIMATE is not used then the planner in FFTW3
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	57 // will destroy the input and output arrays. We must, therefore, create a
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	58 // temporary input array with the same size and 16-byte alignment as
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	59 // the original array when using a different planner strategy.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	60 // Note that we also use any wisdom that is available, either in a
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	61 // FFTW3 system wide file or as supplied by the user.
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	62
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	63 // FIXME: if we can ensure 16 byte alignment in Array<T>
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	64 // (<T> *data) the FFTW3 can use SIMD instructions for further
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	65 // acceleration.
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	66
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	67 // Note that it is profitable to store the FFTW3 plans, for small FFTs.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	68
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	69 fftw_planner::fftw_planner (void)
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	70 : meth (ESTIMATE), rplan (nullptr), rd (0), rs (0), rr (0), rh (0), rn (),
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	71 rsimd_align (false), nthreads (1)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	72 {
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	73 plan[0] = plan[1] = nullptr;
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	74 d[0] = d[1] = s[0] = s[1] = r[0] = r[1] = h[0] = h[1] = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	75 simd_align[0] = simd_align[1] = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	76 inplace[0] = inplace[1] = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	77 n[0] = n[1] = dim_vector ();
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	78
15960 bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	79 #if defined (HAVE_FFTW3_THREADS)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	80 int init_ret = fftw_init_threads ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	81 if (! init_ret)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	82 (*current_liboctave_error_handler) ("Error initializing FFTW threads");
21136 7cac4e7458f2 maint: clean up code around calls to current_liboctave_error_handler. Rik <rik@octave.org> parents: 20955 diff changeset	83
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	84 // Use number of processors available to the current process
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	85 // This can be later changed with fftw ("threads", nthreads).
27032 0ee0bb5eb381 make nproc default to "overridable" to agree with nproc shell utility Mike Miller <mtmiller@octave.org> parents: 26377 diff changeset	86 nthreads = octave_num_processors_wrapper (OCTAVE_NPROC_CURRENT_OVERRIDABLE);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	87 fftw_plan_with_nthreads (nthreads);
15960 bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	88 #endif
bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	89
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	90 // If we have a system wide wisdom file, import it.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	91 fftw_import_system_wisdom ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	92 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	93
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	94 fftw_planner::~fftw_planner (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	95 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	96 fftw_plan *plan_p;
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	97
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	98 plan_p = reinterpret_cast<fftw_plan *> (&rplan);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	99 if (*plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	100 fftw_destroy_plan (*plan_p);
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	101
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	102 plan_p = reinterpret_cast<fftw_plan *> (&plan[0]);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	103 if (*plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	104 fftw_destroy_plan (*plan_p);
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	105
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	106 plan_p = reinterpret_cast<fftw_plan *> (&plan[1]);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	107 if (*plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	108 fftw_destroy_plan (*plan_p);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	109 }
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	110
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	111 bool
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	112 fftw_planner::instance_ok (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	113 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	114 bool retval = true;
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	115
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	116 if (! instance)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	117 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	118 instance = new fftw_planner ();
26369 0249ba4c9589 Fix static analyzer detected V668 issues (bug #55347). Andreas Weber <octave@josoansi.de> parents: 26148 diff changeset	119 singleton_cleanup_list::add (cleanup_instance);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	120 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	121
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	122 return retval;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	123 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	124
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	125 void
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	126 fftw_planner::threads (int nt)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	127 {
21520 e1c921322ba2 Fix configuration conditions in oct-fftw.cc Mike Miller <mtmiller@octave.org> parents: 21451 diff changeset	128 #if defined (HAVE_FFTW3_THREADS)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	129 if (instance_ok () && nt != threads ())
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	130 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	131 instance->nthreads = nt;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	132 fftw_plan_with_nthreads (nt);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	133 // Clear the current plans.
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	134 instance->rplan = instance->plan[0] = instance->plan[1] = nullptr;
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	135 }
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	136 #else
26148 8c72f0345ee8 maint: Silence compiler warnings about unused or too many parameters. Kai T. Ohlhus <k.ohlhus@gmail.com> parents: 25688 diff changeset	137 octave_unused_parameter (nt);
8c72f0345ee8 maint: Silence compiler warnings about unused or too many parameters. Kai T. Ohlhus <k.ohlhus@gmail.com> parents: 25688 diff changeset	138
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	139 (*current_liboctave_warning_handler)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	140 ("unable to change number of threads without FFTW thread support");
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	141 #endif
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	142 }
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	143
763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	144 #define CHECK_SIMD_ALIGNMENT(x) \
6482 7e958a1532c6 [project @ 2007-04-04 02:37:07 by jwe] jwe parents: 6228 diff changeset	145 (((reinterpret_cast<ptrdiff_t> (x)) & 0xF) == 0)
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	146
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	147 void *
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	148 fftw_planner::do_create_plan (int dir, const int rank,
23440 f15f0d8b78c8 pass dim_vector as const reference, not by value John W. Eaton <jwe@octave.org> parents: 23433 diff changeset	149 const dim_vector& dims,
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	150 octave_idx_type howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	151 octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	152 octave_idx_type dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	153 const Complex in, Complex out)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	154 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	155 int which = (dir == FFTW_FORWARD) ? 0 : 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	156 fftw_plan cur_plan_p = reinterpret_cast<fftw_plan > (&plan[which]);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	157 bool create_new_plan = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	158 bool ioalign = CHECK_SIMD_ALIGNMENT (in) && CHECK_SIMD_ALIGNMENT (out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	159 bool ioinplace = (in == out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	160
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	161 // Don't create a new plan if we have a non SIMD plan already but
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	162 // can do SIMD. This prevents endlessly recreating plans if we
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	163 // change the alignment.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	164
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	165 if (plan[which] == nullptr \|\| d[which] != dist \|\| s[which] != stride
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	166 \|\| r[which] != rank \|\| h[which] != howmany
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	167 \|\| ioinplace != inplace[which]
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	168 \|\| ((ioalign != simd_align[which]) ? ! ioalign : false))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	169 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	170 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	171 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	172 // We still might not have the same shape of array.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	173
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	174 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	175 if (dims(i) != n[which](i))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	176 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	177 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	178 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	179 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	180 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	181
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	182 if (create_new_plan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	183 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	184 d[which] = dist;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	185 s[which] = stride;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	186 r[which] = rank;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	187 h[which] = howmany;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	188 simd_align[which] = ioalign;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	189 inplace[which] = ioinplace;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	190 n[which] = dims;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	191
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	192 // Note reversal of dimensions for column major storage in FFTW.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	193 octave_idx_type nn = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	194 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	195
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	196 for (int i = 0, j = rank-1; i < rank; i++, j--)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	197 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	198 tmp[i] = dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	199 nn *= dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	200 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	201
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	202 int plan_flags = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	203 bool plan_destroys_in = true;
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	204
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	205 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	206 {
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	207 case UNKNOWN:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	208 case ESTIMATE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	209 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	210 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	211 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	212 case MEASURE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	213 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	214 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	215 case PATIENT:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	216 plan_flags \|= FFTW_PATIENT;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	217 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	218 case EXHAUSTIVE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	219 plan_flags \|= FFTW_EXHAUSTIVE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	220 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	221 case HYBRID:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	222 if (nn < 8193)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	223 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	224 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	225 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	226 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	227 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	228 }
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	229 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	230 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	231
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	232 if (ioalign)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	233 plan_flags &= ~FFTW_UNALIGNED;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	234 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	235 plan_flags \|= FFTW_UNALIGNED;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	236
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	237 if (*cur_plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	238 fftw_destroy_plan (*cur_plan_p);
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	239
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	240 if (plan_destroys_in)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	241 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	242 // Create matrix with the same size and 16-byte alignment as input
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	243 OCTAVE_LOCAL_BUFFER (Complex, itmp, nn * howmany + 32);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	244 itmp = reinterpret_cast<Complex *>
25688 b2917b7858ba maint: Use Octave convention for spacing of C++ cast statements. Rik <rik@octave.org> parents: 25103 diff changeset	245 (((reinterpret_cast<ptrdiff_t> (itmp) + 15) & ~ 0xF) +
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	246 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	247
27277 db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	248 *cur_plan_p
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	249 = fftw_plan_many_dft (rank, tmp, howmany,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	250 reinterpret_cast<fftw_complex *> (itmp),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	251 nullptr, stride, dist,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	252 reinterpret_cast<fftw_complex *> (out),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	253 nullptr, stride, dist, dir, plan_flags);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	254 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	255 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	256 {
27277 db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	257 *cur_plan_p
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	258 = fftw_plan_many_dft (rank, tmp, howmany,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	259 reinterpret_cast<fftw_complex > (const_cast<Complex > (in)),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	260 nullptr, stride, dist,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	261 reinterpret_cast<fftw_complex *> (out),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	262 nullptr, stride, dist, dir, plan_flags);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	263 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	264
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	265 if (*cur_plan_p == nullptr)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	266 (*current_liboctave_error_handler) ("Error creating fftw plan");
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	267 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	268
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	269 return *cur_plan_p;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	270 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	271
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	272 void *
23440 f15f0d8b78c8 pass dim_vector as const reference, not by value John W. Eaton <jwe@octave.org> parents: 23433 diff changeset	273 fftw_planner::do_create_plan (const int rank, const dim_vector& dims,
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	274 octave_idx_type howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	275 octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	276 octave_idx_type dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	277 const double in, Complex out)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	278 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	279 fftw_plan cur_plan_p = reinterpret_cast<fftw_plan > (&rplan);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	280 bool create_new_plan = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	281 bool ioalign = CHECK_SIMD_ALIGNMENT (in) && CHECK_SIMD_ALIGNMENT (out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	282
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	283 // Don't create a new plan if we have a non SIMD plan already but
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	284 // can do SIMD. This prevents endlessly recreating plans if we
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	285 // change the alignment.
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	286
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	287 if (rplan == nullptr \|\| rd != dist \|\| rs != stride \|\| rr != rank
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	288 \|\| rh != howmany \|\| ((ioalign != rsimd_align) ? ! ioalign : false))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	289 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	290 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	291 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	292 // We still might not have the same shape of array.
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	293
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	294 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	295 if (dims(i) != rn(i))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	296 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	297 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	298 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	299 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	300 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	301
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	302 if (create_new_plan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	303 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	304 rd = dist;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	305 rs = stride;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	306 rr = rank;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	307 rh = howmany;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	308 rsimd_align = ioalign;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	309 rn = dims;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	310
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	311 // Note reversal of dimensions for column major storage in FFTW.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	312 octave_idx_type nn = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	313 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	314
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	315 for (int i = 0, j = rank-1; i < rank; i++, j--)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	316 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	317 tmp[i] = dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	318 nn *= dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	319 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	320
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	321 int plan_flags = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	322 bool plan_destroys_in = true;
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	323
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	324 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	325 {
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	326 case UNKNOWN:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	327 case ESTIMATE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	328 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	329 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	330 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	331 case MEASURE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	332 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	333 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	334 case PATIENT:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	335 plan_flags \|= FFTW_PATIENT;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	336 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	337 case EXHAUSTIVE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	338 plan_flags \|= FFTW_EXHAUSTIVE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	339 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	340 case HYBRID:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	341 if (nn < 8193)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	342 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	343 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	344 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	345 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	346 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	347 }
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	348 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	349 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	350
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	351 if (ioalign)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	352 plan_flags &= ~FFTW_UNALIGNED;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	353 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	354 plan_flags \|= FFTW_UNALIGNED;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	355
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	356 if (*cur_plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	357 fftw_destroy_plan (*cur_plan_p);
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	358
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	359 if (plan_destroys_in)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	360 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	361 // Create matrix with the same size and 16-byte alignment as input
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	362 OCTAVE_LOCAL_BUFFER (double, itmp, nn + 32);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	363 itmp = reinterpret_cast<double *>
25688 b2917b7858ba maint: Use Octave convention for spacing of C++ cast statements. Rik <rik@octave.org> parents: 25103 diff changeset	364 (((reinterpret_cast<ptrdiff_t> (itmp) + 15) & ~ 0xF) +
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	365 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	366
27277 db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	367 *cur_plan_p
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	368 = fftw_plan_many_dft_r2c (rank, tmp, howmany, itmp,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	369 nullptr, stride, dist,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	370 reinterpret_cast<fftw_complex *> (out),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	371 nullptr, stride, dist, plan_flags);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	372 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	373 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	374 {
27277 db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	375 *cur_plan_p
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	376 = fftw_plan_many_dft_r2c (rank, tmp, howmany,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	377 (const_cast<double *> (in)),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	378 nullptr, stride, dist,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	379 reinterpret_cast<fftw_complex *> (out),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	380 nullptr, stride, dist, plan_flags);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	381 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	382
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	383 if (*cur_plan_p == nullptr)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	384 (*current_liboctave_error_handler) ("Error creating fftw plan");
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	385 }
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	386
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	387 return *cur_plan_p;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	388 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	389
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	390 fftw_planner::FftwMethod
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	391 fftw_planner::do_method (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	392 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	393 return meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	394 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	395
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	396 fftw_planner::FftwMethod
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	397 fftw_planner::do_method (FftwMethod _meth)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	398 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	399 FftwMethod ret = meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	400 if (_meth == ESTIMATE \|\| _meth == MEASURE
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	401 \|\| _meth == PATIENT \|\| _meth == EXHAUSTIVE
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	402 \|\| _meth == HYBRID)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	403 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	404 if (meth != _meth)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	405 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	406 meth = _meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	407 if (rplan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	408 fftw_destroy_plan (reinterpret_cast<fftw_plan> (rplan));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	409 if (plan[0])
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	410 fftw_destroy_plan (reinterpret_cast<fftw_plan> (plan[0]));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	411 if (plan[1])
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	412 fftw_destroy_plan (reinterpret_cast<fftw_plan> (plan[1]));
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	413 rplan = plan[0] = plan[1] = nullptr;
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	414 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	415 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	416 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	417 ret = UNKNOWN;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	418 return ret;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	419 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	420
23457 21baad6b35c4 maint: Use C++11 nullptr rather than 0 or NULL when possible. Rik <rik@octave.org> parents: 23440 diff changeset	421 float_fftw_planner *float_fftw_planner::instance = nullptr;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	422
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	423 float_fftw_planner::float_fftw_planner (void)
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	424 : meth (ESTIMATE), rplan (nullptr), rd (0), rs (0), rr (0), rh (0), rn (),
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	425 rsimd_align (false), nthreads (1)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	426 {
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	427 plan[0] = plan[1] = nullptr;
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	428 d[0] = d[1] = s[0] = s[1] = r[0] = r[1] = h[0] = h[1] = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	429 simd_align[0] = simd_align[1] = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	430 inplace[0] = inplace[1] = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	431 n[0] = n[1] = dim_vector ();
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	432
15960 bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	433 #if defined (HAVE_FFTW3F_THREADS)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	434 int init_ret = fftwf_init_threads ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	435 if (! init_ret)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	436 (*current_liboctave_error_handler) ("Error initializing FFTW3F threads");
21136 7cac4e7458f2 maint: clean up code around calls to current_liboctave_error_handler. Rik <rik@octave.org> parents: 20955 diff changeset	437
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	438 // Use number of processors available to the current process
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	439 // This can be later changed with fftw ("threads", nthreads).
27032 0ee0bb5eb381 make nproc default to "overridable" to agree with nproc shell utility Mike Miller <mtmiller@octave.org> parents: 26377 diff changeset	440 nthreads = octave_num_processors_wrapper (OCTAVE_NPROC_CURRENT_OVERRIDABLE);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	441 fftwf_plan_with_nthreads (nthreads);
15960 bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	442 #endif
bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	443
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	444 // If we have a system wide wisdom file, import it.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	445 fftwf_import_system_wisdom ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	446 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	447
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	448 float_fftw_planner::~float_fftw_planner (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	449 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	450 fftwf_plan *plan_p;
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	451
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	452 plan_p = reinterpret_cast<fftwf_plan *> (&rplan);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	453 if (*plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	454 fftwf_destroy_plan (*plan_p);
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	455
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	456 plan_p = reinterpret_cast<fftwf_plan *> (&plan[0]);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	457 if (*plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	458 fftwf_destroy_plan (*plan_p);
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	459
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	460 plan_p = reinterpret_cast<fftwf_plan *> (&plan[1]);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	461 if (*plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	462 fftwf_destroy_plan (*plan_p);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	463 }
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	464
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	465 bool
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	466 float_fftw_planner::instance_ok (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	467 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	468 bool retval = true;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	469
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	470 if (! instance)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	471 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	472 instance = new float_fftw_planner ();
26369 0249ba4c9589 Fix static analyzer detected V668 issues (bug #55347). Andreas Weber <octave@josoansi.de> parents: 26148 diff changeset	473 singleton_cleanup_list::add (cleanup_instance);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	474 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	475
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	476 return retval;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	477 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	478
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	479 void
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	480 float_fftw_planner::threads (int nt)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	481 {
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	482 #if defined (HAVE_FFTW3F_THREADS)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	483 if (instance_ok () && nt != threads ())
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	484 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	485 instance->nthreads = nt;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	486 fftwf_plan_with_nthreads (nt);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	487 // Clear the current plans.
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	488 instance->rplan = instance->plan[0] = instance->plan[1] = nullptr;
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	489 }
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	490 #else
26148 8c72f0345ee8 maint: Silence compiler warnings about unused or too many parameters. Kai T. Ohlhus <k.ohlhus@gmail.com> parents: 25688 diff changeset	491 octave_unused_parameter (nt);
8c72f0345ee8 maint: Silence compiler warnings about unused or too many parameters. Kai T. Ohlhus <k.ohlhus@gmail.com> parents: 25688 diff changeset	492
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	493 (*current_liboctave_warning_handler)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	494 ("unable to change number of threads without FFTW thread support");
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	495 #endif
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	496 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	497
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	498 void *
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	499 float_fftw_planner::do_create_plan (int dir, const int rank,
23440 f15f0d8b78c8 pass dim_vector as const reference, not by value John W. Eaton <jwe@octave.org> parents: 23433 diff changeset	500 const dim_vector& dims,
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	501 octave_idx_type howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	502 octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	503 octave_idx_type dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	504 const FloatComplex *in,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	505 FloatComplex *out)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	506 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	507 int which = (dir == FFTW_FORWARD) ? 0 : 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	508 fftwf_plan cur_plan_p = reinterpret_cast<fftwf_plan > (&plan[which]);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	509 bool create_new_plan = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	510 bool ioalign = CHECK_SIMD_ALIGNMENT (in) && CHECK_SIMD_ALIGNMENT (out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	511 bool ioinplace = (in == out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	512
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	513 // Don't create a new plan if we have a non SIMD plan already but
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	514 // can do SIMD. This prevents endlessly recreating plans if we
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	515 // change the alignment.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	516
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	517 if (plan[which] == nullptr \|\| d[which] != dist \|\| s[which] != stride
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	518 \|\| r[which] != rank \|\| h[which] != howmany
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	519 \|\| ioinplace != inplace[which]
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	520 \|\| ((ioalign != simd_align[which]) ? ! ioalign : false))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	521 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	522 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	523 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	524 // We still might not have the same shape of array.
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	525
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	526 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	527 if (dims(i) != n[which](i))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	528 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	529 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	530 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	531 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	532 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	533
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	534 if (create_new_plan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	535 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	536 d[which] = dist;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	537 s[which] = stride;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	538 r[which] = rank;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	539 h[which] = howmany;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	540 simd_align[which] = ioalign;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	541 inplace[which] = ioinplace;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	542 n[which] = dims;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	543
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	544 // Note reversal of dimensions for column major storage in FFTW.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	545 octave_idx_type nn = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	546 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	547
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	548 for (int i = 0, j = rank-1; i < rank; i++, j--)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	549 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	550 tmp[i] = dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	551 nn *= dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	552 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	553
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	554 int plan_flags = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	555 bool plan_destroys_in = true;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	556
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	557 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	558 {
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	559 case UNKNOWN:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	560 case ESTIMATE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	561 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	562 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	563 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	564 case MEASURE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	565 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	566 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	567 case PATIENT:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	568 plan_flags \|= FFTW_PATIENT;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	569 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	570 case EXHAUSTIVE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	571 plan_flags \|= FFTW_EXHAUSTIVE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	572 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	573 case HYBRID:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	574 if (nn < 8193)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	575 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	576 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	577 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	578 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	579 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	580 }
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	581 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	582 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	583
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	584 if (ioalign)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	585 plan_flags &= ~FFTW_UNALIGNED;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	586 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	587 plan_flags \|= FFTW_UNALIGNED;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	588
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	589 if (*cur_plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	590 fftwf_destroy_plan (*cur_plan_p);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	591
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	592 if (plan_destroys_in)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	593 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	594 // Create matrix with the same size and 16-byte alignment as input
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	595 OCTAVE_LOCAL_BUFFER (FloatComplex, itmp, nn * howmany + 32);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	596 itmp = reinterpret_cast<FloatComplex *>
25688 b2917b7858ba maint: Use Octave convention for spacing of C++ cast statements. Rik <rik@octave.org> parents: 25103 diff changeset	597 (((reinterpret_cast<ptrdiff_t> (itmp) + 15) & ~ 0xF) +
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	598 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	599
27277 db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	600 *cur_plan_p
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	601 = fftwf_plan_many_dft (rank, tmp, howmany,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	602 reinterpret_cast<fftwf_complex *> (itmp),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	603 nullptr, stride, dist,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	604 reinterpret_cast<fftwf_complex *> (out),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	605 nullptr, stride, dist, dir, plan_flags);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	606 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	607 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	608 {
27277 db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	609 *cur_plan_p
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	610 = fftwf_plan_many_dft (rank, tmp, howmany,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	611 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex > (in)),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	612 nullptr, stride, dist,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	613 reinterpret_cast<fftwf_complex *> (out),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	614 nullptr, stride, dist, dir, plan_flags);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	615 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	616
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	617 if (*cur_plan_p == nullptr)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	618 (*current_liboctave_error_handler) ("Error creating fftw plan");
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	619 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	620
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	621 return *cur_plan_p;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	622 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	623
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	624 void *
23440 f15f0d8b78c8 pass dim_vector as const reference, not by value John W. Eaton <jwe@octave.org> parents: 23433 diff changeset	625 float_fftw_planner::do_create_plan (const int rank, const dim_vector& dims,
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	626 octave_idx_type howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	627 octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	628 octave_idx_type dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	629 const float in, FloatComplex out)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	630 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	631 fftwf_plan cur_plan_p = reinterpret_cast<fftwf_plan > (&rplan);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	632 bool create_new_plan = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	633 bool ioalign = CHECK_SIMD_ALIGNMENT (in) && CHECK_SIMD_ALIGNMENT (out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	634
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	635 // Don't create a new plan if we have a non SIMD plan already but
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	636 // can do SIMD. This prevents endlessly recreating plans if we
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	637 // change the alignment.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	638
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	639 if (rplan == nullptr \|\| rd != dist \|\| rs != stride \|\| rr != rank
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	640 \|\| rh != howmany \|\| ((ioalign != rsimd_align) ? ! ioalign : false))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	641 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	642 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	643 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	644 // We still might not have the same shape of array.
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	645
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	646 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	647 if (dims(i) != rn(i))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	648 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	649 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	650 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	651 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	652 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	653
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	654 if (create_new_plan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	655 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	656 rd = dist;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	657 rs = stride;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	658 rr = rank;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	659 rh = howmany;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	660 rsimd_align = ioalign;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	661 rn = dims;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	662
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	663 // Note reversal of dimensions for column major storage in FFTW.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	664 octave_idx_type nn = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	665 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	666
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	667 for (int i = 0, j = rank-1; i < rank; i++, j--)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	668 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	669 tmp[i] = dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	670 nn *= dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	671 }
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	672
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	673 int plan_flags = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	674 bool plan_destroys_in = true;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	675
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	676 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	677 {
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	678 case UNKNOWN:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	679 case ESTIMATE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	680 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	681 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	682 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	683 case MEASURE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	684 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	685 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	686 case PATIENT:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	687 plan_flags \|= FFTW_PATIENT;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	688 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	689 case EXHAUSTIVE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	690 plan_flags \|= FFTW_EXHAUSTIVE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	691 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	692 case HYBRID:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	693 if (nn < 8193)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	694 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	695 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	696 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	697 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	698 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	699 }
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	700 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	701 }
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	702
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	703 if (ioalign)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	704 plan_flags &= ~FFTW_UNALIGNED;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	705 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	706 plan_flags \|= FFTW_UNALIGNED;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	707
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	708 if (*cur_plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	709 fftwf_destroy_plan (*cur_plan_p);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	710
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	711 if (plan_destroys_in)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	712 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	713 // Create matrix with the same size and 16-byte alignment as input
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	714 OCTAVE_LOCAL_BUFFER (float, itmp, nn + 32);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	715 itmp = reinterpret_cast<float *>
25688 b2917b7858ba maint: Use Octave convention for spacing of C++ cast statements. Rik <rik@octave.org> parents: 25103 diff changeset	716 (((reinterpret_cast<ptrdiff_t> (itmp) + 15) & ~ 0xF) +
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	717 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	718
27277 db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	719 *cur_plan_p
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	720 = fftwf_plan_many_dft_r2c (rank, tmp, howmany, itmp,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	721 nullptr, stride, dist,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	722 reinterpret_cast<fftwf_complex *> (out),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	723 nullptr, stride, dist, plan_flags);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	724 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	725 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	726 {
27277 db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	727 *cur_plan_p
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	728 = fftwf_plan_many_dft_r2c (rank, tmp, howmany,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	729 (const_cast<float *> (in)),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	730 nullptr, stride, dist,
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	731 reinterpret_cast<fftwf_complex *> (out),
db687716fed6 style fixes: generally aim to break long lines before operators, not after John W. Eaton <jwe@octave.org> parents: 27032 diff changeset	732 nullptr, stride, dist, plan_flags);
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	733 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	734
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	735 if (*cur_plan_p == nullptr)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	736 (*current_liboctave_error_handler) ("Error creating fftw plan");
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	737 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	738
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	739 return *cur_plan_p;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	740 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	741
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	742 float_fftw_planner::FftwMethod
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	743 float_fftw_planner::do_method (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	744 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	745 return meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	746 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	747
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	748 float_fftw_planner::FftwMethod
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	749 float_fftw_planner::do_method (FftwMethod _meth)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	750 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	751 FftwMethod ret = meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	752 if (_meth == ESTIMATE \|\| _meth == MEASURE
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	753 \|\| _meth == PATIENT \|\| _meth == EXHAUSTIVE
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	754 \|\| _meth == HYBRID)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	755 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	756 if (meth != _meth)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	757 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	758 meth = _meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	759 if (rplan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	760 fftwf_destroy_plan (reinterpret_cast<fftwf_plan> (rplan));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	761 if (plan[0])
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	762 fftwf_destroy_plan (reinterpret_cast<fftwf_plan> (plan[0]));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	763 if (plan[1])
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	764 fftwf_destroy_plan (reinterpret_cast<fftwf_plan> (plan[1]));
23795 980f39c3ab90 Use C++11 nullptr rather than 0 in code (bug #51565). Rik <rik@octave.org> parents: 23475 diff changeset	765 rplan = plan[0] = plan[1] = nullptr;
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	766 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	767 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	768 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	769 ret = UNKNOWN;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	770 return ret;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	771 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	772
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	773 template <typename T>
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	774 static inline void
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	775 convert_packcomplex_1d (T *out, size_t nr, size_t nc,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	776 octave_idx_type stride, octave_idx_type dist)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	777 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	778 octave_quit ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	779
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	780 // Fill in the missing data.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	781
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	782 for (size_t i = 0; i < nr; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	783 for (size_t j = nc/2+1; j < nc; j++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	784 out[jstride + idist] = conj (out[(nc - j)stride + idist]);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	785
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	786 octave_quit ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	787 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	788
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	789 template <typename T>
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	790 static inline void
23433 c9fab0bc983e maint: Use convention 'int& x' for naming references. Rik <rik@octave.org> parents: 23220 diff changeset	791 convert_packcomplex_Nd (T *out, const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	792 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	793 size_t nc = dv(0);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	794 size_t nr = dv(1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	795 size_t np = (dv.ndims () > 2 ? dv.numel () / nc / nr : 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	796 size_t nrp = nr * np;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	797 T ptr1, ptr2;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	798
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	799 octave_quit ();
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	800
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	801 // Create space for the missing elements.
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	802
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	803 for (size_t i = 0; i < nrp; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	804 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	805 ptr1 = out + i * (nc/2 + 1) + nrp*((nc-1)/2);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	806 ptr2 = out + i * nc;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	807 for (size_t j = 0; j < nc/2+1; j++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	808 ptr2++ = ptr1++;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	809 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	810
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	811 octave_quit ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	812
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	813 // Fill in the missing data for the rank = 2 case directly for speed.
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	814
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	815 for (size_t i = 0; i < np; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	816 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	817 for (size_t j = 1; j < nr; j++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	818 for (size_t k = nc/2+1; k < nc; k++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	819 out[k + (j + inr)nc] = conj (out[nc - k + ((i+1)nr - j)nc]);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	820
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	821 for (size_t j = nc/2+1; j < nc; j++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	822 out[j + inrnc] = conj (out[(inr+1)nc - j]);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	823 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	824
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	825 octave_quit ();
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	826
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	827 // Now do the permutations needed for rank > 2 cases.
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	828
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	829 size_t jstart = dv(0) * dv(1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	830 size_t kstep = dv(0);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	831 size_t nel = dv.numel ();
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	832
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	833 for (int inner = 2; inner < dv.ndims (); inner++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	834 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	835 size_t jmax = jstart * dv(inner);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	836 for (size_t i = 0; i < nel; i+=jmax)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	837 for (size_t j = jstart, jj = jmax-jstart; j < jj;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	838 j+=jstart, jj-=jstart)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	839 for (size_t k = 0; k < jstart; k+= kstep)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	840 for (size_t l = nc/2+1; l < nc; l++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	841 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	842 T tmp = out[i+ j + k + l];
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	843 out[i + j + k + l] = out[i + jj + k + l];
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	844 out[i + jj + k + l] = tmp;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	845 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	846 jstart = jmax;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	847 }
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	848
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	849 octave_quit ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	850 }
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	851
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	852 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	853 fftw::fft (const double in, Complex out, size_t npts,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	854 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	855 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	856 dist = (dist < 0 ? npts : dist);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	857
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	858 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	859 void *vplan = fftw_planner::create_plan (1, dv, nsamples,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	860 stride, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	861 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	862
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	863 fftw_execute_dft_r2c (plan, (const_cast<double *>(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	864 reinterpret_cast<fftw_complex *> (out));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	865
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	866 // Need to create other half of the transform.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	867
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	868 convert_packcomplex_1d (out, nsamples, npts, stride, dist);
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	869
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	870 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	871 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	872
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	873 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	874 fftw::fft (const Complex in, Complex out, size_t npts,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	875 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	876 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	877 dist = (dist < 0 ? npts : dist);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	878
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	879 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	880 void *vplan = fftw_planner::create_plan (FFTW_FORWARD, 1, dv,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	881 nsamples, stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	882 dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	883 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	884
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	885 fftw_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	886 reinterpret_cast<fftw_complex > (const_cast<Complex >(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	887 reinterpret_cast<fftw_complex *> (out));
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	888
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	889 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	890 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	891
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	892 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	893 fftw::ifft (const Complex in, Complex out, size_t npts,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	894 size_t nsamples, octave_idx_type stride,
078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	895 octave_idx_type dist)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	896 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	897 dist = (dist < 0 ? npts : dist);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	898
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	899 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	900 void *vplan = fftw_planner::create_plan (FFTW_BACKWARD, 1, dv, nsamples,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	901 stride, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	902 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	903
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	904 fftw_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	905 reinterpret_cast<fftw_complex > (const_cast<Complex >(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	906 reinterpret_cast<fftw_complex *> (out));
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	907
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	908 const Complex scale = npts;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	909 for (size_t j = 0; j < nsamples; j++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	910 for (size_t i = 0; i < npts; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	911 out[istride + jdist] /= scale;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	912
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	913 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	914 }
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	915
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	916 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	917 fftw::fftNd (const double in, Complex out, const int rank,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	918 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	919 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	920 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	921 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	922 dist *= dv(i);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	923
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	924 // Fool with the position of the start of the output matrix, so that
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	925 // creating other half of the matrix won't cause cache problems.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	926
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	927 octave_idx_type offset = (dv.numel () / dv(0)) * ((dv(0) - 1) / 2);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	928
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	929 void *vplan = fftw_planner::create_plan (rank, dv, 1, 1, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	930 in, out + offset);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	931 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	932
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	933 fftw_execute_dft_r2c (plan, (const_cast<double *>(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	934 reinterpret_cast<fftw_complex *> (out+ offset));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	935
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	936 // Need to create other half of the transform.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	937
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	938 convert_packcomplex_Nd (out, dv);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	939
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	940 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	941 }
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	942
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	943 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	944 fftw::fftNd (const Complex in, Complex out, const int rank,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	945 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	946 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	947 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	948 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	949 dist *= dv(i);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	950
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	951 void *vplan = fftw_planner::create_plan (FFTW_FORWARD, rank, dv,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	952 1, 1, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	953 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	954
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	955 fftw_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	956 reinterpret_cast<fftw_complex > (const_cast<Complex >(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	957 reinterpret_cast<fftw_complex *> (out));
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	958
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	959 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	960 }
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	961
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	962 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	963 fftw::ifftNd (const Complex in, Complex out, const int rank,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	964 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	965 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	966 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	967 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	968 dist *= dv(i);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	969
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	970 void *vplan = fftw_planner::create_plan (FFTW_BACKWARD, rank, dv,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	971 1, 1, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	972 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	973
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	974 fftw_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	975 reinterpret_cast<fftw_complex > (const_cast<Complex >(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	976 reinterpret_cast<fftw_complex *> (out));
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	977
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	978 const size_t npts = dv.numel ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	979 const Complex scale = npts;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	980 for (size_t i = 0; i < npts; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	981 out[i] /= scale;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	982
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	983 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	984 }
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	985
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	986 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	987 fftw::fft (const float in, FloatComplex out, size_t npts,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	988 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	989 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	990 dist = (dist < 0 ? npts : dist);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	991
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	992 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	993 void *vplan = float_fftw_planner::create_plan (1, dv, nsamples, stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	994 dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	995 fftwf_plan plan = reinterpret_cast<fftwf_plan> (vplan);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	996
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	997 fftwf_execute_dft_r2c (plan, (const_cast<float *>(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	998 reinterpret_cast<fftwf_complex *> (out));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	999
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1000 // Need to create other half of the transform.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1001
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1002 convert_packcomplex_1d (out, nsamples, npts, stride, dist);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1003
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1004 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1005 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1006
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1007 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1008 fftw::fft (const FloatComplex in, FloatComplex out, size_t npts,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	1009 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1010 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1011 dist = (dist < 0 ? npts : dist);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1012
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1013 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1014 void *vplan = float_fftw_planner::create_plan (FFTW_FORWARD, 1, dv,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1015 nsamples, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1016 in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1017 fftwf_plan plan = reinterpret_cast<fftwf_plan> (vplan);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1018
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1019 fftwf_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1020 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex >(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1021 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1022
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1023 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1024 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1025
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1026 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1027 fftw::ifft (const FloatComplex in, FloatComplex out, size_t npts,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	1028 size_t nsamples, octave_idx_type stride,
078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	1029 octave_idx_type dist)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1030 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1031 dist = (dist < 0 ? npts : dist);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1032
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1033 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1034 void *vplan = float_fftw_planner::create_plan (FFTW_BACKWARD, 1, dv,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1035 nsamples, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1036 in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1037 fftwf_plan plan = reinterpret_cast<fftwf_plan> (vplan);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1038
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1039 fftwf_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1040 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex >(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1041 reinterpret_cast<fftwf_complex *> (out));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1042
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1043 const FloatComplex scale = npts;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1044 for (size_t j = 0; j < nsamples; j++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1045 for (size_t i = 0; i < npts; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1046 out[istride + jdist] /= scale;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1047
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1048 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1049 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1050
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1051 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1052 fftw::fftNd (const float in, FloatComplex out, const int rank,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	1053 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1054 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1055 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1056 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1057 dist *= dv(i);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1058
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1059 // Fool with the position of the start of the output matrix, so that
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1060 // creating other half of the matrix won't cause cache problems.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1061
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1062 octave_idx_type offset = (dv.numel () / dv(0)) * ((dv(0) - 1) / 2);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1063
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1064 void *vplan = float_fftw_planner::create_plan (rank, dv, 1, 1, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1065 in, out + offset);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1066 fftwf_plan plan = reinterpret_cast<fftwf_plan> (vplan);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1067
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1068 fftwf_execute_dft_r2c (plan, (const_cast<float *>(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1069 reinterpret_cast<fftwf_complex *> (out+ offset));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1070
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1071 // Need to create other half of the transform.
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1072
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1073 convert_packcomplex_Nd (out, dv);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1074
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1075 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1076 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1077
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1078 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1079 fftw::fftNd (const FloatComplex in, FloatComplex out, const int rank,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	1080 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1081 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1082 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1083 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1084 dist *= dv(i);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1085
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1086 void *vplan = float_fftw_planner::create_plan (FFTW_FORWARD, rank, dv,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1087 1, 1, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1088 fftwf_plan plan = reinterpret_cast<fftwf_plan> (vplan);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1089
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1090 fftwf_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1091 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex >(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1092 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1093
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1094 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1095 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1096
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1097 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1098 fftw::ifftNd (const FloatComplex in, FloatComplex out, const int rank,
25103 078b795c5219 maint: style check C++ ahead of 4.4 release. Rik <rik@octave.org> parents: 25054 diff changeset	1099 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1100 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1101 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1102 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1103 dist *= dv(i);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1104
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1105 void *vplan = float_fftw_planner::create_plan (FFTW_BACKWARD, rank, dv,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1106 1, 1, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1107 fftwf_plan plan = reinterpret_cast<fftwf_plan> (vplan);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1108
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1109 fftwf_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1110 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex >(in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1111 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1112
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1113 const size_t npts = dv.numel ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1114 const FloatComplex scale = npts;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1115 for (size_t i = 0; i < npts; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1116 out[i] /= scale;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1117
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1118 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1119 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1120
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	1121 #endif
21600 184b85b31e03 more updates for version function mmuetzel <markus.muetzel@gmx.de> parents: 21520 diff changeset	1122
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1123 std::string
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1124 fftw_version (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1125 {
21600 184b85b31e03 more updates for version function mmuetzel <markus.muetzel@gmx.de> parents: 21520 diff changeset	1126 #if defined (HAVE_FFTW)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1127 return ::fftw_version;
21600 184b85b31e03 more updates for version function mmuetzel <markus.muetzel@gmx.de> parents: 21520 diff changeset	1128 #else
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1129 return "none";
21600 184b85b31e03 more updates for version function mmuetzel <markus.muetzel@gmx.de> parents: 21520 diff changeset	1130 #endif
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1131 }
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1132
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1133 std::string
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1134 fftwf_version (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1135 {
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1136 #if defined (HAVE_FFTW)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1137 return ::fftwf_version;
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1138 #else
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1139 return "none";
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1140 #endif
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1141 }
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1142 }

Mercurial > octave

annotate liboctave/numeric/oct-fftw.cc @ 27919:1891570abac8