feat: werewolf and vampire lord sync

tiltedphoques · Mar 16, 2022 · d0a387d · d0a387d
1 parent 85eda1c
commit d0a387d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/Code/client/Games/Skyrim/Magic/MagicTarget.cpp b/Code/client/Games/Skyrim/Magic/MagicTarget.cpp
@@ -6,6 +6,9 @@
 #include <Games/ActorExtension.h>
 #include "EffectItem.h"
 
+#include <Structs/Skyrim/AnimationGraphDescriptor_WerewolfBehavior.h>
+#include <Structs/Skyrim/AnimationGraphDescriptor_VampireLordBehavior.h>
+
 #include <Events/AddTargetEvent.h>
 
 TP_THIS_FUNCTION(TAddTarget, bool, MagicTarget, MagicTarget::AddTargetData& arData);
@@ -36,6 +39,12 @@ bool TP_MAKE_THISCALL(HookAddTarget, MagicTarget, MagicTarget::AddTargetData& ar
     if (!pTargetActorEx)
         return ThisCall(RealAddTarget, apThis, arData);
 
+    if (arData.pEffectItem->IsWerewolfEffect())
+        pTargetActorEx->GraphDescriptorHash = AnimationGraphDescriptor_WerewolfBehavior::m_key;
+
+    if (arData.pEffectItem->IsVampireLordEffect())
+        pTargetActorEx->GraphDescriptorHash = AnimationGraphDescriptor_VampireLordBehavior::m_key;
+
     AddTargetEvent addTargetEvent{};
     addTargetEvent.TargetID = pTargetActor->formID;
     addTargetEvent.SpellID = arData.pSpell->formID;

diff --git a/Code/client/Games/Skyrim/PlayerCharacter.cpp b/Code/client/Games/Skyrim/PlayerCharacter.cpp
@@ -26,11 +26,7 @@ void TP_MAKE_THISCALL(HookSetBeastForm, void, void* apUnk1, void* apUnk2, bool a
 {
     spdlog::warn("Setting beast form: {}", aEntering);
 
-    BSAnimationGraphManager* pManager = nullptr;
-    PlayerCharacter::Get()->animationGraphHolder.GetBSAnimationGraph(&pManager);
-    auto hash = pManager->GetDescriptorKey();
-    auto pDescriptor = AnimationGraphDescriptorManager::Get().GetDescriptor(hash);
-    spdlog::info("Key: {}", hash);
+    // TODO: send respawn event
 
     ThisCall(RealSetBeastForm, apThis, apUnk1, apUnk2, aEntering);
 }

diff --git a/Code/client/Services/Debug/Views/AnimDebugView.cpp b/Code/client/Services/Debug/Views/AnimDebugView.cpp
@@ -11,6 +11,7 @@
 #include <Games/Skyrim/Misc/InventoryEntry.h>
 #include <Games/Skyrim/Misc/MiddleProcess.h>
 #include <Games/Skyrim/Actor.h>
+#include <Games/ActorExtension.h>
 #endif
 
 #if (TP_FALLOUT4)
@@ -69,6 +70,8 @@ void TestService::DrawAnimDebugView()
         return;
     }
 
+    spdlog::info("{}", pActor->GetExtension()->GraphDescriptorHash);
+
     if (ImGui::Button("Clear all"))
     {
         s_varMap.clear();