hellbound/Assets/Scripts/Core/Editor/ConditionalHidePropertyDraw...

75 lines
3.1 KiB
C#

using UnityEngine;
using UnityEditor;
//Original version of the ConditionalHideAttribute created by Brecht Lecluyse (www.brechtos.com)
//Modified by: Sebastian Lague
[CustomPropertyDrawer(typeof(ConditionalHideAttribute))]
public class ConditionalHidePropertyDrawer : PropertyDrawer
{
public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
{
ConditionalHideAttribute condHAtt = (ConditionalHideAttribute)attribute;
bool enabled = GetConditionalHideAttributeResult(condHAtt, property);
if (enabled)
EditorGUI.PropertyField(position, property, label, true);
}
public override float GetPropertyHeight(SerializedProperty property, GUIContent label)
{
ConditionalHideAttribute condHAtt = (ConditionalHideAttribute)attribute;
bool enabled = GetConditionalHideAttributeResult(condHAtt, property);
if (enabled)
return EditorGUI.GetPropertyHeight(property, label);
//We want to undo the spacing added before and after the property
return -EditorGUIUtility.standardVerticalSpacing;
}
bool GetConditionalHideAttributeResult(ConditionalHideAttribute condHAtt, SerializedProperty property)
{
SerializedProperty sourcePropertyValue;
//Get the full relative property path of the sourcefield so we can have nested hiding.Use old method when dealing with arrays
if (!property.isArray)
{
string propertyPath = property.propertyPath; //returns the property path of the property we want to apply the attribute to
string conditionPath = propertyPath.Replace(property.name, condHAtt.conditionalSourceField); //changes the path to the conditionalsource property path
sourcePropertyValue = property.serializedObject.FindProperty(conditionPath);
//if the find failed->fall back to the old system
if (sourcePropertyValue == null)
//original implementation (doens't work with nested serializedObjects)
sourcePropertyValue = property.serializedObject.FindProperty(condHAtt.conditionalSourceField);
}
else
{
//original implementation (doens't work with nested serializedObjects)
sourcePropertyValue = property.serializedObject.FindProperty(condHAtt.conditionalSourceField);
}
if (sourcePropertyValue != null)
return CheckPropertyType(condHAtt, sourcePropertyValue);
return true;
}
bool CheckPropertyType(ConditionalHideAttribute condHAtt, SerializedProperty sourcePropertyValue)
{
//Note: add others for custom handling if desired
switch (sourcePropertyValue.propertyType)
{
case SerializedPropertyType.Boolean:
return sourcePropertyValue.boolValue;
case SerializedPropertyType.Enum:
return sourcePropertyValue.enumValueIndex == condHAtt.enumIndex;
default:
Debug.LogError("Data type of the property used for conditional hiding [" + sourcePropertyValue.propertyType + "] is currently not supported");
return true;
}
}
}