using UnityEditor; using UnityEngine; namespace UnityAtoms.Editor { /// /// A custom property drawer for References. Makes it possible to choose between a Variable and a constant value (not a Atom Contant, but a regular value). /// [CustomPropertyDrawer(typeof(AtomReference), true)] public class AtomReferenceDrawer : PropertyDrawer { private static readonly string[] _popupOptions = { "Use Constant", "Use Variable" }; private static GUIStyle _popupStyle; public override void OnGUI(Rect position, SerializedProperty property, GUIContent label) { if (_popupStyle == null) { _popupStyle = new GUIStyle(GUI.skin.GetStyle("PaneOptions")); _popupStyle.imagePosition = ImagePosition.ImageOnly; } label = EditorGUI.BeginProperty(position, label, property); position = EditorGUI.PrefixLabel(position, label); EditorGUI.BeginChangeCheck(); // Get properties SerializedProperty useConstant = property.FindPropertyRelative("UseConstant"); SerializedProperty constantValue = property.FindPropertyRelative("ConstantValue"); SerializedProperty variable = property.FindPropertyRelative("Variable"); // Calculate rect for configuration button Rect buttonRect = new Rect(position); buttonRect.yMin += _popupStyle.margin.top; buttonRect.width = _popupStyle.fixedWidth + _popupStyle.margin.right; position.xMin = buttonRect.xMax; // Store old indent level and set it to 0, the PrefixLabel takes care of it int indent = EditorGUI.indentLevel; EditorGUI.indentLevel = 0; int result = EditorGUI.Popup(buttonRect, useConstant.boolValue ? 0 : 1, _popupOptions, _popupStyle); useConstant.boolValue = result == 0; EditorGUI.PropertyField(position, useConstant.boolValue ? constantValue : variable, GUIContent.none); if (EditorGUI.EndChangeCheck()) property.serializedObject.ApplyModifiedProperties(); EditorGUI.indentLevel = indent; EditorGUI.EndProperty(); } } }