#include "Image_FFT.h"


void FFT_Shift(double * src,int size_w,int size_h){
    for(int i=0;i<size_w;i++){
        for(int j=0;j<size_h;j++){
            if((i+j)%2)
                src[i*size_w+j]=-src[i*size_w+j];
        }
        
    }
    
}
void ImageFFT(IplImage * src,Complex * dst){
    //FFT_Shift(src, src);
    if(src->depth!=IPL_DEPTH_8U)
        exit(0);
    int width=src->width;
    int height=src->height;
    double *image_data=(double*)malloc(sizeof(double)*width*height);
    for(int i=0;i<width;i++){
        for(int j=0;j<height;j++){
            image_data[i*width+j]=GETPIX(src, i, j);
        }
        
    }
    FFT_Shift(image_data,width, height);//图像中心化
    FFT2D(image_data, dst, width, height);
    free(image_data);
    
}
void Nomalsize(double *src,double *dst,int size_w,int size_h){
    double  max=0.0,min=DBL_MAX;
    for(int i=0;i<size_w*size_h;i++){
        max=src[i]>max?src[i]:max;
        min=src[i]<min?src[i]:min;
    }
    double step=255.0/(max-min);
    //printf("%d",test);
    printf("max:%lf   min:%lf\n",max,min);
    for(int i=0;i<size_w*size_h;i++){
        dst[i]=(src[i]-min)*step;
        dst[i]*=45.9*log((double)(1+dst[i]));
    }
    
}
void getAmplitudespectrum(Complex * src,int size_w,int size_h,IplImage *dst){
    double *despe=(double *)malloc(sizeof(double)*size_w*size_h);
    if(despe==NULL)
        exit(0);
    double real=0.0;
    double imagin=0.0;
    
    for(int i=0;i<size_w;i++)
        for(int j=0;j<size_h;j++){
            real=src[i*size_w+j].real;
            imagin=src[i*size_w+j].imagin;
            despe[i*size_w+j]=sqrt(real*real+imagin*imagin);
            
        }
    Nomalsize(despe, despe, size_w, size_h);
    for(int i=0;i<size_w;i++)
        for(int j=0;j<size_h;j++){
            cvSetReal2D(dst, i, j, despe[i*size_w+j]);
            
        }
    free(despe);
    
}

void ImageIFFT(Complex *src,IplImage *dst,int size_w,int size_h){
    Complex *temp_c=(Complex*)malloc(sizeof(Complex)*size_w*size_h);
    if(temp_c==NULL)
        exit(0);
    for(int i=0;i<size_w*size_h;i++)
        Copy_Complex(&src[i],&temp_c[i]);
    Complex *temp=(Complex*)malloc(sizeof(Complex)*size_w*size_h);
    if(temp==NULL)
        exit(0);
    double *temp_d=(double *)malloc(sizeof(double)*size_w*size_h);
    if(temp_d==NULL)
        exit(0);
    IFFT2D(temp_c,temp,size_w,size_h);
    for(int i=0;i<size_w;i++)
        for(int j=0;j<size_h;j++){
            temp_d[i*size_h+j]=temp[i*size_h+j].real;
        }
    FFT_Shift(temp_d, size_w, size_h);
    for(int i=0;i<size_w;i++)
        for(int j=0;j<size_h;j++){
            cvSetReal2D(dst, i, j, temp_d[i*size_h+j]);
        }
    free(temp);
    free(temp_c);
    free(temp_d);

}