#include "renderer/VertexFormat.h"
#include "unittest/TestSuite.h"

static void testCreate(void) {
	VertexFormat * vertexFormat = VertexFormat_create(0, NULL);
	TestCase_assertPointerNonNULL(vertexFormat);
	TestCase_assertUIntEqual(vertexFormat->attributeCount, 0);
	TestCase_assertPointerNULL(vertexFormat->attributeTypes);
	TestCase_assertUIntEqual(vertexFormat->componentCountTotal, 0);
	TestCase_assertUIntEqual(vertexFormat->bytesPerVertex, 0);
	VertexFormat_dispose(vertexFormat);
	
	VertexAttributeTypeSpec attributeTypeSpec = {"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 3};
	vertexFormat = VertexFormat_create(1, &attributeTypeSpec);
	TestCase_assertPointerNonNULL(vertexFormat);
	TestCase_assertUIntEqual(vertexFormat->attributeCount, 1);
	TestCase_assertPointerNonNULL(vertexFormat->attributeTypes);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[0].name, "position");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].dataType, ATTRIBUTE_TYPE_FLOAT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].usageHint, ATTRIBUTE_USAGE_POSITION);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].componentCount, 3);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].byteOffset, 0);
	TestCase_assertUIntEqual(vertexFormat->componentCountTotal, 3);
	TestCase_assertUIntEqual(vertexFormat->bytesPerVertex, 12);
	VertexFormat_dispose(vertexFormat);
	
	VertexAttributeTypeSpec attributeTypeSpecs1[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 2},
		{"normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	vertexFormat = VertexFormat_create(sizeof(attributeTypeSpecs1) / sizeof(attributeTypeSpecs1[0]), attributeTypeSpecs1);
	TestCase_assertPointerNonNULL(vertexFormat);
	TestCase_assertUIntEqual(vertexFormat->attributeCount, 2);
	TestCase_assertPointerNonNULL(vertexFormat->attributeTypes);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[0].name, "position");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].dataType, ATTRIBUTE_TYPE_FLOAT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].usageHint, ATTRIBUTE_USAGE_POSITION);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].componentCount, 2);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].byteOffset, 0);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[1].name, "normal");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[1].dataType, ATTRIBUTE_TYPE_FLOAT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[1].usageHint, ATTRIBUTE_USAGE_NORMAL);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[1].componentCount, 3);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[1].byteOffset, sizeof(float) * 2);
	TestCase_assertUIntEqual(vertexFormat->componentCountTotal, 5);
	TestCase_assertUIntEqual(vertexFormat->bytesPerVertex, 20);
	VertexFormat_dispose(vertexFormat);
	
	VertexAttributeTypeSpec attributeTypeSpecs2[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 3},
		{"color", ATTRIBUTE_TYPE_UINT, ATTRIBUTE_USAGE_COLOR, 4},
		{"normal", ATTRIBUTE_TYPE_INT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	vertexFormat = VertexFormat_create(sizeof(attributeTypeSpecs2) / sizeof(attributeTypeSpecs2[0]), attributeTypeSpecs2);
	TestCase_assertPointerNonNULL(vertexFormat);
	TestCase_assertUIntEqual(vertexFormat->attributeCount, 3);
	TestCase_assertPointerNonNULL(vertexFormat->attributeTypes);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[0].name, "position");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].dataType, ATTRIBUTE_TYPE_FLOAT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].usageHint, ATTRIBUTE_USAGE_POSITION);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].componentCount, 3);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].byteOffset, 0);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[1].name, "color");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[1].dataType, ATTRIBUTE_TYPE_UINT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[1].usageHint, ATTRIBUTE_USAGE_COLOR);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[1].componentCount, 4);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[1].byteOffset, sizeof(float) * 3);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[2].name, "normal");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[2].dataType, ATTRIBUTE_TYPE_INT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[2].usageHint, ATTRIBUTE_USAGE_NORMAL);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[2].componentCount, 3);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[2].byteOffset, sizeof(float) * 7);
	TestCase_assertUIntEqual(vertexFormat->componentCountTotal, 10);
	TestCase_assertUIntEqual(vertexFormat->bytesPerVertex, 40);
	VertexFormat_dispose(vertexFormat);
}

static void testCopy(void) {
	VertexFormat * originalVertexFormat = VertexFormat_create(0, NULL);
	VertexFormat * vertexFormat = VertexFormat_copy(originalVertexFormat);
	VertexFormat_dispose(originalVertexFormat);
	TestCase_assertPointerNonNULL(vertexFormat);
	TestCase_assertUIntEqual(vertexFormat->attributeCount, 0);
	TestCase_assertPointerNULL(vertexFormat->attributeTypes);
	TestCase_assertUIntEqual(vertexFormat->componentCountTotal, 0);
	TestCase_assertUIntEqual(vertexFormat->bytesPerVertex, 0);
	VertexFormat_dispose(vertexFormat);
	
	VertexAttributeTypeSpec attributeTypeSpec = {"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 3};
	originalVertexFormat = VertexFormat_create(1, &attributeTypeSpec);
	vertexFormat = VertexFormat_copy(originalVertexFormat);
	VertexFormat_dispose(originalVertexFormat);
	TestCase_assertPointerNonNULL(vertexFormat);
	TestCase_assertUIntEqual(vertexFormat->attributeCount, 1);
	TestCase_assertPointerNonNULL(vertexFormat->attributeTypes);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[0].name, "position");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].dataType, ATTRIBUTE_TYPE_FLOAT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].usageHint, ATTRIBUTE_USAGE_POSITION);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].componentCount, 3);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].byteOffset, 0);
	TestCase_assertUIntEqual(vertexFormat->componentCountTotal, 3);
	TestCase_assertUIntEqual(vertexFormat->bytesPerVertex, 12);
	VertexFormat_dispose(vertexFormat);
	
	VertexAttributeTypeSpec attributeTypeSpecs1[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 2},
		{"normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	originalVertexFormat = VertexFormat_create(sizeof(attributeTypeSpecs1) / sizeof(attributeTypeSpecs1[0]), attributeTypeSpecs1);
	vertexFormat = VertexFormat_copy(originalVertexFormat);
	VertexFormat_dispose(originalVertexFormat);
	TestCase_assertPointerNonNULL(vertexFormat);
	TestCase_assertUIntEqual(vertexFormat->attributeCount, 2);
	TestCase_assertPointerNonNULL(vertexFormat->attributeTypes);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[0].name, "position");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].dataType, ATTRIBUTE_TYPE_FLOAT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].usageHint, ATTRIBUTE_USAGE_POSITION);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].componentCount, 2);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].byteOffset, 0);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[1].name, "normal");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[1].dataType, ATTRIBUTE_TYPE_FLOAT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[1].usageHint, ATTRIBUTE_USAGE_NORMAL);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[1].componentCount, 3);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[1].byteOffset, sizeof(float) * 2);
	TestCase_assertUIntEqual(vertexFormat->componentCountTotal, 5);
	TestCase_assertUIntEqual(vertexFormat->bytesPerVertex, 20);
	VertexFormat_dispose(vertexFormat);
	
	VertexAttributeTypeSpec attributeTypeSpecs2[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 3},
		{"color", ATTRIBUTE_TYPE_UINT, ATTRIBUTE_USAGE_COLOR, 4},
		{"normal", ATTRIBUTE_TYPE_INT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	originalVertexFormat = VertexFormat_create(sizeof(attributeTypeSpecs2) / sizeof(attributeTypeSpecs2[0]), attributeTypeSpecs2);
	vertexFormat = VertexFormat_copy(originalVertexFormat);
	VertexFormat_dispose(originalVertexFormat);
	TestCase_assertPointerNonNULL(vertexFormat);
	TestCase_assertUIntEqual(vertexFormat->attributeCount, 3);
	TestCase_assertPointerNonNULL(vertexFormat->attributeTypes);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[0].name, "position");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].dataType, ATTRIBUTE_TYPE_FLOAT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[0].usageHint, ATTRIBUTE_USAGE_POSITION);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].componentCount, 3);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[0].byteOffset, 0);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[1].name, "color");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[1].dataType, ATTRIBUTE_TYPE_UINT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[1].usageHint, ATTRIBUTE_USAGE_COLOR);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[1].componentCount, 4);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[1].byteOffset, sizeof(float) * 3);
	TestCase_assertStringEqual(vertexFormat->attributeTypes[2].name, "normal");
	TestCase_assertIntEqual(vertexFormat->attributeTypes[2].dataType, ATTRIBUTE_TYPE_INT);
	TestCase_assertIntEqual(vertexFormat->attributeTypes[2].usageHint, ATTRIBUTE_USAGE_NORMAL);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[2].componentCount, 3);
	TestCase_assertUIntEqual(vertexFormat->attributeTypes[2].byteOffset, sizeof(float) * 7);
	TestCase_assertUIntEqual(vertexFormat->componentCountTotal, 10);
	TestCase_assertUIntEqual(vertexFormat->bytesPerVertex, 40);
	VertexFormat_dispose(vertexFormat);
}

void testMatchesSpec(void) {
	VertexAttributeTypeSpec attributeTypeSpecs1[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 2},
		{"normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	VertexAttributeTypeSpec attributeTypeSpecs2[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 3},
		{"normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	VertexAttributeTypeSpec attributeTypeSpecs3[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 2},
		{"normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_NORMAL, 4}
	};
	VertexAttributeTypeSpec attributeTypeSpecs4[] = {
		{"some_other_name_for_position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 2},
		{"something_used_as_a_normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	VertexAttributeTypeSpec attributeTypeSpecs5[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 2},
		{"normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_NORMAL, 3},
		{"color", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_COLOR, 4}
	};
	VertexAttributeTypeSpec attributeTypeSpecs6[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_UNSPECIFIED, 2},
		{"normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	VertexAttributeTypeSpec attributeTypeSpecs7[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 2},
		{"normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_TEXTURE_COORDINATE, 3}
	};
	VertexAttributeTypeSpec attributeTypeSpecs8[] = {
		{"position", ATTRIBUTE_TYPE_INT, ATTRIBUTE_USAGE_POSITION, 2},
		{"normal", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	VertexAttributeTypeSpec attributeTypeSpecs9[] = {
		{"position", ATTRIBUTE_TYPE_FLOAT, ATTRIBUTE_USAGE_POSITION, 2},
		{"normal", ATTRIBUTE_TYPE_UINT, ATTRIBUTE_USAGE_NORMAL, 3}
	};
	VertexFormat * vertexFormat = VertexFormat_create(sizeof(attributeTypeSpecs1) / sizeof(attributeTypeSpecs1[0]), attributeTypeSpecs1);
	TestCase_assertBoolTrue(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs1) / sizeof(attributeTypeSpecs1[0]), attributeTypeSpecs1));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, 1, attributeTypeSpecs1));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs2) / sizeof(attributeTypeSpecs2[0]), attributeTypeSpecs2));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs3) / sizeof(attributeTypeSpecs3[0]), attributeTypeSpecs3));
	TestCase_assertBoolTrue(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs4) / sizeof(attributeTypeSpecs4[0]), attributeTypeSpecs4));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs5) / sizeof(attributeTypeSpecs5[0]), attributeTypeSpecs5));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs6) / sizeof(attributeTypeSpecs6[0]), attributeTypeSpecs6));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs7) / sizeof(attributeTypeSpecs7[0]), attributeTypeSpecs7));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs8) / sizeof(attributeTypeSpecs8[0]), attributeTypeSpecs8));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs9) / sizeof(attributeTypeSpecs9[0]), attributeTypeSpecs9));
	VertexFormat_dispose(vertexFormat);
	
	vertexFormat = VertexFormat_create(sizeof(attributeTypeSpecs5) / sizeof(attributeTypeSpecs5[0]), attributeTypeSpecs5);
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs1) / sizeof(attributeTypeSpecs1[0]), attributeTypeSpecs1));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs2) / sizeof(attributeTypeSpecs2[0]), attributeTypeSpecs2));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs3) / sizeof(attributeTypeSpecs3[0]), attributeTypeSpecs3));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs4) / sizeof(attributeTypeSpecs4[0]), attributeTypeSpecs4));
	TestCase_assertBoolTrue(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs5) / sizeof(attributeTypeSpecs5[0]), attributeTypeSpecs5));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs6) / sizeof(attributeTypeSpecs6[0]), attributeTypeSpecs6));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs7) / sizeof(attributeTypeSpecs7[0]), attributeTypeSpecs7));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs8) / sizeof(attributeTypeSpecs8[0]), attributeTypeSpecs8));
	TestCase_assertBoolFalse(VertexFormat_matchesSpec(vertexFormat, sizeof(attributeTypeSpecs9) / sizeof(attributeTypeSpecs9[0]), attributeTypeSpecs9));
	VertexFormat_dispose(vertexFormat);
}

TEST_SUITE(VertexFormatTest,
           testCreate,
           testCopy,
           testMatchesSpec)