20240422

Unity/MeshGeneration/Modifiers/GameObjectModifiers/FeatureBehaviourModifier.cs

@@ -0,0 +1,36 @@
+namespace Mapbox.Unity.MeshGeneration.Modifiers
+{
+	using Mapbox.Unity.MeshGeneration.Data;
+	using Mapbox.Unity.MeshGeneration.Components;
+	using UnityEngine;
+	using System.Collections.Generic;
+
+	[CreateAssetMenu(menuName = "Mapbox/Modifiers/Add Feature Behaviour Modifier")]
+	public class FeatureBehaviourModifier : GameObjectModifier
+	{
+		private Dictionary<GameObject, FeatureBehaviour> _features;
+		private FeatureBehaviour _tempFeature;
+
+		public override void Initialize()
+		{
+			if (_features == null)
+			{
+				_features = new Dictionary<GameObject, FeatureBehaviour>();
+			}
+		}
+
+		public override void Run(VectorEntity ve, UnityTile tile)
+		{
+			if (_features.ContainsKey(ve.GameObject))
+			{
+				_features[ve.GameObject].Initialize(ve);
+			}
+			else
+			{
+				_tempFeature = ve.GameObject.AddComponent<FeatureBehaviour>();
+				_features.Add(ve.GameObject, _tempFeature);
+				_tempFeature.Initialize(ve);
+			}
+		}
+	}
+}