/*
 * Copyright (c) 2003 Matteo Frigo
 * Copyright (c) 2003 Massachusetts Institute of Technology
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

/* header file for fftw3 */
/* $Id: fftw3.h,v 1.4 2006/12/21 11:25:15 glass Exp $ */

#ifndef FFTW3_H
#define FFTW3_H

#include <config.h>
#include <stdio.h>

#ifdef __cplusplus
/* In C++ mode prefer class complex<> */
#include <complex>
using std::complex;
#  define FFTW_DEFINE_COMPLEX(R, C) typedef complex<R> C
extern "C"
{
#else
/* If <complex.h> is included, use the C99 complex type.  Otherwise
   define a type bit-compatible with C99 complex */
#ifdef _Complex_I
#  define FFTW_DEFINE_COMPLEX(R, C) typedef R _Complex C
#else
#  define FFTW_DEFINE_COMPLEX(R, C) typedef R C[2]
#endif
#endif /* __cplusplus */


#define FFTW_CONCAT(prefix, name) prefix ## name
#define FFTW_MANGLE_DOUBLE(name) FFTW_CONCAT(fftw_, name)
#define FFTW_MANGLE_FLOAT(name) FFTW_CONCAT(fftwf_, name)
#define FFTW_MANGLE_LONG_DOUBLE(name) FFTW_CONCAT(fftwl_, name)


enum fftw_r2r_kind_do_not_use_me {
     FFTW_R2HC=0, FFTW_HC2R=1, FFTW_DHT=2,
     FFTW_REDFT00=3, FFTW_REDFT01=4, FFTW_REDFT10=5, FFTW_REDFT11=6,
     FFTW_RODFT00=7, FFTW_RODFT01=8, FFTW_RODFT10=9, FFTW_RODFT11=10
};

struct fftw_iodim_do_not_use_me {
     int n;                     /* dimension size */
     int is;			/* input stride */
     int os;			/* output stride */
};

/*
  huge second-order macro that defines prototypes for all API
  functions.  We expand this macro for each supported precision
 
  X: name-mangling macro
  R: real data type
  C: complex data type
*/

#define FFTW_DEFINE_API(X, R, C)					\
									\
FFTW_DEFINE_COMPLEX(R, C);						\
									\
typedef struct X(plan_s) *X(plan);					\
									\
typedef struct fftw_iodim_do_not_use_me X(iodim);			\
									\
typedef enum fftw_r2r_kind_do_not_use_me X(r2r_kind);			\
									\
void FFTEXP X(execute)(const X(plan) p);				\
									\
X(plan) FFTEXP X(plan_dft)(int rank, const int *n,			\
			   C *in, C *out, int sign, unsigned flags);	\
									\
X(plan) FFTEXP X(plan_dft_1d)(int n, C *in, C *out, int sign,		\
			      unsigned flags);				\
X(plan) FFTEXP X(plan_dft_2d)(int nx, int ny,				\
			      C *in, C *out, int sign, unsigned flags);	\
X(plan) FFTEXP X(plan_dft_3d)(int nx, int ny, int nz,			\
			      C *in, C *out, int sign, unsigned flags);	\
									\
X(plan) FFTEXP X(plan_many_dft)(int rank, const int *n,			\
	                        int howmany,				\
	                        C *in, const int *inembed,		\
	                        int istride, int idist,			\
	                        C *out, const int *onembed,		\
	                        int ostride, int odist,			\
	                        int sign, unsigned flags);		\
									\
X(plan) FFTEXP X(plan_guru_dft)(int rank, const X(iodim) *dims,		\
				int howmany_rank,			\
				const X(iodim) *howmany_dims,		\
				C *in, C *out,				\
				int sign, unsigned flags);		\
X(plan) FFTEXP X(plan_guru_split_dft)(int rank, const X(iodim) *dims,	\
				      int howmany_rank,			\
				      const X(iodim) *howmany_dims,	\
				      R *ri, R *ii, R *ro, R *io,	\
				      unsigned flags);			\
									\
void FFTEXP X(execute_dft)(const X(plan) p, C *in, C *out);		\
void FFTEXP X(execute_split_dft)(const X(plan) p, R *ri, R *ii,		\
				 R *ro, R *io);				\
									\
X(plan) FFTEXP X(plan_many_dft_r2c)(int rank, const int *n,		\
	                            int howmany,			\
        	                    R *in, const int *inembed,		\
	                            int istride, int idist,		\
	                            C *out, const int *onembed,		\
	                            int ostride, int odist,		\
	                            unsigned flags);			\
									\
X(plan) FFTEXP X(plan_dft_r2c)(int rank, const int *n,			\
	                       R *in, C *out, unsigned flags);		\
									\
X(plan) FFTEXP X(plan_dft_r2c_1d)(int n,R *in,C *out,unsigned flags);	\
X(plan) FFTEXP X(plan_dft_r2c_2d)(int nx, int ny,			\
				  R *in, C *out, unsigned flags);	\
X(plan) FFTEXP X(plan_dft_r2c_3d)(int nx, int ny,			\
				  int nz,				\
				  R *in, C *out, unsigned flags);	\
									\
									\
X(plan) FFTEXP X(plan_many_dft_c2r)(int rank, const int *n,		\
				    int howmany,			\
				    C *in, const int *inembed,		\
				    int istride, int idist,		\
				    R *out, const int *onembed,		\
				    int ostride, int odist,		\
				    unsigned flags);			\
									\
X(plan) FFTEXP X(plan_dft_c2r)(int rank, const int *n,			\
	                       C *in, R *out, unsigned flags);		\
									\
X(plan) FFTEXP X(plan_dft_c2r_1d)(int n,C *in,R *out,unsigned flags);	\
X(plan) FFTEXP X(plan_dft_c2r_2d)(int nx, int ny,			\
				  C *in, R *out, unsigned flags);	\
X(plan) FFTEXP X(plan_dft_c2r_3d)(int nx, int ny,			\
				  int nz,				\
				  C *in, R *out, unsigned flags);	\
									\
X(plan) FFTEXP X(plan_guru_dft_r2c)(int rank, const X(iodim) *dims,	\
				    int howmany_rank,			\
				    const X(iodim) *howmany_dims,	\
				    R *in, C *out,			\
				    unsigned flags);			\
X(plan) FFTEXP X(plan_guru_dft_c2r)(int rank, const X(iodim) *dims,	\
				    int howmany_rank,			\
				    const X(iodim) *howmany_dims,	\
				    C *in, R *out,			\
				    unsigned flags);			\
									\
X(plan) FFTEXP X(plan_guru_split_dft_r2c)(int rank, 			\
					  const X(iodim) *dims,		\
					  int howmany_rank,		\
					  const X(iodim) *howmany_dims,	\
					  R *in, R *ro, R *io,		\
					  unsigned flags);		\
X(plan) FFTEXP X(plan_guru_split_dft_c2r)(int rank,			\
				     const X(iodim) *dims,		\
				     int howmany_rank,			\
				     const X(iodim) *howmany_dims,	\
				     R *ri, R *ii, R *out,		\
				     unsigned flags);			\
									\
void FFTEXP X(execute_dft_r2c)(const X(plan) p, R *in, C *out);		\
void FFTEXP X(execute_dft_c2r)(const X(plan) p, C *in, R *out);		\
									\
void FFTEXP X(execute_split_dft_r2c)(const X(plan) p, R *in,		\
				     R *ro, R *io);			\
void FFTEXP X(execute_split_dft_c2r)(const X(plan) p, R *ri,		\
				     R *ii, R *out);			\
									\
X(plan) FFTEXP X(plan_many_r2r)(int rank, const int *n,			\
	                        int howmany,				\
        	                R *in, const int *inembed,		\
                	        int istride, int idist,			\
        	                R *out, const int *onembed,		\
	                        int ostride, int odist,			\
	                        const X(r2r_kind) *kind,		\
				unsigned flags);			\
									\
X(plan) FFTEXP X(plan_r2r)(int rank, const int *n, R *in, R *out,	\
	                   const X(r2r_kind) *kind, unsigned flags);	\
									\
X(plan) FFTEXP X(plan_r2r_1d)(int n, R *in, R *out,			\
	                      X(r2r_kind) kind, unsigned flags);	\
X(plan) FFTEXP X(plan_r2r_2d)(int nx, int ny, R *in, R *out,		\
	                      X(r2r_kind) kindx, X(r2r_kind) kindy,	\
        	              unsigned flags);				\
X(plan) FFTEXP X(plan_r2r_3d)(int nx, int ny, int nz,			\
	                      R *in, R *out, X(r2r_kind) kindx,		\
	                      X(r2r_kind) kindy, X(r2r_kind) kindz,	\
	                      unsigned flags);				\
									\
X(plan) FFTEXP X(plan_guru_r2r)(int rank, const X(iodim) *dims,		\
				int howmany_rank,			\
	                        const X(iodim) *howmany_dims,		\
        	                R *in, R *out,				\
	                        const X(r2r_kind) *kind,		\
				unsigned flags);			\
void FFTEXP X(execute_r2r)(const X(plan) p, R *in, R *out);		\
									\
void FFTEXP X(destroy_plan)(X(plan) p);					\
void FFTEXP X(forget_wisdom)(void);					\
void FFTEXP X(cleanup)(void);						\
									\
void FFTEXP X(plan_with_nthreads)(int nthreads);			\
int  FFTEXP X(init_threads)(void);					\
void FFTEXP X(cleanup_threads)(void);					\
									\
void FFTEXP X(export_wisdom_to_file)(FILE *output_file);		\
char* FFTEXP X(export_wisdom_to_string)(void);				\
void FFTEXP X(export_wisdom)(void (*write_char)(char c, void *),	\
			     void *data);				\
int FFTEXP X(import_system_wisdom)(void);				\
int FFTEXP X(import_wisdom_from_file)(FILE *input_file);		\
int FFTEXP X(import_wisdom_from_string)(const char *input_string);	\
int FFTEXP X(import_wisdom)(int (*read_char)(void *), void *data);	\
									\
void FFTEXP X(fprint_plan)(const X(plan) p, FILE *output_file);		\
void FFTEXP X(print_plan)(const X(plan) p);				\
									\
void* FFTEXP X(malloc)(size_t n);					\
void FFTEXP X(free)(void *p);						\
									\
void FFTEXP X(flops)(const X(plan) p, double *add,			\
		     double *mul, double *fma);				\
									\
extern const char X(version)[];						\
extern const char X(cc)[];						\
extern const char X(codelet_optim)[];

/* end of FFTW_DEFINE_API macro */

FFTW_DEFINE_API(FFTW_MANGLE_DOUBLE, double, fftw_complex)
FFTW_DEFINE_API(FFTW_MANGLE_FLOAT, float, fftwf_complex)
FFTW_DEFINE_API(FFTW_MANGLE_LONG_DOUBLE, long double, fftwl_complex)

#define FFTW_FORWARD (-1)
#define FFTW_BACKWARD (+1)

/* documented flags */
#define FFTW_MEASURE (0U)
#define FFTW_DESTROY_INPUT (1U << 0)
#define FFTW_UNALIGNED (1U << 1)
#define FFTW_CONSERVE_MEMORY (1U << 2)
#define FFTW_EXHAUSTIVE (1U << 3) /* NO_EXHAUSTIVE is default */
#define FFTW_PRESERVE_INPUT (1U << 4) /* cancels FFTW_DESTROY_INPUT */
#define FFTW_PATIENT (1U << 5) /* IMPATIENT is default */
#define FFTW_ESTIMATE (1U << 6)

/* undocumented beyond-guru flags */
#define FFTW_ESTIMATE_PATIENT (1U << 7)
#define FFTW_BELIEVE_PCOST (1U << 8)
#define FFTW_DFT_R2HC_ICKY (1U << 9)
#define FFTW_NONTHREADED_ICKY (1U << 10)
#define FFTW_NO_BUFFERING (1U << 11)
#define FFTW_NO_INDIRECT_OP (1U << 12)
#define FFTW_ALLOW_LARGE_GENERIC (1U << 13) /* NO_LARGE_GENERIC is default */
#define FFTW_NO_RANK_SPLITS (1U << 14)
#define FFTW_NO_VRANK_SPLITS (1U << 15)
#define FFTW_NO_VRECURSE (1U << 16)

#define FFTW_NO_SIMD (1U << 17)

#ifdef __cplusplus
}  /* extern "C" */
#endif /* __cplusplus */

#endif /* FFTW3_H */
