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

121 lines
5.2 KiB
C#
Raw Normal View History

2022-01-24 02:15:23 +00:00
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())
{
2022-02-14 08:53:58 +00:00
if (this.handleType.enumValueIndex == (int)BezierPoint3D.HandleType.Connected)
this.rightHandleLocalPosition.vector3Value = -this.leftHandleLocalPosition.vector3Value;
2022-01-24 02:15:23 +00:00
}
EditorGUI.BeginChangeCheck();
EditorGUILayout.PropertyField(this.rightHandleLocalPosition);
if (EditorGUI.EndChangeCheck())
{
2022-02-14 08:53:58 +00:00
if (this.handleType.enumValueIndex == (int)BezierPoint3D.HandleType.Connected)
this.leftHandleLocalPosition.vector3Value = -this.rightHandleLocalPosition.vector3Value;
2022-01-24 02:15:23 +00:00
}
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;
}
}
}