DCT.rar_二维 DCT 变换_图像DCT_图像DCT变换 c++

#include <memory.h> #include <stdio.h> #include <math.h> #include <time.h> #define PI 3.1415926 #define CLK_TCK CLOCKS_PER_SEC int N; char mode; void To_int(double *F,int *to_int)//四舍五入取整数 { int i; for(i=1;i<=N*N;i++) { if(F[i-1]>=0)to_int[i-1]=(int)(F[i-1]+0.5); else if(F[i-1]<0)to_int[i-1]=(int)(F[i-1]-0.5); } } void DCT(double *f,double *F) { int n,m,x; double *dTemp=new double[N*N];//中间矩阵 double *coff=new double[N*N];//变换系数 coff[0]=1/sqrt(N); for(m=1;m<N;m++) coff[m]=sqrt(2)/sqrt(N); memset(dTemp,0,sizeof(double)*N*N); memset(F,0,sizeof(double)*N*N); //一维变换 for(n=0;n<N;n++) for(m=0;m<N;m++) for(x=0;x<N;x++) dTemp[m*N+n]+=f[x*N+n]*coff[m]*cos((2*x+1)*PI*m/(2*N)); //第二次一维变换 for(m=0;m<N;m++) for(n=0;n<N;n++) for(x=0;x<N;x++) F[m*N+n]+=dTemp[m*N+x]*coff[n]*cos((2*x+1)*PI*n/(2*N)); delete []dTemp; delete []coff; } void iDCT(double *f,double *F) { int m,y,x; double *dTemp=new double[N*N];//中间矩阵 double *coff=new double[N*N];//变换系数 coff[0]=1/sqrt(N); for(m=1;m<N;m++) coff[m]=sqrt(2)/sqrt(N); memset(dTemp,0,sizeof(double)*N*N); memset(F,0,sizeof(double)*N*N); //一维变换 for(x=0;x<N;x++) for(y=0;y<N;y++) for(m=0;m<N;m++) dTemp[x*N+y]+=f[x*N+m]*coff[m]*cos((2*y+1)*PI*m/(2*N)); //第二次一维变换 for(y=0;y<N;y++) for(x=0;x<N;x++) for(m=0;m<N;m++) F[x*N+y]+=dTemp[m*N+y]*coff[m]*cos((2*x+1)*PI*m/(2*N)); delete []dTemp; delete []coff; } int main() { clock_t start,end; start=clock(); int i; long L; FILE* pOriData; FILE* pTranData; FILE* pItranData; printf("请将初始数据存放到OriData.txt文件\n选择变换模式：\n"); printf("A DCT正变换\tB DCT反变换\n"); scanf("%c",&mode); printf("变换维数:"); scanf("%d",&N); double *f=new double[N*N];//初始矩阵 double *F=new double[N*N];//变换后输出矩阵 int *to_int=new int[N*N];//存放取整后数据 memset(F,0,sizeof(double)*N*N);//初始化为0 memset(to_int,0,sizeof(int)*N*N); pOriData=fopen("OriData.txt","rt"); if(!pOriData) { printf("OriData.txt does not exist."); return 0; } for(i=0;i<N*N;i++) { fscanf(pOriData,"%lf,",&f[i]); } fclose(pOriData); printf("循环次数:"); scanf("%d",&L); //输出初始矩阵 printf("输入的矩阵如下:\n"); for(i=1;i<=N*N;i++) { printf("%f\t",f[i-1]); if(i%N==0) printf("\n"); } if(mode=='a') { for(i=0;i<L;i++) DCT(f,F);//变换 //输出变换后矩阵 pTranData=fopen("TranData.txt","wt"); if(!pTranData) { printf("Can't creat file!\n"); return 0; } printf("变换后数据存放:TranData.txt\n"); To_int(F,to_int);//取整数 for(i=1;i<=N*N;i++) { fprintf(pTranData,"%d\t",to_int[i-1]); if(i%N==0) fprintf(pTranData,"\n"); } fclose(pTranData); } else if(mode=='b') { for(i=0;i<L;i++) iDCT(f,F); //输出反变换后矩阵 pItranData=fopen("ItranData.txt","wt"); if(!pItranData) { printf("Can't creat file!\n"); return 0; } printf("反变换后数据存放:ItranData.txt\n"); To_int(F,to_int);//取整数 for(i=1;i<=N*N;i++) { fprintf(pItranData,"%d\t",to_int[i-1]); if(i%N==0) fprintf(pItranData,"\n"); } fclose(pItranData); } else { printf("You type a invalid mode!"); return 0; }//printf("\n"); delete []f; delete []F; end=clock(); printf("耗时:%f\n",(double)(end-start)/CLK_TCK); return 0; }