1
This commit is contained in:
357
Assets/Dreamteck/Splines/Editor/Components/ObjectBenderEditor.cs
Normal file
357
Assets/Dreamteck/Splines/Editor/Components/ObjectBenderEditor.cs
Normal file
@@ -0,0 +1,357 @@
|
||||
namespace Dreamteck.Splines.Editor
|
||||
{
|
||||
using UnityEngine;
|
||||
using System.Collections;
|
||||
using UnityEditor;
|
||||
using System.Collections.Generic;
|
||||
|
||||
[CustomEditor(typeof(ObjectBender), true)]
|
||||
[CanEditMultipleObjects]
|
||||
public class ObjectBenderEditor : SplineUserEditor
|
||||
{
|
||||
List<int> selected = new List<int>();
|
||||
Vector2 scroll = Vector2.zero;
|
||||
bool generatedUvs = false;
|
||||
|
||||
protected override void Awake()
|
||||
{
|
||||
ObjectBender bender = (ObjectBender)target;
|
||||
if(!Application.isPlaying) bender.UpdateReferences();
|
||||
base.Awake();
|
||||
}
|
||||
|
||||
void PropertyEditor(ObjectBender.BendProperty[] properties)
|
||||
{
|
||||
if (selected.Count == 0) return;
|
||||
int applyRotationCount = 0, applyScaleCount = 0, enableCount = 0, bendMeshCount = 0, bendColliderCount = 0, bendSplineCount = 0;
|
||||
bool showMesh = false, showCollider = false, showSpline = false;
|
||||
float colliderUpdateRate = 0f;
|
||||
for(int i = 0; i < selected.Count; i++)
|
||||
{
|
||||
ObjectBender.BendProperty property = properties[selected[i]];
|
||||
if (property.enabled) enableCount++;
|
||||
if (property.applyRotation) applyRotationCount++;
|
||||
if (property.applyScale) applyScaleCount++;
|
||||
if (property.bendMesh) bendMeshCount++;
|
||||
if (property.bendCollider) bendColliderCount++;
|
||||
if (property.bendSpline) bendSplineCount++;
|
||||
if (property.filter != null) showMesh = true;
|
||||
if (property.collider != null) showCollider = true;
|
||||
if (property.splineComputer != null) showSpline = true;
|
||||
colliderUpdateRate += property.colliderUpdateRate;
|
||||
}
|
||||
bool enabled = enableCount == selected.Count;
|
||||
bool applyRotation = applyRotationCount == selected.Count;
|
||||
bool applyScale = applyScaleCount == selected.Count;
|
||||
bool bendMesh = bendMeshCount == selected.Count;
|
||||
bool bendCollider = bendColliderCount == selected.Count;
|
||||
bool bendSpline = bendSplineCount == selected.Count;
|
||||
colliderUpdateRate /= selected.Count;
|
||||
bool lastEnabled = enabled, lastApplyRotation = applyRotation, lastApplyScale = applyScale, lastBendMesh = bendMesh, lastBendCollider = bendCollider, lastBendSpline = bendSpline;
|
||||
float lastColliderUpdateRate = colliderUpdateRate;
|
||||
|
||||
EditorGUIUtility.labelWidth = 90;
|
||||
EditorGUI.BeginChangeCheck();
|
||||
GUI.color = Color.white;
|
||||
EditorGUILayout.BeginVertical(GUI.skin.box, GUILayout.Width(EditorGUIUtility.currentViewWidth - 50));
|
||||
EditorGUILayout.BeginHorizontal();
|
||||
enabled = EditorGUILayout.Toggle(enabled, GUILayout.Width(20));
|
||||
if (selected.Count == 1) EditorGUILayout.LabelField(properties[selected[0]].transform.transform.name);
|
||||
else EditorGUILayout.LabelField("Multiple objects");
|
||||
EditorGUILayout.EndHorizontal();
|
||||
|
||||
EditorGUILayout.BeginHorizontal();
|
||||
|
||||
EditorGUILayout.BeginVertical();
|
||||
applyRotation = EditorGUILayout.Toggle("Apply rotation", applyRotation);
|
||||
applyScale = EditorGUILayout.Toggle("Apply scale", applyScale);
|
||||
if (showSpline) bendSpline = EditorGUILayout.Toggle("Bend Spline", bendSpline);
|
||||
EditorGUILayout.EndVertical();
|
||||
|
||||
EditorGUILayout.BeginVertical();
|
||||
if (showMesh)
|
||||
{
|
||||
bendMesh = EditorGUILayout.Toggle("Bend Mesh", bendMesh);
|
||||
if (bendMesh)
|
||||
{
|
||||
if (showCollider)
|
||||
{
|
||||
bendCollider = EditorGUILayout.Toggle("Bend Collider", bendCollider);
|
||||
if (bendCollider)
|
||||
{
|
||||
EditorGUI.indentLevel++;
|
||||
colliderUpdateRate = EditorGUILayout.FloatField("Update Rate", colliderUpdateRate);
|
||||
EditorGUI.indentLevel--;
|
||||
}
|
||||
}
|
||||
else GUI.Label(new Rect(EditorGUIUtility.currentViewWidth / 2f - 25, 40, EditorGUIUtility.currentViewWidth / 2f - 30, 22), "No Mesh Colliders Available");
|
||||
}
|
||||
}
|
||||
else EditorGUILayout.LabelField("No Meshes Available");
|
||||
EditorGUILayout.EndVertical();
|
||||
|
||||
EditorGUILayout.EndHorizontal();
|
||||
|
||||
EditorGUILayout.EndVertical();
|
||||
if (EditorGUI.EndChangeCheck())
|
||||
{
|
||||
for(int i = 0; i < selected.Count; i++)
|
||||
{
|
||||
if (lastEnabled != enabled) properties[selected[i]].enabled = enabled;
|
||||
if (lastApplyRotation != applyRotation) properties[selected[i]].applyRotation = applyRotation;
|
||||
if (lastApplyScale != applyScale) properties[selected[i]].applyScale = applyScale;
|
||||
if (bendMesh != lastBendMesh) properties[selected[i]].bendMesh = bendMesh;
|
||||
if (bendCollider != lastBendCollider) properties[selected[i]].bendCollider = bendCollider;
|
||||
if (bendSpline != lastBendSpline) properties[selected[i]].bendSpline = bendSpline;
|
||||
if (lastColliderUpdateRate != colliderUpdateRate) properties[selected[i]].colliderUpdateRate = colliderUpdateRate;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void GetChildCount(Transform parent, ref int count)
|
||||
{
|
||||
foreach(Transform child in parent)
|
||||
{
|
||||
count++;
|
||||
GetChildCount(child, ref count);
|
||||
}
|
||||
}
|
||||
|
||||
public override void OnInspectorGUI()
|
||||
{
|
||||
showAveraging = false;
|
||||
base.OnInspectorGUI();
|
||||
}
|
||||
|
||||
protected override void BodyGUI()
|
||||
{
|
||||
base.BodyGUI();
|
||||
ObjectBender bender = (ObjectBender)target;
|
||||
|
||||
serializedObject.Update();
|
||||
SerializedProperty axis = serializedObject.FindProperty("_axis");
|
||||
SerializedProperty normalMode = serializedObject.FindProperty("_normalMode");
|
||||
SerializedProperty forwardMode = serializedObject.FindProperty("_forwardMode");
|
||||
|
||||
for (int i = 0; i < targets.Length; i++)
|
||||
{
|
||||
ObjectBender objBender = (ObjectBender)targets[i];
|
||||
int childCount = 0;
|
||||
GetChildCount(objBender.transform, ref childCount);
|
||||
if (objBender.bendProperties.Length - 1 != childCount && !Application.isPlaying) objBender.UpdateReferences();
|
||||
}
|
||||
EditorGUI.BeginChangeCheck();
|
||||
EditorGUILayout.PropertyField(axis, new GUIContent("Axis"));
|
||||
EditorGUILayout.PropertyField(normalMode, new GUIContent("Up Vector"));
|
||||
|
||||
if (normalMode.intValue == (int)ObjectBender.NormalMode.Custom)
|
||||
{
|
||||
SerializedProperty customNormal = serializedObject.FindProperty("_customNormal");
|
||||
EditorGUILayout.PropertyField(customNormal, new GUIContent("Custom Up"));
|
||||
}
|
||||
EditorGUILayout.PropertyField(forwardMode, new GUIContent("Forward Vector"));
|
||||
if (forwardMode.intValue == (int)ObjectBender.ForwardMode.Custom)
|
||||
{
|
||||
SerializedProperty customForward = serializedObject.FindProperty("_customForward");
|
||||
EditorGUILayout.PropertyField(customForward, new GUIContent("Custom Forward"));
|
||||
}
|
||||
|
||||
if (EditorGUI.EndChangeCheck())
|
||||
{
|
||||
serializedObject.ApplyModifiedProperties();
|
||||
for (int i = 0; i < targets.Length; i++)
|
||||
{
|
||||
ObjectBender objBender = (ObjectBender)targets[i];
|
||||
objBender.Rebuild();
|
||||
}
|
||||
}
|
||||
|
||||
if (targets.Length > 1)
|
||||
{
|
||||
EditorGUILayout.LabelField("Object properties unavailable when multiple benders are selected.", EditorStyles.centeredGreyMiniLabel);
|
||||
return;
|
||||
}
|
||||
if (!bender.bend)
|
||||
{
|
||||
float scrollHeight = Mathf.Min(bender.bendProperties.Length, 15) * 22;
|
||||
scroll = EditorGUILayout.BeginScrollView(scroll, GUILayout.Height(scrollHeight+5));
|
||||
|
||||
for (int i = 0; i < bender.bendProperties.Length; i++)
|
||||
{
|
||||
bool isSelected = selected.Contains(i);
|
||||
if (!bender.bendProperties[i].enabled)
|
||||
{
|
||||
GUI.color = Color.gray;
|
||||
if (isSelected) GUI.color = Color.Lerp(Color.gray, SplinePrefs.highlightColor, 0.5f);
|
||||
}
|
||||
else
|
||||
{
|
||||
if (isSelected) GUI.color = SplinePrefs.highlightColor;
|
||||
else GUI.color = Color.white;
|
||||
}
|
||||
GUILayout.Box(bender.bendProperties[i].transform.transform.name, GUILayout.Height(18), GUILayout.Width(EditorGUIUtility.currentViewWidth - 60));
|
||||
if (GUILayoutUtility.GetLastRect().Contains(Event.current.mousePosition) && Event.current.type == EventType.MouseDown)
|
||||
{
|
||||
if (Event.current.control)
|
||||
{
|
||||
if (!selected.Contains(i)) selected.Add(i);
|
||||
}
|
||||
else if (Event.current.shift && selected.Count > 0)
|
||||
{
|
||||
int from = selected[0];
|
||||
selected.Clear();
|
||||
if (from < i)
|
||||
{
|
||||
for (int n = from; n <= i; n++) selected.Add(n);
|
||||
}
|
||||
else
|
||||
{
|
||||
for (int n = from; n >= i; n--) selected.Add(n);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
selected.Clear();
|
||||
selected.Add(i);
|
||||
}
|
||||
Repaint();
|
||||
SceneView.RepaintAll();
|
||||
}
|
||||
GUI.color = Color.white;
|
||||
}
|
||||
EditorGUILayout.EndScrollView();
|
||||
|
||||
if (selected.Count > 0)
|
||||
{
|
||||
PropertyEditor(bender.bendProperties);
|
||||
}
|
||||
|
||||
if (selected.Count > 0)
|
||||
{
|
||||
if (Event.current.type == EventType.KeyDown)
|
||||
{
|
||||
if (Event.current.keyCode == KeyCode.DownArrow)
|
||||
{
|
||||
if (selected.Count > 1)
|
||||
{
|
||||
int temp = selected[selected.Count - 1];
|
||||
selected.Clear();
|
||||
selected.Add(temp);
|
||||
}
|
||||
selected[0]++;
|
||||
if (selected[0] >= bender.bendProperties.Length) selected[0] = 0;
|
||||
}
|
||||
if (Event.current.keyCode == KeyCode.UpArrow)
|
||||
{
|
||||
if (selected.Count > 1)
|
||||
{
|
||||
int temp = selected[0];
|
||||
selected.Clear();
|
||||
selected.Add(temp);
|
||||
}
|
||||
selected[0]--;
|
||||
if (selected[0] < 0) selected[0] = bender.bendProperties.Length - 1;
|
||||
}
|
||||
|
||||
Repaint();
|
||||
SceneView.RepaintAll();
|
||||
Event.current.Use();
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
string editModeText = "Enter Edit Mode";
|
||||
if (!bender.bend) editModeText = "Bend";
|
||||
if (GUILayout.Button(editModeText))
|
||||
{
|
||||
if (bender.bend) bender.bend = false;
|
||||
else bender.bend = true;
|
||||
}
|
||||
if (bender.bend && !generatedUvs)
|
||||
{
|
||||
if (GUILayout.Button("Generate Lightmap UVS"))
|
||||
{
|
||||
bender.EditorGenerateLightmapUVs();
|
||||
generatedUvs = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
protected override void DuringSceneGUI(SceneView currentSceneView)
|
||||
{
|
||||
base.DuringSceneGUI(currentSceneView);
|
||||
ObjectBender bender = (ObjectBender)target;
|
||||
if (selected.Count > 0)
|
||||
{
|
||||
Handles.BeginGUI();
|
||||
for(int i = 0; i < selected.Count; i++)
|
||||
{
|
||||
Vector2 screenPosition = HandleUtility.WorldToGUIPoint(bender.bendProperties[selected[i]].transform.transform.position);
|
||||
DreamteckEditorGUI.Label(new Rect(screenPosition.x - 120 + bender.bendProperties[selected[i]].transform.transform.name.Length * 4, screenPosition.y, 120, 25), bender.bendProperties[selected[i]].transform.transform.name);
|
||||
}
|
||||
Handles.EndGUI();
|
||||
}
|
||||
for(int i = 0; i < bender.bendProperties.Length; i++)
|
||||
{
|
||||
if(bender.bendProperties[i].bendSpline && bender.bendProperties[i].splineComputer != null)
|
||||
{
|
||||
DSSplineDrawer.DrawSplineComputer(bender.bendProperties[i].splineComputer, 0.0, 1.0, 0.2f);
|
||||
}
|
||||
}
|
||||
|
||||
//Draw bounds
|
||||
if (bender.bend) return;
|
||||
TS_Bounds bound = bender.GetBounds();
|
||||
Vector3 a = bender.transform.TransformPoint(bound.min);
|
||||
Vector3 b = bender.transform.TransformPoint(new Vector3(bound.max.x, bound.min.y, bound.min.z));
|
||||
Vector3 c = bender.transform.TransformPoint(new Vector3(bound.max.x, bound.min.y, bound.max.z));
|
||||
Vector3 d = bender.transform.TransformPoint(new Vector3(bound.min.x, bound.min.y, bound.max.z));
|
||||
|
||||
Vector3 e = bender.transform.TransformPoint(new Vector3(bound.min.x, bound.max.y, bound.min.z));
|
||||
Vector3 f = bender.transform.TransformPoint(new Vector3(bound.max.x, bound.max.y, bound.min.z));
|
||||
Vector3 g = bender.transform.TransformPoint(new Vector3(bound.max.x, bound.max.y, bound.max.z));
|
||||
Vector3 h = bender.transform.TransformPoint(new Vector3(bound.min.x, bound.max.y, bound.max.z));
|
||||
|
||||
Handles.color = Color.gray;
|
||||
Handles.DrawLine(a, b);
|
||||
Handles.DrawLine(b, c);
|
||||
Handles.DrawLine(c, d);
|
||||
Handles.DrawLine(d, a);
|
||||
|
||||
Handles.DrawLine(e, f);
|
||||
Handles.DrawLine(f, g);
|
||||
Handles.DrawLine(g, h);
|
||||
Handles.DrawLine(h, e);
|
||||
|
||||
Handles.DrawLine(a, e);
|
||||
Handles.DrawLine(b, f);
|
||||
Handles.DrawLine(c, g);
|
||||
Handles.DrawLine(d, h);
|
||||
|
||||
Vector3 r = bender.transform.right;
|
||||
Vector3 fr = bender.transform.forward;
|
||||
|
||||
switch (bender.axis)
|
||||
{
|
||||
case ObjectBender.Axis.Z: Handles.color = Color.blue; Handles.DrawLine(r + b, r + c); break;
|
||||
case ObjectBender.Axis.X: Handles.color = Color.red; Handles.DrawLine(b - fr, a - fr); break;
|
||||
case ObjectBender.Axis.Y: Handles.color = Color.green; Handles.DrawLine(b- fr + r, f - fr + r); break;
|
||||
}
|
||||
}
|
||||
|
||||
protected override void OnDestroy()
|
||||
{
|
||||
base.OnDestroy();
|
||||
SplineUser user = (SplineUser)target;
|
||||
if (Application.isEditor && !Application.isPlaying)
|
||||
{
|
||||
if (user == null) OnDelete(); //The object or the component is being deleted
|
||||
else if (user.spline != null)
|
||||
{
|
||||
if(!generatedUvs) user.Rebuild();
|
||||
}
|
||||
}
|
||||
SplineComputerEditor.hold = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user