Enum Mask Property Drawer

EnumMaskPropertyDrawer.cs

// (C) 2014 ERAL
// Distributed under the Boost Software License, Version 1.0.
// (See copy at http://www.boost.org/LICENSE_1_0.txt)

using UnityEngine;

#if UNITY_EDITOR
using UnityEditor;

[CustomPropertyDrawer(typeof(EnumMaskAttribute))]
public class EnumMaskPropertyDrawer : PropertyDrawer {
	
	public override void OnGUI(Rect position, SerializedProperty property, GUIContent label) {
		if (SerializedPropertyType.Enum == property.propertyType) {
			object current = GetCurrent(property);
			if (null != current) {
				EditorGUI.BeginChangeCheck();
				var value = EditorGUI.EnumMaskField(position, label, (System.Enum)current);
				if (EditorGUI.EndChangeCheck()) {
					property.intValue = System.Convert.ToInt32(value);
				}
			}
		} else {
			EditorGUI.LabelField(position, label, new GUIContent("This type has not supported."));
		}
	}

	private static object GetCurrent(SerializedProperty property) {
		object result = property.serializedObject.targetObject;
		var property_names = property.propertyPath.Replace(".Array.data", ".").Split('.');
		foreach (var property_name in property_names) {
			var parent = result;
			var indexer_start = property_name.IndexOf('[');
			if (-1 == indexer_start) {
				var binding_flags = System.Reflection.BindingFlags.Instance | System.Reflection.BindingFlags.Public | System.Reflection.BindingFlags.NonPublic;
				result = parent.GetType().GetField(property_name, binding_flags).GetValue(parent);
			} else if (parent.GetType().IsArray) {
				var indexer_end = property_name.IndexOf(']');
				var index_string = property_name.Substring(indexer_start + 1, indexer_end - indexer_start - 1);
				var index = int.Parse(index_string);
				var array = (System.Array)parent;
				if (index < array.Length) {
					result = array.GetValue(index);
				} else {
					result = null;
					break;
				}
			} else {
				throw new System.MissingFieldException();
			}
		}
		return result;
	}
}
#endif

public class EnumMaskAttribute : PropertyAttribute {
}

Sample.cs

// Created by ERAL
// This is free and unencumbered software released into the public domain.

using UnityEngine;
using System.Collections;

public class Sample : MonoBehaviour {
	[EnumMask]
	public Flags m_Flags;

	[System.Flags]
	public enum Flags {
		foo = 1<<0,
		bar = 1<<1,
		baz = 1<<2,
		qux = 1<<3,
	}
}

sample.png

Can be used to struct & array & private field.

Thanks for this it's very useful. Just a hint to anyone else using it: with the unity inspector in debug mode this is disabled or something. Took me ages to figure that one out!