cannam@167: /* cannam@167: * Copyright (c) 2003, 2007-14 Matteo Frigo cannam@167: * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology cannam@167: * cannam@167: * This program is free software; you can redistribute it and/or modify cannam@167: * it under the terms of the GNU General Public License as published by cannam@167: * the Free Software Foundation; either version 2 of the License, or cannam@167: * (at your option) any later version. cannam@167: * cannam@167: * This program is distributed in the hope that it will be useful, cannam@167: * but WITHOUT ANY WARRANTY; without even the implied warranty of cannam@167: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the cannam@167: * GNU General Public License for more details. cannam@167: * cannam@167: * You should have received a copy of the GNU General Public License cannam@167: * along with this program; if not, write to the Free Software cannam@167: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA cannam@167: * cannam@167: */ cannam@167: cannam@167: /* Functions in the FFTW Fortran API, mangled according to the cannam@167: F77(...) macro. This file is designed to be #included by cannam@167: f77api.c, possibly multiple times in order to support multiple cannam@167: compiler manglings (via redefinition of F77). */ cannam@167: cannam@167: FFTW_VOIDFUNC F77(execute, EXECUTE)(X(plan) * const p) cannam@167: { cannam@167: plan *pln = (*p)->pln; cannam@167: pln->adt->solve(pln, (*p)->prb); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(destroy_plan, DESTROY_PLAN)(X(plan) *p) cannam@167: { cannam@167: X(destroy_plan)(*p); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(cleanup, CLEANUP)(void) cannam@167: { cannam@167: X(cleanup)(); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(forget_wisdom, FORGET_WISDOM)(void) cannam@167: { cannam@167: X(forget_wisdom)(); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(export_wisdom, EXPORT_WISDOM)(void (*f77_write_char)(char *, void *), cannam@167: void *data) cannam@167: { cannam@167: write_char_data ad; cannam@167: ad.f77_write_char = f77_write_char; cannam@167: ad.data = data; cannam@167: X(export_wisdom)(write_char, (void *) &ad); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(import_wisdom, IMPORT_WISDOM)(int *isuccess, cannam@167: void (*f77_read_char)(int *, void *), cannam@167: void *data) cannam@167: { cannam@167: read_char_data ed; cannam@167: ed.f77_read_char = f77_read_char; cannam@167: ed.data = data; cannam@167: *isuccess = X(import_wisdom)(read_char, (void *) &ed); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(import_system_wisdom, IMPORT_SYSTEM_WISDOM)(int *isuccess) cannam@167: { cannam@167: *isuccess = X(import_system_wisdom)(); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(print_plan, PRINT_PLAN)(X(plan) * const p) cannam@167: { cannam@167: X(print_plan)(*p); cannam@167: fflush(stdout); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(flops,FLOPS)(X(plan) *p, double *add, double *mul, double *fma) cannam@167: { cannam@167: X(flops)(*p, add, mul, fma); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(estimate_cost,ESTIMATE_COST)(double *cost, X(plan) * const p) cannam@167: { cannam@167: *cost = X(estimate_cost)(*p); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(cost,COST)(double *cost, X(plan) * const p) cannam@167: { cannam@167: *cost = X(cost)(*p); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(set_timelimit,SET_TIMELIMIT)(double *t) cannam@167: { cannam@167: X(set_timelimit)(*t); cannam@167: } cannam@167: cannam@167: /******************************** DFT ***********************************/ cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft, PLAN_DFT)(X(plan) *p, int *rank, const int *n, cannam@167: C *in, C *out, int *sign, int *flags) cannam@167: { cannam@167: int *nrev = reverse_n(*rank, n); cannam@167: *p = X(plan_dft)(*rank, nrev, in, out, *sign, *flags); cannam@167: X(ifree0)(nrev); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_1d, PLAN_DFT_1D)(X(plan) *p, int *n, C *in, C *out, cannam@167: int *sign, int *flags) cannam@167: { cannam@167: *p = X(plan_dft_1d)(*n, in, out, *sign, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_2d, PLAN_DFT_2D)(X(plan) *p, int *nx, int *ny, cannam@167: C *in, C *out, int *sign, int *flags) cannam@167: { cannam@167: *p = X(plan_dft_2d)(*ny, *nx, in, out, *sign, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_3d, PLAN_DFT_3D)(X(plan) *p, int *nx, int *ny, int *nz, cannam@167: C *in, C *out, cannam@167: int *sign, int *flags) cannam@167: { cannam@167: *p = X(plan_dft_3d)(*nz, *ny, *nx, in, out, *sign, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_many_dft, PLAN_MANY_DFT)(X(plan) *p, int *rank, const int *n, cannam@167: int *howmany, cannam@167: C *in, const int *inembed, cannam@167: int *istride, int *idist, cannam@167: C *out, const int *onembed, cannam@167: int *ostride, int *odist, cannam@167: int *sign, int *flags) cannam@167: { cannam@167: int *nrev = reverse_n(*rank, n); cannam@167: int *inembedrev = reverse_n(*rank, inembed); cannam@167: int *onembedrev = reverse_n(*rank, onembed); cannam@167: *p = X(plan_many_dft)(*rank, nrev, *howmany, cannam@167: in, inembedrev, *istride, *idist, cannam@167: out, onembedrev, *ostride, *odist, cannam@167: *sign, *flags); cannam@167: X(ifree0)(onembedrev); cannam@167: X(ifree0)(inembedrev); cannam@167: X(ifree0)(nrev); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_guru_dft, PLAN_GURU_DFT)(X(plan) *p, int *rank, const int *n, cannam@167: const int *is, const int *os, cannam@167: int *howmany_rank, const int *h_n, cannam@167: const int *h_is, const int *h_os, cannam@167: C *in, C *out, int *sign, int *flags) cannam@167: { cannam@167: X(iodim) *dims = make_dims(*rank, n, is, os); cannam@167: X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os); cannam@167: *p = X(plan_guru_dft)(*rank, dims, *howmany_rank, howmany_dims, cannam@167: in, out, *sign, *flags); cannam@167: X(ifree0)(howmany_dims); cannam@167: X(ifree0)(dims); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_guru_split_dft, PLAN_GURU_SPLIT_DFT)(X(plan) *p, int *rank, const int *n, cannam@167: const int *is, const int *os, cannam@167: int *howmany_rank, const int *h_n, cannam@167: const int *h_is, const int *h_os, cannam@167: R *ri, R *ii, R *ro, R *io, int *flags) cannam@167: { cannam@167: X(iodim) *dims = make_dims(*rank, n, is, os); cannam@167: X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os); cannam@167: *p = X(plan_guru_split_dft)(*rank, dims, *howmany_rank, howmany_dims, cannam@167: ri, ii, ro, io, *flags); cannam@167: X(ifree0)(howmany_dims); cannam@167: X(ifree0)(dims); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(execute_dft, EXECUTE_DFT)(X(plan) * const p, C *in, C *out) cannam@167: { cannam@167: plan_dft *pln = (plan_dft *) (*p)->pln; cannam@167: if ((*p)->sign == FFT_SIGN) cannam@167: pln->apply((plan *) pln, in[0], in[0]+1, out[0], out[0]+1); cannam@167: else cannam@167: pln->apply((plan *) pln, in[0]+1, in[0], out[0]+1, out[0]); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(execute_split_dft, EXECUTE_SPLIT_DFT)(X(plan) * const p, cannam@167: R *ri, R *ii, R *ro, R *io) cannam@167: { cannam@167: plan_dft *pln = (plan_dft *) (*p)->pln; cannam@167: pln->apply((plan *) pln, ri, ii, ro, io); cannam@167: } cannam@167: cannam@167: /****************************** DFT r2c *********************************/ cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_r2c, PLAN_DFT_R2C)(X(plan) *p, int *rank, const int *n, cannam@167: R *in, C *out, int *flags) cannam@167: { cannam@167: int *nrev = reverse_n(*rank, n); cannam@167: *p = X(plan_dft_r2c)(*rank, nrev, in, out, *flags); cannam@167: X(ifree0)(nrev); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_r2c_1d, PLAN_DFT_R2C_1D)(X(plan) *p, int *n, R *in, C *out, cannam@167: int *flags) cannam@167: { cannam@167: *p = X(plan_dft_r2c_1d)(*n, in, out, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_r2c_2d, PLAN_DFT_R2C_2D)(X(plan) *p, int *nx, int *ny, cannam@167: R *in, C *out, int *flags) cannam@167: { cannam@167: *p = X(plan_dft_r2c_2d)(*ny, *nx, in, out, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_r2c_3d, PLAN_DFT_R2C_3D)(X(plan) *p, cannam@167: int *nx, int *ny, int *nz, cannam@167: R *in, C *out, cannam@167: int *flags) cannam@167: { cannam@167: *p = X(plan_dft_r2c_3d)(*nz, *ny, *nx, in, out, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_many_dft_r2c, PLAN_MANY_DFT_R2C)( cannam@167: X(plan) *p, int *rank, const int *n, cannam@167: int *howmany, cannam@167: R *in, const int *inembed, int *istride, int *idist, cannam@167: C *out, const int *onembed, int *ostride, int *odist, cannam@167: int *flags) cannam@167: { cannam@167: int *nrev = reverse_n(*rank, n); cannam@167: int *inembedrev = reverse_n(*rank, inembed); cannam@167: int *onembedrev = reverse_n(*rank, onembed); cannam@167: *p = X(plan_many_dft_r2c)(*rank, nrev, *howmany, cannam@167: in, inembedrev, *istride, *idist, cannam@167: out, onembedrev, *ostride, *odist, cannam@167: *flags); cannam@167: X(ifree0)(onembedrev); cannam@167: X(ifree0)(inembedrev); cannam@167: X(ifree0)(nrev); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_guru_dft_r2c, PLAN_GURU_DFT_R2C)( cannam@167: X(plan) *p, int *rank, const int *n, cannam@167: const int *is, const int *os, cannam@167: int *howmany_rank, const int *h_n, cannam@167: const int *h_is, const int *h_os, cannam@167: R *in, C *out, int *flags) cannam@167: { cannam@167: X(iodim) *dims = make_dims(*rank, n, is, os); cannam@167: X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os); cannam@167: *p = X(plan_guru_dft_r2c)(*rank, dims, *howmany_rank, howmany_dims, cannam@167: in, out, *flags); cannam@167: X(ifree0)(howmany_dims); cannam@167: X(ifree0)(dims); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_guru_split_dft_r2c, PLAN_GURU_SPLIT_DFT_R2C)( cannam@167: X(plan) *p, int *rank, const int *n, cannam@167: const int *is, const int *os, cannam@167: int *howmany_rank, const int *h_n, cannam@167: const int *h_is, const int *h_os, cannam@167: R *in, R *ro, R *io, int *flags) cannam@167: { cannam@167: X(iodim) *dims = make_dims(*rank, n, is, os); cannam@167: X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os); cannam@167: *p = X(plan_guru_split_dft_r2c)(*rank, dims, *howmany_rank, howmany_dims, cannam@167: in, ro, io, *flags); cannam@167: X(ifree0)(howmany_dims); cannam@167: X(ifree0)(dims); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(execute_dft_r2c, EXECUTE_DFT_R2C)(X(plan) * const p, R *in, C *out) cannam@167: { cannam@167: plan_rdft2 *pln = (plan_rdft2 *) (*p)->pln; cannam@167: problem_rdft2 *prb = (problem_rdft2 *) (*p)->prb; cannam@167: pln->apply((plan *) pln, in, in + (prb->r1 - prb->r0), out[0], out[0]+1); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(execute_split_dft_r2c, EXECUTE_SPLIT_DFT_R2C)(X(plan) * const p, cannam@167: R *in, R *ro, R *io) cannam@167: { cannam@167: plan_rdft2 *pln = (plan_rdft2 *) (*p)->pln; cannam@167: problem_rdft2 *prb = (problem_rdft2 *) (*p)->prb; cannam@167: pln->apply((plan *) pln, in, in + (prb->r1 - prb->r0), ro, io); cannam@167: } cannam@167: cannam@167: /****************************** DFT c2r *********************************/ cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_c2r, PLAN_DFT_C2R)(X(plan) *p, int *rank, const int *n, cannam@167: C *in, R *out, int *flags) cannam@167: { cannam@167: int *nrev = reverse_n(*rank, n); cannam@167: *p = X(plan_dft_c2r)(*rank, nrev, in, out, *flags); cannam@167: X(ifree0)(nrev); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_c2r_1d, PLAN_DFT_C2R_1D)(X(plan) *p, int *n, C *in, R *out, cannam@167: int *flags) cannam@167: { cannam@167: *p = X(plan_dft_c2r_1d)(*n, in, out, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_c2r_2d, PLAN_DFT_C2R_2D)(X(plan) *p, int *nx, int *ny, cannam@167: C *in, R *out, int *flags) cannam@167: { cannam@167: *p = X(plan_dft_c2r_2d)(*ny, *nx, in, out, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_dft_c2r_3d, PLAN_DFT_C2R_3D)(X(plan) *p, cannam@167: int *nx, int *ny, int *nz, cannam@167: C *in, R *out, cannam@167: int *flags) cannam@167: { cannam@167: *p = X(plan_dft_c2r_3d)(*nz, *ny, *nx, in, out, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_many_dft_c2r, PLAN_MANY_DFT_C2R)( cannam@167: X(plan) *p, int *rank, const int *n, cannam@167: int *howmany, cannam@167: C *in, const int *inembed, int *istride, int *idist, cannam@167: R *out, const int *onembed, int *ostride, int *odist, cannam@167: int *flags) cannam@167: { cannam@167: int *nrev = reverse_n(*rank, n); cannam@167: int *inembedrev = reverse_n(*rank, inembed); cannam@167: int *onembedrev = reverse_n(*rank, onembed); cannam@167: *p = X(plan_many_dft_c2r)(*rank, nrev, *howmany, cannam@167: in, inembedrev, *istride, *idist, cannam@167: out, onembedrev, *ostride, *odist, cannam@167: *flags); cannam@167: X(ifree0)(onembedrev); cannam@167: X(ifree0)(inembedrev); cannam@167: X(ifree0)(nrev); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_guru_dft_c2r, PLAN_GURU_DFT_C2R)( cannam@167: X(plan) *p, int *rank, const int *n, cannam@167: const int *is, const int *os, cannam@167: int *howmany_rank, const int *h_n, cannam@167: const int *h_is, const int *h_os, cannam@167: C *in, R *out, int *flags) cannam@167: { cannam@167: X(iodim) *dims = make_dims(*rank, n, is, os); cannam@167: X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os); cannam@167: *p = X(plan_guru_dft_c2r)(*rank, dims, *howmany_rank, howmany_dims, cannam@167: in, out, *flags); cannam@167: X(ifree0)(howmany_dims); cannam@167: X(ifree0)(dims); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_guru_split_dft_c2r, PLAN_GURU_SPLIT_DFT_C2R)( cannam@167: X(plan) *p, int *rank, const int *n, cannam@167: const int *is, const int *os, cannam@167: int *howmany_rank, const int *h_n, cannam@167: const int *h_is, const int *h_os, cannam@167: R *ri, R *ii, R *out, int *flags) cannam@167: { cannam@167: X(iodim) *dims = make_dims(*rank, n, is, os); cannam@167: X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os); cannam@167: *p = X(plan_guru_split_dft_c2r)(*rank, dims, *howmany_rank, howmany_dims, cannam@167: ri, ii, out, *flags); cannam@167: X(ifree0)(howmany_dims); cannam@167: X(ifree0)(dims); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(execute_dft_c2r, EXECUTE_DFT_C2R)(X(plan) * const p, C *in, R *out) cannam@167: { cannam@167: plan_rdft2 *pln = (plan_rdft2 *) (*p)->pln; cannam@167: problem_rdft2 *prb = (problem_rdft2 *) (*p)->prb; cannam@167: pln->apply((plan *) pln, out, out + (prb->r1 - prb->r0), in[0], in[0]+1); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(execute_split_dft_c2r, EXECUTE_SPLIT_DFT_C2R)(X(plan) * const p, cannam@167: R *ri, R *ii, R *out) cannam@167: { cannam@167: plan_rdft2 *pln = (plan_rdft2 *) (*p)->pln; cannam@167: problem_rdft2 *prb = (problem_rdft2 *) (*p)->prb; cannam@167: pln->apply((plan *) pln, out, out + (prb->r1 - prb->r0), ri, ii); cannam@167: } cannam@167: cannam@167: /****************************** r2r *********************************/ cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_r2r, PLAN_R2R)(X(plan) *p, int *rank, const int *n, cannam@167: R *in, R *out, cannam@167: int *kind, int *flags) cannam@167: { cannam@167: int *nrev = reverse_n(*rank, n); cannam@167: X(r2r_kind) *k = ints2kinds(*rank, kind); cannam@167: *p = X(plan_r2r)(*rank, nrev, in, out, k, *flags); cannam@167: X(ifree0)(k); cannam@167: X(ifree0)(nrev); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_r2r_1d, PLAN_R2R_1D)(X(plan) *p, int *n, R *in, R *out, cannam@167: int *kind, int *flags) cannam@167: { cannam@167: *p = X(plan_r2r_1d)(*n, in, out, (X(r2r_kind)) *kind, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_r2r_2d, PLAN_R2R_2D)(X(plan) *p, int *nx, int *ny, cannam@167: R *in, R *out, cannam@167: int *kindx, int *kindy, int *flags) cannam@167: { cannam@167: *p = X(plan_r2r_2d)(*ny, *nx, in, out, cannam@167: (X(r2r_kind)) *kindy, (X(r2r_kind)) *kindx, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_r2r_3d, PLAN_R2R_3D)(X(plan) *p, cannam@167: int *nx, int *ny, int *nz, cannam@167: R *in, R *out, cannam@167: int *kindx, int *kindy, int *kindz, cannam@167: int *flags) cannam@167: { cannam@167: *p = X(plan_r2r_3d)(*nz, *ny, *nx, in, out, cannam@167: (X(r2r_kind)) *kindz, (X(r2r_kind)) *kindy, cannam@167: (X(r2r_kind)) *kindx, *flags); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_many_r2r, PLAN_MANY_R2R)( cannam@167: X(plan) *p, int *rank, const int *n, cannam@167: int *howmany, cannam@167: R *in, const int *inembed, int *istride, int *idist, cannam@167: R *out, const int *onembed, int *ostride, int *odist, cannam@167: int *kind, int *flags) cannam@167: { cannam@167: int *nrev = reverse_n(*rank, n); cannam@167: int *inembedrev = reverse_n(*rank, inembed); cannam@167: int *onembedrev = reverse_n(*rank, onembed); cannam@167: X(r2r_kind) *k = ints2kinds(*rank, kind); cannam@167: *p = X(plan_many_r2r)(*rank, nrev, *howmany, cannam@167: in, inembedrev, *istride, *idist, cannam@167: out, onembedrev, *ostride, *odist, cannam@167: k, *flags); cannam@167: X(ifree0)(k); cannam@167: X(ifree0)(onembedrev); cannam@167: X(ifree0)(inembedrev); cannam@167: X(ifree0)(nrev); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(plan_guru_r2r, PLAN_GURU_R2R)( cannam@167: X(plan) *p, int *rank, const int *n, cannam@167: const int *is, const int *os, cannam@167: int *howmany_rank, const int *h_n, cannam@167: const int *h_is, const int *h_os, cannam@167: R *in, R *out, int *kind, int *flags) cannam@167: { cannam@167: X(iodim) *dims = make_dims(*rank, n, is, os); cannam@167: X(iodim) *howmany_dims = make_dims(*howmany_rank, h_n, h_is, h_os); cannam@167: X(r2r_kind) *k = ints2kinds(*rank, kind); cannam@167: *p = X(plan_guru_r2r)(*rank, dims, *howmany_rank, howmany_dims, cannam@167: in, out, k, *flags); cannam@167: X(ifree0)(k); cannam@167: X(ifree0)(howmany_dims); cannam@167: X(ifree0)(dims); cannam@167: } cannam@167: cannam@167: FFTW_VOIDFUNC F77(execute_r2r, EXECUTE_R2R)(X(plan) * const p, R *in, R *out) cannam@167: { cannam@167: plan_rdft *pln = (plan_rdft *) (*p)->pln; cannam@167: pln->apply((plan *) pln, in, out); cannam@167: }