From 66c1b736351fae0595c7fc23467e60c09601dd8a Mon Sep 17 00:00:00 2001
From: Wojtek Figat <wojtek@figat.pl>
Date: Thu, 18 Mar 2021 17:55:07 +0100
Subject: [PATCH] Add `F` to focus camera view in asset previews

---
 Source/Editor/Viewport/Cameras/ArcBallCamera.cs     |  9 +++++++++
 Source/Editor/Viewport/Cameras/FPSCamera.cs         |  8 ++++++++
 Source/Editor/Viewport/Cameras/ViewportCamera.cs    |  2 +-
 .../Viewport/Previews/AnimatedModelPreview.cs       | 13 +++++++++++++
 Source/Editor/Viewport/Previews/ModelPreview.cs     | 13 +++++++++++++
 5 files changed, 44 insertions(+), 1 deletion(-)

diff --git a/Source/Editor/Viewport/Cameras/ArcBallCamera.cs b/Source/Editor/Viewport/Cameras/ArcBallCamera.cs
index 876adc796..e6a72c8b4 100644
--- a/Source/Editor/Viewport/Cameras/ArcBallCamera.cs
+++ b/Source/Editor/Viewport/Cameras/ArcBallCamera.cs
@@ -94,6 +94,15 @@ namespace FlaxEditor.Viewport.Cameras
             Viewport.ViewPosition = _orbitCenter + localPosition;
         }
 
+        /// <inheritdoc />
+        public override void SetArcBallView(Quaternion orientation, Vector3 orbitCenter, float orbitRadius)
+        {
+            base.SetArcBallView(orientation, orbitCenter, orbitRadius);
+
+            _orbitCenter = orbitCenter;
+            _orbitRadius = orbitRadius;
+        }
+
         /// <inheritdoc />
         public override void UpdateView(float dt, ref Vector3 moveDelta, ref Vector2 mouseDelta, out bool centerMouse)
         {
diff --git a/Source/Editor/Viewport/Cameras/FPSCamera.cs b/Source/Editor/Viewport/Cameras/FPSCamera.cs
index e60def838..1079b7560 100644
--- a/Source/Editor/Viewport/Cameras/FPSCamera.cs
+++ b/Source/Editor/Viewport/Cameras/FPSCamera.cs
@@ -166,6 +166,14 @@ namespace FlaxEditor.Viewport.Cameras
             MoveViewport(position, orientation);
         }
         
+        /// <inheritdoc />
+        public override void SetArcBallView(Quaternion orientation, Vector3 orbitCenter, float orbitRadius)
+        {
+            base.SetArcBallView(orientation, orbitCenter, orbitRadius);
+
+            TargetPoint = orbitCenter;
+        }
+
         /// <inheritdoc />
         public override void Update(float deltaTime)
         {
diff --git a/Source/Editor/Viewport/Cameras/ViewportCamera.cs b/Source/Editor/Viewport/Cameras/ViewportCamera.cs
index 85b7a917c..6564d16d0 100644
--- a/Source/Editor/Viewport/Cameras/ViewportCamera.cs
+++ b/Source/Editor/Viewport/Cameras/ViewportCamera.cs
@@ -62,7 +62,7 @@ namespace FlaxEditor.Viewport.Cameras
         /// <param name="orientation">The view rotation.</param>
         /// <param name="orbitCenter">The orbit center location.</param>
         /// <param name="orbitRadius">The orbit radius.</param>
-        public void SetArcBallView(Quaternion orientation, Vector3 orbitCenter, float orbitRadius)
+        public virtual void SetArcBallView(Quaternion orientation, Vector3 orbitCenter, float orbitRadius)
         {
             // Rotate
             Viewport.ViewOrientation = orientation;
diff --git a/Source/Editor/Viewport/Previews/AnimatedModelPreview.cs b/Source/Editor/Viewport/Previews/AnimatedModelPreview.cs
index 6f6103a5d..697a7c1c1 100644
--- a/Source/Editor/Viewport/Previews/AnimatedModelPreview.cs
+++ b/Source/Editor/Viewport/Previews/AnimatedModelPreview.cs
@@ -249,6 +249,19 @@ namespace FlaxEditor.Viewport.Previews
             }
         }
 
+        /// <inheritdoc />
+        public override bool OnKeyDown(KeyboardKeys key)
+        {
+            switch (key)
+            {
+            case KeyboardKeys.F:
+                // Pay respect..
+                ViewportCamera.SetArcBallView(_previewModel.Box);
+                break;
+            }
+            return base.OnKeyDown(key);
+        }
+
         /// <inheritdoc />
         public override void OnDestroy()
         {
diff --git a/Source/Editor/Viewport/Previews/ModelPreview.cs b/Source/Editor/Viewport/Previews/ModelPreview.cs
index 7055edbe9..3f285ddcd 100644
--- a/Source/Editor/Viewport/Previews/ModelPreview.cs
+++ b/Source/Editor/Viewport/Previews/ModelPreview.cs
@@ -85,6 +85,19 @@ namespace FlaxEditor.Viewport.Previews
             }
         }
 
+        /// <inheritdoc />
+        public override bool OnKeyDown(KeyboardKeys key)
+        {
+            switch (key)
+            {
+            case KeyboardKeys.F:
+                // Pay respect..
+                ViewportCamera.SetArcBallView(_previewModel.Box);
+                break;
+            }
+            return base.OnKeyDown(key);
+        }
+
         /// <inheritdoc />
         public override void OnDestroy()
         {