fungus/Assets/Fungus/Flowchart/Editor/LabelEditor.cs

using UnityEditor;
using UnityEditorInternal;
using UnityEngine;
using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;

namespace Fungus
{
	[CustomEditor (typeof(Label))]
	public class LabelEditor : CommandEditor
	{
		protected SerializedProperty keyProp;
		
		static public void LabelField(SerializedProperty property, 
		                              GUIContent labelText, 
		                              Block block)
		{
			List<string> labelKeys = new List<string>();
			List<Label> labelObjects = new List<Label>();
			
			labelKeys.Add("<None>");
			labelObjects.Add(null);
			
			Label selectedLabel = property.objectReferenceValue as Label;

			int index = 0;
			int selectedIndex = 0;
			foreach (Command command in block.commandList)
			{
				Label label = command as Label;
				if (label == null)
				{
					continue;
				}

				labelKeys.Add(label.key);
				labelObjects.Add(label);
				
				index++;
				
				if (label == selectedLabel)
				{
					selectedIndex = index;
				}
			}

			selectedIndex = EditorGUILayout.Popup(labelText.text, selectedIndex, labelKeys.ToArray());

			property.objectReferenceValue = labelObjects[selectedIndex];
		}

		protected virtual void OnEnable()
		{
			keyProp = serializedObject.FindProperty("key");
		}
		
		public override void DrawCommandGUI()
		{
			Label t = target as Label;

			Flowchart flowchart = t.GetFlowchart();
			if (flowchart == null)
			{
				return;
			}
		
			serializedObject.Update();

			EditorGUILayout.PropertyField(keyProp);
			keyProp.stringValue = flowchart.GetUniqueLabelKey(keyProp.stringValue, t);

			serializedObject.ApplyModifiedProperties();
		}
	}
	
}