mirror of https://github.com/AxioDL/metaforce.git
136 lines
4.1 KiB
C++
136 lines
4.1 KiB
C++
#include "CStateMachine.hpp"
|
|
#include "CAi.hpp"
|
|
#include "CStateManager.hpp"
|
|
|
|
namespace urde {
|
|
CStateMachine::CStateMachine(CInputStream& in) {
|
|
CAiTrigger* lastTrig = nullptr;
|
|
u32 stateCount = in.readUint32Big();
|
|
|
|
x0_states.reserve(stateCount);
|
|
|
|
for (u32 i = 0; i < stateCount; ++i) {
|
|
std::string name = in.readString(31, false);
|
|
CAiStateFunc func = CAi::GetStateFunc(name.c_str());
|
|
x0_states.emplace_back(func, name.c_str());
|
|
}
|
|
|
|
x10_triggers.reserve(in.readUint32Big());
|
|
|
|
for (u32 i = 0; i < stateCount; ++i) {
|
|
x0_states[i].SetNumTriggers(in.readUint32Big());
|
|
if (x0_states[i].GetNumTriggers() == 0)
|
|
continue;
|
|
CAiTrigger* firstTrig = x10_triggers.data() + x10_triggers.size();
|
|
x0_states[i].SetTriggers(firstTrig);
|
|
x10_triggers.resize(x10_triggers.size() + x0_states[i].GetNumTriggers());
|
|
|
|
for (u32 j = 0; j < x0_states[i].GetNumTriggers(); ++j) {
|
|
u32 triggerCount = in.readUint32Big();
|
|
u32 lastTriggerIdx = triggerCount - 1;
|
|
for (u32 k = 0; k < triggerCount; ++k) {
|
|
std::string name = in.readString(31, false);
|
|
bool isNot = name.front() == '!';
|
|
CAiTriggerFunc func = CAi::GetTrigerFunc(isNot ? name.c_str() + 1 : name.c_str());
|
|
float arg = in.readFloatBig();
|
|
CAiTrigger* newTrig;
|
|
if (k < lastTriggerIdx) {
|
|
x10_triggers.emplace_back();
|
|
newTrig = &x10_triggers.back();
|
|
} else {
|
|
newTrig = &firstTrig[j];
|
|
}
|
|
if (k == 0)
|
|
newTrig->Setup(func, isNot, arg, &x0_states[in.readUint32Big()]);
|
|
else
|
|
newTrig->Setup(func, isNot, arg, lastTrig);
|
|
lastTrig = newTrig;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
s32 CStateMachine::GetStateIndex(std::string_view state) const {
|
|
auto it = std::find_if(x0_states.begin(), x0_states.end(),
|
|
[state](const CAiState& st) { return (strncmp(st.GetName(), state.data(), 31) == 0); });
|
|
if (it == x0_states.end())
|
|
return 0;
|
|
|
|
return it - x0_states.begin();
|
|
}
|
|
|
|
void CStateMachineState::Update(CStateManager& mgr, CAi& ai, float delta) {
|
|
if (x4_state) {
|
|
x8_time += delta;
|
|
x4_state->CallFunc(mgr, ai, EStateMsg::Update, delta);
|
|
for (int i = 0; i < x4_state->GetNumTriggers(); ++i) {
|
|
CAiTrigger* trig = x4_state->GetTrig(i);
|
|
CAiState* state = nullptr;
|
|
bool andPassed = true;
|
|
while (andPassed && trig) {
|
|
andPassed = false;
|
|
if (trig->CallFunc(mgr, ai)) {
|
|
andPassed = true;
|
|
state = trig->GetState();
|
|
trig = trig->GetAnd();
|
|
}
|
|
}
|
|
if (andPassed && state) {
|
|
x4_state->CallFunc(mgr, ai, EStateMsg::Deactivate, 0.f);
|
|
x4_state = state;
|
|
x8_time = 0.f;
|
|
x18_24_codeTrigger = false;
|
|
xc_random = mgr.GetActiveRandom()->Float();
|
|
x4_state->CallFunc(mgr, ai, EStateMsg::Activate, delta);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void CStateMachineState::SetState(CStateManager& mgr, CAi& ai, s32 idx) {
|
|
CAiState* state = const_cast<CAiState*>(&x0_machine->GetStateVector()[idx]);
|
|
if (x4_state != state) {
|
|
if (x4_state)
|
|
x4_state->CallFunc(mgr, ai, EStateMsg::Deactivate, 0.f);
|
|
x4_state = state;
|
|
x8_time = 0.f;
|
|
xc_random = mgr.GetActiveRandom()->Float();
|
|
x18_24_codeTrigger = false;
|
|
x4_state->CallFunc(mgr, ai, EStateMsg::Activate, 0.f);
|
|
}
|
|
}
|
|
|
|
void CStateMachineState::SetState(CStateManager& mgr, CAi& ai, const CStateMachine* machine, std::string_view state) {
|
|
if (!machine)
|
|
return;
|
|
|
|
if (!x0_machine)
|
|
x0_machine = machine;
|
|
|
|
s32 idx = machine->GetStateIndex(state);
|
|
SetState(mgr, ai, idx);
|
|
}
|
|
|
|
const std::vector<CAiState>* CStateMachineState::GetStateVector() const {
|
|
if (!x0_machine)
|
|
return nullptr;
|
|
|
|
return &x0_machine->GetStateVector();
|
|
}
|
|
|
|
void CStateMachineState::Setup(const CStateMachine* machine) {
|
|
x0_machine = machine;
|
|
x4_state = nullptr;
|
|
x8_time = 0.f;
|
|
xc_random = 0.f;
|
|
x10_delay = 0.f;
|
|
}
|
|
|
|
CFactoryFnReturn FAiFiniteStateMachineFactory(const SObjectTag& tag, CInputStream& in, const CVParamTransfer& vparms,
|
|
CObjectReference*) {
|
|
return TToken<CStateMachine>::GetIObjObjectFor(std::make_unique<CStateMachine>(in));
|
|
}
|
|
|
|
} // namespace urde
|