conv.cpp

来自「使用fft实现的快速卷积」· C++ 代码 · 共 35 行

// CONV.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <complex>
#include "CONV&CORR.h"
#include <iostream>
#include <vector>
using namespace std;
int main(int argc, char* argv[])
{
	printf("Hello World!\n");
	complex<double> TD1[10],TD2[3];
	vector<double> a,f;
	int M = 10;
	int N = 3;
	complex<double> TDout[12];
	int i;
	for(i = 0;i < 10;i++)
	{
		a.push_back(i);
		TD1[i] = complex<double>(a[i],0);
	}
	for(i = 0;i < 3;i++)
	{
		f.push_back(i);
		TD2[i] = complex<double>(f[i],0);
	}
	CONV(TD1,TD2,TDout,M,N);
	for(i = 0;i < 12;i++)
		cout<<TDout[i].real()<<"\t"<<TDout[i].imag()<<endl;
	return 0;
}