limboai/limbo_state.cpp

/* limbo_state.cpp */

#include "limbo_state.h"
#include "core/array.h"
#include "core/class_db.h"
#include "core/error_macros.h"
#include "core/object.h"
#include "core/typedefs.h"
#include "core/variant.h"
#include "limbo_string_names.h"

const String LimboState::EVENT_FINISHED = "finished";

LimboState *LimboState::get_root() const {
	const LimboState *state = this;
	while (state->get_parent() && state->get_parent()->is_class("LimboState")) {
		state = Object::cast_to<LimboState>(get_parent());
	}
	return const_cast<LimboState *>(state);
}

LimboState *LimboState::named(String p_name) {
	set_name(p_name);
	return this;
};

void LimboState::_setup() {
	if (get_script_instance() &&
			get_script_instance()->has_method(LimboStringNames::get_singleton()->_setup)) {
		get_script_instance()->call(LimboStringNames::get_singleton()->_setup);
	}
	emit_signal(LimboStringNames::get_singleton()->setup);
};

void LimboState::_enter() {
	active = true;
	if (get_script_instance() &&
			get_script_instance()->has_method(LimboStringNames::get_singleton()->_enter)) {
		get_script_instance()->call(LimboStringNames::get_singleton()->_enter);
	}
	emit_signal(LimboStringNames::get_singleton()->entered);
};

void LimboState::_exit() {
	if (!active) {
		return;
	}
	if (get_script_instance() &&
			get_script_instance()->has_method(LimboStringNames::get_singleton()->_exit)) {
		get_script_instance()->call(LimboStringNames::get_singleton()->_exit);
	}
	emit_signal(LimboStringNames::get_singleton()->exited);
	active = false;
};

void LimboState::_update(float p_delta) {
	if (get_script_instance() &&
			get_script_instance()->has_method(LimboStringNames::get_singleton()->_update)) {
		get_script_instance()->call(LimboStringNames::get_singleton()->_update, p_delta);
	}
	emit_signal(LimboStringNames::get_singleton()->updated, p_delta);
};

void LimboState::_initialize(Object *p_agent, const Ref<Blackboard> &p_blackboard) {
	ERR_FAIL_COND(p_agent == nullptr);

	agent = p_agent;

	if (!p_blackboard.is_null()) {
		if (!blackboard->get_data().empty()) {
			blackboard->set_parent_scope(p_blackboard);
		} else {
			blackboard = p_blackboard;
		}
	}

	_setup();
};

bool LimboState::dispatch(const String &p_event, const Variant &p_cargo) {
	ERR_FAIL_COND_V(p_event.empty(), false);
	bool result = false;
	if (handlers.size() > 0 && handlers.has(p_event)) {
		if (p_cargo.get_type() == Variant::NIL) {
			result = call(handlers[p_event]);
		} else {
			result = call(handlers[p_event], p_cargo);
		}
	}
	return result;
}

void LimboState::add_event_handler(const String &p_event, const StringName &p_method) {
	ERR_FAIL_COND(p_event.empty());
	ERR_FAIL_COND(!has_method(p_method));
	handlers.insert(p_event, p_method);
}

LimboState *LimboState::call_on_enter(Object *p_object, const StringName &p_method) {
	ERR_FAIL_COND_V(p_object == nullptr, this);
	ERR_FAIL_COND_V(!p_object->has_method(p_method), this);
	connect(LimboStringNames::get_singleton()->entered, p_object, p_method);
	return this;
}

LimboState *LimboState::call_on_exit(Object *p_object, const StringName &p_method) {
	ERR_FAIL_COND_V(p_object == nullptr, this);
	ERR_FAIL_COND_V(!p_object->has_method(p_method), this);
	connect(LimboStringNames::get_singleton()->exited, p_object, p_method);
	return this;
}

LimboState *LimboState::call_on_update(Object *p_object, const StringName &p_method) {
	ERR_FAIL_COND_V(p_object == nullptr, this);
	ERR_FAIL_COND_V(!p_object->has_method(p_method), this);
	connect(LimboStringNames::get_singleton()->updated, p_object, p_method);
	return this;
}

void LimboState::set_guard_func(Object *p_object, const StringName &p_func, const Array &p_binds) {
	ERR_FAIL_COND(p_object == nullptr);
	ERR_FAIL_COND(!p_object->has_method(p_func));

	guard.obj = p_object;
	guard.func = p_func;
	guard.binds = p_binds;
}

void LimboState::clear_guard_func() {
	guard.obj = nullptr;
	guard.func = "";
	guard.binds.clear();
}

void LimboState::_notification(int p_what) {
	switch (p_what) {
		case NOTIFICATION_EXIT_TREE: {
			if (active) {
				_exit();
			}
		} break;
	}
}

void LimboState::_bind_methods() {
	ClassDB::bind_method(D_METHOD("get_root"), &LimboState::get_root);
	ClassDB::bind_method(D_METHOD("get_agent"), &LimboState::get_agent);
	ClassDB::bind_method(D_METHOD("set_agent", "p_agent"), &LimboState::set_agent);
	ClassDB::bind_method(D_METHOD("event_finished"), &LimboState::event_finished);
	ClassDB::bind_method(D_METHOD("is_active"), &LimboState::is_active);
	ClassDB::bind_method(D_METHOD("_setup"), &LimboState::_setup);
	ClassDB::bind_method(D_METHOD("_enter"), &LimboState::_enter);
	ClassDB::bind_method(D_METHOD("_exit"), &LimboState::_exit);
	ClassDB::bind_method(D_METHOD("_update", "p_delta"), &LimboState::_update);
	ClassDB::bind_method(D_METHOD("_initialize", "p_agent", "p_blackboard"), &LimboState::_initialize);
	ClassDB::bind_method(D_METHOD("dispatch", "p_event", "p_cargo"), &LimboState::dispatch, Variant());
	ClassDB::bind_method(D_METHOD("named", "p_name"), &LimboState::named);
	ClassDB::bind_method(D_METHOD("add_event_handler", "p_event", "p_method"), &LimboState::add_event_handler);
	ClassDB::bind_method(D_METHOD("call_on_enter", "p_object", "p_method"), &LimboState::call_on_enter);
	ClassDB::bind_method(D_METHOD("call_on_exit", "p_object", "p_method"), &LimboState::call_on_exit);
	ClassDB::bind_method(D_METHOD("call_on_update", "p_object", "p_method"), &LimboState::call_on_update);
	ClassDB::bind_method(D_METHOD("set_guard_func", "p_object", "p_func", "p_binds"), &LimboState::set_guard_func, Array());
	ClassDB::bind_method(D_METHOD("clear_guard_func"), &LimboState::clear_guard_func);
	ClassDB::bind_method(D_METHOD("get_blackboard"), &LimboState::get_blackboard);

	ClassDB::bind_method(D_METHOD("_set_blackboard_data", "p_blackboard"), &LimboState::_set_blackboard_data);
	ClassDB::bind_method(D_METHOD("_get_blackboard_data"), &LimboState::_get_blackboard_data);

	BIND_VMETHOD(MethodInfo("_setup"));
	BIND_VMETHOD(MethodInfo("_enter"));
	BIND_VMETHOD(MethodInfo("_exit"));
	BIND_VMETHOD(MethodInfo("_update", PropertyInfo(Variant::REAL, "p_delta")));

	ADD_PROPERTY(PropertyInfo(Variant::STRING, "EVENT_FINISHED", PROPERTY_HINT_NONE, "", 0), "", "event_finished");
	ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "agent", PROPERTY_HINT_RESOURCE_TYPE, "Object", 0), "set_agent", "get_agent");
	ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "blackboard", PROPERTY_HINT_RESOURCE_TYPE, "Blackboard", 0), "", "get_blackboard");
	ADD_PROPERTY(PropertyInfo(Variant::DICTIONARY, "_blackboard_data", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_INTERNAL), "_set_blackboard_data", "_get_blackboard_data");

	ADD_SIGNAL(MethodInfo("setup"));
	ADD_SIGNAL(MethodInfo("entered"));
	ADD_SIGNAL(MethodInfo("exited"));
	ADD_SIGNAL(MethodInfo("updated", PropertyInfo(Variant::REAL, "p_delta")));
};

LimboState::LimboState() {
	agent = nullptr;
	active = false;
	blackboard = Ref<Blackboard>(memnew(Blackboard));

	guard.obj = nullptr;

	set_process(false);
	set_physics_process(false);
	set_process_input(false);
}