#include "prop_instance.h"

#include "../mesh_data_resource/nodes/mesh_data_instance.h"

#include "../thread_pool/thread_pool.h"

Ref<PropData> PropInstance::get_prop_data() {
	return _prop_data;
}
void PropInstance::set_prop_data(const Ref<PropData> &data) {
	if (_prop_data == data)
		return;

	_prop_data = data;

	build();
}

bool PropInstance::get_auto_bake() const {
	return _auto_bake;
}
void PropInstance::set_auto_bake(const bool value) {
	_auto_bake = value;
}

bool PropInstance::get_snap_to_mesh() const {
	return _snap_to_mesh;
}
void PropInstance::set_snap_to_mesh(const bool value) {
	_snap_to_mesh = value;
}

Vector3 PropInstance::get_snap_axis() const {
	return _snap_axis;
}
void PropInstance::set_snap_axis(const Vector3 &value) {
	_snap_axis = value;
}

void PropInstance::register_prop_mesh_data_instance(MeshDataInstance *instance) {
	ERR_FAIL_COND(!ObjectDB::instance_validate(instance));

	_mesh_data_instances.push_back(instance);

	if (_auto_bake)
		queue_bake();
}

void PropInstance::unregister_prop_mesh_data_instance(MeshDataInstance *instance) {
	ERR_FAIL_COND(!ObjectDB::instance_validate(instance));

	_mesh_data_instances.erase(instance);

	if (_auto_bake)
		queue_bake();
}

void PropInstance::bake() {
	_baking = true;
	_bake_queued = false;

	_job->clear();

	for (int i = 0; i < _mesh_data_instances.size(); ++i) {
		MeshDataInstance *md = _mesh_data_instances.get(i);

		ERR_CONTINUE(!ObjectDB::instance_validate(md));

		Ref<MeshDataResource> mdr = md->get_mesh();

		if (!mdr.is_valid())
			continue;

		Transform t = md->get_transform();

		Spatial *sp = Object::cast_to<Spatial>(md->get_parent());

		while (sp) {
			t *= sp->get_transform();

			sp = Object::cast_to<Spatial>(sp->get_parent());
		}

		_job->add_mesh_instance(t, mdr, md->get_texture());
	}

	if (_job->get_mesh_instance_count() > 0) {
		ThreadPool::get_singleton()->add_job(_job);
	}
}

void PropInstance::bake_finished() {
	_baking = false;

	if (_bake_queued) {
		call_deferred("bake");
	}
}

void PropInstance::queue_bake() {
	if (!_bake_queued) {
		_bake_queued = true;

		if (!_baking) {
			call_deferred("bake");
		}
	}
}

void PropInstance::build() {
	if (!is_inside_tree()) {
		return;
	}

	for (int i = 0; i < get_child_count(); ++i) {
		get_child(i)->queue_delete();
	}

	if (!_prop_data.is_valid())
		return;

	for (int i = 0; i < _prop_data->get_prop_count(); ++i) {
		Ref<PropDataEntry> e = _prop_data->get_prop(i);

		if (!e.is_valid())
			continue;
	}
}

PropInstance::PropInstance() {
	_baking = false;
	_bake_queued = false;
	_snap_to_mesh = false;
	_snap_axis = Vector3(0, -1, 0);
	_job.instance();
	_job->connect("completed", this, "bake_finished", Vector<Variant>(), Object::CONNECT_DEFERRED);
}
PropInstance::~PropInstance() {
	_mesh_data_instances.clear();
	_job.unref();
	_prop_data.unref();
}

void PropInstance::_bind_methods() {
	ClassDB::bind_method(D_METHOD("get_auto_bake"), &PropInstance::get_auto_bake);
	ClassDB::bind_method(D_METHOD("set_auto_bake", "value"), &PropInstance::set_auto_bake);
	ADD_PROPERTY(PropertyInfo(Variant::BOOL, "auto_bake"), "set_auto_bake", "get_auto_bake");

	ClassDB::bind_method(D_METHOD("get_snap_to_mesh"), &PropInstance::get_snap_to_mesh);
	ClassDB::bind_method(D_METHOD("set_snap_to_mesh", "value"), &PropInstance::set_snap_to_mesh);
	ADD_PROPERTY(PropertyInfo(Variant::BOOL, "snap_to_mesh"), "set_snap_to_mesh", "get_snap_to_mesh");

	ClassDB::bind_method(D_METHOD("get_snap_axis"), &PropInstance::get_snap_axis);
	ClassDB::bind_method(D_METHOD("set_snap_axis", "value"), &PropInstance::set_snap_axis);
	ADD_PROPERTY(PropertyInfo(Variant::VECTOR3, "snap_axis"), "set_snap_axis", "get_snap_axis");

	ClassDB::bind_method(D_METHOD("bake"), &PropInstance::bake);
	ClassDB::bind_method(D_METHOD("queue_bake"), &PropInstance::queue_bake);
	ClassDB::bind_method(D_METHOD("bake_finished"), &PropInstance::bake_finished);
}