limboai/bt/decorators/bt_run_limit.cpp

39 lines
1.1 KiB
C++

/**
* bt_run_limit.cpp
* =============================================================================
* Copyright 2021-2023 Serhii Snitsaruk
*
* Use of this source code is governed by an MIT-style
* license that can be found in the LICENSE file or at
* https://opensource.org/licenses/MIT.
* =============================================================================
*/
#include "bt_run_limit.h"
void BTRunLimit::set_run_limit(int p_value) {
run_limit = p_value;
emit_changed();
}
String BTRunLimit::_generate_name() const {
return vformat("RunLimit x%d", run_limit);
}
int BTRunLimit::_tick(double p_delta) {
ERR_FAIL_COND_V_MSG(get_child_count() == 0, FAILURE, "BT decorator has no child.");
if (get_child(0)->get_status() != RUNNING) {
if (num_runs >= run_limit) {
return FAILURE;
}
num_runs += 1;
}
return get_child(0)->execute(p_delta);
}
void BTRunLimit::_bind_methods() {
ClassDB::bind_method(D_METHOD("set_run_limit", "p_value"), &BTRunLimit::set_run_limit);
ClassDB::bind_method(D_METHOD("get_run_limit"), &BTRunLimit::get_run_limit);
ADD_PROPERTY(PropertyInfo(Variant::INT, "run_limit"), "set_run_limit", "get_run_limit");
}