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

annotate liboctave/oct-fftw.cc @ 11586:12df7854fa7c

strip trailing whitespace from source files

author	John W. Eaton <jwe@octave.org>
date	Thu, 20 Jan 2011 17:24:59 -0500
parents	fd0a3ac60b0e
children	7dd7cccf0757

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
11523 fd0a3ac60b0e update copyright notices John W. Eaton <jwe@octave.org> parents: 11509 diff changeset	3 Copyright (C) 2001-2011 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
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	8 under the terms of the GNU General Public License as published by the
7016 93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 6482 diff changeset	9 Free Software Foundation; either version 3 of the License, or (at your
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 6482 diff changeset	10 option) any later version.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	11
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	12 Octave is distributed in the hope that it will be useful, but WITHOUT
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	13 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	15 for more details.
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
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	23 #ifdef HAVE_CONFIG_H
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	24 #include <config.h>
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
9523 0ce82753dd72 more configure changes for libraries John W. Eaton <jwe@octave.org> parents: 9516 diff changeset	27 #if defined (HAVE_FFTW)
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	28
4775 88b638195bd1 [project @ 2004-02-16 19:57:06 by jwe] jwe parents: 4774 diff changeset	29 #include <iostream>
88b638195bd1 [project @ 2004-02-16 19:57:06 by jwe] jwe parents: 4774 diff changeset	30 #include <vector>
88b638195bd1 [project @ 2004-02-16 19:57:06 by jwe] jwe parents: 4774 diff changeset	31
4786 fc316bde0053 [project @ 2004-02-18 12:52:20 by jwe] jwe parents: 4785 diff changeset	32 #include "lo-error.h"
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	33 #include "oct-fftw.h"
4786 fc316bde0053 [project @ 2004-02-18 12:52:20 by jwe] jwe parents: 4785 diff changeset	34 #include "quit.h"
8377 25bc2d31e1bf improve OCTAVE_LOCAL_BUFFER Jaroslav Hajek <highegg@gmail.com> parents: 7789 diff changeset	35 #include "oct-locbuf.h"
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	36
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	37 octave_fftw_planner *octave_fftw_planner::instance = 0;
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	38
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	39 // Helper class to create and cache fftw plans for both 1d and
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	40 // 2d. This implementation defaults to using FFTW_ESTIMATE to create
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	41 // the plans, which in theory is suboptimal, but provides quit
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	42 // reasonable performance.
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	43
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	44 // Also note that if FFTW_ESTIMATE is not used the planner in FFTW3
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	45 // destroys the input and output arrays. We must therefore create a
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	46 // temporary input array with the same size and 16-byte alignment as
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	47 // the original array and use that for the planner. Note that we also
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	48 // use any wisdom that is available, either in a FFTW3 system wide file
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	49 // or as supplied by the user.
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	50
5775 ace8d8d26933 [project @ 2006-04-24 19:13:06 by jwe] jwe parents: 5307 diff changeset	51 // FIXME -- if we can ensure 16 byte alignment in Array<T>
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	52 // (<T> *data) the FFTW3 can use SIMD instructions for further
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	53 // acceleration.
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	54
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	55 // Note that it is profitable to store the FFTW3 plans, for small
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	56 // ffts.
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	57
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	58 octave_fftw_planner::octave_fftw_planner (void)
11509 fc35194006d6 oct-fftw.cc: more constructor tweaks John W. Eaton <jwe@octave.org> parents: 11501 diff changeset	59 : meth (ESTIMATE), rplan (0), rd (0), rs (0), rr (0), rh (0), rn (),
11501 331fcc41ca23 data member initialization fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	60 rsimd_align (false)
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	61 {
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	62 plan[0] = plan[1] = 0;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	63 d[0] = d[1] = s[0] = s[1] = r[0] = r[1] = h[0] = h[1] = 0;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	64 simd_align[0] = simd_align[1] = false;
5044 50140aa23b32 [project @ 2004-10-11 16:31:51 by jwe] jwe parents: 4809 diff changeset	65 inplace[0] = inplace[1] = false;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	66 n[0] = n[1] = dim_vector ();
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	67
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	68 // If we have a system wide wisdom file, import it.
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	69 fftw_import_system_wisdom ();
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	70 }
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	71
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	72 bool
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	73 octave_fftw_planner::instance_ok (void)
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	74 {
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	75 bool retval = true;
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	76
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	77 if (! instance)
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	78 instance = new octave_fftw_planner ();
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	79
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	80 if (! instance)
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	81 {
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	82 (*current_liboctave_error_handler)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	83 ("unable to create octave_fftw_planner object!");
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	84
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	85 retval = false;
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	86 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	87
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	88 return retval;
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	89 }
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	90
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	91 #define CHECK_SIMD_ALIGNMENT(x) \
6482 7e958a1532c6 [project @ 2007-04-04 02:37:07 by jwe] jwe parents: 6228 diff changeset	92 (((reinterpret_cast<ptrdiff_t> (x)) & 0xF) == 0)
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	93
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	94 fftw_plan
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	95 octave_fftw_planner::do_create_plan (int dir, const int rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	96 const dim_vector dims,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	97 octave_idx_type howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	98 octave_idx_type stride,
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	99 octave_idx_type dist,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	100 const Complex in, Complex out)
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	101 {
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	102 int which = (dir == FFTW_FORWARD) ? 0 : 1;
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	103 fftw_plan *cur_plan_p = &plan[which];
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	104 bool create_new_plan = false;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	105 bool ioalign = CHECK_SIMD_ALIGNMENT (in) && CHECK_SIMD_ALIGNMENT (out);
5044 50140aa23b32 [project @ 2004-10-11 16:31:51 by jwe] jwe parents: 4809 diff changeset	106 bool ioinplace = (in == out);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	107
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	108 // Don't create a new plan if we have a non SIMD plan already but
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	109 // can do SIMD. This prevents endlessly recreating plans if we
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	110 // change the alignment.
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	111
4783 55975a3073be [project @ 2004-02-17 17:16:37 by jwe] jwe parents: 4775 diff changeset	112 if (plan[which] == 0 \|\| d[which] != dist \|\| s[which] != stride
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	113 \|\| r[which] != rank \|\| h[which] != howmany
5044 50140aa23b32 [project @ 2004-10-11 16:31:51 by jwe] jwe parents: 4809 diff changeset	114 \|\| ioinplace != inplace[which]
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	115 \|\| ((ioalign != simd_align[which]) ? !ioalign : false))
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	116 create_new_plan = true;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	117 else
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	118 {
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	119 // We still might not have the same shape of array.
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	120
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	121 for (int i = 0; i < rank; i++)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	122 if (dims(i) != n[which](i))
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	123 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	124 create_new_plan = true;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	125 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	126 }
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	127 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	128
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	129 if (create_new_plan)
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	130 {
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	131 d[which] = dist;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	132 s[which] = stride;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	133 r[which] = rank;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	134 h[which] = howmany;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	135 simd_align[which] = ioalign;
5044 50140aa23b32 [project @ 2004-10-11 16:31:51 by jwe] jwe parents: 4809 diff changeset	136 inplace[which] = ioinplace;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	137 n[which] = dims;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	138
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	139 // Note reversal of dimensions for column major storage in FFTW.
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	140 octave_idx_type nn = 1;
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	141 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	142
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	143 for (int i = 0, j = rank-1; i < rank; i++, j--)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	144 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	145 tmp[i] = dims(j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	146 nn *= dims(j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	147 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	148
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	149 int plan_flags = 0;
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	150 bool plan_destroys_in = true;
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	151
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	152 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	153 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	154 case UNKNOWN:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	155 case ESTIMATE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	156 plan_flags \|= FFTW_ESTIMATE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	157 plan_destroys_in = false;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	158 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	159 case MEASURE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	160 plan_flags \|= FFTW_MEASURE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	161 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	162 case PATIENT:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	163 plan_flags \|= FFTW_PATIENT;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	164 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	165 case EXHAUSTIVE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	166 plan_flags \|= FFTW_EXHAUSTIVE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	167 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	168 case HYBRID:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	169 if (nn < 8193)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	170 plan_flags \|= FFTW_MEASURE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	171 else
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	172 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	173 plan_flags \|= FFTW_ESTIMATE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	174 plan_destroys_in = false;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	175 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	176 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	177 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	178
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	179 if (ioalign)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	180 plan_flags &= ~FFTW_UNALIGNED;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	181 else
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	182 plan_flags \|= FFTW_UNALIGNED;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	183
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	184 if (*cur_plan_p)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	185 fftw_destroy_plan (*cur_plan_p);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	186
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	187 if (plan_destroys_in)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	188 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	189 // Create matrix with the same size and 16-byte alignment as input
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	190 OCTAVE_LOCAL_BUFFER (Complex, itmp, nn * howmany + 32);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	191 itmp = reinterpret_cast<Complex *>
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	192 (((reinterpret_cast<ptrdiff_t>(itmp) + 15) & ~ 0xF) +
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	193 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	194
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	195 *cur_plan_p =
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	196 fftw_plan_many_dft (rank, tmp, howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	197 reinterpret_cast<fftw_complex *> (itmp),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	198 0, stride, dist, reinterpret_cast<fftw_complex *> (out),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	199 0, stride, dist, dir, plan_flags);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	200 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	201 else
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	202 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	203 *cur_plan_p =
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	204 fftw_plan_many_dft (rank, tmp, howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	205 reinterpret_cast<fftw_complex > (const_cast<Complex > (in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	206 0, stride, dist, reinterpret_cast<fftw_complex *> (out),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	207 0, stride, dist, dir, plan_flags);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	208 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	209
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	210 if (*cur_plan_p == 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	211 (*current_liboctave_error_handler) ("Error creating fftw plan");
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	212 }
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	213
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	214 return *cur_plan_p;
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	215 }
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	216
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	217 fftw_plan
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	218 octave_fftw_planner::do_create_plan (const int rank, const dim_vector dims,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	219 octave_idx_type howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	220 octave_idx_type stride,
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	221 octave_idx_type dist,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	222 const double in, Complex out)
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	223 {
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	224 fftw_plan *cur_plan_p = &rplan;
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	225 bool create_new_plan = false;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	226 bool ioalign = CHECK_SIMD_ALIGNMENT (in) && CHECK_SIMD_ALIGNMENT (out);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	227
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	228 // Don't create a new plan if we have a non SIMD plan already but
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	229 // can do SIMD. This prevents endlessly recreating plans if we
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	230 // change the alignment.
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	231
4783 55975a3073be [project @ 2004-02-17 17:16:37 by jwe] jwe parents: 4775 diff changeset	232 if (rplan == 0 \|\| rd != dist \|\| rs != stride \|\| rr != rank
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	233 \|\| rh != howmany \|\| ((ioalign != rsimd_align) ? !ioalign : false))
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	234 create_new_plan = true;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	235 else
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	236 {
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	237 // We still might not have the same shape of array.
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	238
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	239 for (int i = 0; i < rank; i++)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	240 if (dims(i) != rn(i))
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	241 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	242 create_new_plan = true;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	243 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	244 }
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	245 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	246
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	247 if (create_new_plan)
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	248 {
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	249 rd = dist;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	250 rs = stride;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	251 rr = rank;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	252 rh = howmany;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	253 rsimd_align = ioalign;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	254 rn = dims;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	255
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	256 // Note reversal of dimensions for column major storage in FFTW.
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	257 octave_idx_type nn = 1;
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	258 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	259
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	260 for (int i = 0, j = rank-1; i < rank; i++, j--)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	261 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	262 tmp[i] = dims(j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	263 nn *= dims(j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	264 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	265
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	266 int plan_flags = 0;
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	267 bool plan_destroys_in = true;
aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	268
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	269 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	270 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	271 case UNKNOWN:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	272 case ESTIMATE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	273 plan_flags \|= FFTW_ESTIMATE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	274 plan_destroys_in = false;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	275 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	276 case MEASURE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	277 plan_flags \|= FFTW_MEASURE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	278 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	279 case PATIENT:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	280 plan_flags \|= FFTW_PATIENT;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	281 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	282 case EXHAUSTIVE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	283 plan_flags \|= FFTW_EXHAUSTIVE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	284 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	285 case HYBRID:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	286 if (nn < 8193)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	287 plan_flags \|= FFTW_MEASURE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	288 else
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	289 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	290 plan_flags \|= FFTW_ESTIMATE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	291 plan_destroys_in = false;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	292 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	293 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	294 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	295
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	296 if (ioalign)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	297 plan_flags &= ~FFTW_UNALIGNED;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	298 else
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	299 plan_flags \|= FFTW_UNALIGNED;
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	300
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	301 if (*cur_plan_p)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	302 fftw_destroy_plan (*cur_plan_p);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	303
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	304 if (plan_destroys_in)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	305 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	306 // Create matrix with the same size and 16-byte alignment as input
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	307 OCTAVE_LOCAL_BUFFER (double, itmp, nn + 32);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	308 itmp = reinterpret_cast<double *>
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	309 (((reinterpret_cast<ptrdiff_t>(itmp) + 15) & ~ 0xF) +
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	310 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	311
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	312 *cur_plan_p =
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	313 fftw_plan_many_dft_r2c (rank, tmp, howmany, itmp,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	314 0, stride, dist, reinterpret_cast<fftw_complex *> (out),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	315 0, stride, dist, plan_flags);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	316 }
6228 aa5df9ba98d5 [project @ 2007-01-05 22:49:03 by dbateman] dbateman parents: 5775 diff changeset	317 else
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	318 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	319 *cur_plan_p =
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	320 fftw_plan_many_dft_r2c (rank, tmp, howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	321 (const_cast<double *> (in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	322 0, stride, dist, reinterpret_cast<fftw_complex *> (out),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	323 0, stride, dist, plan_flags);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	324 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	325
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	326 if (*cur_plan_p == 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	327 (*current_liboctave_error_handler) ("Error creating fftw plan");
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	328 }
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	329
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	330 return *cur_plan_p;
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	331 }
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	332
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	333 octave_fftw_planner::FftwMethod
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	334 octave_fftw_planner::do_method (void)
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	335 {
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	336 return meth;
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	337 }
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	338
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	339 octave_fftw_planner::FftwMethod
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	340 octave_fftw_planner::do_method (FftwMethod _meth)
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	341 {
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	342 FftwMethod ret = meth;
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	343 if (_meth == ESTIMATE \|\| _meth == MEASURE
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	344 \|\| _meth == PATIENT \|\| _meth == EXHAUSTIVE
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	345 \|\| _meth == HYBRID)
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	346 {
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	347 if (meth != _meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	348 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	349 meth = _meth;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	350 if (rplan)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	351 fftw_destroy_plan (rplan);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	352 if (plan[0])
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	353 fftw_destroy_plan (plan[0]);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	354 if (plan[1])
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	355 fftw_destroy_plan (plan[1]);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	356 rplan = plan[0] = plan[1] = 0;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	357 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	358 }
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	359 else
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	360 ret = UNKNOWN;
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	361 return ret;
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	362 }
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	363
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	364 octave_float_fftw_planner *octave_float_fftw_planner::instance = 0;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	365
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	366 octave_float_fftw_planner::octave_float_fftw_planner (void)
11509 fc35194006d6 oct-fftw.cc: more constructor tweaks John W. Eaton <jwe@octave.org> parents: 11501 diff changeset	367 : meth (ESTIMATE), rplan (0), rd (0), rs (0), rr (0), rh (0), rn (),
11501 331fcc41ca23 data member initialization fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	368 rsimd_align (false)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	369 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	370 plan[0] = plan[1] = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	371 d[0] = d[1] = s[0] = s[1] = r[0] = r[1] = h[0] = h[1] = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	372 simd_align[0] = simd_align[1] = false;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	373 inplace[0] = inplace[1] = false;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	374 n[0] = n[1] = dim_vector ();
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	375
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	376 // If we have a system wide wisdom file, import it.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	377 fftwf_import_system_wisdom ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	378 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	379
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	380 bool
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	381 octave_float_fftw_planner::instance_ok (void)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	382 {
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	383 bool retval = true;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	384
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	385 if (! instance)
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	386 instance = new octave_float_fftw_planner ();
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	387
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	388 if (! instance)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	389 {
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	390 (*current_liboctave_error_handler)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	391 ("unable to create octave_fftw_planner object!");
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	392
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	393 retval = false;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	394 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	395
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	396 return retval;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	397 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	398
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	399 fftwf_plan
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	400 octave_float_fftw_planner::do_create_plan (int dir, const int rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	401 const dim_vector dims,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	402 octave_idx_type howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	403 octave_idx_type stride,
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	404 octave_idx_type dist,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	405 const FloatComplex *in,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	406 FloatComplex *out)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	407 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	408 int which = (dir == FFTW_FORWARD) ? 0 : 1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	409 fftwf_plan *cur_plan_p = &plan[which];
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	410 bool create_new_plan = false;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	411 bool ioalign = CHECK_SIMD_ALIGNMENT (in) && CHECK_SIMD_ALIGNMENT (out);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	412 bool ioinplace = (in == out);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	413
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	414 // Don't create a new plan if we have a non SIMD plan already but
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	415 // can do SIMD. This prevents endlessly recreating plans if we
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	416 // change the alignment.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	417
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	418 if (plan[which] == 0 \|\| d[which] != dist \|\| s[which] != stride
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	419 \|\| r[which] != rank \|\| h[which] != howmany
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	420 \|\| ioinplace != inplace[which]
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	421 \|\| ((ioalign != simd_align[which]) ? !ioalign : false))
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	422 create_new_plan = true;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	423 else
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	424 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	425 // We still might not have the same shape of array.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	426
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	427 for (int i = 0; i < rank; i++)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	428 if (dims(i) != n[which](i))
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	429 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	430 create_new_plan = true;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	431 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	432 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	433 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	434
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	435 if (create_new_plan)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	436 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	437 d[which] = dist;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	438 s[which] = stride;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	439 r[which] = rank;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	440 h[which] = howmany;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	441 simd_align[which] = ioalign;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	442 inplace[which] = ioinplace;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	443 n[which] = dims;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	444
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	445 // Note reversal of dimensions for column major storage in FFTW.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	446 octave_idx_type nn = 1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	447 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	448
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	449 for (int i = 0, j = rank-1; i < rank; i++, j--)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	450 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	451 tmp[i] = dims(j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	452 nn *= dims(j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	453 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	454
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	455 int plan_flags = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	456 bool plan_destroys_in = true;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	457
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	458 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	459 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	460 case UNKNOWN:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	461 case ESTIMATE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	462 plan_flags \|= FFTW_ESTIMATE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	463 plan_destroys_in = false;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	464 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	465 case MEASURE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	466 plan_flags \|= FFTW_MEASURE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	467 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	468 case PATIENT:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	469 plan_flags \|= FFTW_PATIENT;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	470 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	471 case EXHAUSTIVE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	472 plan_flags \|= FFTW_EXHAUSTIVE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	473 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	474 case HYBRID:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	475 if (nn < 8193)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	476 plan_flags \|= FFTW_MEASURE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	477 else
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	478 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	479 plan_flags \|= FFTW_ESTIMATE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	480 plan_destroys_in = false;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	481 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	482 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	483 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	484
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	485 if (ioalign)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	486 plan_flags &= ~FFTW_UNALIGNED;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	487 else
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	488 plan_flags \|= FFTW_UNALIGNED;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	489
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	490 if (*cur_plan_p)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	491 fftwf_destroy_plan (*cur_plan_p);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	492
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	493 if (plan_destroys_in)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	494 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	495 // Create matrix with the same size and 16-byte alignment as input
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	496 OCTAVE_LOCAL_BUFFER (FloatComplex, itmp, nn * howmany + 32);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	497 itmp = reinterpret_cast<FloatComplex *>
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	498 (((reinterpret_cast<ptrdiff_t>(itmp) + 15) & ~ 0xF) +
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	499 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	500
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	501 *cur_plan_p =
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	502 fftwf_plan_many_dft (rank, tmp, howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	503 reinterpret_cast<fftwf_complex *> (itmp),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	504 0, stride, dist, reinterpret_cast<fftwf_complex *> (out),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	505 0, stride, dist, dir, plan_flags);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	506 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	507 else
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	508 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	509 *cur_plan_p =
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	510 fftwf_plan_many_dft (rank, tmp, howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	511 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex > (in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	512 0, stride, dist, reinterpret_cast<fftwf_complex *> (out),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	513 0, stride, dist, dir, plan_flags);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	514 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	515
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	516 if (*cur_plan_p == 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	517 (*current_liboctave_error_handler) ("Error creating fftw plan");
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	518 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	519
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	520 return *cur_plan_p;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	521 }
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	522
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	523 fftwf_plan
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	524 octave_float_fftw_planner::do_create_plan (const int rank,
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	525 const dim_vector dims,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	526 octave_idx_type howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	527 octave_idx_type stride,
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	528 octave_idx_type dist,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	529 const float in, FloatComplex out)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	530 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	531 fftwf_plan *cur_plan_p = &rplan;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	532 bool create_new_plan = false;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	533 bool ioalign = CHECK_SIMD_ALIGNMENT (in) && CHECK_SIMD_ALIGNMENT (out);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	534
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	535 // Don't create a new plan if we have a non SIMD plan already but
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	536 // can do SIMD. This prevents endlessly recreating plans if we
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	537 // change the alignment.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	538
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	539 if (rplan == 0 \|\| rd != dist \|\| rs != stride \|\| rr != rank
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	540 \|\| rh != howmany \|\| ((ioalign != rsimd_align) ? !ioalign : false))
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	541 create_new_plan = true;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	542 else
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	543 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	544 // We still might not have the same shape of array.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	545
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	546 for (int i = 0; i < rank; i++)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	547 if (dims(i) != rn(i))
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	548 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	549 create_new_plan = true;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	550 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	551 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	552 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	553
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	554 if (create_new_plan)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	555 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	556 rd = dist;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	557 rs = stride;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	558 rr = rank;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	559 rh = howmany;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	560 rsimd_align = ioalign;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	561 rn = dims;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	562
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	563 // Note reversal of dimensions for column major storage in FFTW.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	564 octave_idx_type nn = 1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	565 OCTAVE_LOCAL_BUFFER (int, tmp, rank);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	566
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	567 for (int i = 0, j = rank-1; i < rank; i++, j--)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	568 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	569 tmp[i] = dims(j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	570 nn *= dims(j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	571 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	572
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	573 int plan_flags = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	574 bool plan_destroys_in = true;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	575
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	576 switch (meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	577 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	578 case UNKNOWN:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	579 case ESTIMATE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	580 plan_flags \|= FFTW_ESTIMATE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	581 plan_destroys_in = false;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	582 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	583 case MEASURE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	584 plan_flags \|= FFTW_MEASURE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	585 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	586 case PATIENT:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	587 plan_flags \|= FFTW_PATIENT;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	588 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	589 case EXHAUSTIVE:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	590 plan_flags \|= FFTW_EXHAUSTIVE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	591 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	592 case HYBRID:
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	593 if (nn < 8193)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	594 plan_flags \|= FFTW_MEASURE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	595 else
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	596 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	597 plan_flags \|= FFTW_ESTIMATE;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	598 plan_destroys_in = false;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	599 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	600 break;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	601 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	602
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	603 if (ioalign)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	604 plan_flags &= ~FFTW_UNALIGNED;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	605 else
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	606 plan_flags \|= FFTW_UNALIGNED;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	607
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	608 if (*cur_plan_p)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	609 fftwf_destroy_plan (*cur_plan_p);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	610
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	611 if (plan_destroys_in)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	612 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	613 // Create matrix with the same size and 16-byte alignment as input
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	614 OCTAVE_LOCAL_BUFFER (float, itmp, nn + 32);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	615 itmp = reinterpret_cast<float *>
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	616 (((reinterpret_cast<ptrdiff_t>(itmp) + 15) & ~ 0xF) +
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	617 ((reinterpret_cast<ptrdiff_t> (in)) & 0xF));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	618
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	619 *cur_plan_p =
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	620 fftwf_plan_many_dft_r2c (rank, tmp, howmany, itmp,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	621 0, stride, dist, reinterpret_cast<fftwf_complex *> (out),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	622 0, stride, dist, plan_flags);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	623 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	624 else
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	625 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	626 *cur_plan_p =
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	627 fftwf_plan_many_dft_r2c (rank, tmp, howmany,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	628 (const_cast<float *> (in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	629 0, stride, dist, reinterpret_cast<fftwf_complex *> (out),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	630 0, stride, dist, plan_flags);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	631 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	632
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	633 if (*cur_plan_p == 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	634 (*current_liboctave_error_handler) ("Error creating fftw plan");
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	635 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	636
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	637 return *cur_plan_p;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	638 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	639
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	640 octave_float_fftw_planner::FftwMethod
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	641 octave_float_fftw_planner::do_method (void)
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	642 {
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	643 return meth;
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	644 }
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	645
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	646 octave_float_fftw_planner::FftwMethod
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	647 octave_float_fftw_planner::do_method (FftwMethod _meth)
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	648 {
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	649 FftwMethod ret = meth;
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	650 if (_meth == ESTIMATE \|\| _meth == MEASURE
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	651 \|\| _meth == PATIENT \|\| _meth == EXHAUSTIVE
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	652 \|\| _meth == HYBRID)
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	653 {
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	654 if (meth != _meth)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	655 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	656 meth = _meth;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	657 if (rplan)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	658 fftwf_destroy_plan (rplan);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	659 if (plan[0])
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	660 fftwf_destroy_plan (plan[0]);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	661 if (plan[1])
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	662 fftwf_destroy_plan (plan[1]);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	663 rplan = plan[0] = plan[1] = 0;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	664 }
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	665 }
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	666 else
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	667 ret = UNKNOWN;
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	668 return ret;
fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	669 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	670
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	671 template <class T>
4775 88b638195bd1 [project @ 2004-02-16 19:57:06 by jwe] jwe parents: 4774 diff changeset	672 static inline void
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	673 convert_packcomplex_1d (T *out, size_t nr, size_t nc,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	674 octave_idx_type stride, octave_idx_type dist)
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	675 {
10142 829e69ec3110 make OCTAVE_QUIT a function Jaroslav Hajek <highegg@gmail.com> parents: 9523 diff changeset	676 octave_quit ();
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	677
d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	678 // Fill in the missing data.
d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	679
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	680 for (size_t i = 0; i < nr; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	681 for (size_t j = nc/2+1; j < nc; j++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	682 out[jstride + idist] = conj(out[(nc - j)stride + idist]);
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	683
10142 829e69ec3110 make OCTAVE_QUIT a function Jaroslav Hajek <highegg@gmail.com> parents: 9523 diff changeset	684 octave_quit ();
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	685 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	686
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	687 template <class T>
4775 88b638195bd1 [project @ 2004-02-16 19:57:06 by jwe] jwe parents: 4774 diff changeset	688 static inline void
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	689 convert_packcomplex_Nd (T *out, const dim_vector &dv)
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	690 {
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	691 size_t nc = dv(0);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	692 size_t nr = dv(1);
4808 a9ec0ce18568 [project @ 2004-03-02 17:36:28 by jwe] jwe parents: 4788 diff changeset	693 size_t np = (dv.length () > 2 ? dv.numel () / nc / nr : 1);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	694 size_t nrp = nr * np;
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	695 T ptr1, ptr2;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	696
10142 829e69ec3110 make OCTAVE_QUIT a function Jaroslav Hajek <highegg@gmail.com> parents: 9523 diff changeset	697 octave_quit ();
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	698
d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	699 // Create space for the missing elements.
d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	700
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	701 for (size_t i = 0; i < nrp; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	702 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	703 ptr1 = out + i * (nc/2 + 1) + nrp*((nc-1)/2);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	704 ptr2 = out + i * nc;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	705 for (size_t j = 0; j < nc/2+1; j++)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	706 ptr2++ = ptr1++;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	707 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	708
10142 829e69ec3110 make OCTAVE_QUIT a function Jaroslav Hajek <highegg@gmail.com> parents: 9523 diff changeset	709 octave_quit ();
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	710
d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	711 // Fill in the missing data for the rank = 2 case directly for speed.
d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	712
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	713 for (size_t i = 0; i < np; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	714 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	715 for (size_t j = 1; j < nr; j++)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	716 for (size_t k = nc/2+1; k < nc; k++)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	717 out[k + (j + inr)nc] = conj(out[nc - k + ((i+1)nr - j)nc]);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	718
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	719 for (size_t j = nc/2+1; j < nc; j++)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	720 out[j + inrnc] = conj(out[(inr+1)nc - j]);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	721 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	722
10142 829e69ec3110 make OCTAVE_QUIT a function Jaroslav Hajek <highegg@gmail.com> parents: 9523 diff changeset	723 octave_quit ();
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	724
d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	725 // Now do the permutations needed for rank > 2 cases.
d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	726
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	727 size_t jstart = dv(0) * dv(1);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	728 size_t kstep = dv(0);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	729 size_t nel = dv.numel ();
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	730
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	731 for (int inner = 2; inner < dv.length (); inner++)
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	732 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	733 size_t jmax = jstart * dv(inner);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	734 for (size_t i = 0; i < nel; i+=jmax)
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	735 for (size_t j = jstart, jj = jmax-jstart; j < jj;
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	736 j+=jstart, jj-=jstart)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	737 for (size_t k = 0; k < jstart; k+= kstep)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	738 for (size_t l = nc/2+1; l < nc; l++)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	739 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	740 T tmp = out[i+ j + k + l];
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	741 out[i + j + k + l] = out[i + jj + k + l];
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	742 out[i + jj + k + l] = tmp;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	743 }
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	744 jstart = jmax;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	745 }
4785 d3018a33c584 [project @ 2004-02-17 21:58:17 by jwe] jwe parents: 4784 diff changeset	746
10142 829e69ec3110 make OCTAVE_QUIT a function Jaroslav Hajek <highegg@gmail.com> parents: 9523 diff changeset	747 octave_quit ();
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	748 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	749
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	750 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	751 octave_fftw::fft (const double in, Complex out, size_t npts,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	752 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	753 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	754 dist = (dist < 0 ? npts : dist);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	755
10350 12884915a8e4 merge MArray classes & improve Array interface Jaroslav Hajek <highegg@gmail.com> parents: 10314 diff changeset	756 dim_vector dv (npts, 1);
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	757 fftw_plan plan = octave_fftw_planner::create_plan (1, dv, nsamples,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	758 stride, dist, in, out);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	759
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	760 fftw_execute_dft_r2c (plan, (const_cast<double *>(in)),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	761 reinterpret_cast<fftw_complex *> (out));
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	762
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	763 // Need to create other half of the transform.
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	764
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	765 convert_packcomplex_1d (out, nsamples, npts, stride, dist);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	766
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	767 return 0;
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	768 }
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	769
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	770 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	771 octave_fftw::fft (const Complex in, Complex out, size_t npts,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	772 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	773 {
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	774 dist = (dist < 0 ? npts : dist);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	775
10350 12884915a8e4 merge MArray classes & improve Array interface Jaroslav Hajek <highegg@gmail.com> parents: 10314 diff changeset	776 dim_vector dv (npts, 1);
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	777 fftw_plan plan = octave_fftw_planner::create_plan (FFTW_FORWARD, 1, dv,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	778 nsamples, stride,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	779 dist, in, out);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	780
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	781 fftw_execute_dft (plan,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	782 reinterpret_cast<fftw_complex > (const_cast<Complex >(in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	783 reinterpret_cast<fftw_complex *> (out));
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	784
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	785 return 0;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	786 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	787
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	788 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	789 octave_fftw::ifft (const Complex in, Complex out, size_t npts,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	790 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	791 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	792 dist = (dist < 0 ? npts : dist);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	793
10350 12884915a8e4 merge MArray classes & improve Array interface Jaroslav Hajek <highegg@gmail.com> parents: 10314 diff changeset	794 dim_vector dv (npts, 1);
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	795 fftw_plan plan = octave_fftw_planner::create_plan (FFTW_BACKWARD, 1, dv,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	796 nsamples, stride,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	797 dist, in, out);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	798
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	799 fftw_execute_dft (plan,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	800 reinterpret_cast<fftw_complex > (const_cast<Complex >(in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	801 reinterpret_cast<fftw_complex *> (out));
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	802
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	803 const Complex scale = npts;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	804 for (size_t j = 0; j < nsamples; j++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	805 for (size_t i = 0; i < npts; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	806 out[istride + jdist] /= scale;
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	807
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	808 return 0;
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	809 }
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	810
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	811 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	812 octave_fftw::fftNd (const double in, Complex out, const int rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	813 const dim_vector &dv)
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	814 {
5275 23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe] jwe parents: 5044 diff changeset	815 octave_idx_type dist = 1;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	816 for (int i = 0; i < rank; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	817 dist *= dv(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	818
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	819 // Fool with the position of the start of the output matrix, so that
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	820 // creating other half of the matrix won't cause cache problems.
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	821
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	822 octave_idx_type offset = (dv.numel () / dv(0)) * ((dv(0) - 1) / 2);
12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	823
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	824 fftw_plan plan = octave_fftw_planner::create_plan (rank, dv, 1, 1, dist,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	825 in, out + offset);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	826
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	827 fftw_execute_dft_r2c (plan, (const_cast<double *>(in)),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	828 reinterpret_cast<fftw_complex *> (out+ offset));
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	829
4809 b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	830 // Need to create other half of the transform.
b60be7678bdc [project @ 2004-03-02 17:40:08 by jwe] jwe parents: 4808 diff changeset	831
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	832 convert_packcomplex_Nd (out, dv);
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	833
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	834 return 0;
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	835 }
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	836
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	837 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	838 octave_fftw::fftNd (const Complex in, Complex out, const int rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	839 const dim_vector &dv)
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	840 {
5275 23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe] jwe parents: 5044 diff changeset	841 octave_idx_type dist = 1;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	842 for (int i = 0; i < rank; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	843 dist *= dv(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	844
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	845 fftw_plan plan = octave_fftw_planner::create_plan (FFTW_FORWARD, rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	846 dv, 1, 1, dist, in, out);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	847
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	848 fftw_execute_dft (plan,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	849 reinterpret_cast<fftw_complex > (const_cast<Complex >(in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	850 reinterpret_cast<fftw_complex *> (out));
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	851
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	852 return 0;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	853 }
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	854
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	855 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	856 octave_fftw::ifftNd (const Complex in, Complex out, const int rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	857 const dim_vector &dv)
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	858 {
5275 23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe] jwe parents: 5044 diff changeset	859 octave_idx_type dist = 1;
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	860 for (int i = 0; i < rank; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	861 dist *= dv(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	862
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	863 fftw_plan plan = octave_fftw_planner::create_plan (FFTW_BACKWARD, rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	864 dv, 1, 1, dist, in, out);
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	865
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	866 fftw_execute_dft (plan,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	867 reinterpret_cast<fftw_complex > (const_cast<Complex >(in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	868 reinterpret_cast<fftw_complex *> (out));
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	869
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	870 const size_t npts = dv.numel ();
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	871 const Complex scale = npts;
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	872 for (size_t i = 0; i < npts; i++)
4773 ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe] jwe parents: 3874 diff changeset	873 out[i] /= scale;
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	874
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	875 return 0;
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	876 }
adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	877
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	878 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	879 octave_fftw::fft (const float in, FloatComplex out, size_t npts,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	880 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	881 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	882 dist = (dist < 0 ? npts : dist);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	883
10350 12884915a8e4 merge MArray classes & improve Array interface Jaroslav Hajek <highegg@gmail.com> parents: 10314 diff changeset	884 dim_vector dv (npts, 1);
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	885 fftwf_plan plan = octave_float_fftw_planner::create_plan (1, dv, nsamples,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	886 stride, dist,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	887 in, out);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	888
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	889 fftwf_execute_dft_r2c (plan, (const_cast<float *>(in)),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	890 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	891
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	892 // Need to create other half of the transform.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	893
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	894 convert_packcomplex_1d (out, nsamples, npts, stride, dist);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	895
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	896 return 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	897 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	898
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	899 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	900 octave_fftw::fft (const FloatComplex in, FloatComplex out, size_t npts,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	901 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	902 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	903 dist = (dist < 0 ? npts : dist);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	904
10350 12884915a8e4 merge MArray classes & improve Array interface Jaroslav Hajek <highegg@gmail.com> parents: 10314 diff changeset	905 dim_vector dv (npts, 1);
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	906 fftwf_plan plan = octave_float_fftw_planner::create_plan (FFTW_FORWARD, 1,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	907 dv, nsamples,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	908 stride, dist,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	909 in, out);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	910
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	911 fftwf_execute_dft (plan,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	912 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex >(in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	913 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	914
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	915 return 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	916 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	917
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	918 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	919 octave_fftw::ifft (const FloatComplex in, FloatComplex out, size_t npts,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	920 size_t nsamples, octave_idx_type stride, octave_idx_type dist)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	921 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	922 dist = (dist < 0 ? npts : dist);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	923
10350 12884915a8e4 merge MArray classes & improve Array interface Jaroslav Hajek <highegg@gmail.com> parents: 10314 diff changeset	924 dim_vector dv (npts, 1);
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	925 fftwf_plan plan = octave_float_fftw_planner::create_plan (FFTW_BACKWARD, 1,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	926 dv, nsamples,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	927 stride, dist,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	928 in, out);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	929
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	930 fftwf_execute_dft (plan,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	931 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex >(in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	932 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	933
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	934 const FloatComplex scale = npts;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	935 for (size_t j = 0; j < nsamples; j++)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	936 for (size_t i = 0; i < npts; i++)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	937 out[istride + jdist] /= scale;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	938
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	939 return 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	940 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	941
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	942 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	943 octave_fftw::fftNd (const float in, FloatComplex out, const int rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	944 const dim_vector &dv)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	945 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	946 octave_idx_type dist = 1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	947 for (int i = 0; i < rank; i++)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	948 dist *= dv(i);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	949
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	950 // Fool with the position of the start of the output matrix, so that
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	951 // creating other half of the matrix won't cause cache problems.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	952
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	953 octave_idx_type offset = (dv.numel () / dv(0)) * ((dv(0) - 1) / 2);
12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	954
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	955 fftwf_plan plan = octave_float_fftw_planner::create_plan (rank, dv, 1, 1,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	956 dist, in,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	957 out + offset);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	958
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	959 fftwf_execute_dft_r2c (plan, (const_cast<float *>(in)),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	960 reinterpret_cast<fftwf_complex *> (out+ offset));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	961
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	962 // Need to create other half of the transform.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	963
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	964 convert_packcomplex_Nd (out, dv);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	965
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	966 return 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	967 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	968
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	969 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	970 octave_fftw::fftNd (const FloatComplex in, FloatComplex out, const int rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	971 const dim_vector &dv)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	972 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	973 octave_idx_type dist = 1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	974 for (int i = 0; i < rank; i++)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	975 dist *= dv(i);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	976
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	977 fftwf_plan plan = octave_float_fftw_planner::create_plan (FFTW_FORWARD,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	978 rank, dv, 1, 1,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	979 dist, in, out);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	980
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	981 fftwf_execute_dft (plan,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	982 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex >(in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	983 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	984
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	985 return 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	986 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	987
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	988 int
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	989 octave_fftw::ifftNd (const FloatComplex in, FloatComplex out, const int rank,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	990 const dim_vector &dv)
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	991 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	992 octave_idx_type dist = 1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	993 for (int i = 0; i < rank; i++)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	994 dist *= dv(i);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	995
9516 fb933db0c517 convert fftw planner classes to singleton objects John W. Eaton <jwe@octave.org> parents: 8920 diff changeset	996 fftwf_plan plan = octave_float_fftw_planner::create_plan (FFTW_BACKWARD,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	997 rank, dv, 1, 1,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	998 dist, in, out);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	999
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	1000 fftwf_execute_dft (plan,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	1001 reinterpret_cast<fftwf_complex > (const_cast<FloatComplex >(in)),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	1002 reinterpret_cast<fftwf_complex *> (out));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1003
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1004 const size_t npts = dv.numel ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1005 const FloatComplex scale = npts;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1006 for (size_t i = 0; i < npts; i++)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1007 out[i] /= scale;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1008
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1009 return 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1010 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7017 diff changeset	1011
3828 adc217ebe692 [project @ 2001-05-02 14:50:46 by jwe] jwe parents: diff changeset	1012 #endif

Mercurial > octave-nkf

annotate liboctave/oct-fftw.cc @ 11586:12df7854fa7c