#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <time.h>
#include "../header/wavelib.h"

double absmax(double *array, int N) {
	double max;
	int i;

	max = 0.0;
	for (i = 0; i < N; ++i) {
		if (fabs(array[i]) >= max) {
			max = fabs(array[i]);
		}
	}

	return max;
}

double generate_rnd() {
	double rnd;

	rnd = (double) (rand() % 100 + 1);

	return rnd;
}

int main() {
	wave_object obj;
	wt2_object wt;
	int i, k, J, rows, cols,N;
	double *inp, *wavecoeffs,*oup,*diff;
	double *cLL;
	int ir, ic;
	double amax;

	rows = 32;
	cols = 30;
    N = rows*cols;

	char *name = "db2";
	obj = wave_init(name);// Initialize the wavelet
	srand(time(0));
	inp = (double*)calloc(N, sizeof(double));
	oup = (double*)calloc(N, sizeof(double));
	diff = (double*)calloc(N, sizeof(double));
    
	J = 3;

	wt = wt2_init(obj, "dwt", rows,cols, J);

	for (i = 0; i < rows; ++i) {
		for (k = 0; k < cols; ++k) {
			//inp[i*cols + k] = i*cols + k;
			inp[i*cols + k] = generate_rnd();
			oup[i*cols + k] = 0.0;
		}
	}

	wavecoeffs = dwt2(wt, inp);

	cLL = getWT2Coeffs(wt, wavecoeffs, 1, "D", &ir, &ic);

	dispWT2Coeffs(cLL, ir, ic);
	
	idwt2(wt, wavecoeffs, oup);

	for (i = 0; i < rows*cols; ++i) {
		diff[i] = oup[i] - inp[i];
	}

	amax = absmax(diff, rows*cols);

	wt2_summary(wt);

	printf("Abs Max %g \n", amax);

	wave_free(obj);
	wt2_free(wt);
	free(inp);
	free(wavecoeffs);
	free(oup);
	free(diff);
    return 0;
}