lsh4/Assets/DunGen/Code/LocalPropSet.cs

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

namespace DunGen
{
	public enum LocalPropSetCountMode
	{
		Random,
		DepthBased,
		DepthMultiply,
	}

	public delegate int GetPropCountDelegate(LocalPropSet propSet, RandomStream randomStream, Tile tile);

	[AddComponentMenu("DunGen/Random Props/Local Prop Set")]
	public class LocalPropSet : RandomProp
	{
		private static readonly Dictionary<LocalPropSetCountMode, GetPropCountDelegate> GetCountMethods = new Dictionary<LocalPropSetCountMode, GetPropCountDelegate>();

		[AcceptGameObjectTypes(GameObjectFilter.Scene)]
		public GameObjectChanceTable Props = new GameObjectChanceTable();
		public IntRange PropCount = new IntRange(1, 1);
		public LocalPropSetCountMode CountMode;
		public AnimationCurve CountDepthCurve = AnimationCurve.Linear(0, 0, 1, 1);


		public override void Process(RandomStream randomStream, Tile tile, ref List<GameObject> spawnedObjects)
		{
			var propTable = Props.Clone();

			if (!GetCountMethods.TryGetValue(CountMode, out var getCountDelegate))
				throw new NotImplementedException("LocalPropSet count mode \"" + CountMode + "\" is not yet implemented");

			int count = getCountDelegate(this, randomStream, tile);
			var toKeep = new List<GameObject>(count);

			for (int i = 0; i < count; i++)
			{
				// allowNullSelection is true so that we can treat empty entries as "no prop"
				var chosenEntry = propTable.GetRandom(randomStream,
					tile.Placement.IsOnMainPath,
					tile.Placement.NormalizedDepth,
					previouslyChosen: null,
					allowImmediateRepeats: true,
					removeFromTable: true,
					allowNullSelection: true);

				if (chosenEntry != null && chosenEntry.Value != null)
					toKeep.Add(chosenEntry.Value);
			}

			foreach (var prop in Props.Weights)
			{
				if (prop.Value == null)
					continue;

				bool isActive = toKeep.Contains(prop.Value);
				prop.Value.SetActive(isActive);
			}
		}

		#region GetCount Methods

		static LocalPropSet()
		{
			GetCountMethods[LocalPropSetCountMode.Random] = GetCountRandom;
			GetCountMethods[LocalPropSetCountMode.DepthBased] = GetCountDepthBased;
			GetCountMethods[LocalPropSetCountMode.DepthMultiply] = GetCountDepthMultiply;
		}

		private static int GetCountRandom(LocalPropSet propSet, RandomStream randomStream, Tile tile)
		{
			int count = propSet.PropCount.GetRandom(randomStream);
			count = Mathf.Clamp(count, 0, propSet.Props.Weights.Count);

			return count;
		}

		private static int GetCountDepthBased(LocalPropSet propSet, RandomStream randomStream, Tile tile)
		{
			float curveValue = Mathf.Clamp(propSet.CountDepthCurve.Evaluate(tile.Placement.NormalizedPathDepth), 0, 1);
			int count = Mathf.RoundToInt(Mathf.Lerp(propSet.PropCount.Min, propSet.PropCount.Max, curveValue));

			return count;
		}

		private static int GetCountDepthMultiply(LocalPropSet propSet, RandomStream randomStream, Tile tile)
		{
			float curveValue = Mathf.Clamp(propSet.CountDepthCurve.Evaluate(tile.Placement.NormalizedPathDepth), 0, 1);
			int count = GetCountRandom(propSet, randomStream, tile);
			count = Mathf.RoundToInt(count * curveValue);

			return count;
		}

		#endregion
	}
}