AITransitionDrawer.cs

Go to the documentation of this file.

using UnityEngine;
using UnityEditor;

namespace HKI.Core.AI.StateMachine.Editor
{
    [CustomPropertyDrawer(typeof(AITransition))]
    public class AITransitionDrawer : PropertyDrawer
    {
        // Private variables
        const string decisionSPName = "Decision";
        const string trueStateSPName = "TrueState";
        const string falseStateSPName = "FalseState";

        const float decisionLabelWidth = 60.0f;
        const float trueStateLabelWidth = 75.0f;
        const float falseStateLabelWidth = 75.0f;

        // OnGUI function
        public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
        {
            SerializedProperty decisionSP = property.FindPropertyRelative(decisionSPName);
            SerializedProperty trueStateSP = property.FindPropertyRelative(trueStateSPName);
            SerializedProperty falseStateSP = property.FindPropertyRelative(falseStateSPName);

            float fieldWidth = (position.width - decisionLabelWidth - trueStateLabelWidth - falseStateLabelWidth) / 3.0f;
            
            position.width = decisionLabelWidth;
            EditorGUI.LabelField(position, "Decision:");

            position.x += decisionLabelWidth;
            position.width = fieldWidth;
            EditorGUI.PropertyField(position, decisionSP, GUIContent.none);

            position.x += fieldWidth;
            position.width = trueStateLabelWidth;
            EditorGUI.LabelField(position, "True State:");

            position.x += trueStateLabelWidth;
            position.width = fieldWidth;
            EditorGUI.PropertyField(position, trueStateSP, GUIContent.none);

            position.x += fieldWidth;
            position.width = falseStateLabelWidth;
            EditorGUI.LabelField(position, "False State:");

            position.x += falseStateLabelWidth;
            position.width = fieldWidth;
            EditorGUI.PropertyField(position, falseStateSP, GUIContent.none);
        }
    }
}