使用PropertyDrawer自定枚举显示

# 默认的展示方式

 # 自定义后的展示方式

 

# 标签类，enum成员加上这个标签才会按自定义的方式显示

public class CustomEnumAttribute : HeaderAttribute
{
    public CustomEnumAttribute(string header) : base(header) { }
}

# PropertyDrawer类

[CustomPropertyDrawer(typeof(CustomEnumAttribute))]
public class CustomEnumPropertyDrawer : PropertyDrawer
{
    private readonly List<string> m_displayNames = new List<string>();

    public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
    {
        var attr = (CustomEnumAttribute)attribute;
        var propertyLabel = attr.header;
        if (string.IsNullOrEmpty(propertyLabel))
            propertyLabel = label.text;
        
        var type = property.serializedObject.targetObject.GetType();
        FieldInfo field = type.GetField(property.name, BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance);
        if (null == field)
        {
            EditorGUI.LabelField(position, $"{propertyLabel}: {property.name}未找到");
            return;
        }

        var fieldType = field.FieldType;
        if (!fieldType.IsEnum)
        {
            EditorGUI.LabelField(position, $"{propertyLabel}: {property.name}不是枚举类型");
            return;
        }
    
        var enumNames = property.enumNames;
        foreach (var enumName in enumNames)
        {
            var enumItem = fieldType.GetField(enumName);
            var hds = enumItem.GetCustomAttributes(typeof(HeaderAttribute), false); //获取枚举上的Header标签
            m_displayNames.Add(hds.Length <= 0 ? enumName : ((HeaderAttribute) hds[0]).header);
        }

        EditorGUI.BeginChangeCheck();
        var value = EditorGUI.Popup(position, propertyLabel, property.enumValueIndex, m_displayNames.ToArray());
        if (EditorGUI.EndChangeCheck())
        {
            property.enumValueIndex = value;
        }
    }
}

# 使用

public enum Animal
{
    None,
    [Header("哺乳动物/猪")]
    Pig,
    [Header("鱼类/金鱼")]
    GoldFish,
}

public class NewBehaviourScript : MonoBehaviour
{
    [CustomEnum("")]
    public Animal _animal;
    
}