OpenCv中的离散傅立叶变换和逆变换

具体算法和源码，详细注意的地方在备注部分

#include "stdafx.h"
#include <iostream>

#include "./opencv2/opencv.hpp"

#if _DEBUG
#pragma comment(lib, "opencv_core2411d.lib")
#pragma comment(lib, "opencv_highgui2411d.lib")
#else
#pragma comment(lib, "opencv_core2411.lib")
#pragma comment(lib, "opencv_highgui2411.lib")
#endif

using namespace std;
using namespace cv;

int _tmain(int argc, _TCHAR* argv[])
{
	//构建数列
	Mat table(1, 10, CV_8U);
	uchar* p = table.data;
	for(int i=0; i<10; i++)
	{
		p[i] = i;
	}

	//打印测试数列
	cout<<table<<endl;

	Mat complexI;

	/*
	官方教程。
	因为生成的结果是分实部和虚部，原矩阵空间不足，通过merge方法，生成新的矩阵
	*/
	Mat planes[] = {Mat_<float>(table), Mat::zeros(table.size(), CV_32F)};
	merge(planes, 2, complexI);
	dft(complexI, complexI);

	/*
	//通过在dft中添加参数，可以指定生成的结果为复数，因此，输出的complexI为复数。
	Mat defInput;
	table.convertTo(defInput, CV_32F);
	dft(defInput, complexI, DFT_COMPLEX_OUTPUT);
	*/
	
	//拆分并打印生成的实部和虚部
	split(complexI, planes);

	Mat magI = planes[0];
	Mat magY = planes[1];

	cout<<magI<<endl;
	cout<<magY<<endl;

	//逆变换
	Mat invDFT, invDFTcvt;
	idft(complexI, invDFT, DFT_SCALE|DFT_REAL_OUTPUT );
	//dft(complexI, invDFT, DFT_INVERSE|DFT_SCALE|DFT_REAL_OUTPUT ); //通过指定 DFT_INVERSE 标志，可以计算dft的逆运算，和idft相同。DFT_REAL_OUTPUT该参数意思是将计算结果的实部取出
	invDFT.convertTo(invDFTcvt, CV_8U);

	cout<<invDFTcvt<<endl;
	return 0;
}