1 /* dft.c - discrete fourier transformation program */
4 * author: frank.zirkelbach@physik.uni-augsburg.de
10 int main(int argc,char **argv) {
27 strcpy(src.file,argv[1]);
28 strcpy(dst.file,argv[2]);
30 fourier_init(&fourier,1);
36 bmp_cut_grab_bottom(&cut,&src,src.info.width,GRAB);
39 dst.height=cut.height;
42 fourier.data_len[0]=cut.width;
43 fourier.data_len[1]=cut.height;
44 fourier_alloc_data(&fourier);
46 mag=(double *)malloc(fourier.data_len[0]*fourier.data_len[1]*sizeof(double));
48 printf("unable to alloc mag memory\n");
54 for(y=0;y<fourier.data_len[1];y++) {
55 for(x=0;x<fourier.data_len[0];x++) {
57 fourier.data[offt].r=(src.map[offt].r+src.map[offt].g+src.map[offt].b)/3;
59 offy+=fourier.data_len[0];
62 // do the fourier trafo
63 fourier_dft_2d(&fourier);
65 printf("fourier done!\n");
67 // copy back the data, intensity = sqrt(r^2+i^2)
70 for(y=0;y<fourier.data_len[1];y++) {
71 for(x=0;x<fourier.data_len[0];x++) {
72 mag[offt]=sqrt(fourier.ftdata[offt].r*fourier.ftdata[offt].r+fourier.ftdata[offt].i*fourier.ftdata[offt].i);
73 if(mag[offt]>max) max=mag[offt];
78 printf("found max: %f\n",max);
80 if(max!=0) scale=(int)255/max;
84 printf("scaling image intensity: %d\n",scale);
86 for(y=0;y<dst.height;y++) {
87 for(x=0;x<dst.width;x++) {
88 // usual image processing order
89 offy=((y<dst.height/2)?y+dst.height/2:y-dst.height/2);
91 offy+=((x<dst.width/2)?x+dst.width/2:x-dst.width/2);
93 dst.map[offy].r=scale*mag[offt];
94 dst.map[offy].g=dst.map[offy].r;
95 dst.map[offy].b=dst.map[offy].r;
103 bmp_write_file(&dst);
105 fourier_shutdown(&fourier);
110 printf("done ...\n");