SamsonGame/Assets/Sources/Feel/MMTools/Editor/MMAI/AITransitionInspectorDrawer.cs

121 lines
4.6 KiB
C#

using UnityEngine;
using UnityEditor;
using System.Collections;
namespace MoreMountains.Tools
{
[CustomPropertyDrawer(typeof(AITransition))]
public class AITransitionPropertyInspector : PropertyDrawer
{
const float LineHeight = 16f;
/// <summary>
/// Draws a Transition inspector, a transition is one or more action(s), one or more decision(s) and associated true/false states
/// </summary>
/// <param name="rect"></param>
/// <param name="prop"></param>
/// <param name="label"></param>
public override void OnGUI(Rect rect, SerializedProperty prop, GUIContent label)
{
Rect position = rect;
foreach (SerializedProperty a in prop)
{
var height = Mathf.Max(LineHeight, EditorGUI.GetPropertyHeight(a));
position.height = height;
if(a.name == "Decision")
{
// draw the decision dropdown
DrawSelectionDropdown(position, prop);
// draw the base decision field
position.y += height;
EditorGUI.PropertyField(position, a, new GUIContent(a.name));
position.y += height;
/*var @object = a.objectReferenceValue;
AIDecision @typedObject = @object as AIDecision;
if (@typedObject != null && !string.IsNullOrEmpty(@typedObject.Label))
{
EditorGUI.LabelField(position, "Label", @typedObject.Label);
position.y += height;
}
else
{
EditorGUIUtility.GetControlID(FocusType.Passive);
}*/
}
else
{
EditorGUI.PropertyField(position, a, new GUIContent(a.name));
position.y += height;
}
}
}
/// <summary>
/// Draws a selector letting the user pick any decision associated with the AIBrain this transition is on
/// </summary>
/// <param name="position"></param>
/// <param name="prop"></param>
protected virtual void DrawSelectionDropdown(Rect position, SerializedProperty prop)
{
AIDecision thisDecision = prop.objectReferenceValue as AIDecision;
AIDecision[] decisions = (prop.serializedObject.targetObject as AIBrain).GetAttachedDecisions();
int selected = 0;
int i = 1;
string[] options = new string[decisions.Length + 1];
options[0] = "None";
foreach (AIDecision decision in decisions)
{
string name = string.IsNullOrEmpty(decision.Label) ? decision.GetType().Name : decision.Label;
options[i] = i.ToString() + " - " + name;
if (decision == thisDecision)
{
selected = i;
}
i++;
}
EditorGUI.BeginChangeCheck();
selected = EditorGUI.Popup(position, selected, options);
if (EditorGUI.EndChangeCheck())
{
prop.objectReferenceValue = (selected == 0) ? null : decisions[selected - 1];
prop.serializedObject.ApplyModifiedProperties();
EditorUtility.SetDirty(prop.serializedObject.targetObject);
}
}
/// <summary>
/// Determines the height of the transition property
/// </summary>
/// <param name="property"></param>
/// <param name="label"></param>
/// <returns></returns>
public override float GetPropertyHeight(SerializedProperty property, GUIContent label)
{
float height = 0;
foreach (SerializedProperty a in property)
{
var h = Mathf.Max(LineHeight, EditorGUI.GetPropertyHeight(a));
if(a.name == "Decision")
{
height += h * 2;
/*var @object = a.objectReferenceValue;
AIDecision @typedObject = @object as AIDecision;
if (@typedObject != null && !string.IsNullOrEmpty(@typedObject.Label))
{
height += h;
}*/
}
else
{
height += h;
}
}
return height;
}
}
}