Add scene viewport move gizmo workflow

2026-03-29 16:18:13 +08:00
parent 2651bad080
commit 0ea1cb29e6
11 changed files with 749 additions and 13 deletions
--- a/tests/editor/test_scene_viewport_move_gizmo.cpp
+++ b/tests/editor/test_scene_viewport_move_gizmo.cpp
@@ -0,0 +1,96 @@
+#include <gtest/gtest.h>
+
+#include "Core/EditorContext.h"
+#include "Managers/SceneManager.h"
+#include "Viewport/SceneViewportMoveGizmo.h"
+
+namespace XCEngine::Editor {
+namespace {
+
+class SceneViewportMoveGizmoTest : public ::testing::Test {
+protected:
+    void SetUp() override {
+        m_context.GetSceneManager().NewScene("Move Gizmo Test Scene");
+    }
+
+    static SceneViewportOverlayData MakeOverlay() {
+        SceneViewportOverlayData overlay = {};
+        overlay.valid = true;
+        overlay.cameraPosition = Math::Vector3(0.0f, 0.0f, -5.0f);
+        overlay.cameraForward = Math::Vector3::Forward();
+        overlay.cameraRight = Math::Vector3::Right();
+        overlay.cameraUp = Math::Vector3::Up();
+        overlay.verticalFovDegrees = 60.0f;
+        overlay.nearClipPlane = 0.03f;
+        overlay.farClipPlane = 2000.0f;
+        return overlay;
+    }
+
+    static SceneViewportMoveGizmoContext MakeContext(
+        Components::GameObject* selectedObject,
+        const Math::Vector2& mousePosition) {
+        SceneViewportMoveGizmoContext context = {};
+        context.overlay = MakeOverlay();
+        context.viewportSize = Math::Vector2(800.0f, 600.0f);
+        context.mousePosition = mousePosition;
+        context.selectedObject = selectedObject;
+        return context;
+    }
+
+    SceneManager& GetSceneManager() {
+        return dynamic_cast<SceneManager&>(m_context.GetSceneManager());
+    }
+
+    EditorContext m_context;
+};
+
+TEST_F(SceneViewportMoveGizmoTest, UpdateHighlightsXAxisWhenMouseIsNearXAxisHandle) {
+    Components::GameObject* target = GetSceneManager().CreateEntity("Target");
+    ASSERT_NE(target, nullptr);
+    m_context.GetSelectionManager().SetSelectedEntity(target->GetID());
+
+    SceneViewportMoveGizmo gizmo;
+    gizmo.Update(MakeContext(target, Math::Vector2(436.0f, 300.0f)));
+
+    ASSERT_TRUE(gizmo.GetDrawData().visible);
+    EXPECT_TRUE(gizmo.IsHoveringHandle());
+    EXPECT_TRUE(gizmo.GetDrawData().handles[0].hovered);
+    EXPECT_FALSE(gizmo.GetDrawData().handles[1].hovered);
+    EXPECT_FALSE(gizmo.GetDrawData().handles[2].hovered);
+}
+
+TEST_F(SceneViewportMoveGizmoTest, DraggingXAxisOnlyChangesWorldXAndCreatesUndoStep) {
+    Components::GameObject* target = GetSceneManager().CreateEntity("Target");
+    ASSERT_NE(target, nullptr);
+    const uint64_t targetId = target->GetID();
+    m_context.GetSelectionManager().SetSelectedEntity(target->GetID());
+
+    SceneViewportMoveGizmo gizmo;
+    const auto startContext = MakeContext(target, Math::Vector2(436.0f, 300.0f));
+    gizmo.Update(startContext);
+
+    ASSERT_TRUE(gizmo.TryBeginDrag(startContext, m_context.GetUndoManager()));
+    ASSERT_TRUE(gizmo.IsActive());
+
+    const auto dragContext = MakeContext(target, Math::Vector2(500.0f, 300.0f));
+    gizmo.Update(dragContext);
+    gizmo.UpdateDrag(dragContext);
+    gizmo.EndDrag(m_context.GetUndoManager());
+
+    const Math::Vector3 movedPosition = target->GetTransform()->GetPosition();
+    EXPECT_GT(movedPosition.x, 0.05f);
+    EXPECT_NEAR(movedPosition.y, 0.0f, 1e-4f);
+    EXPECT_NEAR(movedPosition.z, 0.0f, 1e-4f);
+    EXPECT_TRUE(m_context.GetUndoManager().CanUndo());
+
+    m_context.GetUndoManager().Undo();
+    Components::GameObject* restoredTarget = GetSceneManager().GetEntity(targetId);
+    ASSERT_NE(restoredTarget, nullptr);
+    const Math::Vector3 restoredPosition = restoredTarget->GetTransform()->GetPosition();
+    EXPECT_NEAR(restoredPosition.x, 0.0f, 1e-4f);
+    EXPECT_NEAR(restoredPosition.y, 0.0f, 1e-4f);
+    EXPECT_NEAR(restoredPosition.z, 0.0f, 1e-4f);
+}
+
+} // namespace
+} // namespace XCEngine::Editor