使用CustomPropertyDrawer实现编辑器扩展只显示部分枚举值和枚举值中文显示

Unity默认的枚举值不支持数字开头，也不支持汉字；而且只能显示全部的枚举项，无法只显示部分选项

新建脚本EnumTest，代码如下

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public class EnumTest : MonoBehaviour
{   
    public EAttackLevel level;
}

public enum EAttackLevel
{
    None,
    Low,
    Med,
    High,
}

 

①我想不显示None选项

②我想枚举选项以数字开头

③我想枚举选项显示汉语

使用CustomPropertyDrawer以上需求都可以：

新建脚本EnumAttackLevelAttribute：

using System.Collections.Generic;
using UnityEditor;
using UnityEngine;

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

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

    public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
    {
        var att = (EnumAttackLevelAttribute)attribute;
        var type = property.serializedObject.targetObject.GetType();
        var field = type.GetField(property.name);
        var enumtype = field.FieldType;
        foreach (var enumName in property.enumNames)
        {
            var enumfield = enumtype.GetField(enumName);
            if (enumfield.Name == "None")//不显示None
            {
                continue;
            }
            var hds = enumfield.GetCustomAttributes(typeof(HeaderAttribute), false);
            m_displayNames.Add(hds.Length <= 0 ? enumName : ((HeaderAttribute)hds[0]).header);//如果加了自定义属性，显示自定义名，否则显示枚举选项名
        }
        EditorGUI.BeginChangeCheck();
        var value = EditorGUI.Popup(position, att.header, property.enumValueIndex, m_displayNames.ToArray());
        if (EditorGUI.EndChangeCheck())
        {
            property.enumValueIndex = value + 1;//因为我们隐藏了一个显示项None，这儿别忘了加1
            Debug.LogError("value " + property.enumValueIndex.ToString());
        }       
    }
}

 

 

 EnumTest脚本改成如下：

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public class EnumTest : MonoBehaviour
{
    [EnumAttackLevel("攻击级别")]
    public EAttackLevel level;
}

public enum EAttackLevel
{
    [Header("0空")]
    None,

    [Header("1低")]
    Low,

    [Header("2中")]
    Med,

    [Header("3高")]
    High,
}

再在Unity中看我们的枚举值：

 

实现所有需求