/*
 * Copyright © 2019 Advanced Micro Devices, Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice (including the next
 * paragraph) shall be included in all copies or substantial portions of the
 * Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
 * IN THE SOFTWARE.
 */

#include "piglit-util-gl.h"

PIGLIT_GL_TEST_CONFIG_BEGIN
	config.supports_gl_compat_version = 21;

	config.window_visual = PIGLIT_GL_VISUAL_RGBA;
	config.khr_no_error_support = PIGLIT_HAS_ERRORS;

PIGLIT_GL_TEST_CONFIG_END

static int max_texture_coords;

/* Compressed formats from arb_direct_state_access/compressedtextureimage.c */
struct format {
	GLenum token;
	const char **extension;
};

static const struct format *format;

static const char *FXT1[] = {
	"GL_3DFX_texture_compression_FXT1",
	NULL
};

static const char *S3TC[] = {
	"GL_EXT_texture_compression_s3tc",
	NULL
};

static const char *S3TC_srgb[] = {
	"GL_EXT_texture_compression_s3tc",
	"GL_EXT_texture_sRGB",
	NULL
};

static const char *RGTC[] = {
	"GL_ARB_texture_compression_rgtc",
	NULL
};

static const char *RGTC_signed[] = {
	"GL_ARB_texture_compression_rgtc",
	"GL_EXT_texture_snorm",
	NULL
};

static const char *BPTC[] = {
	"GL_ARB_texture_compression_bptc",
	NULL
};

static const struct format formats[] = {
	{ GL_COMPRESSED_RGB_FXT1_3DFX, FXT1 },
	{ GL_COMPRESSED_RGBA_FXT1_3DFX, FXT1 },

	{ GL_COMPRESSED_RGB_S3TC_DXT1_EXT, S3TC },
	{ GL_COMPRESSED_RGBA_S3TC_DXT1_EXT, S3TC },
	{ GL_COMPRESSED_RGBA_S3TC_DXT3_EXT, S3TC },
	{ GL_COMPRESSED_RGBA_S3TC_DXT5_EXT, S3TC },

	{ GL_COMPRESSED_SRGB_S3TC_DXT1_EXT, S3TC_srgb },
	{ GL_COMPRESSED_SRGB_ALPHA_S3TC_DXT1_EXT, S3TC_srgb },
	{ GL_COMPRESSED_SRGB_ALPHA_S3TC_DXT3_EXT, S3TC_srgb },
	{ GL_COMPRESSED_SRGB_ALPHA_S3TC_DXT5_EXT, S3TC_srgb },

	{ GL_COMPRESSED_RGBA_BPTC_UNORM, BPTC },
	{ GL_COMPRESSED_SRGB_ALPHA_BPTC_UNORM, BPTC },
	{ GL_COMPRESSED_RGB_BPTC_UNSIGNED_FLOAT, BPTC },
	{ GL_COMPRESSED_RGB_BPTC_SIGNED_FLOAT, BPTC },

	{ GL_COMPRESSED_RED_RGTC1_EXT, RGTC },
	{ GL_COMPRESSED_SIGNED_RED_RGTC1_EXT, RGTC_signed },
	{ GL_COMPRESSED_RED_GREEN_RGTC2_EXT, RGTC },
	{ GL_COMPRESSED_SIGNED_RED_GREEN_RGTC2_EXT, RGTC_signed },
};

static void
usage(int argc, char **argv)
{
	int i;

	fprintf(stderr, "Usage: %s <format>\n", argv[0]);
	fprintf(stderr, "format is one of:\n");
	for (i = 0; i < ARRAY_SIZE(formats); i++) {
		fprintf(stderr, "  %s\n",
			piglit_get_gl_enum_name(formats[i].token));
	}
	exit(1);
}

void
piglit_init(int argc, char **argv)
{
	int i;
	GLenum arg;
	if (argc != 2)
		usage(argc, argv);

	format = NULL;

	arg = piglit_get_gl_enum_from_name(argv[1]);
	for (i = 0; i < ARRAY_SIZE(formats); i++) {
		if (formats[i].token == arg) {
			format = &formats[i];
			break;
		}
	}

	if (!format)
		usage(argc, argv);

	for (i = 0; format->extension[i]; i++)
		piglit_require_extension(format->extension[i]);

	if (format->token == GL_COMPRESSED_RGB_BPTC_SIGNED_FLOAT)
		piglit_set_tolerance_for_bits(7, 7, 7, 7);

	piglit_require_extension("GL_EXT_direct_state_access");
	glGetIntegerv(GL_MAX_TEXTURE_COORDS, &max_texture_coords);
}

/* Returns n (<= 4) different texunits */
static GLenum*
n_texunits(int n)
{
	static GLenum out[4];
	int i, j;
	assert (n <= 4);
	for (i = 0; i < n; i++) {
		out[i] = rand() % max_texture_coords;
		if (n <= max_texture_coords) {
			for (j = 0; j < i; j++) {
				if (out[i] == out[j]) {
					/* Reset i */
					i--;
					break;
				}
			}
		}
	}
	for (i = 0; i < n; i++) {
		out[i] += GL_TEXTURE0;
	}
	return out;
}

static GLenum
dimension_to_target(int n)
{
	assert(n == 1 || n == 2 || n == 3);
	switch (n) {
		case 1: return GL_TEXTURE_1D;
		case 2: return GL_TEXTURE_2D;
		case 3:
		default:
			return GL_TEXTURE_3D;
	}
}

static GLenum use_display_list = GL_NONE;
static GLuint list;

static bool
init_compressed_texture(GLenum texunit, GLenum target, int* compressed_size,
			void** compressed, float** expected_pixels)
{
	int i;
	GLuint tex;
	const int height = (target == GL_TEXTURE_1D) ? 1 : piglit_height;
	const int depth = (target == GL_TEXTURE_3D) ? 2 : 1;
	int save_active_texture;
	float* image = piglit_rgbw_image(GL_RGBA, piglit_width, height * depth,
					 false, GL_UNSIGNED_NORMALIZED);

	if (!expected_pixels) {
		for (i = 0; i < piglit_width * height * depth * 4; i++) {
			image[i] = (float) rand() / RAND_MAX;
		}
	}

	glGenTextures(1, &tex);
	glBindMultiTextureEXT(texunit, target, tex);

	glMultiTexParameteriEXT(texunit, target, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
	glMultiTexParameteriEXT(texunit, target, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
	glMultiTexParameteriEXT(texunit, target, GL_TEXTURE_WRAP_R, GL_CLAMP_TO_EDGE);
	glMultiTexParameteriEXT(texunit, target, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
	glMultiTexParameteriEXT(texunit, target, GL_TEXTURE_MIN_FILTER,GL_NEAREST);

	if (target == GL_TEXTURE_1D) {
		glMultiTexImage1DEXT(texunit, target, 0, format->token,
				     piglit_width, 0,
				     GL_RGBA, GL_FLOAT, image);
	} else if (target == GL_TEXTURE_2D) {
		glMultiTexImage2DEXT(texunit, target, 0, format->token,
				     piglit_width, height, 0,
				     GL_RGBA, GL_FLOAT, image);
	} else {
		/* 2 layers 3D image */
		glMultiTexImage3DEXT(texunit, target, 0, format->token,
				     piglit_width, height, depth, 0,
				     GL_RGBA, GL_FLOAT, image);
	}

	/* An error here probably means format->token isn't supported, so skip the test */
	if (!piglit_check_gl_error(GL_NO_ERROR)) {
		glDeleteTextures(1, &tex);
		free(image);
		return false;
	}

	if (expected_pixels) {
		*expected_pixels = image;
	} else {
		free(image);
	}
	*compressed_size = piglit_compressed_image_size(format->token, piglit_width, height * depth);
	*compressed = malloc(*compressed_size);

	glGetIntegerv(GL_ACTIVE_TEXTURE, &save_active_texture);
	glActiveTexture(texunit);
	glGetCompressedTexImage(target, 0, *compressed);
	glActiveTexture(save_active_texture);

	glDeleteTextures(1, &tex);

	piglit_check_gl_error(GL_NO_ERROR);
	return true;
}

enum piglit_result
test_MultiCompressedTexImageNDEXT(void* data)
{
	const int n = (int)(intptr_t) data;
	const GLenum target = dimension_to_target(n);
	void *expected_compressed;
	float *compressed, *expected_pixels;
	int compressed_size;
	bool pass = true;
	const int depth = (target == GL_TEXTURE_3D) ? 2 : 1;
	const int height = (target == GL_TEXTURE_1D) ? 1 : piglit_height;
	GLenum* texunits = n_texunits(3);
	GLuint tex;

	/* Make sure ActiveTexture is set to something different */
	glActiveTexture(texunits[2]);

	if (!init_compressed_texture(texunits[1],
				     target,
				     &compressed_size,
				     &expected_compressed,
				     &expected_pixels)) {
		return PIGLIT_SKIP;
	}

	glGenTextures(1, &tex);
	glBindMultiTextureEXT(texunits[0], target, tex);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_WRAP_R, GL_CLAMP_TO_EDGE);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_MIN_FILTER,GL_NEAREST);

	if (use_display_list != GL_NONE)
		glNewList(list, use_display_list);

	if (target == GL_TEXTURE_1D) {
		glCompressedMultiTexImage1DEXT(texunits[0], GL_TEXTURE_1D, 0, format->token,
					       piglit_width, 0,
					       compressed_size, expected_compressed);
	} else if (target == GL_TEXTURE_2D) {
		glCompressedMultiTexImage2DEXT(texunits[0], GL_TEXTURE_2D, 0, format->token,
					       piglit_width, height, 0,
					       compressed_size, expected_compressed);
	} else {
		glCompressedMultiTexImage3DEXT(texunits[0], GL_TEXTURE_3D, 0, format->token,
					       piglit_width, height, depth, 0,
					       compressed_size, expected_compressed);
	}

	if (use_display_list != GL_NONE)
		glEndList(list);

	if (use_display_list == GL_COMPILE) {
		int width;
	        glGetTextureLevelParameterivEXT(tex, target, 0,
                                		GL_TEXTURE_WIDTH, &width);
		pass = width == 0 && pass;
		glCallList(list);
	}

	/* Test GetCompressedMultiTexImageEXT */
	compressed = (float*) malloc(compressed_size);
	glGetCompressedMultiTexImageEXT(texunits[0], target, 0, compressed);


	pass = memcmp(compressed, expected_compressed, compressed_size) == 0 && pass;
	free(compressed);

	/* Draw texture */
	glActiveTexture(GL_TEXTURE0);
	glBindTexture(target, tex);
	glEnable(target);
	piglit_draw_rect_tex(-1.0, -1.0, 2.0, 2.0, 0.0, 0.0, 1.0, 1.0);
	glDisable(target);

	pass = piglit_check_gl_error(GL_NO_ERROR) && pass;

	/* Check to make sure the image was drawn correctly */
	pass = piglit_probe_image_rgba(0, 0,
				      piglit_width,
				      (n == 1) ? 1 : piglit_height,
				      expected_pixels) && pass;

	glDeleteTextures(1, &tex);

	free(expected_pixels);
	free(expected_compressed);

	piglit_check_gl_error(GL_NO_ERROR);

	return pass ? PIGLIT_PASS : PIGLIT_FAIL;
}

enum piglit_result
test_MultiCompressedTexSubImageNDEXT(void* data)
{
	const int n = (int)(intptr_t) data;
	const GLenum target = dimension_to_target(n);
	void *expected_compressed;
	float *expected_pixels;
	int compressed_size;
	GLuint tex;
	bool pass = true;
	const int depth = (target == GL_TEXTURE_3D) ? 2 : 1;
	const int height = (target == GL_TEXTURE_1D) ? 1 : piglit_height;
	GLenum* texunits = n_texunits(3);

	/* Make sure ActiveTexture is set to something different */
	glActiveTexture(texunits[2]);

	/* Create a first source texture, with random content */
	if (!init_compressed_texture(texunits[1],
				     target,
				     &compressed_size,
				     &expected_compressed,
				     NULL /* random content */)) {
		return PIGLIT_SKIP;
	}

	glGenTextures(1, &tex);
	glBindMultiTextureEXT(texunits[0], target, tex);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_WRAP_R, GL_CLAMP_TO_EDGE);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
	glMultiTexParameteriEXT(texunits[0], target, GL_TEXTURE_MIN_FILTER,GL_NEAREST);

	if (target == GL_TEXTURE_1D) {
		glCompressedMultiTexImage1DEXT(texunits[0], GL_TEXTURE_1D, 0, format->token,
					       piglit_width, 0,
					       compressed_size, expected_compressed);
	} else if (target == GL_TEXTURE_2D) {
		glCompressedMultiTexImage2DEXT(texunits[0], GL_TEXTURE_2D, 0, format->token,
					       piglit_width, height, 0,
					       compressed_size, expected_compressed);
	} else {
		glCompressedMultiTexImage3DEXT(texunits[0], GL_TEXTURE_3D, 0, format->token,
					       piglit_width, height, depth, 0,
					       compressed_size, expected_compressed);
	}
	free(expected_compressed);

	/* Then create a second one with rgbw content */
	init_compressed_texture(texunits[1],
				target,
				&compressed_size,
				&expected_compressed,
				&expected_pixels);

	if (use_display_list != GL_NONE)
		glNewList(list, use_display_list);

	/* And update our compressed texture using glCompressedTextureSubImage */
	if (target == GL_TEXTURE_1D) {
		glCompressedMultiTexSubImage1DEXT(texunits[0], GL_TEXTURE_1D, 0, 0,
						  piglit_width, format->token,
						  compressed_size, expected_compressed);
	} else if (target == GL_TEXTURE_2D) {
		glCompressedMultiTexSubImage2DEXT(texunits[0], GL_TEXTURE_2D, 0, 0, 0,
						  piglit_width, height, format->token,
						  compressed_size, expected_compressed);
	} else {
		glCompressedMultiTexSubImage3DEXT(texunits[0], GL_TEXTURE_3D, 0, 0, 0, 0,
						  piglit_width, height, depth, format->token,
						  compressed_size, expected_compressed);
	}

	if (use_display_list != GL_NONE)
		glEndList(list);
	if (use_display_list == GL_COMPILE)
		glCallList(list);

	/* Draw texture */
	glActiveTexture(GL_TEXTURE0);
	glEnable(target);
	glBindTexture(target, tex);
	piglit_draw_rect_tex(-1.0, -1.0, 2.0, 2.0, 0.0, 0.0, 1.0, 1.0);
	glDisable(target);
	pass = piglit_check_gl_error(GL_NO_ERROR) && pass;

	/* Check to make sure the image was drawn correctly */
	pass = piglit_probe_image_rgba(0, 0,
				       piglit_width, (n == 1) ? 1 : piglit_height,
				       expected_pixels) && pass;
	glDeleteTextures(1, &tex);

	free(expected_pixels);
	free(expected_compressed);

	return pass ? PIGLIT_PASS : PIGLIT_FAIL;
}

enum piglit_result
piglit_display(void)
{
	int i;
	/* Not testing 1D variants since no compression format
	   supports 1D textures (see _mesa_target_can_be_compressed) */
	struct piglit_subtest tests[] = {
		{
			"CompressedTextureImage3DEXT",
			NULL,
			test_MultiCompressedTexImageNDEXT,
			(void*) 3
		},
		{
			"CompressedTextureImage2DEXT",
			NULL,
			test_MultiCompressedTexImageNDEXT,
			(void*) 2
		},
		{
			"CompressedTextureSubImage3DEXT",
			NULL,
			test_MultiCompressedTexSubImageNDEXT,
			(void*) 3
		},
		{
			"CompressedTextureSubImage2DEXT",
			NULL,
			test_MultiCompressedTexSubImageNDEXT,
			(void*) 2
		},
		{
			NULL
		}
	};

	enum piglit_result result = piglit_run_selected_subtests(tests, NULL, 0, PIGLIT_PASS);
	list = glGenLists(1);

	/* Re-run the same test but using display list GL_COMPILE */
	for (i = 0; tests[i].name; i++) {
		char* test_name_display_list;
		asprintf(&test_name_display_list, "%s + display list GL_COMPILE", tests[i].name);
		tests[i].name = test_name_display_list;
	}
	use_display_list = GL_COMPILE;
	result = piglit_run_selected_subtests(tests, NULL, 0, result);

	/* Re-run the same test but using display list GL_COMPILE_AND_EXECUTE */
	for (i = 0; tests[i].name; i++) {
		char* test_name_display_list;
		asprintf(&test_name_display_list, "%s_AND_EXECUTE", tests[i].name);
		tests[i].name = test_name_display_list;
	}
	use_display_list = GL_COMPILE_AND_EXECUTE;
	result = piglit_run_selected_subtests(tests, NULL, 0, result);

	glDeleteLists(list, 1);

	return result;
}