20240422

Unity/MeshGeneration/Modifiers/MeshModifiers/ReplaceFeatureCollectionModifier.cs

@@ -0,0 +1,147 @@
+namespace Mapbox.Unity.MeshGeneration.Modifiers
+{
+	using UnityEngine;
+	using System.Collections.Generic;
+	using Mapbox.Unity.MeshGeneration.Data;
+	using Mapbox.Unity.Utilities;
+	using Mapbox.Unity.Map;
+	using System;
+
+	[System.Serializable]
+	public class FeatureBundle
+	{
+		//public name param will be displayed in inspector list ui instead of element x...
+		[HideInInspector] public string Name;
+
+		public bool active;
+
+		public GameObject prefab;
+		public bool scaleDownWithWorld = true;
+
+		[Geocode] public List<string> _prefabLocations = new List<string>();
+
+		public List<string> _explicitlyBlockedFeatureIds = new List<string>();
+	}
+
+	/// <summary>
+	/// ReplaceFeatureCollectionModifier aggregates multiple ReplaceFeatureModifier objects into one modifier.
+	/// </summary>
+	[CreateAssetMenu(menuName = "Mapbox/Modifiers/Replace Feature Collection Modifier")]
+	public class ReplaceFeatureCollectionModifier : GameObjectModifier, IReplacementCriteria
+	{
+		public List<FeatureBundle> features = new List<FeatureBundle>();
+
+		private List<ReplaceFeatureModifier> _replaceFeatureModifiers;
+
+		//update all names to make inspector look better...
+		private void OnValidate()
+		{
+			for (int i = 0; i < features.Count; i++)
+			{
+				features[i].Name = (features[i].prefab == null) ? "Feature" : features[i].prefab.name;
+			}
+		}
+
+		public override void Initialize()
+		{
+			base.Initialize();
+
+			if (_replaceFeatureModifiers != null && _replaceFeatureModifiers.Count > 0)
+			{
+				foreach (var replaceFeatureModifier in _replaceFeatureModifiers)
+				{
+					if (replaceFeatureModifier != null)
+					{
+						replaceFeatureModifier.Clear();
+					}
+				}
+			}
+
+			_replaceFeatureModifiers = new List<ReplaceFeatureModifier>();
+			foreach (FeatureBundle feature in features)
+			{
+				ReplaceFeatureModifier replaceFeatureModifier = ScriptableObject.CreateInstance<ReplaceFeatureModifier>();
+
+				replaceFeatureModifier.Active = feature.active;
+				replaceFeatureModifier.SpawnPrefabOptions = new SpawnPrefabOptions()
+				{
+					prefab = feature.prefab,
+					scaleDownWithWorld = feature.scaleDownWithWorld
+				};
+				replaceFeatureModifier.PrefabLocations = new List<string>(feature._prefabLocations);
+				replaceFeatureModifier.BlockedIds = new List<string>(feature._explicitlyBlockedFeatureIds);
+				replaceFeatureModifier.Initialize();
+
+				_replaceFeatureModifiers.Add(replaceFeatureModifier);
+			}
+		}
+
+		public override void FeaturePreProcess(VectorFeatureUnity feature)
+		{
+			foreach (ReplaceFeatureModifier modifier in _replaceFeatureModifiers)
+			{
+				if (modifier == null)
+				{
+					continue;
+				}
+
+				modifier.FeaturePreProcess(feature);
+			}
+		}
+
+		public override void SetProperties(ModifierProperties properties)
+		{
+			foreach (ReplaceFeatureModifier modifier in _replaceFeatureModifiers)
+			{
+				if (modifier == null)
+				{
+					continue;
+				}
+
+				modifier.SetProperties(properties);
+			}
+		}
+
+		public bool ShouldReplaceFeature(VectorFeatureUnity feature)
+		{
+			foreach (ReplaceFeatureModifier modifier in _replaceFeatureModifiers)
+			{
+				if (modifier == null)
+				{
+					continue;
+				}
+
+				if (modifier.ShouldReplaceFeature(feature))
+				{
+					return true;
+				}
+			}
+
+			return false;
+		}
+
+		public override void Run(VectorEntity ve, UnityTile tile)
+		{
+			foreach (ReplaceFeatureModifier modifier in _replaceFeatureModifiers)
+			{
+				modifier.Run(ve, tile);
+			}
+		}
+
+		public override void OnPoolItem(VectorEntity vectorEntity)
+		{
+			foreach (ReplaceFeatureModifier modifier in _replaceFeatureModifiers)
+			{
+				modifier.OnPoolItem(vectorEntity);
+			}
+		}
+
+		public override void Clear()
+		{
+			foreach (var subModules in _replaceFeatureModifiers)
+			{
+				subModules.Clear();
+			}
+		}
+	}
+}