MinhHai
120 lines
3.1 KiB
C#
120 lines
3.1 KiB
C#
using UnityEditor;
|
|
using UnityEngine.UIElements;
|
|
using System.Collections.Generic;
|
|
using EditorAttributes.Editor.Utility;
|
|
|
|
namespace EditorAttributes.Editor
|
|
{
|
|
[CustomPropertyDrawer(typeof(ConditionalFieldAttribute))]
|
|
public class ConditionalFieldDrawer : PropertyDrawerBase
|
|
{
|
|
public override VisualElement CreatePropertyGUI(SerializedProperty property)
|
|
{
|
|
var conditionalAttribute = attribute as ConditionalFieldAttribute;
|
|
|
|
var root = new VisualElement();
|
|
var propertyField = CreatePropertyField(property);
|
|
|
|
var errorBox = new HelpBox();
|
|
|
|
root.Add(propertyField);
|
|
|
|
UpdateVisualElement(root, () =>
|
|
{
|
|
var canDrawProperty = CanDrawProperty(conditionalAttribute, conditionalAttribute.BooleanNames, property, errorBox);
|
|
|
|
switch (conditionalAttribute.ConditionResult)
|
|
{
|
|
case ConditionResult.ShowHide:
|
|
if (canDrawProperty)
|
|
{
|
|
AddElement(root, propertyField);
|
|
}
|
|
else
|
|
{
|
|
RemoveElement(root, propertyField);
|
|
}
|
|
break;
|
|
|
|
case ConditionResult.EnableDisable:
|
|
propertyField.SetEnabled(canDrawProperty);
|
|
break;
|
|
}
|
|
|
|
DisplayErrorBox(root, errorBox);
|
|
});
|
|
|
|
return root;
|
|
}
|
|
|
|
private bool CanDrawProperty(ConditionalFieldAttribute attribute, string[] conditionNames, SerializedProperty property, HelpBox errorBox)
|
|
{
|
|
var booleanList = new List<bool>();
|
|
|
|
foreach (var conditionName in conditionNames)
|
|
{
|
|
var memberInfo = ReflectionUtility.GetValidMemberInfo(conditionName, property);
|
|
var serializedProperty = property.serializedObject.FindProperty(conditionName);
|
|
|
|
if (memberInfo == null)
|
|
{
|
|
errorBox.text = $"The provided condition \"{conditionName}\" could not be found";
|
|
continue;
|
|
}
|
|
|
|
if (ReflectionUtility.GetMemberInfoType(memberInfo) == typeof(bool))
|
|
{
|
|
var propertyValue = (bool)ReflectionUtility.GetMemberInfoValue(memberInfo, property);
|
|
|
|
booleanList.Add(propertyValue);
|
|
}
|
|
else if (serializedProperty != null && serializedProperty.propertyType == SerializedPropertyType.Boolean)
|
|
{
|
|
var propertyValue = serializedProperty.boolValue;
|
|
|
|
booleanList.Add(propertyValue);
|
|
}
|
|
else
|
|
{
|
|
errorBox.text = $"The provided condition \"{conditionName}\" is not a valid boolean";
|
|
}
|
|
}
|
|
|
|
for (int i = 0; i < booleanList.Count; i++)
|
|
{
|
|
if (!(attribute.NegatedValues == null || attribute.NegatedValues.Length == 0))
|
|
{
|
|
if (attribute.NegatedValues[i])
|
|
booleanList[i] = !booleanList[i];
|
|
}
|
|
|
|
switch (attribute.ConditionType)
|
|
{
|
|
case ConditionType.AND:
|
|
if (!booleanList[i]) return false;
|
|
continue;
|
|
|
|
case ConditionType.OR:
|
|
if (booleanList[i]) return true;
|
|
continue;
|
|
|
|
case ConditionType.NAND:
|
|
if (!booleanList[i]) return true;
|
|
continue;
|
|
|
|
case ConditionType.NOR:
|
|
if (booleanList[i]) return false;
|
|
continue;
|
|
}
|
|
}
|
|
|
|
return attribute.ConditionType switch
|
|
{
|
|
ConditionType.AND => true,
|
|
ConditionType.NOR => true,
|
|
_ => false,
|
|
};
|
|
}
|
|
}
|
|
}
|