testsub.c

几个FFT算法

#include "stdlib.h"
#include "stdio.h"
#include "malloc.h"
#include "memory.h"
#include "math.h"
#include "cmpl.h"

void  initTestData(CMPL data[],int len)
{
	int i;
	for (i=0;i<len;i++)
	{
		data[i].Re=(double)(rand() % 10000);
		data[i].Im=(double)(rand() % 10000);
	}

}

void print_FFTArray( CMPL data[], int n, FILE *fp)
{
	int i;
	for (i=0;i<n;i++)
	{
		fprintf(fp,"%.16f,%.16f\n",data[i].Re,data[i].Im);
	}
	fprintf(fp,"\n\n\n");
	return;
}

void dumpData(CMPL data[],int n, char *fileName)
{
	FILE *fp=fopen(fileName,"wb");
	int i;
	for (i=0;i<n;i++)
		fwrite(&(data[i]),sizeof(CMPL),1,fp);
	fclose(fp);
}

void readData(CMPL data[],int n,char *fileName)
{
	FILE *fp=fopen(fileName,"rb");
	int i;
	for (i=0;i<n;i++)
		fread(&(data[i]),sizeof(CMPL),1,fp);
	fclose(fp);

}
#define MIN_ERROR  1e-14

void CompResult(CMPL data1[],CMPL data2[],int n)
{
	double maxErr=0,minErr=0;
	double d;
	int i;
	for (i=0;i<n;i++)
	{
		if ( data2[i].Re!=0 )
		{
			d=(data1[i].Re - data2[i].Re )/data2[i].Re;
			if (d>0)
			{
				if (d>maxErr)
				{
					printf("data[%d].Re= f1=%.16f,f2=%.16f\n",i,data1[i].Re,data2[i].Re);
					maxErr=d;
				}
			}
			else if (d<0) 
			{
				if (d<minErr)
				{
					printf("data[%d].Re f1=%.16f,f2=%.16f\n",i,data1[i].Re,data2[i].Re);
					minErr=d;
				}
			}
		}
		if (data2[i].Im!=0)
		{
			d=(data1[i].Im - data2[i].Im )/data2[i].Im;
			if (d>0)
			{
				if (d>maxErr)
				{
					printf("data[%d].Im f1=%.16f,f2=%.16f\n",i,data1[i].Im,data2[i].Im);
					maxErr=d;
				}
			}
			else if (d<0)
			{
				if (d<minErr)
				{
					printf("data[%d].Im f1=%.16f,f2=%.16f\n",i,data1[i].Im,data2[i].Im);
					minErr=d;
				}

			}
		}
		
	}
	printf("max=%.16f,min=%.16f\n",maxErr,minErr);
}

void combineToCMPL(double ReArr[],double ImArr[],CMPL cmplArr[],int n)
{
	int i;
	for (i=0;i<n;i++)
	{
		cmplArr[i].Re=ReArr[i];
		cmplArr[i].Im=ImArr[i];
	}
}

void combineToCMPL2(double Arr[],CMPL cmplArr[],int n)
{
	int i;
	for (i=0;i<n;i++)
	{
		cmplArr[i].Re=Arr[2*i];
		cmplArr[i].Im=Arr[2*i+1];
	}
}

void ReArrageToDoubleArray(CMPL cmplArr[],double Arr[],int n)
{
	int i;
	for (i=0;i<n;i++)
	{
		Arr[i*2]=cmplArr[i].Re;
		Arr[i*2+1]=cmplArr[i].Im;
	}
}

void SplitToReImArray(CMPL cmplArr[],double ReArr[],double ImArr[],int n)
{
	int i;
	for (i=0;i<n;i++)
	{
		ReArr[i]=cmplArr[i].Re;
		ImArr[i]=cmplArr[i].Im;
	}
}