labs / lab2 / fft / labdft

#include "mydft.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);	

#ifdef _MEX_DFT_
	if (!mydft(&x, &y))	
#elif _MEX_IDFT_
	if (!myidft(&x, &y))
#else
	mexErrMsgTxt("Recompile MEX with _MEX_DFT_ or _MEX_IDFT_ defined");
#endif
		mexErrMsgTxt("Input length must be power of 2");	
	
	MxArrayFromComplexArray(&plhs[0], &y);	

	return;
}