#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,Pattern,Out;
	char* paramNames[]={"in","out","pattern"};
	cmdLineReadable* params[]={&In,&Out,&Pattern};
	cmdLineParse(argc-1,&argv[1],paramNames,sizeof(paramNames)/sizeof(char*),params);
	if(!In.set || !Out.set || !Pattern.set){
		fprintf(stderr,"Usage %s:\n",argv[0]);
		fprintf(stderr,"\t--pattern <input pattern image>\n");
		fprintf(stderr,"\t--in <input image>\n");
		fprintf(stderr,"\t--out <output pattern detectionimage>\n");
		return EXIT_FAILURE;
	}

	int res,pRes;
	SquareGrid<> input[3],pattern[3],mask;

	// Read in the red, green, and blue components of the image
	if(!ReadImage(In.value,input[0],input[1],input[2])){return EXIT_FAILURE;}
	res=input[0].resolution();

	// Read in the red, green, and blue components of the pattern
	SquareGrid<> temp[3];
	if(!ReadImage(Pattern.value,temp[0],temp[1],temp[2])){return EXIT_FAILURE;}
	pRes=temp[0].resolution();

	// Check that the pattern is smaller than the image
	if(res<=pRes){
		fprintf(stderr,"Pattern must be smaller than image: %d < %d\n",pRes,res);
		return EXIT_FAILURE;
	}

	// Copy the pattern to a re-centered, zero-padded image, and set the mask
	pattern[0].resize(res);
	pattern[1].resize(res);
	pattern[2].resize(res);
	mask.resize(res);
	for(int i=0;i<pRes;i++){
		for(int j=0;j<pRes;j++){
			pattern[0](i-pRes/2,j-pRes/2)=temp[0](i,j);
			pattern[1](i-pRes/2,j-pRes/2)=temp[1](i,j);
			pattern[2](i-pRes/2,j-pRes/2)=temp[2](i,j);
			mask(i-pRes/2,j-pRes/2)=1;
		}
	}

	// Do your pattern matching here
	// ...

	// The average value of (x-y)*(x-y) with x,y in [0,1] is 1/6.
	// Multiplying this by the square norm of the mask gives us the
	// expected square L2-difference of two random signals on the mask.
	// We divide the moving square L2-difference by this value to get a
	// function that should roughly be in the range [0,1]
	float scale=float(mask.squareNorm()/6);

	// Normalize by the expected error
	for(int i=0;i<res;i++){
		for(int j=0;j<res;j++){
			input[0](i,j)/=scale;
			input[1](i,j)/=scale;
			input[2](i,j)/=scale;
		}
	}

	// Invert the image so that brighter spots correspond to better matches
	for(int i=0;i<res;i++){
		for(int j=0;j<res;j++){
			input[0](i,j)=1.f-input[0](i,j);
			input[1](i,j)=1.f-input[1](i,j);
			input[2](i,j)=1.f-input[2](i,j);
		}
	}

	// Write the image out to file
	WriteImage(input[0],input[1],input[2],Out.value);


	return EXIT_SUCCESS;
}