#include "lut.h"
#include <stdlib.h>
#include "MPI_Communication/unpack.h"
#include "Util/types.h"
#include <stdint.h>
#include <string.h>
#include "Util/utils.h"

// Work_t
void t_unpack1()
{
	// Hard-coded packed work
	unsigned char packedWork[] =
	{
		// --> ../data/inputTest.bin
		46,46,47,100,97,116,97,47,105,110,112,117,116,84,101,115,116,46,98,105,110,
		// continue with 256 - strlen("../data/inputTest.bin") zeros...
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
		// --> 5, {ranks}+0,0, 15, 0 on 4-bytes integers little-endian
		5,0,0,0, 0,0,0,0, 11,3,12,0, 42,17,2,32, 123,0,0,0, 10,0,77,51, 0,0,0,0, 0,0,0,0,
		15,0,0,0, 0,0,0,0, 2,0,0,0
	};

	uint32_t nbSeriesInChunk = 7;
	Work_t* work = unpack_work(packedWork, nbSeriesInChunk);

	LUT_ASSERT(strcmp(work->inputFileName, "../data/inputTest.bin") == 0);
	LUT_ASSERT(work->nbSeries == 5);
	uint32_t ranks[] =
	{
		0,
		12*65536 + 3*256+11,
		32*16777216 + 2*65536 + 17*256 + 42,
		123,
		51*16777216 + 77*65536 + 0*256 + 10
	};
	for (uint32_t i=0; i < work->nbSeries; i++)
	{
		LUT_ASSERT(work->ranks[i] == ranks[i]);
	}
	LUT_ASSERT(work->nbClusters == 15);
	LUT_ASSERT(work->clustOnMedoids == 0);
	LUT_ASSERT(work->p_for_dissims == 2);
	
	free_work(work);
}

// Result_t
void t_unpack2() {

	// Hard-coded packed result
	unsigned char packedResult[] =
	{
		// 5, {medoids_ID}, {medoids_ranks}
		5,0,0,0,
		11,13,15,0, 11,0,0,0, 42,14,0,17, 7,0,123,0, 10,0,0,51,
		32,0,5,0, 4,11,0,0, 42,0,0,23, 77,5,35,0, 10,1,1,1
	};

	Result_t* result = unpack_result(packedResult);

	LUT_ASSERT(result->nbClusters == 5);
	uint32_t medoids_ID[] =
	{
		15*65536 + 13*256+11,
		11,
		17*16777216 + 0*65536 + 14*256 + 42,
		123*65536 + 0*256 + 7,
		51*16777216 + 0*65536 + 0*256 + 10
	};
	uint32_t medoids_ranks[] =
	{
		5*65536 + 32,
		11*256 + 4,
		23*16777216 + 42,
		35*65536 + 5*256 + 77,
		1*16777216 + 1*65536 + 1*256 + 10
	};
	for (uint32_t i=0; i<result->nbClusters; i++)
	{
		LUT_ASSERT(result->medoids_ID[i] == medoids_ID[i]);
		LUT_ASSERT(result->medoids_ranks[i] == medoids_ranks[i]);
	}

	free_result(result);
}