limboai/bt/composites/bt_parallel.cpp

/* bt_parallel.cpp */

#include "bt_parallel.h"

#include "core/object/class_db.h"

void BTParallel::_enter() {
	for (int i = 0; i < get_child_count(); i++) {
		get_child(i)->cancel();
	}
}

int BTParallel::_tick(double p_delta) {
	int num_succeeded = 0;
	int num_failed = 0;
	int return_status = RUNNING;
	for (int i = 0; i < get_child_count(); i++) {
		int status = 0;
		Ref<BTTask> child = get_child(i);
		if (!repeat && (child->get_status() == FAILURE or child->get_status() == SUCCESS)) {
			status = child->get_status();
		} else {
			status = child->execute(p_delta);
		}
		if (status == FAILURE) {
			num_failed += 1;
			if (num_failed >= num_failures_required && return_status == RUNNING) {
				return_status = FAILURE;
			}
		} else if (status == SUCCESS) {
			num_succeeded += 1;
			if (num_succeeded >= num_successes_required && return_status == RUNNING) {
				return_status = SUCCESS;
			}
		}
	}
	if (!repeat && (num_failed + num_succeeded) == get_child_count() && return_status == RUNNING) {
		return_status = FAILURE;
	}
	return return_status;
}

void BTParallel::_bind_methods() {
	ClassDB::bind_method(D_METHOD("get_num_successes_required"), &BTParallel::get_num_successes_required);
	ClassDB::bind_method(D_METHOD("set_num_successes_required", "p_value"), &BTParallel::set_num_successes_required);
	ClassDB::bind_method(D_METHOD("get_num_failures_required"), &BTParallel::get_num_failures_required);
	ClassDB::bind_method(D_METHOD("set_num_failures_required", "p_value"), &BTParallel::set_num_failures_required);
	ClassDB::bind_method(D_METHOD("get_repeat"), &BTParallel::get_repeat);
	ClassDB::bind_method(D_METHOD("set_repeat", "p_value"), &BTParallel::set_repeat);

	ADD_PROPERTY(PropertyInfo(Variant::INT, "num_successes_required"), "set_num_successes_required", "get_num_successes_required");
	ADD_PROPERTY(PropertyInfo(Variant::INT, "num_failures_required"), "set_num_failures_required", "get_num_failures_required");
	ADD_PROPERTY(PropertyInfo(Variant::BOOL, "repeat"), "set_repeat", "get_repeat");
}