hellbound/Assets/Sources/Lean/Common/Scripts/LeanEditor.cs

467 lines
11 KiB
C#

#if UNITY_EDITOR
using UnityEngine;
using UnityEditor;
using System.Collections.Generic;
namespace Lean.Common
{
/// <summary>This is the base class for all inspectors.</summary>
public abstract class LeanEditor : UnityEditor.Editor
{
private static SerializedObject dataA;
private static SerializedProperty dataB;
private static GUIContent customContent = new GUIContent();
private static List<Color> colors = new List<Color>();
private static List<float> labelWidths = new List<float>();
private static List<bool> mixedValues = new List<bool>();
public void GetTargets<T>(out T tgt, out T[] tgts)
where T : Object
{
tgt = (T)target;
tgts = System.Array.ConvertAll(targets, item => (T)item);
}
public static void SetData(SerializedObject newData)
{
dataA = newData;
dataB = null;
}
public static void SetData(SerializedProperty newData)
{
dataA = null;
dataB = newData;
}
public static void ClearData()
{
dataA = null;
dataB = null;
}
public override void OnInspectorGUI()
{
SetData(serializedObject);
ClearStacks();
Separator();
OnInspector();
Separator();
serializedObject.ApplyModifiedProperties();
ClearData();
}
public virtual void OnSceneGUI()
{
OnScene();
if (GUI.changed == true)
{
EditorUtility.SetDirty(target);
}
}
protected void Each<T>(T[] tgts, System.Action<T> update, bool dirty = false)
where T : Object
{
foreach (var t in tgts)
{
update(t);
if (dirty == true)
{
EditorUtility.SetDirty(t);
}
}
}
protected bool Any<T>(T[] tgts, System.Func<T, bool> check)
where T : Object
{
foreach (var t in tgts)
{
if (check(t) == true)
{
return true;
}
}
return false;
}
protected bool All<T>(T[] tgts, System.Func<T, bool> check)
where T : Object
{
foreach (var t in tgts)
{
if (check(t) == false)
{
return false;
}
}
return true;
}
public static Rect Reserve(float height = 19.0f)
{
var rect =
EditorGUILayout.BeginVertical();
EditorGUILayout.LabelField(string.Empty, GUILayout.Height(height), GUILayout.ExpandWidth(true), GUILayout.MinWidth(0.0f));
EditorGUILayout.EndVertical();
return rect;
}
public static void Info(string message)
{
EditorGUILayout.HelpBox(StripRichText(message), MessageType.Info); // Help boxes can't display rich text for some reason, so strip it
}
public static void Warning(string message)
{
EditorGUILayout.HelpBox(StripRichText(message), MessageType.Warning); // Help boxes can't display rich text for some reason, so strip it
}
public static void Error(string message)
{
EditorGUILayout.HelpBox(StripRichText(message), MessageType.Error); // Help boxes can't display rich text for some reason, so strip it
}
public static void Separator()
{
EditorGUILayout.Separator();
}
public static void BeginIndent()
{
EditorGUI.indentLevel += 1;
}
public static void EndIndent()
{
EditorGUI.indentLevel -= 1;
}
public static bool Button(string text)
{
return GUILayout.Button(text);
}
public static bool HelpButton(string helpText, UnityEditor.MessageType type, string buttonText, float buttonWidth)
{
var clicked = false;
EditorGUILayout.BeginHorizontal();
{
EditorGUILayout.HelpBox(helpText, type);
var style = new GUIStyle(GUI.skin.button); style.wordWrap = true;
clicked = GUILayout.Button(buttonText, style, GUILayout.ExpandHeight(true), GUILayout.Width(buttonWidth));
}
EditorGUILayout.EndHorizontal();
return clicked;
}
public static void ClearStacks()
{
while (colors.Count > 0)
{
EndColor();
}
while (labelWidths.Count > 0)
{
EndLabelWidth();
}
while (mixedValues.Count > 0)
{
EndMixedValue();
}
}
public static void BeginMixedValue(bool mixed = true)
{
mixedValues.Add(EditorGUI.showMixedValue);
EditorGUI.showMixedValue = mixed;
}
public static void EndMixedValue()
{
if (mixedValues.Count > 0)
{
var index = mixedValues.Count - 1;
EditorGUI.showMixedValue = mixedValues[index];
mixedValues.RemoveAt(index);
}
}
public static void BeginDisabled(bool disabled = true)
{
EditorGUI.BeginDisabledGroup(disabled);
}
public static void EndDisabled()
{
EditorGUI.EndDisabledGroup();
}
public static void BeginError(bool error = true)
{
BeginColor(Color.red, error);
}
public static void EndError()
{
EndColor();
}
public static void BeginColor(Color color, bool show = true)
{
colors.Add(GUI.color);
GUI.color = show == true ? color : colors[0];
}
public static void EndColor()
{
if (colors.Count > 0)
{
var index = colors.Count - 1;
GUI.color = colors[index];
colors.RemoveAt(index);
}
}
public static void BeginLabelWidth(float width)
{
labelWidths.Add(EditorGUIUtility.labelWidth);
EditorGUIUtility.labelWidth = width;
}
public static void EndLabelWidth()
{
if (labelWidths.Count > 0)
{
var index = labelWidths.Count - 1;
EditorGUIUtility.labelWidth = labelWidths[index];
labelWidths.RemoveAt(index);
}
}
public static bool DrawFoldout(string overrideText, string overrideTooltip, string propertyPath = "m_Name")
{
var property = GetPropertyAndSetCustomContent(propertyPath, overrideTooltip, overrideText);
property.isExpanded = EditorGUILayout.Foldout(property.isExpanded, customContent);
return property.isExpanded;
}
public static bool DrawExpand(string propertyPath, ref bool modified, string overrideTooltip = null, string overrideText = null)
{
var property = GetPropertyAndSetCustomContent(propertyPath, overrideTooltip, overrideText);
var rect = EditorGUILayout.BeginVertical(); EditorGUILayout.LabelField(string.Empty, GUILayout.Height(EditorGUI.GetPropertyHeight(property))); EditorGUILayout.EndVertical();
var rectF = rect; rectF.height = 16;
property.isExpanded = EditorGUI.Foldout(rectF, property.isExpanded, GUIContent.none);
EditorGUI.BeginChangeCheck();
EditorGUI.PropertyField(rect, property, customContent, true);
modified = EditorGUI.EndChangeCheck();
return property.isExpanded;
}
public static bool DrawExpand(string propertyPath, string overrideTooltip = null, string overrideText = null)
{
var modified = false; return DrawExpand(propertyPath, ref modified, overrideTooltip, overrideText);
}
public static bool Draw(string propertyPath, string overrideTooltip = null, string overrideText = null)
{
var property = GetPropertyAndSetCustomContent(propertyPath, overrideTooltip, overrideText);
EditorGUI.BeginChangeCheck();
EditorGUILayout.PropertyField(property, customContent, true);
return EditorGUI.EndChangeCheck();
}
public static void Draw(string propertyPath, ref bool dirty, string overrideTooltip = null, string overrideText = null)
{
if (Draw(propertyPath, overrideTooltip, overrideText) == true)
{
dirty = true;
}
}
public static void Draw(string propertyPath, ref bool dirty1, ref bool dirty2, string overrideTooltip = null, string overrideText = null)
{
if (Draw(propertyPath, overrideTooltip, overrideText) == true)
{
dirty1 = true;
dirty2 = true;
}
}
public static bool DrawMinMax(string propertyPath, float min, float max, string overrideTooltip = null, string overrideText = null)
{
var property = GetPropertyAndSetCustomContent(propertyPath, overrideTooltip, overrideText);
var value = property.vector2Value;
EditorGUI.BeginChangeCheck();
EditorGUILayout.MinMaxSlider(customContent, ref value.x, ref value.y, min, max);
if (EditorGUI.EndChangeCheck() == true)
{
property.vector2Value = value;
return true;
}
return false;
}
public static bool DrawIntPopup(int[] values, GUIContent[] contents, string propertyPath, string overrideTooltip = null, string overrideText = null)
{
var property = GetPropertyAndSetCustomContent(propertyPath, overrideTooltip, overrideText);
EditorGUI.BeginChangeCheck();
EditorGUILayout.IntPopup(property, contents, values, customContent);
return EditorGUI.EndChangeCheck();
}
public static void DrawIntPopup(int[] values, GUIContent[] contents, string propertyPath, ref bool modified, string overrideTooltip = null, string overrideText = null)
{
if (DrawIntPopup(values, contents, propertyPath, overrideTooltip, overrideText) == true)
{
modified = true;
}
}
public static bool DrawLayer(string propertyPath, string overrideTooltip = null, string overrideText = null)
{
var property = GetPropertyAndSetCustomContent(propertyPath, overrideTooltip, overrideText);
var value = property.intValue;
EditorGUI.BeginChangeCheck();
value = EditorGUILayout.LayerField(customContent, value);
if (EditorGUI.EndChangeCheck() == true)
{
property.intValue = value;
return true;
}
return false;
}
public static bool DrawEulerAngles(string propertyPath, string overrideTooltip = null, string overrideText = null)
{
var property = GetPropertyAndSetCustomContent(propertyPath, overrideTooltip, overrideText);
var value = property.quaternionValue.eulerAngles;
EditorGUI.BeginChangeCheck();
BeginMixedValue(property.hasMultipleDifferentValues);
value = EditorGUILayout.Vector3Field(customContent, value);
EndMixedValue();
if (EditorGUI.EndChangeCheck() == true)
{
property.quaternionValue = Quaternion.Euler(value);
return true;
}
return false;
}
public static SerializedProperty GetProperty(string propertyPath)
{
var property = default(SerializedProperty);
if (dataA != null)
{
property = dataA.FindProperty(propertyPath);
}
else if (dataB != null)
{
property = dataB.FindPropertyRelative(propertyPath);
}
return property;
}
protected void DirtyAndUpdate()
{
for (var i = targets.Length - 1; i >= 0; i--)
{
EditorUtility.SetDirty(targets[i]);
}
serializedObject.Update();
}
private static SerializedProperty GetPropertyAndSetCustomContent(string propertyPath, string overrideTooltip, string overrideText)
{
var property = GetProperty(propertyPath);
customContent.text = string.IsNullOrEmpty(overrideText ) == false || property == null ? overrideText : property.displayName;
customContent.tooltip = string.IsNullOrEmpty(overrideTooltip) == false || property == null ? overrideTooltip : property.tooltip;
customContent.tooltip = StripRichText(customContent.tooltip); // Tooltips can't display rich text for some reason, so strip it
return property;
}
private static string StripRichText(string s)
{
if (s != null)
{
s = s.Replace("<b>", "").Replace("</b>", "");
}
return s;
}
protected virtual void OnInspector()
{
}
protected virtual void OnScene()
{
}
}
}
#endif