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

annotate liboctave/numeric/oct-fftw.cc @ 23440:f15f0d8b78c8

pass dim_vector as const reference, not by value * ls-hdf5.cc, ls-hdf5.h, oct-fftw.cc, oct-fftw.h: Pass dim_vector as const reference, not by value.

author	John W. Eaton <jwe@octave.org>
date	Wed, 26 Apr 2017 00:55:06 -0400
parents	c9fab0bc983e
children	21baad6b35c4

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
23219 3ac9f9ecfae5 maint: Update copyright dates. John W. Eaton <jwe@octave.org> parents: 23083 diff changeset	3 Copyright (C) 2001-2017 John W. Eaton
7017 a1dbe9d80eee [project @ 2007-10-12 21:27:11 by jwe] jwe parents: 7016 diff changeset	4
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	5 This file is part of Octave.
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	6
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	7 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	8 under the terms of the GNU General Public License as published by
3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	9 the Free Software Foundation; either version 3 of the License, or
3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	10 (at your option) any later version.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	11
22755 3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	12 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	13 WITHOUT ANY WARRANTY; without even the implied warranty of
3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
3a2b891d0b33 maint: Standardize Copyright formatting. Rik <rik@octave.org> parents: 22402 diff changeset	15 GNU General Public License for more details.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	16
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	17 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	18 along with Octave; see the file COPYING. If not, see
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 6482 diff changeset	19 <http://www.gnu.org/licenses/>.
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 */
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	22
21724 aba2e6293dd8 use "#if ..." consistently instead of "#ifdef" and "#ifndef" John W. Eaton <jwe@octave.org> parents: 21688 diff changeset	23 #if defined (HAVE_CONFIG_H)
21301 40de9f8f23a6 Use '#include "config.h"' rather than <config.h>. Rik <rik@octave.org> parents: 21202 diff changeset	24 # include "config.h"
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	25 #endif
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	26
4775 88b638195bd1 [project @ 2004-02-16 19:57:06 by jwe] jwe parents: 4774 diff changeset	27 #include <iostream>
21688 68ab693d16e4 * oct-fftw.cc: Allow compilation without FFTW. John W. Eaton <jwe@octave.org> parents: 21660 diff changeset	28 #include <string>
4775 88b638195bd1 [project @ 2004-02-16 19:57:06 by jwe] jwe parents: 4774 diff changeset	29 #include <vector>
88b638195bd1 [project @ 2004-02-16 19:57:06 by jwe] jwe parents: 4774 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"
4786 fc316bde0053 [project @ 2004-02-18 12:52:20 by jwe] jwe parents: 4785 diff changeset	37 #include "quit.h"
8377 25bc2d31e1bf improve OCTAVE_LOCAL_BUFFER Jaroslav Hajek <highegg@gmail.com> parents: 7789 diff changeset	38 #include "oct-locbuf.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
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	49 fftw_planner *fftw_planner::instance = 0;
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)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	70 : meth (ESTIMATE), rplan (0), rd (0), rs (0), rr (0), rh (0), rn (),
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 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	73 plan[0] = plan[1] = 0;
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).
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	86 nthreads = octave_num_processors_wrapper (OCTAVE_NPROC_CURRENT);
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 ();
13983 7dd7cccf0757 clean up memory allocated for singletons before exit John W. Eaton <jwe@octave.org> parents: 11586 diff changeset	119
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	120 if (instance)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	121 singleton_cleanup_list::add (cleanup_instance);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	122 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	123
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	124 if (! instance)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	125 (*current_liboctave_error_handler)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	126 ("unable to create fftw_planner object!");
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	127
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	128 return retval;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	129 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	130
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	131 void
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	132 fftw_planner::threads (int nt)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	133 {
21520 e1c921322ba2 Fix configuration conditions in oct-fftw.cc Mike Miller <mtmiller@octave.org> parents: 21451 diff changeset	134 #if defined (HAVE_FFTW3_THREADS)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	135 if (instance_ok () && nt != threads ())
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	136 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	137 instance->nthreads = nt;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	138 fftw_plan_with_nthreads (nt);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	139 // Clear the current plans.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	140 instance->rplan = instance->plan[0] = instance->plan[1] = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	141 }
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	142 #else
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	143 (*current_liboctave_warning_handler)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	144 ("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	145 #endif
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	146 }
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	147
763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	148 #define CHECK_SIMD_ALIGNMENT(x) \
6482 7e958a1532c6 [project @ 2007-04-04 02:37:07 by jwe] jwe parents: 6228 diff changeset	149 (((reinterpret_cast<ptrdiff_t> (x)) & 0xF) == 0)
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	150
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	151 void *
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	152 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	153 const dim_vector& dims,
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	154 octave_idx_type howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	155 octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	156 octave_idx_type dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	157 const Complex in, Complex out)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	158 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	159 int which = (dir == FFTW_FORWARD) ? 0 : 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	160 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	161 bool create_new_plan = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	162 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	163 bool ioinplace = (in == out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	164
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	165 // 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	166 // 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	167 // change the alignment.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	168
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	169 if (plan[which] == 0 \|\| d[which] != dist \|\| s[which] != stride
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	170 \|\| r[which] != rank \|\| h[which] != howmany
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	171 \|\| ioinplace != inplace[which]
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	172 \|\| ((ioalign != simd_align[which]) ? ! ioalign : false))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	173 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	174 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	175 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	176 // 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	177
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	178 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	179 if (dims(i) != n[which](i))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	180 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	181 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	182 break;
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 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	185
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	186 if (create_new_plan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	187 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	188 d[which] = dist;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	189 s[which] = stride;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	190 r[which] = rank;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	191 h[which] = howmany;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	192 simd_align[which] = ioalign;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	193 inplace[which] = ioinplace;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	194 n[which] = dims;
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 // 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	197 octave_idx_type nn = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	198 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	199
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	200 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	201 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	202 tmp[i] = dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	203 nn *= dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	204 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	205
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	206 int plan_flags = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	207 bool plan_destroys_in = true;
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	208
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	209 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	210 {
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	211 case UNKNOWN:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	212 case ESTIMATE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	213 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	214 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	215 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	216 case MEASURE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	217 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	218 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	219 case PATIENT:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	220 plan_flags \|= FFTW_PATIENT;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	221 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	222 case EXHAUSTIVE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	223 plan_flags \|= FFTW_EXHAUSTIVE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	224 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	225 case HYBRID:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	226 if (nn < 8193)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	227 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	228 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	229 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	230 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	231 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	232 }
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	233 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	234 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	235
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	236 if (ioalign)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	237 plan_flags &= ~FFTW_UNALIGNED;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	238 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	239 plan_flags \|= FFTW_UNALIGNED;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	240
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	241 if (*cur_plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	242 fftw_destroy_plan (*cur_plan_p);
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	243
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	244 if (plan_destroys_in)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	245 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	246 // 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	247 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	248 itmp = reinterpret_cast<Complex *>
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	249 (((reinterpret_cast<ptrdiff_t>(itmp) + 15) & ~ 0xF) +
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	250 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	251
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	252 *cur_plan_p =
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	253 fftw_plan_many_dft (rank, tmp, howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	254 reinterpret_cast<fftw_complex *> (itmp),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	255 0, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	256 reinterpret_cast<fftw_complex *> (out),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	257 0, stride, dist, dir, plan_flags);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	258 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	259 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	260 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	261 *cur_plan_p =
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	262 fftw_plan_many_dft (rank, tmp, howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	263 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	264 0, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	265 reinterpret_cast<fftw_complex *> (out),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	266 0, stride, dist, dir, plan_flags);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	267 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	268
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	269 if (*cur_plan_p == 0)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	270 (*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	271 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	272
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	273 return *cur_plan_p;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	274 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	275
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	276 void *
23440 f15f0d8b78c8 pass dim_vector as const reference, not by value John W. Eaton <jwe@octave.org> parents: 23433 diff changeset	277 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	278 octave_idx_type howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	279 octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	280 octave_idx_type dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	281 const double in, Complex 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 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	284 bool create_new_plan = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	285 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	286
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	287 // 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	288 // 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	289 // change the alignment.
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	290
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	291 if (rplan == 0 \|\| rd != dist \|\| rs != stride \|\| rr != rank
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	292 \|\| rh != howmany \|\| ((ioalign != rsimd_align) ? ! ioalign : false))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	293 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	294 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	295 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	296 // 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	297
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	298 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	299 if (dims(i) != rn(i))
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 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	302 break;
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 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	305
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	306 if (create_new_plan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	307 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	308 rd = dist;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	309 rs = stride;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	310 rr = rank;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	311 rh = howmany;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	312 rsimd_align = ioalign;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	313 rn = dims;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	314
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	315 // 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	316 octave_idx_type nn = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	317 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	318
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	319 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	320 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	321 tmp[i] = dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	322 nn *= dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	323 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	324
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	325 int plan_flags = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	326 bool plan_destroys_in = true;
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	327
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	328 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	329 {
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	330 case UNKNOWN:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	331 case ESTIMATE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	332 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	333 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	334 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	335 case MEASURE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	336 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	337 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	338 case PATIENT:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	339 plan_flags \|= FFTW_PATIENT;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	340 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	341 case EXHAUSTIVE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	342 plan_flags \|= FFTW_EXHAUSTIVE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	343 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	344 case HYBRID:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	345 if (nn < 8193)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	346 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	347 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	348 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	349 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	350 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	351 }
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	352 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	353 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	354
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	355 if (ioalign)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	356 plan_flags &= ~FFTW_UNALIGNED;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	357 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	358 plan_flags \|= FFTW_UNALIGNED;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	359
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	360 if (*cur_plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	361 fftw_destroy_plan (*cur_plan_p);
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	362
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	363 if (plan_destroys_in)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	364 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	365 // 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	366 OCTAVE_LOCAL_BUFFER (double, itmp, nn + 32);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	367 itmp = reinterpret_cast<double *>
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	368 (((reinterpret_cast<ptrdiff_t>(itmp) + 15) & ~ 0xF) +
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	369 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	370
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	371 *cur_plan_p =
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	372 fftw_plan_many_dft_r2c (rank, tmp, howmany, itmp,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	373 0, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	374 reinterpret_cast<fftw_complex *> (out),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	375 0, stride, dist, plan_flags);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	376 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	377 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	378 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	379 *cur_plan_p =
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	380 fftw_plan_many_dft_r2c (rank, tmp, howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	381 (const_cast<double *> (in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	382 0, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	383 reinterpret_cast<fftw_complex *> (out),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	384 0, stride, dist, plan_flags);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	385 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	386
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	387 if (*cur_plan_p == 0)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	388 (*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	389 }
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	390
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	391 return *cur_plan_p;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	392 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	393
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	394 fftw_planner::FftwMethod
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	395 fftw_planner::do_method (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	396 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	397 return meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	398 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	399
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	400 fftw_planner::FftwMethod
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	401 fftw_planner::do_method (FftwMethod _meth)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	402 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	403 FftwMethod ret = meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	404 if (_meth == ESTIMATE \|\| _meth == MEASURE
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	405 \|\| _meth == PATIENT \|\| _meth == EXHAUSTIVE
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	406 \|\| _meth == HYBRID)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	407 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	408 if (meth != _meth)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	409 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	410 meth = _meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	411 if (rplan)
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> (rplan));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	413 if (plan[0])
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	414 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	415 if (plan[1])
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	416 fftw_destroy_plan (reinterpret_cast<fftw_plan> (plan[1]));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	417 rplan = plan[0] = plan[1] = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	418 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	419 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	420 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	421 ret = UNKNOWN;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	422 return ret;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	423 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	424
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	425 float_fftw_planner *float_fftw_planner::instance = 0;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	426
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	427 float_fftw_planner::float_fftw_planner (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	428 : meth (ESTIMATE), rplan (0), rd (0), rs (0), rr (0), rh (0), rn (),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	429 rsimd_align (false), nthreads (1)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	430 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	431 plan[0] = plan[1] = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	432 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	433 simd_align[0] = simd_align[1] = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	434 inplace[0] = inplace[1] = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	435 n[0] = n[1] = dim_vector ();
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	436
15960 bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	437 #if defined (HAVE_FFTW3F_THREADS)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	438 int init_ret = fftwf_init_threads ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	439 if (! init_ret)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	440 (*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	441
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	442 // 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	443 // This can be later changed with fftw ("threads", nthreads).
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	444 nthreads = octave_num_processors_wrapper (OCTAVE_NPROC_CURRENT);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	445 fftwf_plan_with_nthreads (nthreads);
15960 bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	446 #endif
bde7731b2b83 added FFTW multithreaded library support Andreas Weber <andy.weber.aw@gmail.com> parents: 15271 diff changeset	447
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	448 // 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	449 fftwf_import_system_wisdom ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	450 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	451
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	452 float_fftw_planner::~float_fftw_planner (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	453 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	454 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	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 *> (&rplan);
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[0]);
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);
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	463
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	464 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	465 if (*plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	466 fftwf_destroy_plan (*plan_p);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	467 }
14078 941d19370065 Cleanup FFTW wisdom plans in class destructor and prevent a memory leak. Rik <octave@nomad.inbox5.com> parents: 13994 diff changeset	468
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	469 bool
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	470 float_fftw_planner::instance_ok (void)
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 bool retval = true;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	473
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	474 if (! instance)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	475 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	476 instance = new float_fftw_planner ();
13994 99f039289e95 also clean up float fftw planner John W. Eaton <jwe@octave.org> parents: 13983 diff changeset	477
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	478 if (instance)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	479 singleton_cleanup_list::add (cleanup_instance);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	480 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	481
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	482 if (! instance)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	483 (*current_liboctave_error_handler)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	484 ("unable to create fftw_planner object!");
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	485
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	486 return retval;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	487 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	488
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	489 void
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	490 float_fftw_planner::threads (int nt)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	491 {
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	492 #if defined (HAVE_FFTW3F_THREADS)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	493 if (instance_ok () && nt != threads ())
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	494 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	495 instance->nthreads = nt;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	496 fftwf_plan_with_nthreads (nt);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	497 // Clear the current plans.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	498 instance->rplan = instance->plan[0] = instance->plan[1] = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	499 }
21451 763e30285935 eliminate configuration macros from oct-fftw.h (bug #41027) John W. Eaton <jwe@octave.org> parents: 21301 diff changeset	500 #else
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	501 (*current_liboctave_warning_handler)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	502 ("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	503 #endif
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	504 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	505
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	506 void *
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	507 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	508 const dim_vector& dims,
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	509 octave_idx_type howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	510 octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	511 octave_idx_type dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	512 const FloatComplex *in,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	513 FloatComplex *out)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	514 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	515 int which = (dir == FFTW_FORWARD) ? 0 : 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	516 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	517 bool create_new_plan = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	518 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	519 bool ioinplace = (in == out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	520
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	521 // 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	522 // 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	523 // change the alignment.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	524
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	525 if (plan[which] == 0 \|\| d[which] != dist \|\| s[which] != stride
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	526 \|\| r[which] != rank \|\| h[which] != howmany
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	527 \|\| ioinplace != inplace[which]
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	528 \|\| ((ioalign != simd_align[which]) ? ! ioalign : false))
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 else
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 // 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	533
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	534 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	535 if (dims(i) != n[which](i))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	536 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	537 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	538 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	539 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	540 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	541
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	542 if (create_new_plan)
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 d[which] = dist;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	545 s[which] = stride;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	546 r[which] = rank;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	547 h[which] = howmany;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	548 simd_align[which] = ioalign;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	549 inplace[which] = ioinplace;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	550 n[which] = dims;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	551
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	552 // 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	553 octave_idx_type nn = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	554 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	555
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	556 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	557 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	558 tmp[i] = dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	559 nn *= dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	560 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	561
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	562 int plan_flags = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	563 bool plan_destroys_in = true;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	564
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	565 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	566 {
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	567 case UNKNOWN:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	568 case ESTIMATE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	569 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	570 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	571 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	572 case MEASURE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	573 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	574 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	575 case PATIENT:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	576 plan_flags \|= FFTW_PATIENT;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	577 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	578 case EXHAUSTIVE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	579 plan_flags \|= FFTW_EXHAUSTIVE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	580 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	581 case HYBRID:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	582 if (nn < 8193)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	583 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	584 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	585 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	586 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	587 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	588 }
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	589 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	590 }
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 (ioalign)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	593 plan_flags &= ~FFTW_UNALIGNED;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	594 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	595 plan_flags \|= FFTW_UNALIGNED;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	596
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	597 if (*cur_plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	598 fftwf_destroy_plan (*cur_plan_p);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	599
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	600 if (plan_destroys_in)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	601 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	602 // 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	603 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	604 itmp = reinterpret_cast<FloatComplex *>
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	605 (((reinterpret_cast<ptrdiff_t>(itmp) + 15) & ~ 0xF) +
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	606 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	607
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	608 *cur_plan_p =
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	609 fftwf_plan_many_dft (rank, tmp, howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	610 reinterpret_cast<fftwf_complex *> (itmp),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	611 0, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	612 reinterpret_cast<fftwf_complex *> (out),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	613 0, stride, dist, dir, plan_flags);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	614 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	615 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	616 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	617 *cur_plan_p =
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	618 fftwf_plan_many_dft (rank, tmp, howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	619 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	620 0, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	621 reinterpret_cast<fftwf_complex *> (out),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	622 0, stride, dist, dir, plan_flags);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	623 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	624
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	625 if (*cur_plan_p == 0)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	626 (*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	627 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	628
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	629 return *cur_plan_p;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	630 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	631
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	632 void *
23440 f15f0d8b78c8 pass dim_vector as const reference, not by value John W. Eaton <jwe@octave.org> parents: 23433 diff changeset	633 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	634 octave_idx_type howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	635 octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	636 octave_idx_type dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	637 const float in, FloatComplex out)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	638 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	639 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	640 bool create_new_plan = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	641 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	642
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	643 // 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	644 // 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	645 // change the alignment.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	646
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	647 if (rplan == 0 \|\| rd != dist \|\| rs != stride \|\| rr != rank
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	648 \|\| rh != howmany \|\| ((ioalign != rsimd_align) ? ! ioalign : false))
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 else
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 // 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	653
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	654 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	655 if (dims(i) != rn(i))
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	656 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	657 create_new_plan = true;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	658 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	659 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	660 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	661
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	662 if (create_new_plan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	663 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	664 rd = dist;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	665 rs = stride;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	666 rr = rank;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	667 rh = howmany;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	668 rsimd_align = ioalign;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	669 rn = dims;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	670
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	671 // 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	672 octave_idx_type nn = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	673 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	674
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	675 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	676 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	677 tmp[i] = dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	678 nn *= dims(j);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	679 }
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	680
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	681 int plan_flags = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	682 bool plan_destroys_in = true;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	683
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	684 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	685 {
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	686 case UNKNOWN:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	687 case ESTIMATE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	688 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	689 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	690 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	691 case MEASURE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	692 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	693 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	694 case PATIENT:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	695 plan_flags \|= FFTW_PATIENT;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	696 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	697 case EXHAUSTIVE:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	698 plan_flags \|= FFTW_EXHAUSTIVE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	699 break;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	700 case HYBRID:
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	701 if (nn < 8193)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	702 plan_flags \|= FFTW_MEASURE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	703 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	704 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	705 plan_flags \|= FFTW_ESTIMATE;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	706 plan_destroys_in = false;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	707 }
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	708 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	709 }
23058 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 (ioalign)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	712 plan_flags &= ~FFTW_UNALIGNED;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	713 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	714 plan_flags \|= FFTW_UNALIGNED;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	715
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	716 if (*cur_plan_p)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	717 fftwf_destroy_plan (*cur_plan_p);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	718
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	719 if (plan_destroys_in)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	720 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	721 // 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	722 OCTAVE_LOCAL_BUFFER (float, itmp, nn + 32);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	723 itmp = reinterpret_cast<float *>
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	724 (((reinterpret_cast<ptrdiff_t>(itmp) + 15) & ~ 0xF) +
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	725 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	726
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	727 *cur_plan_p =
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	728 fftwf_plan_many_dft_r2c (rank, tmp, howmany, itmp,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	729 0, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	730 reinterpret_cast<fftwf_complex *> (out),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	731 0, stride, dist, plan_flags);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	732 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	733 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	734 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	735 *cur_plan_p =
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	736 fftwf_plan_many_dft_r2c (rank, tmp, howmany,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	737 (const_cast<float *> (in)),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	738 0, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	739 reinterpret_cast<fftwf_complex *> (out),
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	740 0, stride, dist, plan_flags);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	741 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	742
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	743 if (*cur_plan_p == 0)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	744 (*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	745 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	746
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	747 return *cur_plan_p;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	748 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	749
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	750 float_fftw_planner::FftwMethod
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	751 float_fftw_planner::do_method (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	752 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	753 return meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	754 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	755
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	756 float_fftw_planner::FftwMethod
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	757 float_fftw_planner::do_method (FftwMethod _meth)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	758 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	759 FftwMethod ret = meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	760 if (_meth == ESTIMATE \|\| _meth == MEASURE
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	761 \|\| _meth == PATIENT \|\| _meth == EXHAUSTIVE
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	762 \|\| _meth == HYBRID)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	763 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	764 if (meth != _meth)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	765 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	766 meth = _meth;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	767 if (rplan)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	768 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	769 if (plan[0])
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	770 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	771 if (plan[1])
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	772 fftwf_destroy_plan (reinterpret_cast<fftwf_plan> (plan[1]));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	773 rplan = plan[0] = plan[1] = 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	774 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	775 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	776 else
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	777 ret = UNKNOWN;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	778 return ret;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	779 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	780
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	781 template <typename T>
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	782 static inline void
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	783 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	784 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	785 {
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
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	788 // Fill in the missing data.
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	789
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	790 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	791 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	792 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	793
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	794 octave_quit ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	795 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	796
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	797 template <typename T>
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	798 static inline void
23433 c9fab0bc983e maint: Use convention 'int& x' for naming references. Rik <rik@octave.org> parents: 23220 diff changeset	799 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	800 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	801 size_t nc = dv(0);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	802 size_t nr = dv(1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	803 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	804 size_t nrp = nr * np;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	805 T ptr1, ptr2;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	806
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	807 octave_quit ();
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	808
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	809 // 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	810
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	811 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	812 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	813 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	814 ptr2 = out + i * nc;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	815 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	816 ptr2++ = ptr1++;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	817 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	818
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	819 octave_quit ();
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 // 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	822
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	823 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	824 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	825 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	826 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	827 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	828
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	829 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	830 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	831 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	832
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	833 octave_quit ();
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	834
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	835 // 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	836
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	837 size_t jstart = dv(0) * dv(1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	838 size_t kstep = dv(0);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	839 size_t nel = dv.numel ();
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	840
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	841 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	842 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	843 size_t jmax = jstart * dv(inner);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	844 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	845 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	846 j+=jstart, jj-=jstart)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	847 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	848 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	849 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	850 T tmp = out[i+ j + k + l];
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	851 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	852 out[i + jj + k + l] = tmp;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	853 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	854 jstart = jmax;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	855 }
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	856
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	857 octave_quit ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	858 }
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	859
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	860 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	861 fftw::fft (const double in, Complex out, size_t npts,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	862 size_t nsamples, 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	863 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	864 dist = (dist < 0 ? npts : dist);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	865
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	866 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	867 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	868 stride, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	869 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	870
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	871 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	872 reinterpret_cast<fftw_complex *> (out));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	873
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	874 // 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	875
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	876 convert_packcomplex_1d (out, nsamples, npts, stride, dist);
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	877
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	878 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	879 }
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	880
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	881 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	882 fftw::fft (const Complex in, Complex out, size_t npts,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	883 size_t nsamples, 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	884 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	885 dist = (dist < 0 ? npts : dist);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	886
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	887 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	888 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	889 nsamples, stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	890 dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	891 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	892
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	893 fftw_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	894 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	895 reinterpret_cast<fftw_complex *> (out));
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	896
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	897 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	898 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	899
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	900 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	901 fftw::ifft (const Complex in, Complex out, size_t npts,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	902 size_t nsamples, octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	903 octave_idx_type dist)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	904 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	905 dist = (dist < 0 ? npts : dist);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	906
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	907 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	908 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	909 stride, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	910 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	911
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	912 fftw_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	913 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	914 reinterpret_cast<fftw_complex *> (out));
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 const Complex scale = npts;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	917 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	918 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	919 out[istride + jdist] /= scale;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	920
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	921 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	922 }
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 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	925 fftw::fftNd (const double in, Complex out, const int rank,
23433 c9fab0bc983e maint: Use convention 'int& x' for naming references. Rik <rik@octave.org> parents: 23220 diff changeset	926 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	927 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	928 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	929 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	930 dist *= dv(i);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	931
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	932 // 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	933 // 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	934
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	935 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	936
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	937 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	938 in, out + offset);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	939 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	940
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	941 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	942 reinterpret_cast<fftw_complex *> (out+ offset));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	943
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	944 // Need to create other half of the transform.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	945
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	946 convert_packcomplex_Nd (out, dv);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	947
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	948 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	949 }
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	950
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	951 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	952 fftw::fftNd (const Complex in, Complex out, const int rank,
23433 c9fab0bc983e maint: Use convention 'int& x' for naming references. Rik <rik@octave.org> parents: 23220 diff changeset	953 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	954 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	955 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	956 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	957 dist *= dv(i);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	958
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	959 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	960 1, 1, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	961 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	962
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	963 fftw_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	964 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	965 reinterpret_cast<fftw_complex *> (out));
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	966
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	967 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	968 }
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 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	971 fftw::ifftNd (const Complex in, Complex out, const int rank,
23433 c9fab0bc983e maint: Use convention 'int& x' for naming references. Rik <rik@octave.org> parents: 23220 diff changeset	972 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	973 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	974 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	975 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	976 dist *= dv(i);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	977
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	978 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	979 1, 1, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	980 fftw_plan plan = reinterpret_cast<fftw_plan> (vplan);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	981
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	982 fftw_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	983 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	984 reinterpret_cast<fftw_complex *> (out));
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	985
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	986 const size_t npts = dv.numel ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	987 const Complex scale = npts;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	988 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	989 out[i] /= scale;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	990
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	991 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	992 }
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	993
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	994 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	995 fftw::fft (const float in, FloatComplex out, size_t npts,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	996 size_t nsamples, 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	997 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	998 dist = (dist < 0 ? npts : dist);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	999
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1000 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1001 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	1002 dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1003 fftwf_plan plan = reinterpret_cast<fftwf_plan> (vplan);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	1004
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1005 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	1006 reinterpret_cast<fftwf_complex *> (out));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1007
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1008 // 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	1009
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1010 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	1011
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1012 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1013 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1014
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1015 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1016 fftw::fft (const FloatComplex in, FloatComplex out, size_t npts,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1017 size_t nsamples, 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	1018 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1019 dist = (dist < 0 ? npts : dist);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1020
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1021 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1022 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	1023 nsamples, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1024 in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1025 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	1026
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1027 fftwf_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1028 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	1029 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1030
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1031 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1032 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1033
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1034 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1035 fftw::ifft (const FloatComplex in, FloatComplex out, size_t npts,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1036 size_t nsamples, octave_idx_type stride,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1037 octave_idx_type dist)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1038 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1039 dist = (dist < 0 ? npts : dist);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1040
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1041 dim_vector dv (npts, 1);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1042 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	1043 nsamples, stride, dist,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1044 in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1045 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	1046
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1047 fftwf_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1048 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	1049 reinterpret_cast<fftwf_complex *> (out));
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1050
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1051 const FloatComplex scale = npts;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1052 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	1053 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	1054 out[istride + jdist] /= scale;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1055
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1056 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1057 }
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 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1060 fftw::fftNd (const float in, FloatComplex out, const int rank,
23433 c9fab0bc983e maint: Use convention 'int& x' for naming references. Rik <rik@octave.org> parents: 23220 diff changeset	1061 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1062 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1063 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1064 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1065 dist *= dv(i);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1066
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1067 // 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	1068 // 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	1069
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1070 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	1071
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1072 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	1073 in, out + offset);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1074 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	1075
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1076 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	1077 reinterpret_cast<fftwf_complex *> (out+ offset));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1078
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1079 // 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	1080
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1081 convert_packcomplex_Nd (out, dv);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1082
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1083 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1084 }
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 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1087 fftw::fftNd (const FloatComplex in, FloatComplex out, const int rank,
23433 c9fab0bc983e maint: Use convention 'int& x' for naming references. Rik <rik@octave.org> parents: 23220 diff changeset	1088 const dim_vector& dv)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1089 {
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1090 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1091 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1092 dist *= dv(i);
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 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	1095 1, 1, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1096 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	1097
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1098 fftwf_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1099 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	1100 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1101
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1102 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1103 }
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 int
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1106 fftw::ifftNd (const FloatComplex in, FloatComplex out, const int rank,
23433 c9fab0bc983e maint: Use convention 'int& x' for naming references. Rik <rik@octave.org> parents: 23220 diff changeset	1107 const dim_vector& dv)
23058 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 octave_idx_type dist = 1;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1110 for (int i = 0; i < rank; i++)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1111 dist *= dv(i);
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 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	1114 1, 1, dist, in, out);
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1115 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	1116
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1117 fftwf_execute_dft (plan,
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1118 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	1119 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1120
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1121 const size_t npts = dv.numel ();
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1122 const FloatComplex scale = npts;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1123 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	1124 out[i] /= scale;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1125
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1126 return 0;
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1127 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1128
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	1129 #endif
21600 184b85b31e03 more updates for version function mmuetzel <markus.muetzel@gmx.de> parents: 21520 diff changeset	1130
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1131 std::string
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1132 fftw_version (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1133 {
21600 184b85b31e03 more updates for version function mmuetzel <markus.muetzel@gmx.de> parents: 21520 diff changeset	1134 #if defined (HAVE_FFTW)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1135 return ::fftw_version;
21600 184b85b31e03 more updates for version function mmuetzel <markus.muetzel@gmx.de> parents: 21520 diff changeset	1136 #else
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1137 return "none";
21600 184b85b31e03 more updates for version function mmuetzel <markus.muetzel@gmx.de> parents: 21520 diff changeset	1138 #endif
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1139 }
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1140
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1141 std::string
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1142 fftwf_version (void)
5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1143 {
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1144 #if defined (HAVE_FFTW)
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1145 return ::fftwf_version;
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1146 #else
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1147 return "none";
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1148 #endif
23058 5839d16828cb move fftw classes inside octave namespace John W. Eaton <jwe@octave.org> parents: 22755 diff changeset	1149 }
21603 37cbdf7bffa9 Add fftwf_version (bug #45659) mmuetzel <markus.muetzel@gmx.de> parents: 21600 diff changeset	1150 }

Mercurial > octave

annotate liboctave/numeric/oct-fftw.cc @ 23440:f15f0d8b78c8