mirror of
https://github.com/RHeavenStudioPlus/HeavenStudioPlus.git
synced 2024-12-05 15:57:32 +00:00
121 lines
No EOL
5.2 KiB
C#
121 lines
No EOL
5.2 KiB
C#
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;
|
|
}
|
|
}
|
|
} |