264 lines
8.2 KiB
C#
Executable File
264 lines
8.2 KiB
C#
Executable File
using UnityEngine;
|
|
using UnityEditor;
|
|
using System.Collections;
|
|
|
|
[CustomEditor(typeof(BezierCurve))]
|
|
public class BezierCurveEditor : Editor
|
|
{
|
|
BezierCurve curve;
|
|
SerializedProperty resolutionProp;
|
|
SerializedProperty closeProp;
|
|
SerializedProperty pointsProp;
|
|
SerializedProperty colorProp;
|
|
|
|
private static bool showPoints = true;
|
|
|
|
void OnEnable()
|
|
{
|
|
curve = (BezierCurve)target;
|
|
|
|
resolutionProp = serializedObject.FindProperty("resolution");
|
|
closeProp = serializedObject.FindProperty("_close");
|
|
pointsProp = serializedObject.FindProperty("points");
|
|
colorProp = serializedObject.FindProperty("drawColor");
|
|
}
|
|
|
|
public override void OnInspectorGUI()
|
|
{
|
|
serializedObject.Update();
|
|
|
|
EditorGUILayout.PropertyField(resolutionProp);
|
|
EditorGUILayout.PropertyField(closeProp);
|
|
EditorGUILayout.PropertyField(colorProp);
|
|
|
|
showPoints = EditorGUILayout.Foldout(showPoints, "Points");
|
|
|
|
if(showPoints)
|
|
{
|
|
int pointCount = pointsProp.arraySize;
|
|
|
|
for(int i = 0; i < pointCount; i++)
|
|
{
|
|
DrawPointInspector(curve[i], i);
|
|
}
|
|
|
|
if(GUILayout.Button("Add Point"))
|
|
{
|
|
Undo.RegisterSceneUndo("Add Point");
|
|
|
|
GameObject pointObject = new GameObject("Point "+pointsProp.arraySize);
|
|
pointObject.transform.parent = curve.transform;
|
|
pointObject.transform.localPosition = Vector3.zero;
|
|
BezierPoint newPoint = pointObject.AddComponent<BezierPoint>();
|
|
|
|
newPoint.curve = curve;
|
|
newPoint.handle1 = Vector3.right*0.1f;
|
|
newPoint.handle2 = -Vector3.right*0.1f;
|
|
|
|
pointsProp.InsertArrayElementAtIndex(pointsProp.arraySize);
|
|
pointsProp.GetArrayElementAtIndex(pointsProp.arraySize - 1).objectReferenceValue = newPoint;
|
|
}
|
|
}
|
|
|
|
if(GUI.changed)
|
|
{
|
|
serializedObject.ApplyModifiedProperties();
|
|
EditorUtility.SetDirty(target);
|
|
}
|
|
}
|
|
|
|
void OnSceneGUI()
|
|
{
|
|
for(int i = 0; i < curve.pointCount; i++)
|
|
{
|
|
DrawPointSceneGUI(curve[i]);
|
|
}
|
|
}
|
|
|
|
void DrawPointInspector(BezierPoint point, int index)
|
|
{
|
|
SerializedObject serObj = new SerializedObject(point);
|
|
|
|
SerializedProperty handleStyleProp = serObj.FindProperty("handleStyle");
|
|
SerializedProperty handle1Prop = serObj.FindProperty("_handle1");
|
|
SerializedProperty handle2Prop = serObj.FindProperty("_handle2");
|
|
|
|
EditorGUILayout.BeginHorizontal();
|
|
|
|
if(GUILayout.Button("X", GUILayout.Width(20)))
|
|
{
|
|
Undo.RegisterSceneUndo("Remove Point");
|
|
pointsProp.MoveArrayElement(curve.GetPointIndex(point), curve.pointCount - 1);
|
|
pointsProp.arraySize--;
|
|
DestroyImmediate(point.gameObject);
|
|
return;
|
|
}
|
|
|
|
EditorGUILayout.ObjectField(point.gameObject, typeof(GameObject), true);
|
|
|
|
if(index != 0 && GUILayout.Button(@"/\", GUILayout.Width(25)))
|
|
{
|
|
UnityEngine.Object other = pointsProp.GetArrayElementAtIndex(index - 1).objectReferenceValue;
|
|
pointsProp.GetArrayElementAtIndex(index - 1).objectReferenceValue = point;
|
|
pointsProp.GetArrayElementAtIndex(index).objectReferenceValue = other;
|
|
}
|
|
|
|
if(index != pointsProp.arraySize - 1 && GUILayout.Button(@"\/", GUILayout.Width(25)))
|
|
{
|
|
UnityEngine.Object other = pointsProp.GetArrayElementAtIndex(index + 1).objectReferenceValue;
|
|
pointsProp.GetArrayElementAtIndex(index + 1).objectReferenceValue = point;
|
|
pointsProp.GetArrayElementAtIndex(index).objectReferenceValue = other;
|
|
}
|
|
|
|
EditorGUILayout.EndHorizontal();
|
|
|
|
EditorGUI.indentLevel++;
|
|
EditorGUI.indentLevel++;
|
|
|
|
int newType = (int)((object)EditorGUILayout.EnumPopup("Handle Type", (BezierPoint.HandleStyle)handleStyleProp.enumValueIndex));
|
|
|
|
if(newType != handleStyleProp.enumValueIndex)
|
|
{
|
|
handleStyleProp.enumValueIndex = newType;
|
|
if(newType == 0)
|
|
{
|
|
if(handle1Prop.vector3Value != Vector3.zero) handle2Prop.vector3Value = -handle1Prop.vector3Value;
|
|
else if(handle2Prop.vector3Value != Vector3.zero) handle1Prop.vector3Value = -handle2Prop.vector3Value;
|
|
else
|
|
{
|
|
handle1Prop.vector3Value = new Vector3(0.1f, 0, 0);
|
|
handle2Prop.vector3Value = new Vector3(-0.1f, 0, 0);
|
|
}
|
|
}
|
|
|
|
else if(newType == 1)
|
|
{
|
|
if(handle1Prop.vector3Value == Vector3.zero && handle2Prop.vector3Value == Vector3.zero)
|
|
{
|
|
handle1Prop.vector3Value = new Vector3(0.1f, 0, 0);
|
|
handle2Prop.vector3Value = new Vector3(-0.1f, 0, 0);
|
|
}
|
|
}
|
|
|
|
else if(newType == 2)
|
|
{
|
|
handle1Prop.vector3Value = Vector3.zero;
|
|
handle2Prop.vector3Value = Vector3.zero;
|
|
}
|
|
}
|
|
|
|
Vector3 newPointPos = EditorGUILayout.Vector3Field("Position : ", point.transform.localPosition);
|
|
if(newPointPos != point.transform.localPosition)
|
|
{
|
|
Undo.RegisterUndo(point.transform, "Move Bezier Point");
|
|
point.transform.localPosition = newPointPos;
|
|
}
|
|
|
|
if(handleStyleProp.enumValueIndex == 0)
|
|
{
|
|
Vector3 newPosition;
|
|
|
|
newPosition = EditorGUILayout.Vector3Field("Handle 1", handle1Prop.vector3Value);
|
|
if(newPosition != handle1Prop.vector3Value)
|
|
{
|
|
handle1Prop.vector3Value = newPosition;
|
|
handle2Prop.vector3Value = -newPosition;
|
|
}
|
|
|
|
newPosition = EditorGUILayout.Vector3Field("Handle 2", handle2Prop.vector3Value);
|
|
if(newPosition != handle2Prop.vector3Value)
|
|
{
|
|
handle1Prop.vector3Value = -newPosition;
|
|
handle2Prop.vector3Value = newPosition;
|
|
}
|
|
}
|
|
|
|
else if(handleStyleProp.enumValueIndex == 1)
|
|
{
|
|
EditorGUILayout.PropertyField(handle1Prop);
|
|
EditorGUILayout.PropertyField(handle2Prop);
|
|
}
|
|
|
|
EditorGUI.indentLevel--;
|
|
EditorGUI.indentLevel--;
|
|
|
|
if(GUI.changed)
|
|
{
|
|
serObj.ApplyModifiedProperties();
|
|
EditorUtility.SetDirty(serObj.targetObject);
|
|
}
|
|
}
|
|
|
|
static void DrawPointSceneGUI(BezierPoint point)
|
|
{
|
|
Handles.Label(point.position + new Vector3(0, HandleUtility.GetHandleSize(point.position) * 0.4f, 0), point.gameObject.name);
|
|
|
|
Handles.color = Color.green;
|
|
Vector3 newPosition = Handles.FreeMoveHandle(point.position, point.transform.rotation, HandleUtility.GetHandleSize(point.position)*0.1f, Vector3.zero, Handles.RectangleCap);
|
|
|
|
if(newPosition != point.position)
|
|
{
|
|
Undo.RegisterUndo(point.transform, "Move Point");
|
|
point.transform.position = newPosition;
|
|
}
|
|
|
|
if(point.handleStyle != BezierPoint.HandleStyle.None)
|
|
{
|
|
Handles.color = Color.cyan;
|
|
Vector3 newGlobal1 = Handles.FreeMoveHandle(point.globalHandle1, point.transform.rotation, HandleUtility.GetHandleSize(point.globalHandle1)*0.075f, Vector3.zero, Handles.CircleCap);
|
|
if(point.globalHandle1 != newGlobal1)
|
|
{
|
|
Undo.RegisterUndo(point, "Move Handle");
|
|
point.globalHandle1 = newGlobal1;
|
|
if(point.handleStyle == BezierPoint.HandleStyle.Connected) point.globalHandle2 = -(newGlobal1 - point.position) + point.position;
|
|
}
|
|
|
|
Vector3 newGlobal2 = Handles.FreeMoveHandle(point.globalHandle2, point.transform.rotation, HandleUtility.GetHandleSize(point.globalHandle2)*0.075f, Vector3.zero, Handles.CircleCap);
|
|
if(point.globalHandle2 != newGlobal2)
|
|
{
|
|
Undo.RegisterUndo(point, "Move Handle");
|
|
point.globalHandle2 = newGlobal2;
|
|
if(point.handleStyle == BezierPoint.HandleStyle.Connected) point.globalHandle1 = -(newGlobal2 - point.position) + point.position;
|
|
}
|
|
|
|
Handles.color = Color.yellow;
|
|
Handles.DrawLine(point.position, point.globalHandle1);
|
|
Handles.DrawLine(point.position, point.globalHandle2);
|
|
}
|
|
}
|
|
|
|
public static void DrawOtherPoints(BezierCurve curve, BezierPoint caller)
|
|
{
|
|
foreach(BezierPoint p in curve.GetAnchorPoints())
|
|
{
|
|
if(p != caller) DrawPointSceneGUI(p);
|
|
}
|
|
}
|
|
|
|
[MenuItem("GameObject/Create Other/Bezier Curve")]
|
|
public static void CreateCurve(MenuCommand command)
|
|
{
|
|
GameObject curveObject = new GameObject("BezierCurve");
|
|
Undo.RegisterUndo(curveObject, "Undo Create Curve");
|
|
BezierCurve curve = curveObject.AddComponent<BezierCurve>();
|
|
|
|
BezierPoint p1 = curve.AddPointAt(Vector3.forward * 0.5f);
|
|
p1.handleStyle = BezierPoint.HandleStyle.Connected;
|
|
p1.handle1 = new Vector3(-0.28f, 0, 0);
|
|
|
|
BezierPoint p2 = curve.AddPointAt(Vector3.right * 0.5f);
|
|
p2.handleStyle = BezierPoint.HandleStyle.Connected;
|
|
p2.handle1 = new Vector3(0, 0, 0.28f);
|
|
|
|
BezierPoint p3 = curve.AddPointAt(-Vector3.forward * 0.5f);
|
|
p3.handleStyle = BezierPoint.HandleStyle.Connected;
|
|
p3.handle1 = new Vector3(0.28f, 0, 0);
|
|
|
|
BezierPoint p4 = curve.AddPointAt(-Vector3.right * 0.5f);
|
|
p4.handleStyle = BezierPoint.HandleStyle.Connected;
|
|
p4.handle1 = new Vector3(0, 0, -0.28f);
|
|
|
|
curve.close = true;
|
|
}
|
|
}
|