#include <stdio.h>
#include <stdlib.h>
#include <Fourier.h>
#include <cmdLineParser.h>
#include <Image.h>

template<class Real>
int ReadImage(char* fileName,SquareGrid<Real>& red,SquareGrid<Real>& green,SquareGrid<Real>& blue){
	char* ext=GetFileExtension(fileName);
	Image32 img;
	if(!strcasecmp(ext,"bmp")){BMPReadImage(fileName,img);}
	else if(!strcasecmp(ext,"jpg") || !strcasecmp(ext,"jpeg")){JPEGReadImage(fileName,img);}
	else{
		fprintf(stderr,"Unsupported image extension: %s\n",ext);
		delete[] ext;
		return 0;
	}
	delete[] ext;
	int w,h,r;
	w=img.width();
	h=img.height();
	r=(w>h)?w:h;

	red.resize(r);
	green.resize(r);
	blue.resize(r);

	for(int i=0;i<w;i++){
		for(int j=0;j<h;j++){
			Pixel32 p=img(i,j);
			red		(i,j)=Real(p.r)/Real(255.0);
			green	(i,j)=Real(p.g)/Real(255.0);
			blue	(i,j)=Real(p.b)/Real(255.0);
		}
	}
	return 1;
}
template<class Real>
int WriteImage(SquareGrid<Real>& red,SquareGrid<Real>& green,SquareGrid<Real>& blue,char* fileName){
	if(red.resolution()!=green.resolution() || red.resolution()!=blue.resolution()){
		fprintf(stderr,"Resolutions differ: red[%d] green[%d] blue[%d]\n",red.resolution(),green.resolution(),blue.resolution());
		return 0;
	}
	Image32 img;
	int r=red.resolution();
	img.setSize(r,r);

	for(int i=0;i<r;i++){
		for(int j=0;j<r;j++){
			Pixel32 p;
			Real r,g,b;
			r=red(i,j);
			g=green(i,j);
			b=blue(i,j);
			if(r<0){r=0;}
			if(r>1){r=1;}
			if(g<0){g=0;}
			if(g>1){g=1;}
			if(b<0){b=0;}
			if(b>1){b=1;}
			p.r=(unsigned char)(r*255.0);
			p.g=(unsigned char)(g*255.0);
			p.b=(unsigned char)(b*255.0);
			img(i,j)=p;
		}
	}

	char* ext=GetFileExtension(fileName);
	if(!strcasecmp(ext,"bmp")){BMPWriteImage(img,fileName);}
	else if(!strcasecmp(ext,"jpg") || !strcasecmp(ext,"jpeg")){JPEGWriteImage(img,fileName);}
	else{
		fprintf(stderr,"Unsupported image extension: %s\n",ext);
		delete[] ext;
		return 0;
	}
	delete[] ext;
	return 1;
}
int main(int argc,char* argv[]){
	// Read the parameters in from the command line
	cmdLineString In,Out,Extension;
	cmdLineInt Steps;
	cmdLineFloat Duration;
	char* paramNames[]={"in","out","steps","duration","extension"};
	cmdLineReadable* params[]={&In,&Out,&Steps,&Duration,&Extension};
	cmdLineParse(argc-1,&argv[1],paramNames,sizeof(paramNames)/sizeof(char*),params);
	if(!In.set || !Out.set || !Steps.set || !Duration.set || !Extension.set){
		fprintf(stderr,"Usage %s:\n",argv[0]);
		fprintf(stderr,"\t--in <input image>\n");
		fprintf(stderr,"\t--out <output image header>\n");
		fprintf(stderr,"\t--duration <duration of smoothing>\n");
		fprintf(stderr,"\t--steps <number of frames>\n");
		fprintf(stderr,"\t--extension <output file type>\n");
		return EXIT_FAILURE;
	}

	SquareGrid<> rGrid,bGrid,gGrid;
	FourierKey2D<> rKey,gKey,bKey,rTemp,bTemp,gTemp;
	FourierTransform<> xForm;

	// Read in the image and compute the Fourier transform of the different channels
	ReadImage(In.value,rGrid,gGrid,bGrid);
	xForm.ForwardFourier(rGrid,rKey);
	xForm.ForwardFourier(gGrid,gKey);
	xForm.ForwardFourier(bGrid,bKey);

	// Allocate space for the temporary Fourier coefficients
	rTemp.resize(rKey.resolution());
	gTemp.resize(gKey.resolution());
	bTemp.resize(bKey.resolution());
	for(int i=0;i<Steps.value;i++){
		float t=(Duration.value*i)/Steps.value;

		// Do the wave-propagation here, computing the values
		// of the Fourier coefficients at time t, and writing
		// them into the FourierKey2D objects (rTemp,gTemp,bTemp)
		// ...

		xForm.InverseFourier(rTemp,rGrid);
		xForm.InverseFourier(gTemp,gGrid);
		xForm.InverseFourier(bTemp,bGrid);

		// Output the frame for this time-step
		char outputName[512];
		sprintf(outputName,"%s.%03d.%s",Out.value,i,Extension.value);
		WriteImage(rGrid,gGrid,bGrid,outputName);
	}

	return EXIT_SUCCESS;
}