#include <stdio.h>
#include <stdlib.h>
#include "Fourier.h"
#include "SphericalGrid.h"
#include "cmdLineParser.h"
#include <math.h>

int main(int argc,char* argv[]){
	// Read the parameters in from the command line
	cmdLineString In,Out;
	cmdLineReadable Ref,Axial;
	cmdLineInt Rot;
	char* paramNames[]={"in","out","ref","rot","axial"};
	cmdLineReadable* params[]={&In,&Out,&Ref,&Rot,&Axial};
	cmdLineParse(argc-1,&argv[1],paramNames,sizeof(paramNames)/sizeof(char*),params);
	if(!In.set){
		fprintf(stderr,"You must specify an input file\n");
		return EXIT_FAILURE;
	}
	if(!Out.set){
		fprintf(stderr,"You must specify an output file\n");
		return EXIT_FAILURE;
	}

	char fileName[512];
	SphericalGrid<> in,sym;
	RotationGrid<> euler;
	FourierKeyS2<> keyS2;
	FourierKeySO3<> keySO3;
	HarmonicTransform<> hForm;
	WignerTransform<> wForm;
	float axis[3];
	float matrix[3][3];
	float x,y,z;

	// Read in the input spherical function
	if(!in.read(In.value)){
		fprintf(stderr,"Failed to read in: %s\n",In.value);
		return EXIT_FAILURE;
	}

	// Kill off the 0th order frequency component and normalize so that
	// the function has unit norm.
	hForm.ForwardFourier(in,keyS2);
	keyS2(0,0)=0;
	hForm.InverseFourier(keyS2,in);
	float l=float(sqrt(in.squareNorm()));
	for(int i=0;i<in.resolution();i++){for(int j=0;j<in.resolution();j++){in(i,j)/=l;}}

	// Allocate memory
	sym.resize(in.resolution());
	keySO3.resize(in.resolution());

	// Compute the spherical harmonic transform
	hForm.ForwardFourier(in,keyS2);

	if(Ref.set){
		// Do your reflective symmetry detection here, setting the reflective
		// symmetry descriptor values into the SphericalGrid "sym"
		// ...

		sprintf(fileName,"%s.ref.sgrid",Out.value);
		sym.write(fileName);
	}


	if(Rot.set || Axial.set){
		// Compute the correlation table here
		// ....

		if(Rot.set){
			for(int k=2;k<=Rot.value;k++){
				// Compute your k-fold symmetry descriptor here, setting the rotational
				// symmetry descriptor values into the SphericalGrid "sym"

				sprintf(fileName,"%s.rot.%d.sgrid",Out.value,k);
				sym.write(fileName);
			}
		}
		if(Axial.set){
			// Compute your axial symmetry descriptor here, setting the axial
			// symmetry descriptor values into the SphericalGrid "sym"

			sprintf(fileName,"%s.axial.sgrid",Out.value);
			sym.write(fileName);
		}
	}

	return EXIT_SUCCESS;
}