#include "voxel.h"
#include "voxel_library.h"
#include "voxel_mesher.h"

Voxel::Voxel()
	: Reference(),
	  _id(-1),
	  _material_id(0),
	  _is_transparent(false),
	  _library(NULL),
	  _color(1.f, 1.f, 1.f) {}

Ref<Voxel> Voxel::set_name(String name) {
	_name = name;
	return Ref<Voxel>(this);
}

Ref<Voxel> Voxel::set_id(int id) {
	ERR_FAIL_COND_V(id < 0 || id >= 256, Ref<Voxel>(this));
	// Cannot modify ID after creation
	ERR_FAIL_COND_V(_id != -1, Ref<Voxel>(this));
	_id = id;
	return Ref<Voxel>(this);
}

Ref<Voxel> Voxel::set_color(Color color) {
	_color = color;
	return Ref<Voxel>(this);
}

Ref<Voxel> Voxel::set_material_id(unsigned int id) {
	ERR_FAIL_COND_V(id >= VoxelMesher::MAX_MATERIALS, Ref<Voxel>(this));
	_material_id = id;
	return Ref<Voxel>(this);
}

Ref<Voxel> Voxel::set_transparent(bool t) {
	_is_transparent = t;
	return Ref<Voxel>(this);
}

Ref<Voxel> Voxel::set_cube_geometry(float sy) {
	const Vector3 vertices[SIDE_COUNT][6] = {
		{ // LEFT
				Vector3(0, 0, 0),
				Vector3(0, sy, 0),
				Vector3(0, sy, 1),
				Vector3(0, 0, 0),
				Vector3(0, sy, 1),
				Vector3(0, 0, 1) },
		{ // RIGHT
				Vector3(1, 0, 0),
				Vector3(1, sy, 1),
				Vector3(1, sy, 0),
				Vector3(1, 0, 0),
				Vector3(1, 0, 1),
				Vector3(1, sy, 1) },
		{ // BOTTOM
				Vector3(0, 0, 0),
				Vector3(1, 0, 1),
				Vector3(1, 0, 0),
				Vector3(0, 0, 0),
				Vector3(0, 0, 1),
				Vector3(1, 0, 1) },
		{ // TOP
				Vector3(0, sy, 0),
				Vector3(1, sy, 0),
				Vector3(1, sy, 1),
				Vector3(0, sy, 0),
				Vector3(1, sy, 1),
				Vector3(0, sy, 1) },
		{ // BACK
				Vector3(0, 0, 0),
				Vector3(1, 0, 0),
				Vector3(1, sy, 0),
				Vector3(0, 0, 0),
				Vector3(1, sy, 0),
				Vector3(0, sy, 0) },
		{ // FRONT
				Vector3(1, 0, 1),
				Vector3(0, 0, 1),
				Vector3(1, sy, 1),
				Vector3(0, 0, 1),
				Vector3(0, sy, 1),
				Vector3(1, sy, 1) }
	};

	for (unsigned int side = 0; side < SIDE_COUNT; ++side) {
		_model_side_vertices[side].resize(6);
		PoolVector<Vector3>::Write w = _model_side_vertices[side].write();
		for (unsigned int i = 0; i < 6; ++i) {
			w[i] = vertices[side][i];
		}
	}

	return Ref<Voxel>(this);
}

Ref<Voxel> Voxel::_set_cube_uv_sides(const Vector2 atlas_pos[6]) {
	ERR_FAIL_COND_V(_library == NULL, Ref<Voxel>());

	float e = 0.001;
	const Vector2 uv[4] = {
		Vector2(e, e),
		Vector2(1.f - e, e),
		Vector2(e, 1.f - e),
		Vector2(1.f - e, 1.f - e),
	};

	const int uv6[SIDE_COUNT][6] = {
		// LEFT
		{ 2, 0, 1, 2, 1, 3 },
		// RIGHT
		{ 2, 1, 0, 2, 3, 1 },
		// BOTTOM
		{ 0, 3, 1, 0, 2, 3 },
		// TOP
		{ 0, 1, 3, 0, 3, 2 },
		// BACK
		{ 2, 3, 1, 2, 1, 0 },
		// FRONT
		{ 3, 2, 1, 2, 0, 1 }
	};

	float s = 1.0 / (float)_library->get_atlas_size();

	for (unsigned int side = 0; side < SIDE_COUNT; ++side) {
		_model_side_uv[side].resize(6);
		PoolVector<Vector2>::Write w = _model_side_uv[side].write();
		for (unsigned int i = 0; i < 6; ++i) {
			w[i] = (atlas_pos[side] + uv[uv6[side][i]]) * s;
		}
	}

	return Ref<Voxel>(this);
}

Ref<Voxel> Voxel::set_cube_uv_all_sides(Vector2 atlas_pos) {
	const Vector2 positions[6] = {
		atlas_pos,
		atlas_pos,
		atlas_pos,
		atlas_pos,
		atlas_pos,
		atlas_pos
	};
	return _set_cube_uv_sides(positions);
}

Ref<Voxel> Voxel::set_cube_uv_tbs_sides(Vector2 top_atlas_pos, Vector2 side_atlas_pos, Vector2 bottom_atlas_pos) {
	const Vector2 positions[6] = {
		side_atlas_pos,
		side_atlas_pos,
		bottom_atlas_pos,
		top_atlas_pos,
		side_atlas_pos,
		side_atlas_pos,
	};
	return _set_cube_uv_sides(positions);
}

//Ref<Voxel> Voxel::set_xquad_geometry(Vector2 atlas_pos) {
//    // TODO
//    return Ref<Voxel>(this);
//}

void Voxel::_bind_methods() {

	ClassDB::bind_method(D_METHOD("set_name", "name"), &Voxel::set_name);
	ClassDB::bind_method(D_METHOD("get_name"), &Voxel::get_name);

	ClassDB::bind_method(D_METHOD("set_id", "id"), &Voxel::set_id);
	ClassDB::bind_method(D_METHOD("get_id"), &Voxel::get_id);

	ClassDB::bind_method(D_METHOD("set_color", "color"), &Voxel::set_color);
	ClassDB::bind_method(D_METHOD("get_color"), &Voxel::get_color);

	ClassDB::bind_method(D_METHOD("set_transparent", "color"), &Voxel::set_transparent, DEFVAL(true));
	ClassDB::bind_method(D_METHOD("is_transparent"), &Voxel::is_transparent);

	ClassDB::bind_method(D_METHOD("set_material_id", "id"), &Voxel::set_material_id);
	ClassDB::bind_method(D_METHOD("get_material_id"), &Voxel::get_material_id);

	ClassDB::bind_method(D_METHOD("set_cube_geometry", "height"), &Voxel::set_cube_geometry, DEFVAL(1.f));
	ClassDB::bind_method(D_METHOD("set_cube_uv_all_sides", "atlas_pos"), &Voxel::set_cube_uv_all_sides);
	ClassDB::bind_method(D_METHOD("set_cube_uv_tbs_sides", "top_atlas_pos", "side_atlas_pos", "bottom_atlas_pos"), &Voxel::set_cube_uv_tbs_sides);

	BIND_CONSTANT(CHANNEL_TYPE)
	BIND_CONSTANT(CHANNEL_ISOLEVEL)
	BIND_CONSTANT(CHANNEL_DATA)
}