Files
Fishing2/Assets/vInspector/VInspectorDrawers.cs

500 lines
16 KiB
C#

#if UNITY_EDITOR
using System.Collections;
using System.Linq;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.UIElements;
using UnityEditor.UIElements;
using System.Reflection;
using UnityEditor;
using UnityEditorInternal;
using Type = System.Type;
using Attribute = System.Attribute;
using static VInspector.VInspectorState;
using static VInspector.Libs.VUtils;
using static VInspector.Libs.VGUI;
// using static VTools.VDebug;
namespace VInspector
{
[CustomPropertyDrawer(typeof(SerializedDictionary<,>), true)]
public class SerializedDictionaryDrawer : PropertyDrawer
{
public override void OnGUI(Rect rect, SerializedProperty prop, GUIContent label)
{
var indentedRect = EditorGUI.IndentedRect(rect);
void header()
{
var headerRect = indentedRect.SetHeight(EditorGUIUtility.singleLineHeight);
void foldout()
{
var fullHeaderRect = headerRect.MoveX(3).AddWidthFromRight(17);
if (fullHeaderRect.IsHovered())
fullHeaderRect.Draw(Greyscale(1, .07f));
SetGUIColor(Color.clear);
SetGUIEnabled(true);
if (GUI.Button(fullHeaderRect.AddWidth(-50), ""))
prop.isExpanded = !prop.isExpanded;
ResetGUIColor();
ResetGUIEnabled();
var triangleRect = rect.SetHeight(EditorGUIUtility.singleLineHeight);
SetGUIEnabled(true);
EditorGUI.Foldout(triangleRect, prop.isExpanded, "");
ResetGUIEnabled();
}
void label()
{
SetLabelBold();
SetLabelFontSize(12);
SetGUIColor(Greyscale(.9f));
SetGUIEnabled(true);
GUI.Label(headerRect, prop.displayName);
ResetGUIEnabled();
ResetGUIColor();
ResetLabelStyle();
}
void count()
{
if (kvpsProp_byPropPath[prop.propertyPath].hasMultipleDifferentValues) return;
kvpsProp_byPropPath[prop.propertyPath].arraySize = EditorGUI.DelayedIntField(headerRect.SetWidthFromRight(48 + EditorGUI.indentLevel * 15), kvpsProp_byPropPath[prop.propertyPath].arraySize);
}
void repeatedKeysWarning()
{
if (!curEvent.isRepaint) return;
var hasRepeatedKeys = false;
var hasNullKeys = false;
for (int i = 0; i < kvpsProp_byPropPath[prop.propertyPath].arraySize; i++)
{
hasRepeatedKeys |= kvpsProp_byPropPath[prop.propertyPath].GetArrayElementAtIndex(i).FindPropertyRelative("isKeyRepeated").boolValue;
hasNullKeys |= kvpsProp_byPropPath[prop.propertyPath].GetArrayElementAtIndex(i).FindPropertyRelative("isKeyNull").boolValue;
}
if (!hasRepeatedKeys && !hasNullKeys) return;
var warningTextRect = headerRect.AddWidthFromRight(-prop.displayName.GetLabelWidth(isBold: true));
var warningIconRect = warningTextRect.SetHeightFromMid(20).SetWidth(20);
var warningText = (hasRepeatedKeys && hasNullKeys) ? "Repeated and null keys"
: hasRepeatedKeys ? "Repeated keys"
: hasNullKeys ? "Null keys" : "";
GUI.Label(warningIconRect, EditorGUIUtility.IconContent("Warning"));
SetGUIColor(new Color(1, .9f, .03f) * 1.1f);
GUI.Label(warningTextRect.MoveX(16), warningText);
ResetGUIColor();
}
foldout();
label();
count();
repeatedKeysWarning();
}
void list_()
{
if (!prop.isExpanded) return;
SetupList(prop);
lists_byPropPath[prop.propertyPath].DoList(indentedRect.AddHeightFromBottom(-EditorGUIUtility.singleLineHeight - 3));
}
SetupProps(prop);
header();
list_();
}
public override float GetPropertyHeight(SerializedProperty prop, GUIContent label)
{
SetupProps(prop);
var height = EditorGUIUtility.singleLineHeight;
if (prop.isExpanded)
{
SetupList(prop);
height += lists_byPropPath[prop.propertyPath].GetHeight() + 3;
}
return height;
}
float GetListElementHeight(int index, SerializedProperty prop)
{
var kvpProp = kvpsProp_byPropPath[prop.propertyPath].GetArrayElementAtIndex(index);
var keyProp = kvpProp.FindPropertyRelative("Key");
var valueProp = kvpProp.FindPropertyRelative("Value");
float propHeight(SerializedProperty prop)
{
// var height = typeof(Editor).Assembly.GetType("UnityEditor.ScriptAttributeUtility").InvokeMethod("GetHandler", prop).InvokeMethod<float>("GetHeight", prop, GUIContent.none, true);
var height = EditorGUI.GetPropertyHeight(prop);
if (!IsSingleLine(prop) && prop.type != "EventReference")
height -= 10;
return height;
}
return Mathf.Max(propHeight(keyProp), propHeight(valueProp));
}
void DrawListElement(Rect rect, int index, bool isActive, bool isFocused, SerializedProperty prop)
{
Rect keyRect;
Rect valueRect;
Rect dividerRect;
var kvpProp = kvpsProp_byPropPath[prop.propertyPath].GetArrayElementAtIndex(index);
var keyProp = kvpProp.FindPropertyRelative("Key");
var valueProp = kvpProp.FindPropertyRelative("Value");
void drawProp(Rect rect, SerializedProperty prop)
{
if (IsSingleLine(prop)) { EditorGUI.PropertyField(rect.SetHeight(EditorGUIUtility.singleLineHeight), prop, GUIContent.none); return; }
prop.isExpanded = true;
GUI.BeginGroup(rect);
if (prop.type == "EventReference") // don't hide first line for FMOD EventReference
{
EditorGUIUtility.labelWidth = 1;
EditorGUI.PropertyField(rect.SetPos(0, 0), prop, GUIContent.none);
EditorGUIUtility.labelWidth = 0;
}
else
EditorGUI.PropertyField(rect.SetPos(0, -20), prop, true);
GUI.EndGroup();
}
void rects()
{
var dividerWidh = 6f;
var dividerPos = dividerPosProp.floatValue.Clamp(.2f, .8f);
var fullRect = rect.AddWidthFromRight(-1).AddHeightFromMid(-2);
keyRect = fullRect.SetWidth(fullRect.width * dividerPos - dividerWidh / 2);
valueRect = fullRect.SetWidthFromRight(fullRect.width * (1 - dividerPos) - dividerWidh / 2);
dividerRect = fullRect.MoveX(fullRect.width * dividerPos - dividerWidh / 2).SetWidth(dividerWidh).Resize(-1);
}
void key()
{
drawProp(keyRect, keyProp);
}
void warning()
{
var isKeyRepeated = kvpProp.FindPropertyRelative("isKeyRepeated").boolValue;
var isKeyNull = kvpProp.FindPropertyRelative("isKeyNull").boolValue;
if (!isKeyRepeated && !isKeyNull) return;
var warningRect = keyRect.SetWidthFromRight(20).SetHeight(20).MoveY(-1);
if (kvpProp.FindPropertyRelative("Key").propertyType == SerializedPropertyType.ObjectReference)
warningRect = warningRect.MoveX(-17);
GUI.Label(warningRect, EditorGUIUtility.IconContent("Warning"));
}
void value()
{
drawProp(valueRect, valueProp);
}
void divider()
{
EditorGUIUtility.AddCursorRect(dividerRect, MouseCursor.ResizeHorizontal);
if (!rect.IsHovered()) return;
if (dividerRect.IsHovered())
{
if (curEvent.isMouseDown)
isDividerDragged = true;
if (curEvent.isMouseUp || curEvent.isMouseMove || curEvent.isMouseLeaveWindow)
isDividerDragged = false;
}
if (isDividerDragged && curEvent.isMouseDrag)
dividerPosProp.floatValue += curEvent.mouseDelta.x / rect.width;
}
rects();
key();
warning();
value();
divider();
}
void DrawDictionaryIsEmpty(Rect rect) => GUI.Label(rect, "Dictionary is empty");
IEnumerable<SerializedProperty> GetChildren(SerializedProperty prop, bool enterVisibleGrandchildren)
{
var startPath = prop.propertyPath;
var enterVisibleChildren = true;
while (prop.NextVisible(enterVisibleChildren) && prop.propertyPath.StartsWith(startPath))
{
yield return prop;
enterVisibleChildren = enterVisibleGrandchildren;
}
}
bool IsSingleLine(SerializedProperty prop) => prop.propertyType != SerializedPropertyType.Generic || !prop.hasVisibleChildren || prop.type == "AssetReference";
public void SetupList(SerializedProperty prop)
{
if (lists_byPropPath.ContainsKey(prop.propertyPath)) return;
SetupProps(prop);
lists_byPropPath[prop.propertyPath] = new ReorderableList(kvpsProp_byPropPath[prop.propertyPath].serializedObject, kvpsProp_byPropPath[prop.propertyPath], true, false, true, true);
lists_byPropPath[prop.propertyPath].drawElementCallback = (q, w, e, r) => DrawListElement(q, w, e, r, prop);
lists_byPropPath[prop.propertyPath].elementHeightCallback = (q) => GetListElementHeight(q, prop);
lists_byPropPath[prop.propertyPath].drawNoneElementCallback = DrawDictionaryIsEmpty;
}
Dictionary<string, ReorderableList> lists_byPropPath = new();
// ReorderableList list;
bool isDividerDragged;
public void SetupProps(SerializedProperty prop)
{
if (kvpsProp_byPropPath.ContainsKey(prop.propertyPath)) return;
kvpsProp_byPropPath[prop.propertyPath] = prop.FindPropertyRelative("serializedKvps");
this.dividerPosProp = prop.FindPropertyRelative("dividerPos");
}
Dictionary<string, SerializedProperty> kvpsProp_byPropPath = new();
// SerializedProperty kvpsProp;
SerializedProperty dividerPosProp;
}
[CustomPropertyDrawer(typeof(VariantsAttribute))]
public class VariantsDrawer : PropertyDrawer
{
public override void OnGUI(Rect rect, SerializedProperty prop, GUIContent label)
{
var variantsAttribtue = (VariantsAttribute)attribute;
if (variantsAttribtue.variants.Length == 1 && variantsAttribtue.variants[0] is string dynamicCollectionName)
{
var target = prop.serializedObject.targetObject;
IEnumerable ienum = null;
if (target.GetType().GetMember(dynamicCollectionName, maxBindingFlags).FirstOrDefault() is MemberInfo collectionMember)
if (collectionMember is MethodInfo methodInfo)
ienum = methodInfo.Invoke(target, null) as IEnumerable;
else
ienum = target.GetMemberValue(dynamicCollectionName) as IEnumerable;
if (ienum != null)
{
var variantsList = new List<object>();
foreach (var r in ienum)
variantsList.Add(r);
variantsAttribtue.variants = variantsList.ToArray();
}
}
var variantsArray = variantsAttribtue.variants;
EditorGUI.BeginProperty(rect, label, prop);
var iCur = prop.hasMultipleDifferentValues ? -1 : variantsArray.ToList().IndexOf(prop.GetBoxedValue());
var iNew = EditorGUI.IntPopup(rect, label.text, iCur, variantsArray.Select(r => r.ToString()).ToArray(), Enumerable.Range(0, variantsArray.Length).ToArray());
if (iNew != -1)
prop.SetBoxedValue(variantsArray[iNew]);
else if (!prop.hasMultipleDifferentValues)
prop.SetBoxedValue(variantsArray[0]);
EditorGUI.EndProperty();
}
}
[CustomPropertyDrawer(typeof(MinMaxSliderAttribute))]
public class MinMaxSliderDrawer : PropertyDrawer
{
public override void OnGUI(Rect rect, SerializedProperty prop, GUIContent label)
{
var fieldWidth = 52;
var controlsRect = rect.AddWidthFromRight(-EditorGUIUtility.labelWidth);
var minFieldRect = controlsRect.SetWidth(fieldWidth).AddWidthFromRight(-2 + EditorGUI.indentLevel * 15);
var maxFieldRect = controlsRect.SetWidthFromRight(fieldWidth).AddWidthFromRight(-2 + EditorGUI.indentLevel * 15);
var sliderRect = controlsRect.AddWidthFromMid(-fieldWidth * 2 - 4).AddWidthFromRight(-2 + EditorGUI.indentLevel * 15);
var isInt = prop.propertyType == SerializedPropertyType.Vector2Int;
var min = isInt ? prop.vector2IntValue.x : prop.vector2Value.x;
var max = isInt ? prop.vector2IntValue.y : prop.vector2Value.y;
var minLimit = ((MinMaxSliderAttribute)attribute).min;
var maxLimit = ((MinMaxSliderAttribute)attribute).max;
EditorGUI.PrefixLabel(rect, label);
EditorGUI.BeginProperty(rect, label, prop);
if (sliderRect.width > 14)
{
EditorGUI.BeginChangeCheck();
EditorGUI.MinMaxSlider(sliderRect, ref min, ref max, minLimit, maxLimit);
if (EditorGUI.EndChangeCheck())
{
var abs = (maxLimit - minLimit).Abs();
var decimals = abs > 190 ?
0 : abs > 19 ?
1 : abs > 1.9f ?
2 :
3;
min = (float)System.Math.Round(min, decimals);
max = (float)System.Math.Round(max, decimals);
// same rounding logic as for [Range]
}
}
min = EditorGUI.DelayedFloatField(minFieldRect, min).Max(minLimit).Min(maxLimit);
max = EditorGUI.DelayedFloatField(maxFieldRect, max).Max(min).Max(minLimit).Min(maxLimit);
if (isInt)
prop.vector2IntValue = new Vector2Int(min.RoundToInt(), max.RoundToInt());
else
prop.vector2Value = new Vector2(min, max);
EditorGUI.EndProperty();
}
}
[CustomPropertyDrawer(typeof(TagAttribute))]
public class TagDrawer : PropertyDrawer
{
public override void OnGUI(Rect rect, SerializedProperty prop, GUIContent label)
{
EditorGUI.BeginProperty(rect, label, prop);
prop.stringValue = EditorGUI.TagField(rect, label, prop.stringValue);
EditorGUI.EndProperty();
}
}
[CustomPropertyDrawer(typeof(LayerAttribute))]
public class LayerDrawer : PropertyDrawer
{
public override void OnGUI(Rect rect, SerializedProperty prop, GUIContent label)
{
EditorGUI.BeginProperty(rect, label, prop);
prop.intValue = EditorGUI.LayerField(rect, label, prop.intValue);
EditorGUI.EndProperty();
}
}
}
#endif