#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include <pict.h>
#ifdef THINK_C
#include <console.h>
#endif
#ifdef MYGETOPT
char getopt( int	argc, char	*argv[], char *format );
#endif

#include "cplx.h"

#define	FACTOR	((double)(IH*IW))

#define Boolean char
#define Output	if (G_prefs.Verbose) printf
#define IH	pictIn->image_H
#define IW	pictIn->image_W
#define IN	pictIn->n_comp

#define MIN(a,b) ((a)>(b)?(b):(a))
/*#define ctod(a) ((double)(a)/255.0)*/
#define ctod(a) ((double)(a))

typedef struct {
	char	*AppName;
	int	direct;
	int	Verbose;
	char	*outname;
	} pref;

pref    G_prefs;

void	fftpict(char *orgname,char *dstname);
void    fatalError(char *str);
void	usage(void);

void    fatalError(char *str)
{
    fprintf(stderr,
	"%s: Fatal Error: %s. bye...\n",G_prefs.AppName,str);
	    exit(-2);
}

void	usage(void)
{
	fprintf(stderr,"Uso: %s [-o output] [-i] file [file...]\n",
	G_prefs.AppName);
	exit(1);
}

main( int	argc, char	*argv[] )
{
	int	i,opcion;
	char	scname[256];
	int	theError=0;
	int	OutputOpt=0;
	
	extern char     *optarg;
	extern int      optind,opterr;


	G_prefs.AppName=argv[0];
	G_prefs.direct=1;
	G_prefs.Verbose=0;
	G_prefs.outname=NULL;

#ifdef THINK_C
	argc = ccommand(&argv);
#endif
	while ( (opcion=getopt(argc,argv,"hivo:")) != EOF )
	{
		switch(opcion)
		{
			case 'h':	usage();
					break;
			case 'i':	G_prefs.direct=0;
					break;
			case 'v':	G_prefs.Verbose=1;
					break;
			case 'o':	G_prefs.outname=optarg;
					break;
			defatult:	usage();
		}
	}

	if (strcmp("ifftp", argv[0])) G_prefs.direct=1;

	if (optind==argc) usage();

	for (i=optind; i<argc; i++)
	{

		if ( G_prefs.Verbose )
		{
			printf("%s:\t%s\t->\t%s\n",
			G_prefs.AppName,argv[i],
			G_prefs.outname==NULL?"stdout":G_prefs.outname);
			fflush(stdout);
		}
		
		fftpict(argv[i], G_prefs.outname);
	}
}

void	fftpict(char *orgname,char *dstname)
{
	Pict    pictIn;
	Pict    pictOut;
        int     ReadMode = P_READ;
        int     WriteMode = P_WRITE;

        int	mode,f,i,j, ic;
	int	gx, gy;
        U_char  *r_ptr,*g_ptr,*b_ptr,*a_ptr,*po;
	complex	*cimg[3], **pcimg[3]; 

	pictIn = pict_open(orgname,ReadMode,0,0,0,NULL);

	pictIn->pixels =
		(U_char *)malloc((size_t)IW*(size_t)pictIn->n_comp+1);

	if (pictIn->pixels==NULL) 
			fatalError("Se acabo la memoria");

	for(i=0; i<3; i++)
	{
		if((	cimg[i] =
		(complex *) calloc( (size_t)(IW*IH), sizeof( complex)))
		==NULL) 
			fatalError("Se acabo la memoria");
		if((	pcimg[i] =
		(complex **) calloc( (size_t)(IH), sizeof( complex*)))
		==NULL) 
			fatalError("Se acabo la memoria");
	
		for(j=0;j<IH; j++)
			pcimg[i][j]= cimg[i] + j*IW;
	}

	ic = 0;

	for (i=0; i<IH; i++)
	{
		a_ptr = pictIn->pixels;
		r_ptr = a_ptr+IW;
		g_ptr = r_ptr+IW;
		b_ptr = g_ptr+IW;
		
		pict_get_line(pictIn,pictIn->pixels);

		switch(pictIn->clr_mode)
		{
			case M_MONO:
			case M_RGB8:
				for (j=0; j<IW; j++, ic++)
				{
			Re(cimg[0][ic]) = ctod(pictIn->map[*a_ptr][0]);
			Re(cimg[1][ic]) = ctod(pictIn->map[*a_ptr][1]);
			Re(cimg[2][ic]) = ctod(pictIn->map[*a_ptr++][2]);
				}
				break;

			case M_RGB24:
				for (j=0; j<IW; j++, ic++)
				{
			Re(cimg[0][ic]) = ctod(*a_ptr++);
			Re(cimg[1][ic]) = ctod(*r_ptr++); 
			Re(cimg[2][ic]) = ctod(*g_ptr++);
				}
				break;
			case M_RGBA32:
				for (j=0; j<IW; j++, ic++)
				{
			Re(cimg[0][ic]) = ctod(*r_ptr++);
			Re(cimg[1][ic]) = ctod(*g_ptr++); 
			Re(cimg[2][ic]) = ctod(*b_ptr++);
				}
				break;
		}
	}


	gx = size2g( IW );
	gy = size2g( IH );


	initfft( gx, gy );

	Output("%s:\tfft %d x %d\n",G_prefs.AppName,  gx, gy);

	fft2d( pcimg[0], gx, gy);
	fft2d( pcimg[1], gx, gy);
	fft2d( pcimg[2], gx, gy);

	for(i=0; i< IH; i++)
	for(j=0; j< IW; j++)
	{
		Re(pcimg[0][i][j]) /= FACTOR;
		Im(pcimg[0][i][j]) /= FACTOR;
		Re(pcimg[1][i][j]) /= FACTOR;
		Im(pcimg[1][i][j]) /= FACTOR;
		Re(pcimg[2][i][j]) /= FACTOR;
		Im(pcimg[2][i][j]) /= FACTOR;
	}

	pict_close(pictIn);

	for(i=0; i< IH; i++)
	for(j=0; j< IW; j++)
	{
		Im(pcimg[0][i][j]) = -Im(pcimg[0][i][j]);
		Im(pcimg[1][i][j]) = -Im(pcimg[1][i][j]);
		Im(pcimg[2][i][j]) = -Im(pcimg[2][i][j]);
	}
	/* mierda */

/*
	for(i=0; i< IH/2; i++)
	for(j=0; j< IW/2; j++)
	{
		Re(pcimg[0][IH-1-i][j     ]) = Re(pcimg[0][i][j]);
		Re(pcimg[0][i     ][IW-1-j]) = Re(pcimg[0][i][j]);
		Re(pcimg[0][IH-1-i][IW-1-j]) = Re(pcimg[0][i][j]);
		Im(pcimg[0][IH-1-i][j     ]) = Im(pcimg[0][i][j]);
		Im(pcimg[0][i     ][IW-1-j]) = Im(pcimg[0][i][j]);
		Im(pcimg[0][IH-1-i][IW-1-j]) = Im(pcimg[0][i][j]);
		Re(pcimg[1][IH-1-i][j     ]) = Re(pcimg[1][i][j]);
		Re(pcimg[1][i     ][IW-1-j]) = Re(pcimg[1][i][j]);
		Re(pcimg[1][IH-1-i][IW-1-j]) = Re(pcimg[1][i][j]);
		Im(pcimg[1][IH-1-i][j     ]) = Im(pcimg[1][i][j]);
		Im(pcimg[1][i     ][IW-1-j]) = Im(pcimg[1][i][j]);
		Im(pcimg[1][IH-1-i][IW-1-j]) = Im(pcimg[1][i][j]);
		Re(pcimg[2][IH-1-i][j     ]) = Re(pcimg[2][i][j]);
		Re(pcimg[2][i     ][IW-1-j]) = Re(pcimg[2][i][j]);
		Re(pcimg[2][IH-1-i][IW-1-j]) = Re(pcimg[2][i][j]);
		Im(pcimg[2][IH-1-i][j     ]) = Im(pcimg[2][i][j]);
		Im(pcimg[2][i     ][IW-1-j]) = Im(pcimg[2][i][j]);
		Im(pcimg[2][IH-1-i][IW-1-j]) = Im(pcimg[2][i][j]);
	}
	*/

	for(i=1; i< IH; i+=2)
	for(j=1; j< IW; j+=2)
	{
		Re(pcimg[0][i][j]) = 0;
		Im(pcimg[0][i][j]) = 0;
		Re(pcimg[1][i][j]) = 0;
		Im(pcimg[1][i][j]) = 0;
		Re(pcimg[2][i][j]) = 0;
		Im(pcimg[2][i][j]) = 0;
	}


	fft2d( pcimg[0], gx, gy);
	fft2d( pcimg[1], gx, gy);
	fft2d( pcimg[2], gx, gy);

	mode = 3;
	pictOut = pict_open("fft.pict",WriteMode, IH,IW,mode,NULL);
        if ((pictOut->pixels =
                (U_char*)malloc(pictOut->image_W*pictOut->n_comp))==NULL)
                exit(1);

        for(j=0;j<IH;j++)
        {
                po=pictOut->pixels;
                for(i=0;i<IW;i++)
                       *po++ = (unsigned char)(Re(pcimg[0][j][i]));
                for(i=0;i<IW;i++)
                       *po++ = (unsigned char)(Re(pcimg[1][j][i]));
                for(i=0;i<IW;i++)
                       *po++ = (unsigned char)(Re(pcimg[2][j][i]));
                pict_put_line(pictOut,pictOut->pixels);
        }

	pict_close(pictOut);
	pict_free(pictIn);
}