mirror of https://github.com/AxioDL/metaforce.git
126 lines
3.3 KiB
C++
126 lines
3.3 KiB
C++
#pragma once
|
|
|
|
#include <vector>
|
|
|
|
#include "Runtime/CToken.hpp"
|
|
#include "Runtime/GCNTypes.hpp"
|
|
#include "Runtime/IObj.hpp"
|
|
#include "Runtime/IObjFactory.hpp"
|
|
#include "Runtime/IOStreams.hpp"
|
|
#include "Runtime/World/CAiFuncMap.hpp"
|
|
|
|
namespace urde {
|
|
class CAiState;
|
|
class CStateManager;
|
|
|
|
class CAiTrigger {
|
|
CAiTriggerFunc x0_func;
|
|
float xc_arg = 0.f;
|
|
CAiTrigger* x10_andTrig = nullptr;
|
|
CAiState* x14_state = nullptr;
|
|
bool x18_lNot = false;
|
|
|
|
public:
|
|
CAiTrigger() = default;
|
|
CAiTrigger* GetAnd() const { return x10_andTrig; }
|
|
CAiState* GetState() const { return x14_state; }
|
|
bool CallFunc(CStateManager& mgr, CAi& ai) const {
|
|
if (x0_func) {
|
|
bool ret = (ai.*x0_func)(mgr, xc_arg);
|
|
return x18_lNot == !ret;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void Setup(CAiTriggerFunc func, bool lnot, float arg, CAiTrigger* andTrig) {
|
|
x0_func = func;
|
|
x18_lNot = lnot;
|
|
xc_arg = arg;
|
|
x10_andTrig = andTrig;
|
|
}
|
|
void Setup(CAiTriggerFunc func, bool lnot, float arg, CAiState* state) {
|
|
x0_func = func;
|
|
x18_lNot = lnot;
|
|
xc_arg = arg;
|
|
x14_state = state;
|
|
}
|
|
};
|
|
|
|
class CAiState {
|
|
friend class CStateMachineState;
|
|
CAiStateFunc x0_func;
|
|
char xc_name[32] = {};
|
|
u32 x2c_numTriggers;
|
|
CAiTrigger* x30_firstTrigger;
|
|
|
|
public:
|
|
CAiState(CAiStateFunc func, const char* name) {
|
|
x0_func = func;
|
|
strncpy(xc_name, name, 31);
|
|
}
|
|
|
|
s32 GetNumTriggers() const { return x2c_numTriggers; }
|
|
CAiTrigger* GetTrig(s32 i) const { return &x30_firstTrigger[i]; }
|
|
const char* GetName() const { return xc_name; }
|
|
void SetTriggers(CAiTrigger* triggers) { x30_firstTrigger = triggers; }
|
|
void SetNumTriggers(s32 numTriggers) { x2c_numTriggers = numTriggers; }
|
|
void CallFunc(CStateManager& mgr, CAi& ai, EStateMsg msg, float delta) const {
|
|
if (x0_func)
|
|
(ai.*x0_func)(mgr, msg, delta);
|
|
}
|
|
};
|
|
|
|
class CStateMachine {
|
|
std::vector<CAiState> x0_states;
|
|
std::vector<CAiTrigger> x10_triggers;
|
|
|
|
public:
|
|
explicit CStateMachine(CInputStream& in);
|
|
|
|
s32 GetStateIndex(std::string_view state) const;
|
|
const std::vector<CAiState>& GetStateVector() const { return x0_states; }
|
|
};
|
|
|
|
class CStateMachineState {
|
|
friend class CPatterned;
|
|
const CStateMachine* x0_machine = nullptr;
|
|
CAiState* x4_state = nullptr;
|
|
float x8_time = 0.f;
|
|
float xc_random = 0.f;
|
|
float x10_delay = 0.f;
|
|
float x14_;
|
|
union {
|
|
struct {
|
|
bool x18_24_codeTrigger : 1;
|
|
};
|
|
u32 dummy = 0;
|
|
};
|
|
|
|
public:
|
|
CStateMachineState() = default;
|
|
|
|
CAiState* GetActorState() const { return x4_state; }
|
|
|
|
void Update(CStateManager& mgr, CAi& ai, float delta);
|
|
void SetState(CStateManager&, CAi&, s32);
|
|
void SetState(CStateManager&, CAi&, const CStateMachine*, std::string_view);
|
|
const std::vector<CAiState>* GetStateVector() const;
|
|
void Setup(const CStateMachine* machine);
|
|
void SetDelay(float delay) { x10_delay = delay; }
|
|
float GetTime() const { return x8_time; }
|
|
float GetRandom() const { return xc_random; }
|
|
float GetDelay() const { return x10_delay; }
|
|
void SetCodeTrigger() { x18_24_codeTrigger = true; }
|
|
|
|
const char* GetName() const {
|
|
if (x4_state)
|
|
return x4_state->GetName();
|
|
return nullptr;
|
|
}
|
|
};
|
|
|
|
CFactoryFnReturn FAiFiniteStateMachineFactory(const SObjectTag& tag, CInputStream& in, const CVParamTransfer& vparms,
|
|
CObjectReference*);
|
|
|
|
} // namespace urde
|