limboai/bt/bt_state.h

/* bt_state.h */

#ifndef BT_STATE_H
#define BT_STATE_H

#include "core/object.h"
#include "modules/limboai/bt/behavior_tree.h"
#include "modules/limboai/bt/bt_task.h"
#include "modules/limboai/limbo_state.h"

class BTState : public LimboState {
	GDCLASS(BTState, LimboState);

private:
	Ref<BehaviorTree> behavior_tree;
	Ref<BTTask> root_task;
	String success_event;
	String failure_event;

protected:
	static void _bind_methods();

	virtual void _setup();
	virtual void _enter() {}
	virtual void _exit();
	virtual void _update(float p_delta);

public:
	void set_behavior_tree(const Ref<BehaviorTree> &p_value) { behavior_tree = p_value; }
	Ref<BehaviorTree> get_behavior_tree() const { return behavior_tree; }

	void set_success_event(String p_success_event) { success_event = p_success_event; }
	String get_success_event() const { return success_event; }

	void set_failure_event(String p_failure_event) { failure_event = p_failure_event; }
	String get_failure_event() const { return failure_event; }

	BTState();
};

#endif // BT_STATE_H