#include <stdlib.h>
#include <stdbool.h>
#include "gl/glad/glad.h"
#include <GLFW/glfw3.h>
#include <lua.h>
#include <honeysuckle.h>
#include "util/util.h"
#include "gl.h"

int gl_create_buffer(lua_State *L);
int gl_bind_buffer(lua_State *L);
int gl_buffer_data(lua_State *L);

int gl_vertex_array_create(lua_State *L);
int gl_vertex_array_bind(lua_State *L);
int gl_vertex_attrib_pointer(lua_State *L);
int gl_vertex_array_enable_attrib(lua_State *L);


void setup_data(lua_State *L, int gl_index)
{
	int tbl = hs_create_table(L,
		/* functions */
		hs_str_cfunc("GenBuffers", gl_create_buffer),
		hs_str_cfunc("BindBuffer", gl_bind_buffer),
		hs_str_cfunc("BufferData", gl_buffer_data),

		hs_str_cfunc("GenVertexArrays", gl_vertex_array_create),
		hs_str_cfunc("BindVertexArray", gl_vertex_array_bind),
		hs_str_cfunc("VertexAttribPointer", gl_vertex_attrib_pointer),
		hs_str_cfunc("EnableVertexAttribArray", gl_vertex_array_enable_attrib),

		/******** enums ********/
		/* buffer bind targets */
		hs_str_int("ARRAY_BUFFER", GL_ARRAY_BUFFER),
		hs_str_int("ELEMENT_ARRAY_BUFFER", GL_ELEMENT_ARRAY_BUFFER),

		/* buffer usage patters */
		hs_str_int("STREAM_DRAW", GL_STREAM_DRAW),
		hs_str_int("STATIC_DRAW", GL_STATIC_DRAW),
		hs_str_int("DYNAMIC_DRAW", GL_DYNAMIC_DRAW),
	);	

	append_table(L, gl_index, tbl);
	lua_pop(L, 1);
}


int gl_create_buffer(lua_State *L)
{
	int buf;
	glGenBuffers(1, &buf);
	lua_pushinteger(L, buf);
	return 1;
}


int gl_bind_buffer(lua_State *L)
{
	lua_Integer target, buf;
	hs_parse_args(L, hs_int(target), hs_int(buf));
	glBindBuffer(target, buf);
	return 0;
}


#define GET_BUFFER_TYPE(type, name, conversion) \
	void * get_buffer_ ## name (lua_State *L, size_t *sz, int tbl) { \
		size_t len = lua_objlen(L, tbl); \
		*sz = len * sizeof(type); \
		type *buf = malloc(*sz); \
		if (buf == NULL) \
			hs_throw_error(L, "failed to allocate intermediary buffer"); \
		for (int i=0; i<len; i++) { \
			lua_rawgeti(L, tbl, i+1); \
			if (!lua_isnumber(L, -1)) \
				hs_throw_error(L, "all elements must be numbers (failed at index %d)", i); \
			buf[i] = conversion(L, -1); \
			lua_pop(L, 1); \
		} \
		return buf; \
	}

GET_BUFFER_TYPE(unsigned int, uint, lua_tointeger)
GET_BUFFER_TYPE(int, int, lua_tointeger)
GET_BUFFER_TYPE(float, float, lua_tonumber)

int gl_buffer_data(lua_State *L)
{
	lua_Integer target, type, usage;
	int table;
	hs_parse_args(L, hs_int(target), hs_int(type), hs_tbl(table), hs_int(usage));

	/* build raw buffer */
	void *buf; size_t len;
	switch(type) {
	case GL_UNSIGNED_INT:
		buf = get_buffer_uint(L, &len, table);
		break;

	case GL_INT:
		buf = get_buffer_int(L, &len, table);
		break;

	case GL_FLOAT:
		buf = get_buffer_float(L, &len, table);
		break;

	default:
		hs_throw_error(L, "invalid type");
	}

	/* call */
	glBufferData(target, len, buf, usage);
	free(buf);
	return 0;
}


int gl_vertex_array_create(lua_State *L)
{
	int array;
	glGenVertexArrays(1, &array);
	lua_pushinteger(L, array);
	return 1;
}


int gl_vertex_array_bind(lua_State *L)
{
	lua_Integer array;
	hs_parse_args(L, hs_int(array));
	glBindVertexArray(array);
	return 0;
}


int gl_vertex_attrib_pointer(lua_State *L)
{
	lua_Integer index, size, stride, offset;
	bool normalized;
	hs_parse_args(L, hs_int(index), hs_int(size), hs_bool(normalized), hs_int(stride), hs_int(offset));
	glVertexAttribPointer(index, size, GL_FLOAT, 
	                      normalized, stride*sizeof(float), 
	                      (void*) (offset*sizeof(float)));
	return 0;
}


int gl_vertex_array_enable_attrib(lua_State *L)
{
	lua_Integer index;
	hs_parse_args(L, hs_int(index));
	glEnableVertexAttribArray(index);
	return 0;
}