// Fourier.h: interface for the CFourier class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_FOURIER_H__28DCCEE4_216D_4552_BDF9_DC09A15A11A7__INCLUDED_)
#define AFX_FOURIER_H__28DCCEE4_216D_4552_BDF9_DC09A15A11A7__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000


#include <math.h>
#include <stdlib.h>
#include <stdio.h>
#include <COMPLEX>
#include <MEMORY.H>
using namespace std;
#define PI 3.1415926536

class CFourier  
{
public:
	CFourier();
	virtual ~CFourier();
public:
	long w;	//處理數據的寬度
	long h;	//處理數據的高度

	//1 dimension--------------------------------------
	//一維short數據fft
	BOOL Fourier1(short* TDdata,long lWidth,complex<float>*FDresult);
	//一維int數據fft
	BOOL Fourier1(int* TDdata,long lWidth,complex<float>*FDresult);
	//一維float數據fft
	BOOL Fourier1(float* TDdata,long lWidth,complex<float>*FDresult);
	//一維complex<float>數據fft
	BOOL Fourier1(complex<float>* TDdata,long lWidth,complex<float>*FDresult);
	//一維complex<float>數據ifft
	BOOL InFourier1(complex<float>*FDdata,long lWidth,complex<float>*TDresult);
	
	//2 dimension---------------------------------------
	//二維short數據fft
	BOOL Fourier2(short* TDdata,long lWidth,long lHeight,complex<float>* FDresult);
	//二維int數據fft
	BOOL Fourier2(int* TDdata,long lWidth,long lHeight,complex<float>*FDresult);
	//二維float數據fft
	BOOL Fourier2(float* TDdata,long lWidth,long lHeight,complex<float>*FDresult);
	//二維complex<float>數據fft
	BOOL Fourier2(complex<float>*TDdata,long lWidth,long lHeight,complex<float>*FDresult);
	//
	//二維short數據中的一維fft，當flag=1,只對列變換，flag=2，只對行變換
	BOOL Fourier2(short* TDdata,long lWidth,long lHeight,complex<float>*FDresult,int flag);
	//二維int數據中的一維fft，當flag=1,只對列變換，flag=2，只對行變換
	BOOL Fourier2(int* TDdata,long lWidth,long lHeight,complex<float>*FDresult,int flag);
	//二維float數據中的一維fft，當flag=1,只對列變換，flag=2，只對行變換
	BOOL Fourier2(float* TDdata,long lWidth,long lHeight,complex<float>*FDresult,int flag);
	//二維complex<float>數據中的一維fft，當flag=1,只對列變換，flag=2，只對行變換
	BOOL Fourier2(complex<float>*TDdata,long lWidth,long lHeight,complex<float>*FDresult,int flag);

	//二維complex<float>數據ifft
	BOOL InFourier2(complex<float>*FDdata,long lWidth,long lHeight,complex<float>*TDresult);
	BOOL InFourier2(complex<float>*FDdata,long lWidth,long lHeight);
	//二維complex<float>數據中的一維ifft
	BOOL InFourier2(complex<float>*FDdata,long lWidth,long lHeight,complex<float>*TDresult,int flag);
	
	//遷移函數
	void fftshift(complex<float>*Data,long lWidth,long lHeight,int flag);

protected:
	void FFT(complex<float>*TD,int r);
	void IFFT(complex<float>*FD,int r);
};

#endif // !defined(AFX_FOURIER_H__28DCCEE4_216D_4552_BDF9_DC09A15A11A7__INCLUDED_)