HeavenStudioPlus/Assets/Plugins/NaughtyBezierCurves/Scripts/Editor/BezierPoint3DEditor.cs

using UnityEngine;
using UnityEditor;

namespace NaughtyBezierCurves.Editor
{
    [CustomEditor(typeof(BezierPoint3D), true)]
    [CanEditMultipleObjects]
    public class BezierPoint3DEditor : UnityEditor.Editor
    {
        public const float CircleCapSize = 0.075f;
        public const float RectangeCapSize = 0.1f;
        public const float SphereCapSize = 0.15f;

        public static float pointCapSize = RectangeCapSize;
        public static float handleCapSize = CircleCapSize;

        private BezierPoint3D point;
        private SerializedProperty handleType;
        private SerializedProperty leftHandleLocalPosition;
        private SerializedProperty rightHandleLocalPosition;

        protected virtual void OnEnable()
        {
            this.point = (BezierPoint3D)this.target;
            this.handleType = this.serializedObject.FindProperty("handleType");
            this.leftHandleLocalPosition = this.serializedObject.FindProperty("leftHandleLocalPosition");
            this.rightHandleLocalPosition = this.serializedObject.FindProperty("rightHandleLocalPosition");
        }

        public override void OnInspectorGUI()
        {
            this.serializedObject.Update();

            EditorGUILayout.PropertyField(this.handleType);

            EditorGUI.BeginChangeCheck();
            EditorGUILayout.PropertyField(this.leftHandleLocalPosition);
            if (EditorGUI.EndChangeCheck())
            {
                if (this.handleType.enumValueIndex == (int)BezierPoint3D.HandleType.Connected)
                    this.rightHandleLocalPosition.vector3Value = -this.leftHandleLocalPosition.vector3Value;
            }

            EditorGUI.BeginChangeCheck();
            EditorGUILayout.PropertyField(this.rightHandleLocalPosition);
            if (EditorGUI.EndChangeCheck())
            {
                if (this.handleType.enumValueIndex == (int)BezierPoint3D.HandleType.Connected)
                    this.leftHandleLocalPosition.vector3Value = -this.rightHandleLocalPosition.vector3Value;
            }

            this.serializedObject.ApplyModifiedProperties();
        }

        protected virtual void OnSceneGUI()
        {
            BezierPoint3DEditor.handleCapSize = BezierPoint3DEditor.CircleCapSize;
            BezierCurve3DEditor.DrawPointsSceneGUI(this.point.Curve, this.point);

            BezierPoint3DEditor.handleCapSize = BezierPoint3DEditor.SphereCapSize;
            BezierPoint3DEditor.DrawPointSceneGUI(this.point, Handles.DotHandleCap, Handles.SphereHandleCap);
        }

        public static void DrawPointSceneGUI(BezierPoint3D point)
        {
            DrawPointSceneGUI(point, Handles.RectangleHandleCap, Handles.CircleHandleCap);
        }

        public static void DrawPointSceneGUI(BezierPoint3D point, Handles.CapFunction drawPointFunc, Handles.CapFunction drawHandleFunc)
        {
            // Draw a label for the point
            Handles.color = Color.black;
            Handles.Label(point.Position + new Vector3(0f, HandleUtility.GetHandleSize(point.Position) * 0.4f, 0f), point.gameObject.name);

            // Draw the center of the control point
            Handles.color = Color.yellow;
            Vector3 newPointPosition = Handles.FreeMoveHandle(point.Position, point.transform.rotation,
                HandleUtility.GetHandleSize(point.Position) * BezierPoint3DEditor.pointCapSize, Vector3.one * 0.5f, drawPointFunc);

            if (point.Position != newPointPosition)
            {
                Undo.RegisterCompleteObjectUndo(point.transform, "Move Point");
                point.Position = newPointPosition;
            }

            // Draw the left and right handles
            Handles.color = Color.white;
            Handles.DrawLine(point.Position, point.LeftHandlePosition);
            Handles.DrawLine(point.Position, point.RightHandlePosition);

            Handles.color = Color.cyan;
            Vector3 newLeftHandlePosition = Handles.FreeMoveHandle(point.LeftHandlePosition, point.transform.rotation,
                HandleUtility.GetHandleSize(point.LeftHandlePosition) * BezierPoint3DEditor.handleCapSize, Vector3.zero, drawHandleFunc);

            if (point.LeftHandlePosition != newLeftHandlePosition)
            {
                Undo.RegisterCompleteObjectUndo(point, "Move Left Handle");
                point.LeftHandlePosition = newLeftHandlePosition;
            }

            Vector3 newRightHandlePosition = Handles.FreeMoveHandle(point.RightHandlePosition, point.transform.rotation,
                HandleUtility.GetHandleSize(point.RightHandlePosition) * BezierPoint3DEditor.handleCapSize, Vector3.zero, drawHandleFunc);

            if (point.RightHandlePosition != newRightHandlePosition)
            {
                Undo.RegisterCompleteObjectUndo(point, "Move Right Handle");
                point.RightHandlePosition = newRightHandlePosition;
            }
        }

        private static bool MouseButtonDown(int button)
        {
            return Event.current.type == EventType.MouseDown && Event.current.button == button;
        }

        private static bool MouseButtonUp(int button)
        {
            return Event.current.type == EventType.MouseUp && Event.current.button == button;
        }
    }
}