diff --git a/Source/Editor/Gizmo/DirectionGizmo.cs b/Source/Editor/Gizmo/DirectionGizmo.cs new file mode 100644 index 000000000..176832c5e --- /dev/null +++ b/Source/Editor/Gizmo/DirectionGizmo.cs @@ -0,0 +1,266 @@ +using System.Collections.Generic; +using FlaxEditor.Gizmo; +using FlaxEditor.Viewport; +using FlaxEngine; +using FlaxEngine.GUI; + +namespace FlaxEditor.Gizmo; + +[HideInEditor] +public class DirectionGizmo : ContainerControl +{ + private IGizmoOwner _owner; + private ViewportProjection _viewportProjection; + private EditorViewport _viewport; + private Vector3 _gizmoCenter; + private float _axisLength = 75.0f; + private float _textAxisLength = 95.0f; + private float _spriteRadius = 12.0f; + + private AxisData _xAxisData; + private AxisData _yAxisData; + private AxisData _zAxisData; + private AxisData _negXAxisData; + private AxisData _negYAxisData; + private AxisData _negZAxisData; + + private List _axisData = new List(); + private int _hoveredAxisIndex = -1; + + private SpriteHandle _posHandle; + private SpriteHandle _negHandle; + + private FontReference _fontReference; + + // Store sprite positions for hover detection + private List<(Float2 position, AxisDirection direction)> _spritePositions = new List<(Float2, AxisDirection)>(); + + private struct ViewportProjection + { + private Matrix _viewProjection; + private BoundingFrustum _frustum; + private FlaxEngine.Viewport _viewport; + private Vector3 _origin; + + public void Init(EditorViewport editorViewport) + { + // Inline EditorViewport.ProjectPoint to save on calculation for large set of points + _viewport = new FlaxEngine.Viewport(0, 0, editorViewport.Width, editorViewport.Height); + _frustum = editorViewport.ViewFrustum; + _viewProjection = _frustum.Matrix; + _origin = editorViewport.Task.View.Origin; + } + + public void ProjectPoint(Vector3 worldSpaceLocation, out Float2 viewportSpaceLocation) + { + worldSpaceLocation -= _origin; + _viewport.Project(ref worldSpaceLocation, ref _viewProjection, out var projected); + viewportSpaceLocation = new Float2((float)projected.X, (float)projected.Y); + } + } + + private struct AxisData + { + public Float2 Delta; + public float Distance; + public string Label; + public Color AxisColor; + public bool Negative; + public AxisDirection Direction; + } + + private enum AxisDirection + { + PosX, + PosY, + PosZ, + NegX, + NegY, + NegZ + } + + /// + /// Constructor of the Direction Gizmo + /// + /// The owner of this object. + public DirectionGizmo(IGizmoOwner owner) + { + _owner = owner; + _viewport = owner.Viewport; + _viewportProjection.Init(owner.Viewport); + + _xAxisData = new AxisData { Delta = new Float2(0, 0), Distance = 0, Label = "X", AxisColor = new Color(1.0f, 0.0f, 0.02745f, 1.0f), Negative = false, Direction = AxisDirection.PosX }; + _yAxisData = new AxisData { Delta = new Float2(0, 0), Distance = 0, Label = "Y", AxisColor = new Color(0.239215f, 1.0f, 0.047058f, 1.0f), Negative = false, Direction = AxisDirection.PosY }; + _zAxisData = new AxisData { Delta = new Float2(0, 0), Distance = 0, Label = "Z", AxisColor = new Color(0.0f, 0.0235294f, 1.0f, 1.0f), Negative = false, Direction = AxisDirection.PosZ }; + + _negXAxisData = new AxisData { Delta = new Float2(0, 0), Distance = 0, Label = "-X", AxisColor = new Color(1.0f, 0.0f, 0.02745f, 1.0f), Negative = true, Direction = AxisDirection.NegX }; + _negYAxisData = new AxisData { Delta = new Float2(0, 0), Distance = 0, Label = "-Y", AxisColor = new Color(0.239215f, 1.0f, 0.047058f, 1.0f), Negative = true, Direction = AxisDirection.NegY }; + _negZAxisData = new AxisData { Delta = new Float2(0, 0), Distance = 0, Label = "-Z", AxisColor = new Color(0.0f, 0.0235294f, 1.0f, 1.0f), Negative = true, Direction = AxisDirection.NegZ }; + _axisData.EnsureCapacity(6); + _spritePositions.EnsureCapacity(6); + + _posHandle = Editor.Instance.Icons.VisjectBoxClosed32; + _negHandle = Editor.Instance.Icons.VisjectBoxOpen32; + + _fontReference = new FontReference(Style.Current.FontSmall); + _fontReference.Size = 8; + } + + private bool IsPointInSprite(Float2 point, Float2 spriteCenter) + { + Float2 delta = point - spriteCenter; + float distanceSq = delta.LengthSquared; + float radiusSq = _spriteRadius * _spriteRadius; + return distanceSq <= radiusSq; + } + + /// + public override void OnMouseMove(Float2 location) + { + _hoveredAxisIndex = -1; + + // Check which axis is being hovered - check from closest to farthest for proper layering + for (int i = _spritePositions.Count - 1; i >= 0; i--) + { + if (IsPointInSprite(location, _spritePositions[i].position)) + { + _hoveredAxisIndex = i; + break; + } + } + + base.OnMouseMove(location); + } + + /// + public override bool OnMouseDown(Float2 location, MouseButton button) + { + if (base.OnMouseDown(location, button)) + return true; + + // Check which axis is being clicked - check from closest to farthest for proper layering + for (int i = _spritePositions.Count - 1; i >= 0; i--) + { + if (IsPointInSprite(location, _spritePositions[i].position)) + { + OrientViewToAxis(_spritePositions[i].direction); + return true; + } + } + + return false; + } + + private void OrientViewToAxis(AxisDirection direction) + { + Quaternion orientation = direction switch + { + AxisDirection.PosX => Quaternion.Euler(0, 90, 0), + AxisDirection.NegX => Quaternion.Euler(0, -90, 0), + AxisDirection.PosY => Quaternion.Euler(-90, 0, 0), + AxisDirection.NegY => Quaternion.Euler(90, 0, 0), + AxisDirection.PosZ => Quaternion.Euler(0, 0, 0), + AxisDirection.NegZ => Quaternion.Euler(0, 180, 0), + _ => Quaternion.Identity + }; + + _viewport.ViewOrientation = orientation; + } + + /// + /// Used to Draw the gizmo. + /// + public override void DrawSelf() + { + base.DrawSelf(); + + _viewportProjection.Init(_owner.Viewport); + _gizmoCenter = _viewport.Task.View.WorldPosition + _viewport.Task.View.Direction * 1500; + _viewportProjection.ProjectPoint(_gizmoCenter, out var gizmoCenterScreen); + + var relativeCenter = Size * 0.5f; + + // Project unit vectors + _viewportProjection.ProjectPoint(_gizmoCenter + Vector3.Right, out var xProjected); + _viewportProjection.ProjectPoint(_gizmoCenter + Vector3.Up, out var yProjected); + _viewportProjection.ProjectPoint(_gizmoCenter + Vector3.Forward, out var zProjected); + _viewportProjection.ProjectPoint(_gizmoCenter - Vector3.Right, out var negXProjected); + _viewportProjection.ProjectPoint(_gizmoCenter - Vector3.Up, out var negYProjected); + _viewportProjection.ProjectPoint(_gizmoCenter - Vector3.Forward, out var negZProjected); + + // Normalize by viewport height to keep size independent of FOV and viewport dimensions + float heightNormalization = _viewport.Height / 720.0f; // 720 = reference height + + Float2 xDelta = (xProjected - gizmoCenterScreen) / heightNormalization; + Float2 yDelta = (yProjected - gizmoCenterScreen) / heightNormalization; + Float2 zDelta = (zProjected - gizmoCenterScreen) / heightNormalization; + Float2 negXDelta = (negXProjected - gizmoCenterScreen) / heightNormalization; + Float2 negYDelta = (negYProjected - gizmoCenterScreen) / heightNormalization; + Float2 negZDelta = (negZProjected - gizmoCenterScreen) / heightNormalization; + + // Calculate distances from camera to determine draw order + Vector3 cameraPosition = _viewport.Task.View.Position; + float xDistance = (float)Vector3.Distance(cameraPosition, _gizmoCenter + Vector3.Right); + float yDistance = (float)Vector3.Distance(cameraPosition, _gizmoCenter + Vector3.Up); + float zDistance = (float)Vector3.Distance(cameraPosition, _gizmoCenter + Vector3.Forward); + float negXDistance = (float)Vector3.Distance(cameraPosition, _gizmoCenter - Vector3.Right); + float negYDistance = (float)Vector3.Distance(cameraPosition, _gizmoCenter - Vector3.Up); + float negZDistance = (float)Vector3.Distance(cameraPosition, _gizmoCenter - Vector3.Forward); + + _xAxisData.Delta = xDelta; + _xAxisData.Distance = xDistance; + _yAxisData.Delta = yDelta; + _yAxisData.Distance = yDistance; + _zAxisData.Delta = zDelta; + _zAxisData.Distance = zDistance; + _negXAxisData.Delta = negXDelta; + _negXAxisData.Distance = negXDistance; + _negYAxisData.Delta = negYDelta; + _negYAxisData.Distance = negYDistance; + _negZAxisData.Delta = negZDelta; + _negZAxisData.Distance = negZDistance; + + // Sort for correct draw order. + _axisData.Clear(); + _axisData.AddRange([_xAxisData, _yAxisData, _zAxisData, _negXAxisData, _negYAxisData, _negZAxisData]); + _axisData.Sort( (a, b) => -a.Distance.CompareTo(b.Distance)); + + // Rebuild sprite positions list for hover detection + _spritePositions.Clear(); + + Render2D.DrawSprite(_posHandle, new Rectangle(0, 0, Size), Color.Black.AlphaMultiplied(0.1f)); + + // Draw in order from farthest to closest + for (int i = 0; i < _axisData.Count; i++) + { + var axis = _axisData[i]; + Float2 tipScreen = relativeCenter + axis.Delta * _axisLength; + Float2 tipTextScreen = relativeCenter + axis.Delta * _textAxisLength; + bool isHovered = _hoveredAxisIndex == i; + + // Store sprite position for hover detection + _spritePositions.Add((tipTextScreen, axis.Direction)); + + if (!axis.Negative) + { + Render2D.DrawLine(relativeCenter, tipScreen, axis.AxisColor, 2.0f); + Render2D.DrawSprite(_posHandle, new Rectangle(tipTextScreen - new Float2(_spriteRadius), new Float2(_spriteRadius * 2)), axis.AxisColor); + + var font = _fontReference.GetFont(); + Color textColor = isHovered ? Color.White : Color.Black; + Render2D.DrawText(font, axis.Label, textColor, tipTextScreen - font.MeasureText(axis.Label) * 0.5f); + } + else + { + Render2D.DrawSprite(_posHandle, new Rectangle(tipTextScreen - new Float2(_spriteRadius), new Float2(_spriteRadius * 2)), axis.AxisColor.RGBMultiplied(0.65f)); + Render2D.DrawSprite(_negHandle, new Rectangle(tipTextScreen - new Float2(_spriteRadius), new Float2(_spriteRadius * 2)), axis.AxisColor); + + if (isHovered) + { + var font = _fontReference.GetFont(); + Render2D.DrawText(font, axis.Label, Color.White, tipTextScreen - font.MeasureText(axis.Label) * 0.5f); + } + } + } + } +} diff --git a/Source/Editor/Viewport/MainEditorGizmoViewport.cs b/Source/Editor/Viewport/MainEditorGizmoViewport.cs index 3d366809c..a76c8a16a 100644 --- a/Source/Editor/Viewport/MainEditorGizmoViewport.cs +++ b/Source/Editor/Viewport/MainEditorGizmoViewport.cs @@ -108,6 +108,7 @@ namespace FlaxEditor.Viewport private readonly ViewportDebugDrawData _debugDrawData = new ViewportDebugDrawData(32); private EditorSpritesRenderer _editorSpritesRenderer; private ViewportRubberBandSelector _rubberBandSelector; + private DirectionGizmo _directionGizmo; private bool _gameViewActive; private ViewFlags _preGameViewFlags; @@ -225,6 +226,12 @@ namespace FlaxEditor.Viewport // Add rubber band selector _rubberBandSelector = new ViewportRubberBandSelector(this); + _directionGizmo = new DirectionGizmo(this); + _directionGizmo.AnchorPreset = AnchorPresets.TopRight; + _directionGizmo.Parent = this; + _directionGizmo.LocalY += 25; + _directionGizmo.LocalX -= 150; + _directionGizmo.Size = new Float2(150, 150); // Add grid Grid = new GridGizmo(this);