#include "constantbuffer.h"

static void setInt(uint8_t *constants, int offset, int value) {
	int *ints = (int *)(&constants[offset]);
	ints[0] = value;
}

static void setFloat(uint8_t *constants, int offset, float value) {
	float *floats = (float *)(&constants[offset]);
	floats[0] = value;
}

static void setFloat2(uint8_t *constants, int offset, float value1, float value2) {
	float *floats = (float *)(&constants[offset]);
	floats[0] = value1;
	floats[1] = value2;
}

static void setFloat3(uint8_t *constants, int offset, float value1, float value2, float value3) {
	float *floats = (float *)(&constants[offset]);
	floats[0] = value1;
	floats[1] = value2;
	floats[2] = value3;
}

static void setFloat4(uint8_t *constants, int offset, float value1, float value2, float value3, float value4) {
	float *floats = (float *)(&constants[offset]);
	floats[0] = value1;
	floats[1] = value2;
	floats[2] = value3;
	floats[3] = value4;
}

static void setFloats(uint8_t *constants, int offset, float *values, int count) {
	float *floats = (float *)(&constants[offset]);
	for (int i = 0; i < count; ++i) {
		floats[i] = values[i];
	}
}

static void setBool(uint8_t *constants, int offset, bool value) {
	int *ints = (int *)(&constants[offset]);
	ints[0] = value ? 1 : 0;
}

static void setMatrix4(uint8_t *constants, int offset, kinc_matrix4x4_t *value) {
	float *floats = (float *)(&constants[offset]);
	for (int y = 0; y < 4; ++y) {
		for (int x = 0; x < 4; ++x) {
			floats[x + y * 4] = kinc_matrix4x4_get(value, x, y);
		}
	}
}

static void setMatrix3(uint8_t *constants, int offset, kinc_matrix3x3_t *value) {
	float *floats = (float *)(&constants[offset]);
	for (int y = 0; y < 3; ++y) {
		for (int x = 0; x < 3; ++x) {
			floats[x + y * 4] = kinc_matrix3x3_get(value, x, y);
		}
	}
}

void kinc_g5_constant_buffer_set_int(kinc_g5_constant_buffer_t *buffer, int offset, int value) {
	setInt(buffer->data, offset, value);
}

void kinc_g5_constant_buffer_set_float(kinc_g5_constant_buffer_t *buffer, int offset, float value) {
	setFloat(buffer->data, offset, value);
}

void kinc_g5_constant_buffer_set_float2(kinc_g5_constant_buffer_t *buffer, int offset, float value1, float value2) {
	setFloat2(buffer->data, offset, value1, value2);
}

void kinc_g5_constant_buffer_set_float3(kinc_g5_constant_buffer_t *buffer, int offset, float value1, float value2, float value3) {
	setFloat3(buffer->data, offset, value1, value2, value3);
}

void kinc_g5_constant_buffer_set_float4(kinc_g5_constant_buffer_t *buffer, int offset, float value1, float value2, float value3, float value4) {
	setFloat4(buffer->data, offset, value1, value2, value3, value4);
}

void kinc_g5_constant_buffer_set_floats(kinc_g5_constant_buffer_t *buffer, int offset, float *values, int count) {
	setFloats(buffer->data, offset, values, count);
}

void kinc_g5_constant_buffer_set_bool(kinc_g5_constant_buffer_t *buffer, int offset, bool value) {
	setBool(buffer->data, offset, value);
}

void kinc_g5_constant_buffer_set_matrix4(kinc_g5_constant_buffer_t *buffer, int offset, kinc_matrix4x4_t *value) {
	if (kinc_g5_transposeMat4) {
		kinc_matrix4x4_t m = *value;
		kinc_matrix4x4_transpose(&m);
		setMatrix4(buffer->data, offset, &m);
	}
	else {
		setMatrix4(buffer->data, offset, value);
	}
}

void kinc_g5_constant_buffer_set_matrix3(kinc_g5_constant_buffer_t *buffer, int offset, kinc_matrix3x3_t *value) {
	if (kinc_g5_transposeMat3) {
		kinc_matrix3x3_t m = *value;
		kinc_matrix3x3_transpose(&m);
		setMatrix3(buffer->data, offset, &m);
	}
	else {
		setMatrix3(buffer->data, offset, value);
	}
}