From eaa63e679ead5c5c49257444c1098b34b4dc35da Mon Sep 17 00:00:00 2001
From: Wojtek Figat <wojtek@figat.pl>
Date: Wed, 18 Jan 2023 15:36:42 +0100
Subject: [PATCH] Fix selected UI control outline when using custom Game
 viewport aspect

#897
---
 Source/Editor/Windows/GameWindow.cs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/Source/Editor/Windows/GameWindow.cs b/Source/Editor/Windows/GameWindow.cs
index f17deb4bd..aaed484a1 100644
--- a/Source/Editor/Windows/GameWindow.cs
+++ b/Source/Editor/Windows/GameWindow.cs
@@ -816,15 +816,23 @@ namespace FlaxEditor.Windows
             }
 
             // Selected UI controls outline
+            bool drawAnySelectedControl = false;
             for (var i = 0; i < Editor.Instance.SceneEditing.Selection.Count; i++)
             {
                 if (Editor.Instance.SceneEditing.Selection[i].EditableObject is UIControl controlActor && controlActor && controlActor.Control != null)
                 {
+                    if (!drawAnySelectedControl)
+                    {
+                        drawAnySelectedControl = true;
+                        Render2D.PushTransform(ref _viewport._cachedTransform);
+                    }
                     var control = controlActor.Control;
                     var bounds = Rectangle.FromPoints(control.PointToParent(_viewport, Float2.Zero), control.PointToParent(_viewport, control.Size));
                     Render2D.DrawRectangle(bounds, Editor.Instance.Options.Options.Visual.SelectionOutlineColor0, Editor.Instance.Options.Options.Visual.UISelectionOutlineSize);
                 }
             }
+            if (drawAnySelectedControl)
+                Render2D.PopTransform();
 
             // Play mode hints and overlay
             if (Editor.StateMachine.IsPlayMode)