limboai/hsm/limbo_hsm.h

/**
 * limbo_hsm.h
 * =============================================================================
 * 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.
 * =============================================================================
 */

#ifndef LIMBO_HSM_H
#define LIMBO_HSM_H

#include "limbo_state.h"

#include "core/object/object.h"
#include "core/templates/hash_map.h"

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

public:
	enum UpdateMode : unsigned int {
		IDLE, // automatically call update() during NOTIFICATION_PROCESS
		PHYSICS, // automatically call update() during NOTIFICATION_PHYSICS
		MANUAL, // manually update state machine: user must call update(delta)
	};

private:
	UpdateMode update_mode;
	LimboState *initial_state;
	LimboState *active_state;
	HashMap<uint64_t, LimboState *> transitions;

	_FORCE_INLINE_ uint64_t _get_transition_key(Node *p_from_state, const String &p_event) {
		uint64_t key = hash_djb2_one_64(Variant::OBJECT);
		key = hash_djb2_one_64(Variant(p_from_state).hash(), key);
		key = hash_djb2_one_64(p_event.hash(), key);
		return key;
	}

protected:
	static void _bind_methods();

	void _notification(int p_what);

	virtual void _initialize(Node *p_agent, const Ref<Blackboard> &p_blackboard) override;
	virtual void _enter() override;
	virtual void _exit() override;
	virtual void _update(double p_delta) override;

	void _change_state(LimboState *p_state);

public:
	void set_update_mode(UpdateMode p_mode) { update_mode = p_mode; }
	UpdateMode get_update_mode() const { return update_mode; }

	LimboState *get_active_state() const { return active_state; };
	LimboState *get_leaf_state() const;
	void set_active(bool p_active);

	void set_initial_state(Node *p_state);
	LimboState *get_initial_state() const { return initial_state; }

	virtual void initialize(Node *p_agent, const Ref<Blackboard> &p_parent_scope = nullptr);
	virtual bool dispatch(const String &p_event, const Variant &p_cargo) override;

	void update(double p_delta) { _update(p_delta); }
	void add_transition(Node *p_from_state, Node *p_to_state, const String &p_event);
	// void add_transition_from_any_state(Node *p_to_state, const String &p_event);
	LimboState *anystate() const { return nullptr; };

	LimboHSM();
};

#endif // LIMBO_HSM_H