cannam@167
|
1 /*
|
cannam@167
|
2 * Copyright (c) 2003, 2007-14 Matteo Frigo
|
cannam@167
|
3 * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology
|
cannam@167
|
4 *
|
cannam@167
|
5 * This program is free software; you can redistribute it and/or modify
|
cannam@167
|
6 * it under the terms of the GNU General Public License as published by
|
cannam@167
|
7 * the Free Software Foundation; either version 2 of the License, or
|
cannam@167
|
8 * (at your option) any later version.
|
cannam@167
|
9 *
|
cannam@167
|
10 * This program is distributed in the hope that it will be useful,
|
cannam@167
|
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
cannam@167
|
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
cannam@167
|
13 * GNU General Public License for more details.
|
cannam@167
|
14 *
|
cannam@167
|
15 * You should have received a copy of the GNU General Public License
|
cannam@167
|
16 * along with this program; if not, write to the Free Software
|
cannam@167
|
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
cannam@167
|
18 *
|
cannam@167
|
19 */
|
cannam@167
|
20
|
cannam@167
|
21 /* Functions in the FFTW Fortran API, mangled according to the
|
cannam@167
|
22 F77(...) macro. This file is designed to be #included by
|
cannam@167
|
23 f77api.c, possibly multiple times in order to support multiple
|
cannam@167
|
24 compiler manglings (via redefinition of F77). */
|
cannam@167
|
25
|
cannam@167
|
26 FFTW_VOIDFUNC F77(execute, EXECUTE)(X(plan) * const p)
|
cannam@167
|
27 {
|
cannam@167
|
28 plan *pln = (*p)->pln;
|
cannam@167
|
29 pln->adt->solve(pln, (*p)->prb);
|
cannam@167
|
30 }
|
cannam@167
|
31
|
cannam@167
|
32 FFTW_VOIDFUNC F77(destroy_plan, DESTROY_PLAN)(X(plan) *p)
|
cannam@167
|
33 {
|
cannam@167
|
34 X(destroy_plan)(*p);
|
cannam@167
|
35 }
|
cannam@167
|
36
|
cannam@167
|
37 FFTW_VOIDFUNC F77(cleanup, CLEANUP)(void)
|
cannam@167
|
38 {
|
cannam@167
|
39 X(cleanup)();
|
cannam@167
|
40 }
|
cannam@167
|
41
|
cannam@167
|
42 FFTW_VOIDFUNC F77(forget_wisdom, FORGET_WISDOM)(void)
|
cannam@167
|
43 {
|
cannam@167
|
44 X(forget_wisdom)();
|
cannam@167
|
45 }
|
cannam@167
|
46
|
cannam@167
|
47 FFTW_VOIDFUNC F77(export_wisdom, EXPORT_WISDOM)(void (*f77_write_char)(char *, void *),
|
cannam@167
|
48 void *data)
|
cannam@167
|
49 {
|
cannam@167
|
50 write_char_data ad;
|
cannam@167
|
51 ad.f77_write_char = f77_write_char;
|
cannam@167
|
52 ad.data = data;
|
cannam@167
|
53 X(export_wisdom)(write_char, (void *) &ad);
|
cannam@167
|
54 }
|
cannam@167
|
55
|
cannam@167
|
56 FFTW_VOIDFUNC F77(import_wisdom, IMPORT_WISDOM)(int *isuccess,
|
cannam@167
|
57 void (*f77_read_char)(int *, void *),
|
cannam@167
|
58 void *data)
|
cannam@167
|
59 {
|
cannam@167
|
60 read_char_data ed;
|
cannam@167
|
61 ed.f77_read_char = f77_read_char;
|
cannam@167
|
62 ed.data = data;
|
cannam@167
|
63 *isuccess = X(import_wisdom)(read_char, (void *) &ed);
|
cannam@167
|
64 }
|
cannam@167
|
65
|
cannam@167
|
66 FFTW_VOIDFUNC F77(import_system_wisdom, IMPORT_SYSTEM_WISDOM)(int *isuccess)
|
cannam@167
|
67 {
|
cannam@167
|
68 *isuccess = X(import_system_wisdom)();
|
cannam@167
|
69 }
|
cannam@167
|
70
|
cannam@167
|
71 FFTW_VOIDFUNC F77(print_plan, PRINT_PLAN)(X(plan) * const p)
|
cannam@167
|
72 {
|
cannam@167
|
73 X(print_plan)(*p);
|
cannam@167
|
74 fflush(stdout);
|
cannam@167
|
75 }
|
cannam@167
|
76
|
cannam@167
|
77 FFTW_VOIDFUNC F77(flops,FLOPS)(X(plan) *p, double *add, double *mul, double *fma)
|
cannam@167
|
78 {
|
cannam@167
|
79 X(flops)(*p, add, mul, fma);
|
cannam@167
|
80 }
|
cannam@167
|
81
|
cannam@167
|
82 FFTW_VOIDFUNC F77(estimate_cost,ESTIMATE_COST)(double *cost, X(plan) * const p)
|
cannam@167
|
83 {
|
cannam@167
|
84 *cost = X(estimate_cost)(*p);
|
cannam@167
|
85 }
|
cannam@167
|
86
|
cannam@167
|
87 FFTW_VOIDFUNC F77(cost,COST)(double *cost, X(plan) * const p)
|
cannam@167
|
88 {
|
cannam@167
|
89 *cost = X(cost)(*p);
|
cannam@167
|
90 }
|
cannam@167
|
91
|
cannam@167
|
92 FFTW_VOIDFUNC F77(set_timelimit,SET_TIMELIMIT)(double *t)
|
cannam@167
|
93 {
|
cannam@167
|
94 X(set_timelimit)(*t);
|
cannam@167
|
95 }
|
cannam@167
|
96
|
cannam@167
|
97 /******************************** DFT ***********************************/
|
cannam@167
|
98
|
cannam@167
|
99 FFTW_VOIDFUNC F77(plan_dft, PLAN_DFT)(X(plan) *p, int *rank, const int *n,
|
cannam@167
|
100 C *in, C *out, int *sign, int *flags)
|
cannam@167
|
101 {
|
cannam@167
|
102 int *nrev = reverse_n(*rank, n);
|
cannam@167
|
103 *p = X(plan_dft)(*rank, nrev, in, out, *sign, *flags);
|
cannam@167
|
104 X(ifree0)(nrev);
|
cannam@167
|
105 }
|
cannam@167
|
106
|
cannam@167
|
107 FFTW_VOIDFUNC F77(plan_dft_1d, PLAN_DFT_1D)(X(plan) *p, int *n, C *in, C *out,
|
cannam@167
|
108 int *sign, int *flags)
|
cannam@167
|
109 {
|
cannam@167
|
110 *p = X(plan_dft_1d)(*n, in, out, *sign, *flags);
|
cannam@167
|
111 }
|
cannam@167
|
112
|
cannam@167
|
113 FFTW_VOIDFUNC F77(plan_dft_2d, PLAN_DFT_2D)(X(plan) *p, int *nx, int *ny,
|
cannam@167
|
114 C *in, C *out, int *sign, int *flags)
|
cannam@167
|
115 {
|
cannam@167
|
116 *p = X(plan_dft_2d)(*ny, *nx, in, out, *sign, *flags);
|
cannam@167
|
117 }
|
cannam@167
|
118
|
cannam@167
|
119 FFTW_VOIDFUNC F77(plan_dft_3d, PLAN_DFT_3D)(X(plan) *p, int *nx, int *ny, int *nz,
|
cannam@167
|
120 C *in, C *out,
|
cannam@167
|
121 int *sign, int *flags)
|
cannam@167
|
122 {
|
cannam@167
|
123 *p = X(plan_dft_3d)(*nz, *ny, *nx, in, out, *sign, *flags);
|
cannam@167
|
124 }
|
cannam@167
|
125
|
cannam@167
|
126 FFTW_VOIDFUNC F77(plan_many_dft, PLAN_MANY_DFT)(X(plan) *p, int *rank, const int *n,
|
cannam@167
|
127 int *howmany,
|
cannam@167
|
128 C *in, const int *inembed,
|
cannam@167
|
129 int *istride, int *idist,
|
cannam@167
|
130 C *out, const int *onembed,
|
cannam@167
|
131 int *ostride, int *odist,
|
cannam@167
|
132 int *sign, int *flags)
|
cannam@167
|
133 {
|
cannam@167
|
134 int *nrev = reverse_n(*rank, n);
|
cannam@167
|
135 int *inembedrev = reverse_n(*rank, inembed);
|
cannam@167
|
136 int *onembedrev = reverse_n(*rank, onembed);
|
cannam@167
|
137 *p = X(plan_many_dft)(*rank, nrev, *howmany,
|
cannam@167
|
138 in, inembedrev, *istride, *idist,
|
cannam@167
|
139 out, onembedrev, *ostride, *odist,
|
cannam@167
|
140 *sign, *flags);
|
cannam@167
|
141 X(ifree0)(onembedrev);
|
cannam@167
|
142 X(ifree0)(inembedrev);
|
cannam@167
|
143 X(ifree0)(nrev);
|
cannam@167
|
144 }
|
cannam@167
|
145
|
cannam@167
|
146 FFTW_VOIDFUNC F77(plan_guru_dft, PLAN_GURU_DFT)(X(plan) *p, int *rank, const int *n,
|
cannam@167
|
147 const int *is, const int *os,
|
cannam@167
|
148 int *howmany_rank, const int *h_n,
|
cannam@167
|
149 const int *h_is, const int *h_os,
|
cannam@167
|
150 C *in, C *out, int *sign, int *flags)
|
cannam@167
|
151 {
|
cannam@167
|
152 X(iodim) *dims = make_dims(*rank, n, is, os);
|
cannam@167
|
153 X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os);
|
cannam@167
|
154 *p = X(plan_guru_dft)(*rank, dims, *howmany_rank, howmany_dims,
|
cannam@167
|
155 in, out, *sign, *flags);
|
cannam@167
|
156 X(ifree0)(howmany_dims);
|
cannam@167
|
157 X(ifree0)(dims);
|
cannam@167
|
158 }
|
cannam@167
|
159
|
cannam@167
|
160 FFTW_VOIDFUNC F77(plan_guru_split_dft, PLAN_GURU_SPLIT_DFT)(X(plan) *p, int *rank, const int *n,
|
cannam@167
|
161 const int *is, const int *os,
|
cannam@167
|
162 int *howmany_rank, const int *h_n,
|
cannam@167
|
163 const int *h_is, const int *h_os,
|
cannam@167
|
164 R *ri, R *ii, R *ro, R *io, int *flags)
|
cannam@167
|
165 {
|
cannam@167
|
166 X(iodim) *dims = make_dims(*rank, n, is, os);
|
cannam@167
|
167 X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os);
|
cannam@167
|
168 *p = X(plan_guru_split_dft)(*rank, dims, *howmany_rank, howmany_dims,
|
cannam@167
|
169 ri, ii, ro, io, *flags);
|
cannam@167
|
170 X(ifree0)(howmany_dims);
|
cannam@167
|
171 X(ifree0)(dims);
|
cannam@167
|
172 }
|
cannam@167
|
173
|
cannam@167
|
174 FFTW_VOIDFUNC F77(execute_dft, EXECUTE_DFT)(X(plan) * const p, C *in, C *out)
|
cannam@167
|
175 {
|
cannam@167
|
176 plan_dft *pln = (plan_dft *) (*p)->pln;
|
cannam@167
|
177 if ((*p)->sign == FFT_SIGN)
|
cannam@167
|
178 pln->apply((plan *) pln, in[0], in[0]+1, out[0], out[0]+1);
|
cannam@167
|
179 else
|
cannam@167
|
180 pln->apply((plan *) pln, in[0]+1, in[0], out[0]+1, out[0]);
|
cannam@167
|
181 }
|
cannam@167
|
182
|
cannam@167
|
183 FFTW_VOIDFUNC F77(execute_split_dft, EXECUTE_SPLIT_DFT)(X(plan) * const p,
|
cannam@167
|
184 R *ri, R *ii, R *ro, R *io)
|
cannam@167
|
185 {
|
cannam@167
|
186 plan_dft *pln = (plan_dft *) (*p)->pln;
|
cannam@167
|
187 pln->apply((plan *) pln, ri, ii, ro, io);
|
cannam@167
|
188 }
|
cannam@167
|
189
|
cannam@167
|
190 /****************************** DFT r2c *********************************/
|
cannam@167
|
191
|
cannam@167
|
192 FFTW_VOIDFUNC F77(plan_dft_r2c, PLAN_DFT_R2C)(X(plan) *p, int *rank, const int *n,
|
cannam@167
|
193 R *in, C *out, int *flags)
|
cannam@167
|
194 {
|
cannam@167
|
195 int *nrev = reverse_n(*rank, n);
|
cannam@167
|
196 *p = X(plan_dft_r2c)(*rank, nrev, in, out, *flags);
|
cannam@167
|
197 X(ifree0)(nrev);
|
cannam@167
|
198 }
|
cannam@167
|
199
|
cannam@167
|
200 FFTW_VOIDFUNC F77(plan_dft_r2c_1d, PLAN_DFT_R2C_1D)(X(plan) *p, int *n, R *in, C *out,
|
cannam@167
|
201 int *flags)
|
cannam@167
|
202 {
|
cannam@167
|
203 *p = X(plan_dft_r2c_1d)(*n, in, out, *flags);
|
cannam@167
|
204 }
|
cannam@167
|
205
|
cannam@167
|
206 FFTW_VOIDFUNC F77(plan_dft_r2c_2d, PLAN_DFT_R2C_2D)(X(plan) *p, int *nx, int *ny,
|
cannam@167
|
207 R *in, C *out, int *flags)
|
cannam@167
|
208 {
|
cannam@167
|
209 *p = X(plan_dft_r2c_2d)(*ny, *nx, in, out, *flags);
|
cannam@167
|
210 }
|
cannam@167
|
211
|
cannam@167
|
212 FFTW_VOIDFUNC F77(plan_dft_r2c_3d, PLAN_DFT_R2C_3D)(X(plan) *p,
|
cannam@167
|
213 int *nx, int *ny, int *nz,
|
cannam@167
|
214 R *in, C *out,
|
cannam@167
|
215 int *flags)
|
cannam@167
|
216 {
|
cannam@167
|
217 *p = X(plan_dft_r2c_3d)(*nz, *ny, *nx, in, out, *flags);
|
cannam@167
|
218 }
|
cannam@167
|
219
|
cannam@167
|
220 FFTW_VOIDFUNC F77(plan_many_dft_r2c, PLAN_MANY_DFT_R2C)(
|
cannam@167
|
221 X(plan) *p, int *rank, const int *n,
|
cannam@167
|
222 int *howmany,
|
cannam@167
|
223 R *in, const int *inembed, int *istride, int *idist,
|
cannam@167
|
224 C *out, const int *onembed, int *ostride, int *odist,
|
cannam@167
|
225 int *flags)
|
cannam@167
|
226 {
|
cannam@167
|
227 int *nrev = reverse_n(*rank, n);
|
cannam@167
|
228 int *inembedrev = reverse_n(*rank, inembed);
|
cannam@167
|
229 int *onembedrev = reverse_n(*rank, onembed);
|
cannam@167
|
230 *p = X(plan_many_dft_r2c)(*rank, nrev, *howmany,
|
cannam@167
|
231 in, inembedrev, *istride, *idist,
|
cannam@167
|
232 out, onembedrev, *ostride, *odist,
|
cannam@167
|
233 *flags);
|
cannam@167
|
234 X(ifree0)(onembedrev);
|
cannam@167
|
235 X(ifree0)(inembedrev);
|
cannam@167
|
236 X(ifree0)(nrev);
|
cannam@167
|
237 }
|
cannam@167
|
238
|
cannam@167
|
239 FFTW_VOIDFUNC F77(plan_guru_dft_r2c, PLAN_GURU_DFT_R2C)(
|
cannam@167
|
240 X(plan) *p, int *rank, const int *n,
|
cannam@167
|
241 const int *is, const int *os,
|
cannam@167
|
242 int *howmany_rank, const int *h_n,
|
cannam@167
|
243 const int *h_is, const int *h_os,
|
cannam@167
|
244 R *in, C *out, int *flags)
|
cannam@167
|
245 {
|
cannam@167
|
246 X(iodim) *dims = make_dims(*rank, n, is, os);
|
cannam@167
|
247 X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os);
|
cannam@167
|
248 *p = X(plan_guru_dft_r2c)(*rank, dims, *howmany_rank, howmany_dims,
|
cannam@167
|
249 in, out, *flags);
|
cannam@167
|
250 X(ifree0)(howmany_dims);
|
cannam@167
|
251 X(ifree0)(dims);
|
cannam@167
|
252 }
|
cannam@167
|
253
|
cannam@167
|
254 FFTW_VOIDFUNC F77(plan_guru_split_dft_r2c, PLAN_GURU_SPLIT_DFT_R2C)(
|
cannam@167
|
255 X(plan) *p, int *rank, const int *n,
|
cannam@167
|
256 const int *is, const int *os,
|
cannam@167
|
257 int *howmany_rank, const int *h_n,
|
cannam@167
|
258 const int *h_is, const int *h_os,
|
cannam@167
|
259 R *in, R *ro, R *io, int *flags)
|
cannam@167
|
260 {
|
cannam@167
|
261 X(iodim) *dims = make_dims(*rank, n, is, os);
|
cannam@167
|
262 X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os);
|
cannam@167
|
263 *p = X(plan_guru_split_dft_r2c)(*rank, dims, *howmany_rank, howmany_dims,
|
cannam@167
|
264 in, ro, io, *flags);
|
cannam@167
|
265 X(ifree0)(howmany_dims);
|
cannam@167
|
266 X(ifree0)(dims);
|
cannam@167
|
267 }
|
cannam@167
|
268
|
cannam@167
|
269 FFTW_VOIDFUNC F77(execute_dft_r2c, EXECUTE_DFT_R2C)(X(plan) * const p, R *in, C *out)
|
cannam@167
|
270 {
|
cannam@167
|
271 plan_rdft2 *pln = (plan_rdft2 *) (*p)->pln;
|
cannam@167
|
272 problem_rdft2 *prb = (problem_rdft2 *) (*p)->prb;
|
cannam@167
|
273 pln->apply((plan *) pln, in, in + (prb->r1 - prb->r0), out[0], out[0]+1);
|
cannam@167
|
274 }
|
cannam@167
|
275
|
cannam@167
|
276 FFTW_VOIDFUNC F77(execute_split_dft_r2c, EXECUTE_SPLIT_DFT_R2C)(X(plan) * const p,
|
cannam@167
|
277 R *in, R *ro, R *io)
|
cannam@167
|
278 {
|
cannam@167
|
279 plan_rdft2 *pln = (plan_rdft2 *) (*p)->pln;
|
cannam@167
|
280 problem_rdft2 *prb = (problem_rdft2 *) (*p)->prb;
|
cannam@167
|
281 pln->apply((plan *) pln, in, in + (prb->r1 - prb->r0), ro, io);
|
cannam@167
|
282 }
|
cannam@167
|
283
|
cannam@167
|
284 /****************************** DFT c2r *********************************/
|
cannam@167
|
285
|
cannam@167
|
286 FFTW_VOIDFUNC F77(plan_dft_c2r, PLAN_DFT_C2R)(X(plan) *p, int *rank, const int *n,
|
cannam@167
|
287 C *in, R *out, int *flags)
|
cannam@167
|
288 {
|
cannam@167
|
289 int *nrev = reverse_n(*rank, n);
|
cannam@167
|
290 *p = X(plan_dft_c2r)(*rank, nrev, in, out, *flags);
|
cannam@167
|
291 X(ifree0)(nrev);
|
cannam@167
|
292 }
|
cannam@167
|
293
|
cannam@167
|
294 FFTW_VOIDFUNC F77(plan_dft_c2r_1d, PLAN_DFT_C2R_1D)(X(plan) *p, int *n, C *in, R *out,
|
cannam@167
|
295 int *flags)
|
cannam@167
|
296 {
|
cannam@167
|
297 *p = X(plan_dft_c2r_1d)(*n, in, out, *flags);
|
cannam@167
|
298 }
|
cannam@167
|
299
|
cannam@167
|
300 FFTW_VOIDFUNC F77(plan_dft_c2r_2d, PLAN_DFT_C2R_2D)(X(plan) *p, int *nx, int *ny,
|
cannam@167
|
301 C *in, R *out, int *flags)
|
cannam@167
|
302 {
|
cannam@167
|
303 *p = X(plan_dft_c2r_2d)(*ny, *nx, in, out, *flags);
|
cannam@167
|
304 }
|
cannam@167
|
305
|
cannam@167
|
306 FFTW_VOIDFUNC F77(plan_dft_c2r_3d, PLAN_DFT_C2R_3D)(X(plan) *p,
|
cannam@167
|
307 int *nx, int *ny, int *nz,
|
cannam@167
|
308 C *in, R *out,
|
cannam@167
|
309 int *flags)
|
cannam@167
|
310 {
|
cannam@167
|
311 *p = X(plan_dft_c2r_3d)(*nz, *ny, *nx, in, out, *flags);
|
cannam@167
|
312 }
|
cannam@167
|
313
|
cannam@167
|
314 FFTW_VOIDFUNC F77(plan_many_dft_c2r, PLAN_MANY_DFT_C2R)(
|
cannam@167
|
315 X(plan) *p, int *rank, const int *n,
|
cannam@167
|
316 int *howmany,
|
cannam@167
|
317 C *in, const int *inembed, int *istride, int *idist,
|
cannam@167
|
318 R *out, const int *onembed, int *ostride, int *odist,
|
cannam@167
|
319 int *flags)
|
cannam@167
|
320 {
|
cannam@167
|
321 int *nrev = reverse_n(*rank, n);
|
cannam@167
|
322 int *inembedrev = reverse_n(*rank, inembed);
|
cannam@167
|
323 int *onembedrev = reverse_n(*rank, onembed);
|
cannam@167
|
324 *p = X(plan_many_dft_c2r)(*rank, nrev, *howmany,
|
cannam@167
|
325 in, inembedrev, *istride, *idist,
|
cannam@167
|
326 out, onembedrev, *ostride, *odist,
|
cannam@167
|
327 *flags);
|
cannam@167
|
328 X(ifree0)(onembedrev);
|
cannam@167
|
329 X(ifree0)(inembedrev);
|
cannam@167
|
330 X(ifree0)(nrev);
|
cannam@167
|
331 }
|
cannam@167
|
332
|
cannam@167
|
333 FFTW_VOIDFUNC F77(plan_guru_dft_c2r, PLAN_GURU_DFT_C2R)(
|
cannam@167
|
334 X(plan) *p, int *rank, const int *n,
|
cannam@167
|
335 const int *is, const int *os,
|
cannam@167
|
336 int *howmany_rank, const int *h_n,
|
cannam@167
|
337 const int *h_is, const int *h_os,
|
cannam@167
|
338 C *in, R *out, int *flags)
|
cannam@167
|
339 {
|
cannam@167
|
340 X(iodim) *dims = make_dims(*rank, n, is, os);
|
cannam@167
|
341 X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os);
|
cannam@167
|
342 *p = X(plan_guru_dft_c2r)(*rank, dims, *howmany_rank, howmany_dims,
|
cannam@167
|
343 in, out, *flags);
|
cannam@167
|
344 X(ifree0)(howmany_dims);
|
cannam@167
|
345 X(ifree0)(dims);
|
cannam@167
|
346 }
|
cannam@167
|
347
|
cannam@167
|
348 FFTW_VOIDFUNC F77(plan_guru_split_dft_c2r, PLAN_GURU_SPLIT_DFT_C2R)(
|
cannam@167
|
349 X(plan) *p, int *rank, const int *n,
|
cannam@167
|
350 const int *is, const int *os,
|
cannam@167
|
351 int *howmany_rank, const int *h_n,
|
cannam@167
|
352 const int *h_is, const int *h_os,
|
cannam@167
|
353 R *ri, R *ii, R *out, int *flags)
|
cannam@167
|
354 {
|
cannam@167
|
355 X(iodim) *dims = make_dims(*rank, n, is, os);
|
cannam@167
|
356 X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os);
|
cannam@167
|
357 *p = X(plan_guru_split_dft_c2r)(*rank, dims, *howmany_rank, howmany_dims,
|
cannam@167
|
358 ri, ii, out, *flags);
|
cannam@167
|
359 X(ifree0)(howmany_dims);
|
cannam@167
|
360 X(ifree0)(dims);
|
cannam@167
|
361 }
|
cannam@167
|
362
|
cannam@167
|
363 FFTW_VOIDFUNC F77(execute_dft_c2r, EXECUTE_DFT_C2R)(X(plan) * const p, C *in, R *out)
|
cannam@167
|
364 {
|
cannam@167
|
365 plan_rdft2 *pln = (plan_rdft2 *) (*p)->pln;
|
cannam@167
|
366 problem_rdft2 *prb = (problem_rdft2 *) (*p)->prb;
|
cannam@167
|
367 pln->apply((plan *) pln, out, out + (prb->r1 - prb->r0), in[0], in[0]+1);
|
cannam@167
|
368 }
|
cannam@167
|
369
|
cannam@167
|
370 FFTW_VOIDFUNC F77(execute_split_dft_c2r, EXECUTE_SPLIT_DFT_C2R)(X(plan) * const p,
|
cannam@167
|
371 R *ri, R *ii, R *out)
|
cannam@167
|
372 {
|
cannam@167
|
373 plan_rdft2 *pln = (plan_rdft2 *) (*p)->pln;
|
cannam@167
|
374 problem_rdft2 *prb = (problem_rdft2 *) (*p)->prb;
|
cannam@167
|
375 pln->apply((plan *) pln, out, out + (prb->r1 - prb->r0), ri, ii);
|
cannam@167
|
376 }
|
cannam@167
|
377
|
cannam@167
|
378 /****************************** r2r *********************************/
|
cannam@167
|
379
|
cannam@167
|
380 FFTW_VOIDFUNC F77(plan_r2r, PLAN_R2R)(X(plan) *p, int *rank, const int *n,
|
cannam@167
|
381 R *in, R *out,
|
cannam@167
|
382 int *kind, int *flags)
|
cannam@167
|
383 {
|
cannam@167
|
384 int *nrev = reverse_n(*rank, n);
|
cannam@167
|
385 X(r2r_kind) *k = ints2kinds(*rank, kind);
|
cannam@167
|
386 *p = X(plan_r2r)(*rank, nrev, in, out, k, *flags);
|
cannam@167
|
387 X(ifree0)(k);
|
cannam@167
|
388 X(ifree0)(nrev);
|
cannam@167
|
389 }
|
cannam@167
|
390
|
cannam@167
|
391 FFTW_VOIDFUNC F77(plan_r2r_1d, PLAN_R2R_1D)(X(plan) *p, int *n, R *in, R *out,
|
cannam@167
|
392 int *kind, int *flags)
|
cannam@167
|
393 {
|
cannam@167
|
394 *p = X(plan_r2r_1d)(*n, in, out, (X(r2r_kind)) *kind, *flags);
|
cannam@167
|
395 }
|
cannam@167
|
396
|
cannam@167
|
397 FFTW_VOIDFUNC F77(plan_r2r_2d, PLAN_R2R_2D)(X(plan) *p, int *nx, int *ny,
|
cannam@167
|
398 R *in, R *out,
|
cannam@167
|
399 int *kindx, int *kindy, int *flags)
|
cannam@167
|
400 {
|
cannam@167
|
401 *p = X(plan_r2r_2d)(*ny, *nx, in, out,
|
cannam@167
|
402 (X(r2r_kind)) *kindy, (X(r2r_kind)) *kindx, *flags);
|
cannam@167
|
403 }
|
cannam@167
|
404
|
cannam@167
|
405 FFTW_VOIDFUNC F77(plan_r2r_3d, PLAN_R2R_3D)(X(plan) *p,
|
cannam@167
|
406 int *nx, int *ny, int *nz,
|
cannam@167
|
407 R *in, R *out,
|
cannam@167
|
408 int *kindx, int *kindy, int *kindz,
|
cannam@167
|
409 int *flags)
|
cannam@167
|
410 {
|
cannam@167
|
411 *p = X(plan_r2r_3d)(*nz, *ny, *nx, in, out,
|
cannam@167
|
412 (X(r2r_kind)) *kindz, (X(r2r_kind)) *kindy,
|
cannam@167
|
413 (X(r2r_kind)) *kindx, *flags);
|
cannam@167
|
414 }
|
cannam@167
|
415
|
cannam@167
|
416 FFTW_VOIDFUNC F77(plan_many_r2r, PLAN_MANY_R2R)(
|
cannam@167
|
417 X(plan) *p, int *rank, const int *n,
|
cannam@167
|
418 int *howmany,
|
cannam@167
|
419 R *in, const int *inembed, int *istride, int *idist,
|
cannam@167
|
420 R *out, const int *onembed, int *ostride, int *odist,
|
cannam@167
|
421 int *kind, int *flags)
|
cannam@167
|
422 {
|
cannam@167
|
423 int *nrev = reverse_n(*rank, n);
|
cannam@167
|
424 int *inembedrev = reverse_n(*rank, inembed);
|
cannam@167
|
425 int *onembedrev = reverse_n(*rank, onembed);
|
cannam@167
|
426 X(r2r_kind) *k = ints2kinds(*rank, kind);
|
cannam@167
|
427 *p = X(plan_many_r2r)(*rank, nrev, *howmany,
|
cannam@167
|
428 in, inembedrev, *istride, *idist,
|
cannam@167
|
429 out, onembedrev, *ostride, *odist,
|
cannam@167
|
430 k, *flags);
|
cannam@167
|
431 X(ifree0)(k);
|
cannam@167
|
432 X(ifree0)(onembedrev);
|
cannam@167
|
433 X(ifree0)(inembedrev);
|
cannam@167
|
434 X(ifree0)(nrev);
|
cannam@167
|
435 }
|
cannam@167
|
436
|
cannam@167
|
437 FFTW_VOIDFUNC F77(plan_guru_r2r, PLAN_GURU_R2R)(
|
cannam@167
|
438 X(plan) *p, int *rank, const int *n,
|
cannam@167
|
439 const int *is, const int *os,
|
cannam@167
|
440 int *howmany_rank, const int *h_n,
|
cannam@167
|
441 const int *h_is, const int *h_os,
|
cannam@167
|
442 R *in, R *out, int *kind, int *flags)
|
cannam@167
|
443 {
|
cannam@167
|
444 X(iodim) *dims = make_dims(*rank, n, is, os);
|
cannam@167
|
445 X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os);
|
cannam@167
|
446 X(r2r_kind) *k = ints2kinds(*rank, kind);
|
cannam@167
|
447 *p = X(plan_guru_r2r)(*rank, dims, *howmany_rank, howmany_dims,
|
cannam@167
|
448 in, out, k, *flags);
|
cannam@167
|
449 X(ifree0)(k);
|
cannam@167
|
450 X(ifree0)(howmany_dims);
|
cannam@167
|
451 X(ifree0)(dims);
|
cannam@167
|
452 }
|
cannam@167
|
453
|
cannam@167
|
454 FFTW_VOIDFUNC F77(execute_r2r, EXECUTE_R2R)(X(plan) * const p, R *in, R *out)
|
cannam@167
|
455 {
|
cannam@167
|
456 plan_rdft *pln = (plan_rdft *) (*p)->pln;
|
cannam@167
|
457 pln->apply((plan *) pln, in, out);
|
cannam@167
|
458 }
|