/** Unit tests for IO formats.
 *
 * @author Steffen Vogel <stvogel@eonerc.rwth-aachen.de>
 * @copyright 2017, Institute for Automation of Complex Power Systems, EONERC
 * @license GNU General Public License (version 3)
 *
 * VILLASnode
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *********************************************************************************/

#include <stdio.h>
#include <float.h>

#include <criterion/criterion.h>
#include <criterion/parameterized.h>
#include <criterion/logging.h>

#include "utils.h"
#include "timing.h"
#include "sample.h"
#include "plugin.h"
#include "pool.h"
#include "io.h"
#include "io/raw.h"

#define NUM_SAMPLES 10
#define NUM_VALUES 10

void cr_assert_eq_sample(struct sample *s, struct sample *t)
{
	cr_assert_eq(s->length, t->length);
	cr_assert_eq(s->sequence, t->sequence);
//	cr_assert_eq(s->format, t->format);

	cr_assert_eq(s->ts.origin.tv_sec, t->ts.origin.tv_sec);
	cr_assert_eq(s->ts.origin.tv_nsec, t->ts.origin.tv_nsec);

	for (int i = 0; i < MIN(s->length, t->length); i++) {
		switch (sample_get_data_format(t, i)) {
			case SAMPLE_DATA_FORMAT_FLOAT:
				cr_assert_float_eq(s->data[i].f, t->data[i].f, 1e-3, "Sample data mismatch at index %d: %f != %f", i, s->data[i].f, t->data[i].f);
				break;
			case SAMPLE_DATA_FORMAT_INT:
				cr_assert_eq(s->data[i].i, t->data[i].i);
				break;
		}
	}
}

ParameterizedTestParameters(io, highlevel)
{
	static char formats[][32] = {
#ifdef WITH_HDF5
		"hdf5",
#endif
		"raw-int8",
		"raw-int16-be",
		"raw-int16-le",
		"raw-int32-be",
		"raw-int32-le",
		"raw-int64-be",
		"raw-int64-le",
		"raw-flt32",
		"raw-flt64",
		"villas",
		"json",
		"msg",
		"gtnet",
		"gtnet-fake",
	};

	return cr_make_param_array(char[32], formats, ARRAY_LEN(formats));
}

ParameterizedTest(char *fmt, io, highlevel)
{
	int ret, cnt;

	struct io io;
	struct io_format *f;

	struct pool p = { .state = STATE_DESTROYED };
	struct sample *smps[NUM_SAMPLES];
	struct sample *smpt[NUM_SAMPLES];

	ret = pool_init(&p, 2 * NUM_SAMPLES, SAMPLE_LEN(NUM_VALUES), &memtype_hugepage);
	cr_assert_eq(ret, 0);
	
	info("Testing: %s", fmt);

	/* Prepare a sample with arbitrary data */
	ret = sample_alloc(&p, smps, NUM_SAMPLES);
	cr_assert_eq(ret, NUM_SAMPLES);

	ret = sample_alloc(&p, smpt, NUM_SAMPLES);
	cr_assert_eq(ret, NUM_SAMPLES);

	for (int i = 0; i < NUM_SAMPLES; i++) {
		smpt[i]->capacity = NUM_VALUES;
		
		smps[i]->length = NUM_VALUES;
		smps[i]->sequence = 235 + i;
		smps[i]->format = 0; /* all float */
		smps[i]->ts.origin = time_now();

		for (int j = 0; j < smps[i]->length; j++)
			smps[i]->data[j].f = j * 0.1 + i * 100;
			//smps[i]->data[j].i = -500  + j*100;
	}

	/* Open a file for IO */
	char *fn, dir[64];
	strncpy(dir, "/tmp/villas.XXXXXX", sizeof(dir));
	
	mkdtemp(dir);
	ret = asprintf(&fn, "%s/file", dir);
	cr_assert_gt(ret, 0);

	f = io_format_lookup(fmt);
	cr_assert_not_null(f, "Format '%s' does not exist", fmt);

	ret = io_init(&io, f, IO_FORMAT_ALL);
	cr_assert_eq(ret, 0);

	ret = io_open(&io, fn, "w+");
	cr_assert_eq(ret, 0);

	ret = io_print(&io, smps, NUM_SAMPLES);
	cr_assert_eq(ret, NUM_SAMPLES);

	ret = io_close(&io);
	cr_assert_eq(ret, 0);

#if 1 /* Show the file contents */
	char cmd[128];
	if (!strcmp(fmt, "json") || !strcmp(fmt, "villas"))
		snprintf(cmd, sizeof(cmd), "cat %s", fn);
	else
		snprintf(cmd, sizeof(cmd), "hexdump -C %s", fn);
	system(cmd);
#endif
	
	ret = io_open(&io, fn, "r");
	cr_assert_eq(ret, 0);

	cnt = io_scan(&io, smpt, NUM_SAMPLES);

	/* The RAW format has certain limitations: 
	 *  - limited accuracy if smaller datatypes are used
	 *  - no support for vectors / framing
	 *
	 * We need to take these limitations into account before comparing.
	 */
	if (f->sscan == raw_sscan) {
		cr_assert_eq(cnt, 1);
		cr_assert_eq(smpt[0]->length, smpt[0]->capacity);
		
		if (io.flags & RAW_FAKE) {
		}
		else {
			smpt[0]->sequence = smps[0]->sequence;
			smpt[0]->ts.origin = smps[0]->ts.origin;
		}

		int bits = 1 << (io.flags >> 24);
		for (int j = 0; j < smpt[0]->length; j++) {
			if (io.flags & RAW_FLT) {
				switch (bits) {
					case  32: smps[0]->data[j].f = (float)  smps[0]->data[j].f; break;
					case  64: smps[0]->data[j].f = (double) smps[0]->data[j].f; break;
				}
			}
			else {
				switch (bits) {
					case   8: smps[0]->data[j].i = (  int8_t) smps[0]->data[j].i; break;
					case  16: smps[0]->data[j].i = ( int16_t) smps[0]->data[j].i; break;
					case  32: smps[0]->data[j].i = ( int32_t) smps[0]->data[j].i; break;
					case  64: smps[0]->data[j].i = ( int64_t) smps[0]->data[j].i; break;
				}
			}
		}
	}
	else
		cr_assert_eq(cnt, NUM_SAMPLES, "Read only %d of %d samples back", cnt, NUM_SAMPLES);
	
	for (int i = 0; i < cnt; i++)
		cr_assert_eq_sample(smps[i], smpt[i]);

	ret = io_close(&io);
	cr_assert_eq(ret, 0);

	ret = io_destroy(&io);
	cr_assert_eq(ret, 0);

	ret = unlink(fn);
	cr_assert_eq(ret, 0);
	
	ret = rmdir(dir);
	cr_assert_eq(ret, 0);
	
	free(fn);

	sample_free(smps, NUM_SAMPLES);
	sample_free(smpt, NUM_SAMPLES);

	ret = pool_destroy(&p);
	cr_assert_eq(ret, 0);
}