Скачиваний:
57
Добавлен:
16.04.2013
Размер:
743 б
Скачать
#include "myfft.h"

void mexFunction(int nlhs, mxArray *plhs[],
                 int nrhs, const mxArray *prhs[])
{	      
	int nx;
	ComplexArray x, y;

	if (nrhs != 1)
		mexErrMsgTxt("One input required");

	if (nlhs > 1)
		mexErrMsgTxt("Too many output arguments");
	
	if ( mxGetM(prhs[0]) != 1 )
		mexErrMsgTxt("Input must be row vector");		

	ComplexArrayFromMxArray(&x, prhs[0]);	

	ComplexArrayCreate(&y, x.count);

	ComplexArrayCopy(&x, &y);

#ifdef _MEX_FFT_
	if (!myfft(&y))	
#elif _MEX_IFFT_
	if (!myifft(&y))
#else
	mexErrMsgTxt("Recompile MEX with _MEX_FFT_ or _MEX_IFFT_ defined");
#endif
		mexErrMsgTxt("Input length must be power of 2");	
	
	MxArrayFromComplexArray(&plhs[0], &y);	

	return;
}
Соседние файлы в папке fft