121 lines
4.6 KiB
C#
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;
|
|
}
|
|
}
|
|
} |