UObjectHook: Auto-set BP MotionController transforms

praydog · Dec 31, 2023 · 7c34ec3 · 7c34ec3
1 parent 8a35381
commit 7c34ec3
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 0 deletions.
diff --git a/src/mods/UObjectHook.cpp b/src/mods/UObjectHook.cpp
@@ -23,6 +23,7 @@
 #include <sdk/FBoolProperty.hpp>
 #include <sdk/FObjectProperty.hpp>
 #include <sdk/FArrayProperty.hpp>
+#include <sdk/UMotionControllerComponent.hpp>
 
 #include "uobjecthook/SDKDumper.hpp"
 #include "VR.hpp"
@@ -474,6 +475,8 @@ void UObjectHook::tick_attachments(Rotator<float>* view_rotation, const float wo
         return result;
     };
 
+    update_motion_controller_components(left_hand_position, left_hand_euler, right_hand_position, right_hand_euler);
+
     auto comps = with_mutex([this]() { return m_motion_controller_attached_components; });
 
     sdk::TArray<sdk::UPrimitiveComponent*> overlapped_components{};
@@ -1309,6 +1312,57 @@ void UObjectHook::update_persistent_states() {
     }
 }
 
+void UObjectHook::update_motion_controller_components(const glm::vec3& left_hand_location, const glm::vec3& left_hand_euler,
+                                                      const glm::vec3& right_hand_location, const glm::vec3& right_hand_euler) 
+{
+    const auto mc_c = sdk::UMotionControllerComponent::static_class();
+
+    if (mc_c == nullptr) {
+        SPDLOG_ERROR_ONCE("[UObjectHook] Failed to find MotionControllerComponent class, cannot update motion controller components");
+        return;
+    }
+
+    const auto motion_controllers = this->get_objects_by_class(mc_c);
+
+    if (motion_controllers.empty()) {
+        return;
+    }
+
+    for (auto obj : motion_controllers) {
+        auto mc = (sdk::UMotionControllerComponent*)obj;
+        if (!this->exists(mc)) {
+            continue;
+        }
+
+        if (mc->get_player_index() > 0) {
+            continue;
+        }
+
+        if (mc->has_motion_source()) {
+            const auto& motion_source = mc->get_motion_source();
+            const auto motion_source_str = motion_source.to_string();
+
+            if (motion_source_str == L"Left") {
+                mc->set_world_location(left_hand_location, false, false);
+                mc->set_world_rotation(left_hand_euler, false);
+            } else if (motion_source_str == L"Right") {
+                mc->set_world_location(right_hand_location, false, false);
+                mc->set_world_rotation(right_hand_euler, false);
+            } /*else {
+                SPDLOG_INFO("[UObjectHook] Unknown motion source {}", utility::narrow(motion_source_str));
+            }*/
+        } else {
+            if (mc->get_hand() == sdk::EControllerHand::Left) {
+                mc->set_world_location(left_hand_location, false, false);
+                mc->set_world_rotation(left_hand_euler, false);
+            } else if (mc->get_hand() == sdk::EControllerHand::Right) {
+                mc->set_world_location(right_hand_location, false, false);
+                mc->set_world_rotation(right_hand_euler, false);
+            }
+        }
+    }
+}
+
 sdk::UObject* UObjectHook::StatePath::resolve_base_object() const {
     if (!this->has_valid_base()) {
         return nullptr;

diff --git a/src/mods/UObjectHook.hpp b/src/mods/UObjectHook.hpp
@@ -125,6 +125,8 @@ class UObjectHook : public Mod {
     std::shared_ptr<PersistentCameraState> deserialize_camera(const nlohmann::json& data);
     std::shared_ptr<PersistentCameraState> deserialize_camera_state();
     void update_persistent_states();
+    void update_motion_controller_components(const glm::vec3& left_hand_location, const glm::vec3& left_hand_euler,
+                                             const glm::vec3& right_hand_location, const glm::vec3& right_hand_euler);
 
     static void* add_object(void* rcx, void* rdx, void* r8, void* r9);
     static void* destructor(sdk::UObjectBase* object, void* rdx, void* r8, void* r9);