73 lines
2.6 KiB
C#
73 lines
2.6 KiB
C#
|
using System;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using UnityEngine;
|
||
|
|
|
||
|
|
namespace DunGen
|
||
|
|
{
|
||
|
|
[AddComponentMenu("DunGen/Random Props/Local Prop Set")]
|
||
|
|
public class LocalPropSet : RandomProp
|
||
|
|
{
|
||
|
|
private static readonly Dictionary<LocalPropSetCountMode, GetPropCountDelegate> GetCountMethods;
|
||
|
|
|
||
|
|
[AcceptGameObjectTypes(GameObjectFilter.Scene)]
|
||
|
|
public GameObjectChanceTable Props = new GameObjectChanceTable();
|
||
|
|
|
||
|
|
public IntRange PropCount = new IntRange(1, 1);
|
||
|
|
|
||
|
|
public LocalPropSetCountMode CountMode;
|
||
|
|
|
||
|
|
public AnimationCurve CountDepthCurve = AnimationCurve.Linear(0f, 0f, 1f, 1f);
|
||
|
|
|
||
|
|
public override void Process(RandomStream randomStream, Tile tile)
|
||
|
|
{
|
||
|
|
GameObjectChanceTable gameObjectChanceTable = Props.Clone();
|
||
|
|
if (!GetCountMethods.TryGetValue(CountMode, out var value))
|
||
|
|
{
|
||
|
|
throw new NotImplementedException("LocalPropSet count mode \"" + CountMode.ToString() + "\" is not yet implemented");
|
||
|
|
}
|
||
|
|
int num = value(this, randomStream, tile);
|
||
|
|
List<GameObject> list = new List<GameObject>(num);
|
||
|
|
for (int i = 0; i < num; i++)
|
||
|
|
{
|
||
|
|
GameObjectChance random = gameObjectChanceTable.GetRandom(randomStream, tile.Placement.IsOnMainPath, tile.Placement.NormalizedDepth, null, allowImmediateRepeats: true, removeFromTable: true);
|
||
|
|
if (random != null && random.Value != null)
|
||
|
|
{
|
||
|
|
list.Add(random.Value);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
foreach (GameObjectChance weight in Props.Weights)
|
||
|
|
{
|
||
|
|
if (!list.Contains(weight.Value))
|
||
|
|
{
|
||
|
|
UnityUtil.Destroy(weight.Value);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static LocalPropSet()
|
||
|
|
{
|
||
|
|
GetCountMethods = new Dictionary<LocalPropSetCountMode, GetPropCountDelegate>();
|
||
|
|
GetCountMethods[LocalPropSetCountMode.Random] = GetCountRandom;
|
||
|
|
GetCountMethods[LocalPropSetCountMode.DepthBased] = GetCountDepthBased;
|
||
|
|
GetCountMethods[LocalPropSetCountMode.DepthMultiply] = GetCountDepthMultiply;
|
||
|
|
}
|
||
|
|
|
||
|
|
private static int GetCountRandom(LocalPropSet propSet, RandomStream randomStream, Tile tile)
|
||
|
|
{
|
||
|
|
return Mathf.Clamp(propSet.PropCount.GetRandom(randomStream), 0, propSet.Props.Weights.Count);
|
||
|
|
}
|
||
|
|
|
||
|
|
private static int GetCountDepthBased(LocalPropSet propSet, RandomStream randomStream, Tile tile)
|
||
|
|
{
|
||
|
|
float t = Mathf.Clamp(propSet.CountDepthCurve.Evaluate(tile.Placement.NormalizedPathDepth), 0f, 1f);
|
||
|
|
return Mathf.RoundToInt(Mathf.Lerp(propSet.PropCount.Min, propSet.PropCount.Max, t));
|
||
|
|
}
|
||
|
|
|
||
|
|
private static int GetCountDepthMultiply(LocalPropSet propSet, RandomStream randomStream, Tile tile)
|
||
|
|
{
|
||
|
|
float num = Mathf.Clamp(propSet.CountDepthCurve.Evaluate(tile.Placement.NormalizedPathDepth), 0f, 1f);
|
||
|
|
return Mathf.RoundToInt((float)GetCountRandom(propSet, randomStream, tile) * num);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|