#include <stdbool.h>
#include <tiffio.h>
#include "greatest.h"
#include "libdng.h"

static enum greatest_test_res
check_str_tag(TIFF *im, uint32_t tag, const char *name, const char *expected)
{
	char *temp;
	if (TIFFGetField(im, tag, &temp) != 1) {
			FAILm(name);
	}
		ASSERT_STR_EQm(name, expected, temp);
		PASS();
}

static enum greatest_test_res
check_int_tag(TIFF *im, uint32_t tag, const char *name, int expected)
{
	uint32_t temp = 0;
	if (TIFFGetField(im, tag, &temp) != 1) {
			FAILm(name);
	}
		ASSERT_EQ_FMTm(name, expected, temp, "%d");
		PASS();
}

static enum greatest_test_res
check_float_tag(TIFF *im, uint32_t tag, const char *name, float expected)
{
	float temp;
	if (TIFFGetField(im, tag, &temp) != 1) {
			FAILm(name);
	}
		ASSERT_EQ_FMTm(name, expected, temp, "%f");
		PASS();
}


TEST generate_simple_dng(void)
{
	libdng_info info = {0};
	libdng_new(&info);
		ASSERT_EQm("Set mode", 1, libdng_set_mode_from_name(&info, "RGGB"));
		ASSERT_EQm("Set make", 1, libdng_set_make_model(&info, "Make", "Model"));
		ASSERT_EQm("Set software", 1, libdng_set_software(&info, "Software"));
		ASSERT_EQm("Set orientation", 1, libdng_set_orientation(&info, 4));
		ASSERT_EQm("Set exposuretime", 1, libdng_set_exposure_time(&info, 0.04f));
		ASSERT_EQm("Set fnumber", 1, libdng_set_fnumber(&info, 2.8f));
		ASSERT_EQm("Set fnumber", 1, libdng_set_focal_length(&info, 50.0f, 1.5f));
		ASSERT_EQm("Set framerate", 1, libdng_set_frame_rate(&info, 30.0f));
		ASSERT_EQm("Set distortion", 1, libdng_set_distortion(&info, 1.0f, 2.0f, 3.0f));
		ASSERT_EQm("Set vignette", 1, libdng_set_vignette(&info, 1.0f, -1.0f, 6.0f));
		ASSERT_EQm("Set color matrix 1", 1, libdng_set_color_matrix_1(&info, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f));
		ASSERT_EQm("Set color matrix 2", 1, libdng_set_color_matrix_2(&info, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f));
		ASSERT_EQm("Set forward matrix 1", 1, libdng_set_forward_matrix_1(&info, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f));
		ASSERT_EQm("Set forward matrix 2", 1, libdng_set_forward_matrix_2(&info, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f));
	uint8_t *data = malloc(1280 * 720);
		ASSERT_EQm("Write DNG", 1, libdng_write(&info, "test.dng", 1280, 720, data, 1280 * 720));
	free(data);
	libdng_free(&info);

	// Use LibTIFF directly to verify results
	TIFF *im = TIFFOpen("test.dng", "r");
	if (im == NULL) {
			FAILm("Could not open result");
	}

	toff_t exif_offset;
	if (TIFFGetField(im, TIFFTAG_EXIFIFD, &exif_offset) != 1) {
			FAILm("Could not find EXIF data");
	}

	// Check IFD0 with most metadata and the thumbnail image
		CHECK_CALL(check_int_tag(im, TIFFTAG_ORIENTATION, "ORIENTATION", 4));
		CHECK_CALL(check_int_tag(im, TIFFTAG_BITSPERSAMPLE, "THUMB_BPS", 8));
		CHECK_CALL(check_int_tag(im, TIFFTAG_SAMPLESPERPIXEL, "THUMB_CHANNELS", 3));
		CHECK_CALL(check_int_tag(im, TIFFTAG_PHOTOMETRIC, "THUMB_PHOTOMETRIC", PHOTOMETRIC_RGB));

		CHECK_CALL(check_str_tag(im, TIFFTAG_MAKE, "MAKE", "Make"));
		CHECK_CALL(check_str_tag(im, TIFFTAG_MODEL, "MODEL", "Model"));
		CHECK_CALL(check_str_tag(im, TIFFTAG_UNIQUECAMERAMODEL, "UCM", "Make Model"));
		CHECK_CALL(check_str_tag(im, TIFFTAG_SOFTWARE, "SOFTWARE", "Software"));
		CHECK_CALL(check_int_tag(im, TIFFTAG_ORIENTATION, "ORIENTATION", 4));

	// Check the custom tags
#define MPTAG_VERSION 49982
#define MPTAG_DISTORTION 49983
#define MPTAG_VIGNETTE 49984
	unsigned short count;
	float *fvalues;
	uint8_t *u8values;

	if (TIFFGetField(im, MPTAG_VERSION, &count, &u8values) != 1) {
			FAILm("Could not read MPTAG_VERSION");
	}
		ASSERT_EQ_FMTm("MPTAG_VERSION count", 4, count, "%d");
		ASSERT_EQ_FMTm("MPTAG_VERSION[0]", 1, u8values[0], "%d");
		ASSERT_EQ_FMTm("MPTAG_VERSION[1]", 0, u8values[1], "%d");
		ASSERT_EQ_FMTm("MPTAG_VERSION[2]", 0, u8values[2], "%d");
		ASSERT_EQ_FMTm("MPTAG_VERSION[3]", 0, u8values[3], "%d");

	if (TIFFGetField(im, MPTAG_DISTORTION, &fvalues) != 1) {
			FAILm("Could not read MPTAG_DISTORTION");
	}
		ASSERT_EQ_FMTm("MPTAG_DISTORTION[0]", 1.0f, fvalues[0], "%f");
		ASSERT_EQ_FMTm("MPTAG_DISTORTION[1]", 2.0f, fvalues[1], "%f");
		ASSERT_EQ_FMTm("MPTAG_DISTORTION[2]", 3.0f, fvalues[2], "%f");

	if (TIFFGetField(im, MPTAG_VIGNETTE, &fvalues) != 1) {
			FAILm("Could not read MPTAG_VIGNETTE");
	}
		ASSERT_EQ_FMTm("MPTAG_VIGNETTE[0]", 1.0f, fvalues[0], "%f");
		ASSERT_EQ_FMTm("MPTAG_VIGNETTE[1]", -1.0f, fvalues[1], "%f");
		ASSERT_EQ_FMTm("MPTAG_VIGNETTE[2]", 6.0f, fvalues[2], "%f");

	// Switch to IFD1 which has the raw data
	int subifd_count = 0;
	void *ptr;
	toff_t subifd_offsets[2];
	TIFFGetField(im, TIFFTAG_SUBIFD, &subifd_count, &ptr);
	memcpy(subifd_offsets, ptr, subifd_count * sizeof(subifd_offsets[0]));
	TIFFSetSubDirectory(im, subifd_offsets[0]);

	// Check IFD1 metadata
		CHECK_CALL(check_int_tag(im, TIFFTAG_IMAGEWIDTH, "RAW_WIDTH", 1280));
		CHECK_CALL(check_int_tag(im, TIFFTAG_IMAGELENGTH, "RAW_HEIGHT", 720));
		CHECK_CALL(check_int_tag(im, TIFFTAG_BITSPERSAMPLE, "RAW_BPS", 8));
		CHECK_CALL(check_int_tag(im, TIFFTAG_SAMPLESPERPIXEL, "RAW_CHANNELS", 1));
		CHECK_CALL(check_int_tag(im, TIFFTAG_PHOTOMETRIC, "RAW_PHOTOMETRIC", PHOTOMETRIC_CFA));

	// Switch to the EXIF block with the generic picture metadata
	TIFFReadEXIFDirectory(im, exif_offset);
		CHECK_CALL(check_float_tag(im, EXIFTAG_EXPOSURETIME, "EXPOSURETIME", 0.04f));
		CHECK_CALL(check_float_tag(im, EXIFTAG_FNUMBER, "FNUMBER", 2.8f));
		CHECK_CALL(check_float_tag(im, EXIFTAG_FOCALLENGTH, "FOCALLENGTH", 50.0f));
		CHECK_CALL(check_int_tag(im, EXIFTAG_FOCALLENGTHIN35MMFILM, "FOCALLENGTHIN35MMFILM", 75));
		PASS();
}

TEST read_dng(void)
{
	// Generate test timestamp
	struct tm testtime;
	strptime("2024-08-27 01:02:03", "%Y-%m-%d %H:%M:%S", &testtime);

	// Generate a file to read
	libdng_info info = {0};
	libdng_new(&info);
	libdng_set_datetime(&info, testtime);
	libdng_set_mode_from_name(&info, "RGGB");
	libdng_set_make_model(&info, "Make", "Model");
	libdng_set_software(&info, "Software");
	libdng_set_orientation(&info, LIBDNG_ORIENTATION_RIGHTBOT);
	libdng_set_exposure_time(&info, 1.4f);
	libdng_set_exposure_program(&info, LIBDNG_EXPOSUREPROGRAM_PORTRAIT);
	libdng_set_fnumber(&info, 1.8f);
	libdng_set_focal_length(&info, 75.0f, 1.6f);

	uint8_t *data = malloc(1280 * 720);
	libdng_write(&info, "test.dng", 1280, 720, data, 1280 * 720);
	free(data);
	libdng_free(&info);

	// Read the file again
	libdng_info dng = {0};
	libdng_read(&dng, "test.dng");
		ASSERT_EQm("Bit depth", 8, dng.bit_depth);
		ASSERT_EQm("CFFA[0]", 0, dng.cfapattern[0]);
		ASSERT_EQm("CFFA[1]", 1, dng.cfapattern[1]);
		ASSERT_EQm("CFFA[2]", 1, dng.cfapattern[2]);
		ASSERT_EQm("CFFA[3]", 2, dng.cfapattern[3]);
		ASSERT_STR_EQm("Make", "Make", dng.camera_make);
		ASSERT_STR_EQm("Model", "Model", dng.camera_model);
		ASSERT_STR_EQm("Software", "Software", dng.software);
		ASSERT_EQm("Orientation", LIBDNG_ORIENTATION_RIGHTBOT, dng.orientation);
		ASSERT_EQ_FMTm("Exposure time", 1.4f, dng.exposure_time, "%f");
		ASSERT_EQm("Exposure program", LIBDNG_EXPOSUREPROGRAM_PORTRAIT, dng.exposure_program);
		ASSERT_EQ_FMTm("FNumber", 1.8f, dng.fnumber, "%f");
		ASSERT_EQ_FMTm("Focal length", 75.0f, dng.focal_length, "%f");
		ASSERT_EQ_FMTm("Crop factor", 1.6f, dng.crop_factor, "%f");

	char orig[sizeof "2011-10-08T07:07:09Z"];
	char parsed[sizeof "2011-10-08T07:07:09Z"];
	strftime(orig, sizeof orig, "%FT%TZ", &testtime);
	strftime(parsed, sizeof parsed, "%FT%TZ", &dng.datetime);
		ASSERT_STR_EQm("DateTime", orig, parsed);

		ASSERT_EQm("Year", testtime.tm_year, dng.datetime.tm_year);
		ASSERT_EQm("Month", testtime.tm_mon, dng.datetime.tm_mon);
		ASSERT_EQm("Day", testtime.tm_yday, dng.datetime.tm_yday);
		ASSERT_EQ_FMTm("Hour", testtime.tm_hour, dng.datetime.tm_hour, "%d");
		ASSERT_EQm("Minute", testtime.tm_min, dng.datetime.tm_min);
		ASSERT_EQm("Second", testtime.tm_sec, dng.datetime.tm_sec);

		PASS();
}

SUITE (test_suite)
{
		RUN_TEST(generate_simple_dng);
		RUN_TEST(read_dng);
}

GREATEST_MAIN_DEFS();

int
main(int argc, char **argv)
{
	GREATEST_MAIN_BEGIN();
	libdng_init();
	RUN_SUITE(test_suite);
	GREATEST_MAIN_END();
}