fft1d.h

来自「A good resource for 2D FFT and DCT」· C头文件 代码 · 共 44 行

// FFT1D.h: interface for the CFFT1D class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_FFT1D_H__66DC4545_6EB8_49D0_A399_872D00D6102C__INCLUDED_)
#define AFX_FFT1D_H__66DC4545_6EB8_49D0_A399_872D00D6102C__INCLUDED_

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

////////////////////////////////////////////////////
// FFT1D.h
////////////////////////////////////////////////////
#include "Complex.h"

class CFFT1D
{
private:
	unsigned int nDimension;
	CComplex* pInput;
	CComplex* pOutput;

	void BitReverseCopy();
	bool IsPowerOfTwo (unsigned int x);
	unsigned int NumberOfBitsNeeded ( unsigned int PowerOfTwo );
	unsigned int ReverseBits ( unsigned int index, unsigned int NumBits );

public:
	CFFT1D(unsigned int size = 2);
	~CFFT1D();

	CComplex GetOutputAt(unsigned int index);
	bool SetOutputAt(unsigned int index, double real = 0, double imag = 0);
	CComplex GetInputAt(unsigned int index);
	bool SetInputAt(unsigned int index, double real = 0, double imag = 0);

	void ForwardTransform();
	void ReverseTransform();
};


#endif // !defined(AFX_FFT1D_H__66DC4545_6EB8_49D0_A399_872D00D6102C__INCLUDED_)