/* See LICENSE file for copyright and license details. */
#include "libclut.h"

#include <stddef.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>

struct clut {
	size_t red_size;
	size_t green_size;
	size_t blue_size;
	uint16_t *red;
	uint16_t *green;
	uint16_t *blue;
};

struct dclut {
	size_t red_size;
	size_t green_size;
	size_t blue_size;
	double *red;
	double *green;
	double *blue;
};

static int
clutcmp(const struct clut *a, const struct clut *b, uint16_t tol)
{
	size_t i;
	if (!tol)
		return memcmp(a->red, b->red, 3 * 256 * sizeof(uint16_t));
	for (i = 0; i < 3 * 256; i++) {
		if (a->red[i] > b->red[i]) {
			if (a->red[i] - b->red[i] > tol)
				return +1;
		} else if (a->red[i] < b->red[i]) {
			if (b->red[i] - a->red[i] > tol)
				return -1;
		}
	}
	return 0;
}

static int
dclutcmp(const struct dclut *a, const struct dclut *b, double tol)
{
	size_t i;
	for (i = 0; i < 3 * 256; i++) {
		if (a->red[i] > b->red[i]) {
			if (a->red[i] - b->red[i] > tol)
				return +1;
		} else if (a->red[i] < b->red[i]) {
			if (b->red[i] - a->red[i] > tol)
				return -1;
		}
	}
	return 0;
}

static void
dumpcluts(const struct clut *a, const struct clut *b, int tol)
{
	size_t i;
	uint16_t *r1 = a->red, *r2 = b->red, *g1 = a->green, *g2 = b->green, *b1 = a->blue, *b2 = b->blue;
	for (i = 0; i < 256; i++) {
		if ((tol >= 0) &&
		    (((r1[i] > r2[i]) ? (r1[i] - r2[i]) : (r2[i] - r1[i])) <= tol) &&
		    (((g1[i] > g2[i]) ? (g1[i] - g2[i]) : (g2[i] - g1[i])) <= tol) &&
		    (((b1[i] > b2[i]) ? (b1[i] - b2[i]) : (b2[i] - b1[i])) <= tol))
			continue;
		printf("%3zu (%02zx)  ::  %04x - %04x  ----  %04x - %04x  ----  %04x - %04x\n",
		       i, i, r1[i], r2[i], g1[i], g2[i], b1[i], b2[i]);
	}
}

static void
dumpdcluts(const struct dclut *a, const struct dclut *b, double tol)
{
	size_t i;
	double *r1 = a->red, *r2 = b->red, *g1 = a->green, *g2 = b->green, *b1 = a->blue, *b2 = b->blue;
	for (i = 0; i < 256; i++) {
		if ((tol >= 0) &&
		    (((r1[i] > r2[i]) ? (r1[i] - r2[i]) : (r2[i] - r1[i])) <= tol) &&
		    (((g1[i] > g2[i]) ? (g1[i] - g2[i]) : (g2[i] - g1[i])) <= tol) &&
		    (((b1[i] > b2[i]) ? (b1[i] - b2[i]) : (b2[i] - b1[i])) <= tol))
			continue;
		printf("%3zu (%02zx)  ::  %lf - %lf  ----  %lf - %lf  ----  %lf - %lf\n",
		       i, i, r1[i], r2[i], g1[i], g2[i], b1[i], b2[i]);
	}
}

static double
make_double(double x)
{
	return x * 2;
}


/**
 * Test libclut
 * 
 * @return  0: All tests passed
 *          1: At least one test fail
 *          2: An error occurred
 */
int
main(int argc, char *argv[])
{
#define HALF      ((double)5 / 10)
#define TENTHS(x) ((double)x / 10)

	libclut_colour_space_conversion_matrix_t M, Minv;
	libclut_rgb_colour_space_t srgb  = LIBCLUT_RGB_COLOUR_SPACE_SRGB_INITIALISER;
	libclut_rgb_colour_space_t wgrgb = LIBCLUT_RGB_COLOUR_SPACE_WIDE_GAMUT_RGB_INITIALISER;
	struct clut t1, t2, t3;
	struct dclut d1, d2;
	size_t i, j;
	int rc = 0;
	double param, r, g, b, x, y, z;

	t1.  red_size = t2.  red_size = t3.  red_size = d1.  red_size = d2.  red_size = 256;
	t1.green_size = t2.green_size = t3.green_size = d1.green_size = d2.green_size = 256;
	t1. blue_size = t2. blue_size = t3. blue_size = d1. blue_size = d2. blue_size = 256;
	if (!(t1.red = malloc(3 * 256 * sizeof(uint16_t))))  goto fail;
	if (!(t2.red = malloc(3 * 256 * sizeof(uint16_t))))  goto fail;
	if (!(t3.red = malloc(3 * 256 * sizeof(uint16_t))))  goto fail;
	if (!(d1.red = malloc(3 * 256 * sizeof(double))))  goto fail;
	if (!(d2.red = malloc(3 * 256 * sizeof(double))))  goto fail;
	t1.blue = (t1.green = t1.red + 256) + 256;
	t2.blue = (t2.green = t2.red + 256) + 256;
	t3.blue = (t3.green = t3.red + 256) + 256;
	d1.blue = (d1.green = d1.red + 256) + 256;
	d2.blue = (d2.green = d2.red + 256) + 256;

	libclut_start_over(&t1, UINT16_MAX, uint16_t, 1, 1, 1);
	for (i = 0; i < 256; i++)
		t2.blue[i] = t2.green[i] = t2.red[i] = (uint16_t)((i << 8) | i);
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_start_over failed\n"), rc = 1;

	for (i = 0, j = 255; i < 256; i++, j--) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)i;
		t2.blue[j] = t2.green[j] = t2.red[j] = (uint16_t)i;
	}
	libclut_negative(&t1, UINT16_MAX, uint16_t, 1, 1, 1);
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_negative failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)i;
		t2.blue[i] = t2.green[i] = t2.red[i] = (uint16_t)(UINT16_MAX - (uint16_t)i);
	}
	libclut_rgb_invert(&t1, UINT16_MAX, uint16_t, 1, 1, 1);
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_rgb_invert failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)((i << 8) | i);
		t2.blue[i] = (uint16_t)(1 + (t2.green[i] = (uint16_t)(1 + (t2.red[i] = (uint16_t)(1000 + i)))));
	}
	libclut_rgb_limits(&t1, UINT16_MAX, uint16_t, 1000, 1255, 1001, 1256, 1002, 1257);
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_rgb_limits failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)i;
		t2.red[i]   = (uint16_t)(t1.blue[i]  * 2);
		t2.green[i] = (uint16_t)(t1.green[i] * 3);
		t2.blue[i]  = (uint16_t)(t1.blue[i]  * 4);
	}
	libclut_rgb_brightness(&t1, UINT16_MAX, uint16_t, 2, 3, 4);
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_rgb_brightness failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)i;
		t2.blue[i] = t2.green[i] = (uint16_t)(2 * (t2.red[i] = (uint16_t)i));
	}
	libclut_manipulate(&t1, UINT16_MAX, uint16_t, (double (*)(double))(NULL), make_double, make_double);
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_manipulate failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)i;
		t2.blue[i] = t2.green[i] = t2.red[i] = (uint16_t)(pow((double)i / UINT16_MAX, (double)10 / 11) * UINT16_MAX);
	}
	libclut_gamma(&t1, UINT16_MAX, uint16_t, TENTHS(11), TENTHS(11), TENTHS(11));
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_gamma failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)((i << 8) | i);
		t2.blue[i] = t2.green[i] = t2.red[i] = (uint16_t)((t1.red[i] - UINT16_MAX / 2) / 2 + UINT16_MAX / 2);
	}
	libclut_rgb_contrast(&t1, UINT16_MAX, uint16_t, HALF, HALF, HALF);
	if (clutcmp(&t1, &t2, 1))
		printf("libclut_rgb_contrast failed\n"), rc = 1;

	param = 2;
	for (i = 0; i < 256; i++) {
		double x = (double)i / 255;
		if (i % 255) {
			x = 1 / x - 1;
			x = log(x);
			x = HALF - x / param;
		}
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)((i << 8) | i);
		t2.blue[i] = t2.green[i] = t2.red[i] = (uint16_t)(x * UINT16_MAX);
	}
	libclut_sigmoid(&t1, UINT16_MAX, uint16_t, &param, &param, &param);
	t1.blue[0]   = t1.green[0]   = t1.red[0]   = t2.red[0];
	t1.blue[255] = t1.green[255] = t1.red[255] = t2.red[255];
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_sigmoid failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)((i << 8) | i);
		t2.blue[i] = t2.green[i] = t2.red[i] = t1.red[i & 0xF0] | (t1.red[i & 0xF0] >> 4);
	}
	libclut_lower_resolution(&t1, UINT16_MAX, uint16_t, 16, 0, 16, 0, 16, 0);
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_lower_resolution (x) failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)((i << 8) | i);
		t2.blue[i] = t2.green[i] = t2.red[i] = (uint16_t)((uint16_t)((double)i / 255 * 15 + HALF) * UINT16_MAX / 15);
	}
	libclut_lower_resolution(&t1, UINT16_MAX, uint16_t, 0, 16, 0, 16, 0, 16);
	if (clutcmp(&t1, &t2, 0))
		printf("libclut_lower_resolution (y) failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)((i << 8) | i);
		t2.blue[i] = t2.green[i] = t2.red[i] = (uint16_t)((i << 8) | i);
	}
	libclut_linearise(&t1, UINT16_MAX, uint16_t, 1, 1, 1);
	if (!clutcmp(&t1, &t2, 1))
		printf("libclut_linearise failed\n"), rc = 1;
	libclut_standardise(&t1, UINT16_MAX, uint16_t, 1, 1, 1);
	if (clutcmp(&t1, &t2, 12))
		printf("libclut_linearise/libclut_standardise failed\n"), rc = 1;
	/* High error rate, especially at low values, are expected due
	 * to low precision and truncated values rather rounded values. */

	for (i = 0; i < 256; i++) {
		d1.blue[i] = d1.green[i] = d1.red[i] = (i & 1) ? -1 : 2;
		d2.blue[i] = d2.green[i] = d2.red[i] = (i & 1) ?  0 : 1;
	}
	libclut_clip(&d1, 1, double, 1, 1, 1);
	if (dclutcmp(&d1, &d2, 0))
		printf("libclut_clip failed\n"), rc = 1;

	for (i = 0; i < 256; i++) {
		t1.blue[i] = t1.green[i] = t1.red[i] = (uint16_t)(((255 - i) << 8) | (255 - i));
		t2.blue[i] = t2.green[i] = t2.red[i] = (uint16_t)(pow((double)i / 255, (double)10 / 11) * UINT16_MAX);
	}
	for (i = 0; i < 256; i++)
		t3.blue[i] = t3.green[i] = t3.red[i] = t2.red[255 - i];
	libclut_apply(&t1, UINT16_MAX, uint16_t, &t2, UINT16_MAX, uint16_t, 1, 1, 1);
	if (clutcmp(&t1, &t3, 0))
		printf("libclut_apply failed\n"), rc = 1;

	if (libclut_model_get_rgb_conversion_matrix(&srgb, &wgrgb, M, Minv)) {
		printf("libclut_model_get_rgb_conversion_matrix failed\n"), rc = 1;
		goto rgb_conversion_done;
	}
	if (0.5587 > M[0][0] || M[0][0] > 0.5589 ||
	    0.3587 > M[0][1] || M[0][1] > 0.3588 ||
	    -.0101 > M[0][2] || M[0][2] > -.0099 ||
	    0.0938 > M[1][0] || M[1][0] > 0.0939 ||
	    0.8564 > M[1][1] || M[1][1] > 0.8566 ||
	    0.0746 > M[1][2] || M[1][2] > 0.0747 ||
	    0.0186 > M[2][0] || M[2][0] > 0.0188 ||
	    0.0967 > M[2][1] || M[2][1] > 0.0968 ||
	    1.2237 > M[2][2] || M[2][2] > 1.2239) {
		printf("libclut_model_get_rgb_conversion_matrix failed\n"), rc = 1;
		goto rgb_conversion_done;
	}

	libclut_model_convert_rgb(0.1, 0.5, 0.9, M, &r, &g, &b);
	if (0.3024 > r || r > 0.3025 ||
	    0.5301 > g || g > 0.5302 ||
	    0.9931 > b || b > 0.9932) {
		printf("libclut_model_convert_rgb failed\n"), rc = 1;
		goto rgb_conversion_done;
	}

	libclut_model_convert_rgb(0.1, 0.5, 0.9, Minv, &r, &g, &b);
	if (-1.3341 > r || r > -1.3339 ||
	     0.4916 > g || g >  0.4917 ||
	     0.8144 > b || b >  0.8145) {
		printf("libclut_model_convert_rgb failed\n"), rc = 1;
		goto rgb_conversion_done;
	}

	libclut_convert_rgb_inplace(&t1, 1, double, M, trunc); /* TODO test */
	libclut_convert_rgb(&t1, 1, double, M, trunc, &t2);    /* TODO test */

	if (libclut_model_get_rgb_conversion_matrix(&srgb, NULL, M, Minv)) {
		printf("libclut_model_get_rgb_conversion_matrix failed\n"), rc = 1;
		goto rgb_conversion_done;
	}
	libclut_model_rgb_to_ciexyz(0.1, 0.5, 0.9, M, &x, &y, &z);
	if (0.2227 > x || x > 0.2228 ||
	    0.2120 > y || y > 0.2121 ||
	    0.7739 > z || z > 0.7741) {
		printf("libclut_model_get_rgb_conversion_matrix or libclut_model_rgb_to_ciexyz failed\n"), rc = 1;
		goto rgb_conversion_done;
	}
	libclut_model_ciexyz_to_rgb(x, y, z, Minv, &r, &g, &b);
	if (0.0999 > r || r > 1.0001 ||
	    0.4999 > g || g > 0.5001 ||
	    0.8999 > b || b > 0.9001) {
		printf("libclut_model_get_rgb_conversion_matrix or libclut_model_ciexyz_to_rgb failed\n"), rc = 1;
		goto rgb_conversion_done;
	}

	if (libclut_model_get_rgb_conversion_matrix(NULL, &srgb, Minv, M)) {
		printf("libclut_model_get_rgb_conversion_matrix failed\n"), rc = 1;
		goto rgb_conversion_done;
	}
	libclut_model_rgb_to_ciexyz(0.1, 0.5, 0.9, M, &r, &g, &b);
	libclut_model_ciexyz_to_rgb(r, g, b, Minv, &r, &g, &b);
	if (0.0999 > r || r > 1.0001 ||
	    0.4999 > g || g > 0.5001 ||
	    0.8999 > b || b > 0.9001) {
		printf("libclut_model_get_rgb_conversion_matrix failed\n"), rc = 1;
		goto rgb_conversion_done;
	}

	if (libclut_model_get_rgb_conversion_matrix(NULL, NULL, M, Minv)) {
		printf("libclut_model_get_rgb_conversion_matrix failed\n"), rc = 1;
		goto rgb_conversion_done;
	}
	if (0.999999 > M[0][0] || M[0][0] > 1.000001 ||
	    -.999999 > M[0][1] || M[0][1] > 0.000001 ||
	    -.999999 > M[0][2] || M[0][2] > 0.000001 ||
	    -.999999 > M[1][0] || M[1][0] > 0.000001 ||
	    0.999999 > M[1][1] || M[1][1] > 1.000001 ||
	    -.999999 > M[1][2] || M[1][2] > 0.000001 ||
	    -.999999 > M[2][0] || M[2][0] > 0.000001 ||
	    -.999999 > M[2][1] || M[2][1] > 0.000001 ||
	    0.999999 > M[2][2] || M[2][2] > 1.000001) {
		printf("libclut_model_get_rgb_conversion_matrix failed\n"), rc = 1;
		goto rgb_conversion_done;
	}
	if (0.999999 > Minv[0][0] || Minv[0][0] > 1.000001 ||
	    -.999999 > Minv[0][1] || Minv[0][1] > 0.000001 ||
	    -.999999 > Minv[0][2] || Minv[0][2] > 0.000001 ||
	    -.999999 > Minv[1][0] || Minv[1][0] > 0.000001 ||
	    0.999999 > Minv[1][1] || Minv[1][1] > 1.000001 ||
	    -.999999 > Minv[1][2] || Minv[1][2] > 0.000001 ||
	    -.999999 > Minv[2][0] || Minv[2][0] > 0.000001 ||
	    -.999999 > Minv[2][1] || Minv[2][1] > 0.000001 ||
	    0.999999 > Minv[2][2] || Minv[2][2] > 1.000001) {
		printf("libclut_model_get_rgb_conversion_matrix failed\n"), rc = 1;
		goto rgb_conversion_done;
	}

	libclut_model_get_rgb_conversion_matrix(&srgb, &wgrgb, M, NULL); /* Just testing that we don't get a segfault. */
rgb_conversion_done:

	libclut_model_ciexyz_to_cieluv(0.4, 1.0, 0.7, 0.33, 1, 0.32, &x, &y, &z); /* TODO test */
	libclut_model_cieluv_to_ciexyz(x, y, z, 0.33, 1, 0.32, &x, &y, &z);
	if (0.3999 > x || x > 0.4001 ||
	    0.9999 > y || y > 1.0001 ||
	    0.6999 > z || z > 0.7001)
		printf("libclut_model_cieluv_to_ciexyz failed\n"), rc = 1;

	libclut_model_cieluv_to_cielch(0.5, 0.6, &r, &g); /* TODO test */
	libclut_model_cielch_to_cieluv(r, g, &r, &g);
	if (0.4999 > r || r > 0.5001 ||
	    0.5999 > g || g > 0.6001)
		printf("libclut_model_cielch_to_cieluv failed\n"), rc = 1;

	libclut_model_srgb_to_yiq(0.1, 0.6, 0.9, &r, &g, &b); /* TODO test */
	libclut_model_yiq_to_srgb(r, g, b, &r, &g, &b);
	if (0.0999 > r || r > 0.1001 ||
	    0.5999 > g || g > 0.6001 ||
	    0.8999 > b || b > 0.9001)
		printf("libclut_model_yiq_to_srgb failed\n"), rc = 1;

	libclut_model_srgb_to_ydbdr(0.1, 0.6, 0.9, &r, &g, &b); /* TODO test */
	libclut_model_ydbdr_to_srgb(r, g, b, &r, &g, &b);
	if (0.0999 > r || r > 0.1001 ||
	    0.5999 > g || g > 0.6001 ||
	    0.8999 > b || b > 0.9001)
		printf("libclut_model_ydbdr_to_srgb failed\n"), rc = 1;

	libclut_model_ydbdr_to_yuv(0.1, 0.6, &r, &g); /* TODO test */
	libclut_model_yuv_to_ydbdr(r, g, &r, &g);
	if (0.0999 > r || r > 0.1001 ||
	    0.5999 > g || g > 0.6001)
		printf("libclut_model_yuv_to_ydbdr failed\n"), rc = 1;

	libclut_model_srgb_to_ypbpr(0.1, 0.6, 0.9, &r, &g, &b); /* TODO test */
	libclut_model_ypbpr_to_srgb(r, g, b, &r, &g, &b);
	if (0.0999 > r || r > 0.1001 ||
	    0.5999 > g || g > 0.6001 ||
	    0.8999 > b || b > 0.9001)
		printf("libclut_model_ypbpr_to_srgb failed\n"), rc = 1;

	libclut_model_srgb_to_ycgco(0.1, 0.6, 0.9, &r, &g, &b); /* TODO test */
	libclut_model_ycgco_to_srgb(r, g, b, &r, &g, &b);
	if (0.0999 > r || r > 0.1001 ||
	    0.5999 > g || g > 0.6001 ||
	    0.8999 > b || b > 0.9001)
		printf("libclut_model_ycgco_to_srgb failed\n"), rc = 1;

	libclut_model_ciexyz_to_cie_1960_ucs(0.4, 0.7, 0.6, &x, &y, &z); /* TODO test */
	libclut_model_cie_1960_ucs_to_ciexyz(x, y, z, &x, &y, &z);
	if (0.3999 > x || x > 0.4001 ||
	    0.6999 > y || y > 0.7001 ||
	    0.5999 > z || z > 0.6001)
		printf("libclut_model_cie_1960_ucs_to_ciexyz failed\n"), rc = 1;

	libclut_model_cie_1960_ucs_to_cieuvw(0.1, 0.7, 0.9, 0.3, 0.4, &x, &y, &z); /* TODO test */
	libclut_model_cieuvw_to_cie_1960_ucs(x, y, z, 0.3, 0.4, &x, &y, &z);
	if (0.0999 > x || x > 0.1001 ||
	    0.6999 > y || y > 0.7001 ||
	    0.8999 > z || z > 0.9001)
		printf("libclut_model_cieuvw_to_cie_1960_ucs failed\n"), rc = 1;

	if (!rc)
		printf("everything is fine\n");
	free(t1.red);
	free(t2.red);
	free(t3.red);
	free(d1.red);
	free(d2.red);
	return rc;
 fail:
	perror(*argv);
	return 2;
	(void) argc;
	(void) dumpcluts;
	(void) dumpdcluts;
}

/*
  TODO test these too: (also add this to the makefile)
  
  libclut_translate
  libclut_cie_contrast
  libclut_cie_brightness
  libclut_cie_invert
  libclut_cie_limits
  libclut_cie_manipulate
  libclut_cie_apply
*/