Add CBSLoopReaction

2025-12-13 16:06:07 +00:00 · 2022-11-22 01:25:57 +02:00
parent 286180ee0c
commit 36653c19b0
4 changed files with 173 additions and 7 deletions
--- a/src/MetroidPrime/BodyState/CBSLoopReaction.cpp
+++ b/src/MetroidPrime/BodyState/CBSLoopReaction.cpp
@@ -0,0 +1,143 @@
+#include "MetroidPrime/BodyState/CBSLoopReaction.hpp"
+
+#include "MetroidPrime/BodyState/CBodyController.hpp"
+#include "MetroidPrime/CAnimPlaybackParms.hpp"
+#include "MetroidPrime/CStateManager.hpp"
+
+#include "Kyoto/Animation/CPASAnimParmData.hpp"
+#include "Kyoto/Animation/CPASDatabase.hpp"
+
+CBSLoopReaction::CBSLoopReaction()
+: x4_state(pas::kLS_Invalid), x8_reactionType(pas::kRT_Invalid), xc_24_loopHit(false) {}
+
+void CBSLoopReaction::Start(CBodyController& bc, CStateManager& mgr) {
+  CBodyStateCmdMgr& commandMgr = bc.CommandMgr();
+
+  CBodyStateCmd* cmd = commandMgr.GetCmd(kBSC_LoopReaction);
+  if (cmd) {
+    x8_reactionType = static_cast< const CBCLoopReactionCmd* >(cmd)->GetReactionType();
+    xc_24_loopHit = false;
+  } else {
+    cmd = commandMgr.GetCmd(kBSC_LoopHitReaction);
+    x8_reactionType = static_cast< const CBCLoopHitReactionCmd* >(cmd)->GetReactionType();
+    xc_24_loopHit = true;
+  }
+
+  x4_state = pas::kLS_Begin;
+  const CPASAnimParmData parms(pas::kAS_LoopReaction, CPASAnimParm::FromEnum(x8_reactionType),
+                               CPASAnimParm::FromEnum(x4_state));
+  rstl::pair< float, int > best = bc.GetPASDatabase().FindBestAnimation(parms, *mgr.Random(), -1);
+
+  if (best.first > FLT_EPSILON) {
+    const CAnimPlaybackParms playParms(best.second, -1, 1.f, true);
+    bc.SetCurrentAnimation(playParms, false, false);
+  } else {
+    x4_state = pas::kLS_Loop;
+    CPASAnimParmData loopParms(pas::kAS_LoopReaction, CPASAnimParm::FromEnum(x8_reactionType),
+                               CPASAnimParm::FromEnum(x4_state));
+    bc.LoopBestAnimation(parms, *mgr.Random());
+  }
+}
+
+pas::EAnimationState CBSLoopReaction::UpdateBody(float dt, CBodyController& bc,
+                                                 CStateManager& mgr) {
+  pas::EAnimationState st = GetBodyStateTransition(dt, bc);
+  if (st == pas::kAS_Invalid) {
+    CBodyStateCmdMgr& commandMgr = bc.CommandMgr();
+
+    switch (x4_state) {
+    case pas::kLS_Begin:
+      if (commandMgr.GetCmd(kBSC_ExitState)) {
+        if (PlayExitAnimation(bc, mgr)) {
+          x4_state = pas::kLS_End;
+        } else {
+          x4_state = pas::kLS_Invalid;
+          st = pas::kAS_Locomotion;
+        }
+      } else {
+        if (bc.IsAnimationOver()) {
+          const CPASAnimParmData parms(pas::kAS_LoopReaction,
+                                       CPASAnimParm::FromEnum(s32(x8_reactionType)),
+                                       CPASAnimParm::FromEnum(1));
+          bc.LoopBestAnimation(parms, *mgr.Random());
+          x4_state = pas::kLS_Loop;
+        } else if (commandMgr.GetTargetVector().IsNonZero()) {
+          bc.FaceDirection(commandMgr.GetTargetVector(), dt);
+        }
+      }
+      break;
+    case pas::kLS_Loop:
+      if (commandMgr.GetCmd(kBSC_ExitState)) {
+        if (PlayExitAnimation(bc, mgr)) {
+          x4_state = pas::kLS_End;
+        } else {
+          x4_state = pas::kLS_Invalid;
+          st = pas::kAS_Locomotion;
+        }
+      } else if (commandMgr.GetTargetVector().IsNonZero()) {
+        bc.FaceDirection(commandMgr.GetTargetVector(), dt);
+      }
+      break;
+    case pas::kLS_End:
+      if (bc.IsAnimationOver()) {
+        x4_state = pas::kLS_Invalid;
+        st = pas::kAS_Locomotion;
+      }
+      break;
+    default:
+      break;
+    }
+  }
+  return st;
+}
+
+bool CBSLoopReaction::PlayExitAnimation(CBodyController& bc, CStateManager& mgr) const {
+  const CPASDatabase& db = bc.GetPASDatabase();
+
+  const CPASAnimParmData parms(pas::kAS_LoopReaction, CPASAnimParm::FromEnum(int(x8_reactionType)),
+                               CPASAnimParm::FromEnum(2));
+  const rstl::pair< float, int > best = db.FindBestAnimation(parms, *mgr.Random(), -1);
+  if (best.first > 0.f) {
+    const CAnimPlaybackParms playParms(best.second, -1, 1.f, true);
+    bc.SetCurrentAnimation(playParms, false, false);
+    return true;
+  }
+  return false;
+}
+pas::EAnimationState CBSLoopReaction::GetBodyStateTransition(float dt, CBodyController& bc) {
+  CBodyStateCmdMgr& commandMgr = bc.CommandMgr();
+  if (commandMgr.GetCmd(kBSC_Hurled)) {
+    return pas::kAS_Hurled;
+  }
+  if (commandMgr.GetCmd(kBSC_KnockDown)) {
+    return pas::kAS_Fall;
+  }
+  if (commandMgr.GetCmd(kBSC_KnockBack)) {
+    return pas::kAS_KnockBack;
+  }
+  if (!xc_24_loopHit && commandMgr.GetCmd(kBSC_Locomotion)) {
+    return pas::kAS_Locomotion;
+  }
+
+  if (x4_state == pas::kLS_End) {
+    if (commandMgr.GetCmd(kBSC_MeleeAttack)) {
+      return pas::kAS_MeleeAttack;
+    }
+    if (commandMgr.GetCmd(kBSC_ProjectileAttack)) {
+      return pas::kAS_ProjectileAttack;
+    }
+    if (commandMgr.GetCmd(kBSC_LoopAttack)) {
+      return pas::kAS_LoopAttack;
+    }
+    if (commandMgr.GetCmd(kBSC_Step)) {
+      return pas::kAS_Step;
+    }
+    if (commandMgr.GetMoveVector().IsNonZero()) {
+      return pas::kAS_Locomotion;
+    }
+    if (commandMgr.GetFaceVector().IsNonZero()) {
+      return pas::kAS_Turn;
+    }
+  }
+  return pas::kAS_Invalid;
+}