From 6b1cc5bf2fec3829c031be2dd64550205259c8ef Mon Sep 17 00:00:00 2001
From: Ruan Lucas <79365912+RuanLucasGD@users.noreply.github.com>
Date: Sat, 5 Aug 2023 12:07:40 -0400
Subject: [PATCH] adjust again...

---
 Source/Engine/Level/Actors/Spline.cpp | 47 +++++++++++++++++++++++++++
 Source/Engine/Level/Actors/Spline.h   |  4 +++
 2 files changed, 51 insertions(+)

diff --git a/Source/Engine/Level/Actors/Spline.cpp b/Source/Engine/Level/Actors/Spline.cpp
index 2923463b0..dbd7ef045 100644
--- a/Source/Engine/Level/Actors/Spline.cpp
+++ b/Source/Engine/Level/Actors/Spline.cpp
@@ -455,6 +455,53 @@ void Spline::SetKeyframes(MArray* data)
 
 #endif
 
+#if USE_EDITOR
+
+#include "Engine/Debug/DebugDraw.h"
+
+namespace
+{
+    void DrawSpline(Spline* spline, const Color& color, const Transform& transform, bool depthTest)
+    {
+        const int32 count = spline->Curve.GetKeyframes().Count();
+        if (count == 0)
+            return;
+        Spline::Keyframe* prev = spline->Curve.GetKeyframes().Get();
+        Vector3 prevPos = transform.LocalToWorld(prev->Value.Translation);
+        DEBUG_DRAW_WIRE_SPHERE(BoundingSphere(prevPos, 5.0f), color, 0.0f, depthTest);
+        for (int32 i = 1; i < count; i++)
+        {
+            Spline::Keyframe* next = prev + 1;
+            Vector3 nextPos = transform.LocalToWorld(next->Value.Translation);
+            DEBUG_DRAW_WIRE_SPHERE(BoundingSphere(nextPos, 5.0f), color, 0.0f, depthTest);
+            const float d = (next->Time - prev->Time) / 3.0f;
+            DEBUG_DRAW_BEZIER(prevPos, prevPos + prev->TangentOut.Translation * d, nextPos + next->TangentIn.Translation * d, nextPos, color, 0.0f, depthTest);
+            prev = next;
+            prevPos = nextPos;
+        }
+    }
+}
+
+void Spline::OnDebugDraw()
+{
+    const Color color = GetSplineColor();
+    DrawSpline(this, color.AlphaMultiplied(0.7f), _transform, true);
+
+    // Base
+    Actor::OnDebugDraw();
+}
+
+void Spline::OnDebugDrawSelected()
+{
+    const Color color = GetSplineColor();
+    DrawSpline(this, color.AlphaMultiplied(0.3f), _transform, false);
+
+    // Base
+    Actor::OnDebugDrawSelected();
+}
+
+#endif
+
 void Spline::OnTransformChanged()
 {
     // Base
diff --git a/Source/Engine/Level/Actors/Spline.h b/Source/Engine/Level/Actors/Spline.h
index 426304fb9..ea3cf8569 100644
--- a/Source/Engine/Level/Actors/Spline.h
+++ b/Source/Engine/Level/Actors/Spline.h
@@ -369,6 +369,10 @@ private:
 
 public:
     // [Actor]
+#if USE_EDITOR
+    void OnDebugDraw() override;
+    void OnDebugDrawSelected() override;
+#endif
     void OnTransformChanged() override;
     void Initialize() override;
     void Serialize(SerializeStream& stream, const void* otherObj) override;