lsh4/Assets/DunGen/Code/Editor/Inspectors/TileSetInspector.cs

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

namespace DunGen.Editor
{
	[CustomEditor(typeof(TileSet))]
	public sealed class TileSetInspector : UnityEditor.Editor
	{
		private readonly List<bool> tileShowWeights = new List<bool>();
		private readonly List<List<bool>> lockPrefabShowWeights = new List<List<bool>>();
		private bool showLockPrefabs = false;


		private void OnEnable()
		{
			TileSet tileSet = target as TileSet;

			for (int i = 0; i < tileSet.TileWeights.Weights.Count; i++)
				tileShowWeights.Add(false);

			for (int i = 0; i < tileSet.LockPrefabs.Count; i++)
			{
				lockPrefabShowWeights.Add(new List<bool>());

				for (int j = 0; j < tileSet.LockPrefabs[i].LockPrefabs.Weights.Count; j++)
					lockPrefabShowWeights[i].Add(false);
			}
		}

		public override void OnInspectorGUI()
		{
			TileSet tileSet = target as TileSet;

			if (tileSet == null)
				return;

			EditorUtil.DrawGameObjectChanceTableGUI("Tile", tileSet.TileWeights, tileShowWeights, false, true, target);


			EditorGUILayout.BeginVertical("box");
			showLockPrefabs = EditorGUILayout.Foldout(showLockPrefabs, "Locked Door Prefabs", true);

			if (showLockPrefabs)
			{
				int toDeleteIndex = -1;

				for (int i = 0; i < tileSet.LockPrefabs.Count; i++)
				{
					var l = tileSet.LockPrefabs[i];

					EditorGUILayout.BeginVertical("box");

					EditorGUILayout.BeginHorizontal();

					l.Socket = (DoorwaySocket)EditorGUILayout.ObjectField(new GUIContent("Socket Type", "The socket type that this locked door can be placed on. If left blank, the locked door can be placed on a doorway of any socket type"), l.Socket, typeof(DoorwaySocket), false);

					if (GUILayout.Button("x", EditorStyles.miniButton, InspectorConstants.SmallButtonWidth))
						toDeleteIndex = i;

					EditorGUILayout.EndHorizontal();

					if (i > lockPrefabShowWeights.Count - 1)
						lockPrefabShowWeights.Add(new List<bool>());

					EditorUtil.DrawGameObjectChanceTableGUI("Prefab", l.LockPrefabs, lockPrefabShowWeights[i], false, true, target);

					EditorGUILayout.EndVertical();
				}

				if (toDeleteIndex > -1)
				{
					Undo.RecordObject(target, "Remove Lock Prefab");

					tileSet.LockPrefabs.RemoveAt(toDeleteIndex);
					lockPrefabShowWeights.RemoveAt(toDeleteIndex);

					Undo.FlushUndoRecordObjects();
				}

				EditorGUILayout.Space();
				EditorGUILayout.Space();
				EditorGUILayout.Space();

				if (GUILayout.Button("Add"))
				{
					tileSet.LockPrefabs.Add(new LockedDoorwayAssociation());
					lockPrefabShowWeights.Add(new List<bool>());
				}
			}

			EditorGUILayout.EndVertical();

			if (GUI.changed)
				EditorUtility.SetDirty(tileSet);
		}
	}
}