Formalize scene viewport interaction frame helpers

2026-04-04 01:42:35 +08:00
parent a920ca7a6a
commit e636abb76d
5 changed files with 531 additions and 79 deletions
--- a/editor/src/Viewport/SceneViewportInteractionFrame.h
+++ b/editor/src/Viewport/SceneViewportInteractionFrame.h
@@ -0,0 +1,132 @@
+#pragma once
+
+#include "Core/IEditorContext.h"
+#include "IViewportHostService.h"
+#include "SceneViewportEditorModes.h"
+#include "SceneViewportHudOverlay.h"
+#include "SceneViewportInteractionResolver.h"
+#include "SceneViewportTransformGizmoCoordinator.h"
+
+namespace XCEngine {
+namespace Editor {
+
+struct SceneViewportToolState {
+    bool usingViewMoveTool = false;
+    bool usingTransformTool = false;
+    bool showingMoveGizmo = false;
+    bool showingRotateGizmo = false;
+    bool showingScaleGizmo = false;
+    bool useCenterPivot = false;
+    bool localSpace = false;
+    SceneViewportTransformGizmoFrameOptions gizmoFrameOptions = {};
+};
+
+inline SceneViewportToolState BuildSceneViewportToolState(
+    SceneViewportToolMode toolMode,
+    SceneViewportPivotMode pivotMode,
+    SceneViewportTransformSpaceMode transformSpaceMode) {
+    SceneViewportToolState state = {};
+    state.usingViewMoveTool = toolMode == SceneViewportToolMode::ViewMove;
+    state.usingTransformTool = toolMode == SceneViewportToolMode::Transform;
+    state.showingMoveGizmo = toolMode == SceneViewportToolMode::Move || state.usingTransformTool;
+    state.showingRotateGizmo = toolMode == SceneViewportToolMode::Rotate || state.usingTransformTool;
+    state.showingScaleGizmo = toolMode == SceneViewportToolMode::Scale || state.usingTransformTool;
+    state.useCenterPivot = pivotMode == SceneViewportPivotMode::Center;
+    state.localSpace = transformSpaceMode == SceneViewportTransformSpaceMode::Local;
+    state.gizmoFrameOptions = BuildSceneViewportTransformGizmoFrameOptions(
+        state.useCenterPivot,
+        state.localSpace,
+        state.usingTransformTool,
+        state.showingMoveGizmo,
+        state.showingRotateGizmo,
+        state.showingScaleGizmo);
+    return state;
+}
+
+struct SceneViewportFrameGeometry {
+    Math::Vector2 viewportSize = Math::Vector2::Zero();
+    Math::Vector2 localMousePosition = Math::Vector2::Zero();
+};
+
+inline SceneViewportFrameGeometry BuildSceneViewportFrameGeometry(
+    const ImVec2& viewportSize,
+    const ImVec2& viewportMin,
+    const ImVec2& absoluteMousePosition) {
+    SceneViewportFrameGeometry geometry = {};
+    geometry.viewportSize = Math::Vector2(viewportSize.x, viewportSize.y);
+    geometry.localMousePosition = Math::Vector2(
+        absoluteMousePosition.x - viewportMin.x,
+        absoluteMousePosition.y - viewportMin.y);
+    return geometry;
+}
+
+struct SceneViewportInteractionFrameState {
+    bool hasInteractiveViewport = false;
+    SceneViewportOverlayData overlay = {};
+    SceneViewportTransformGizmoFrameUpdate gizmoFrameUpdate = {};
+    SceneViewportTransformGizmoFrameState gizmoFrameState = {};
+    const SceneViewportOverlayFrameData* overlayFrameData = nullptr;
+    SceneViewportActiveGizmoKind activeGizmoKind = SceneViewportActiveGizmoKind::None;
+    bool gizmoActive = false;
+    SceneViewportHudOverlayData hudOverlay = {};
+};
+
+inline SceneViewportInteractionFrameState BuildSceneViewportInteractionFrameState(
+    IEditorContext& context,
+    IViewportHostService& viewportHostService,
+    bool hasInteractiveViewport,
+    const SceneViewportFrameGeometry& geometry,
+    const SceneViewportTransformGizmoFrameOptions& gizmoFrameOptions,
+    SceneViewportMoveGizmo& moveGizmo,
+    SceneViewportRotateGizmo& rotateGizmo,
+    SceneViewportScaleGizmo& scaleGizmo,
+    const SceneViewportOverlayFrameData& emptyOverlayFrameData) {
+    SceneViewportInteractionFrameState state = {};
+    state.hasInteractiveViewport = hasInteractiveViewport;
+    state.overlayFrameData = &emptyOverlayFrameData;
+
+    if (!hasInteractiveViewport) {
+        CancelSceneViewportTransformGizmoFrame(context, moveGizmo, rotateGizmo, scaleGizmo);
+        state.hudOverlay = BuildSceneViewportHudOverlayData(state.overlay);
+        return state;
+    }
+
+    state.overlay = viewportHostService.GetSceneViewOverlayData();
+    state.gizmoFrameUpdate = RefreshAndSubmitSceneViewportTransformGizmoFrame(
+        viewportHostService,
+        BuildSceneViewportTransformGizmoRefreshRequest(
+            context,
+            state.overlay,
+            geometry.viewportSize,
+            geometry.localMousePosition,
+            gizmoFrameOptions),
+        moveGizmo,
+        rotateGizmo,
+        scaleGizmo);
+    state.gizmoFrameState = state.gizmoFrameUpdate.frameState;
+    state.overlayFrameData = &viewportHostService.GetSceneViewEditorOverlayFrameData(context);
+    state.activeGizmoKind = state.gizmoFrameUpdate.overlaySubmission.activeGizmoKind;
+    state.gizmoActive = state.gizmoFrameUpdate.overlaySubmission.GizmoActive();
+    state.hudOverlay = BuildSceneViewportHudOverlayData(state.overlay);
+    return state;
+}
+
+inline SceneViewportInteractionResolveRequest BuildSceneViewportInteractionResolveRequest(
+    const SceneViewportInteractionFrameState& frameState,
+    const SceneViewportFrameGeometry& geometry,
+    const ImVec2& viewportMin,
+    const ImVec2& viewportMax,
+    const ImVec2& absoluteMousePosition) {
+    SceneViewportInteractionResolveRequest request = {};
+    request.overlayFrameData = frameState.overlayFrameData;
+    request.viewportSize = geometry.viewportSize;
+    request.localMousePosition = geometry.localMousePosition;
+    request.hudOverlay = &frameState.hudOverlay;
+    request.viewportMin = viewportMin;
+    request.viewportMax = viewportMax;
+    request.absoluteMousePosition = absoluteMousePosition;
+    return request;
+}
+
+} // namespace Editor
+} // namespace XCEngine