added logic for handling badgoalsmulti game object

Kinds-of-Intelligence-CFI · Sep 11, 2024 · 058c71d · 058c71d
1 parent a0e1d9e
commit 058c71d
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/Assets/Scripts/ArenaBuilders.cs b/Assets/Scripts/ArenaBuilders.cs
@@ -32,6 +32,7 @@ public class ArenaBuilder
         private Collider _agentCollider;
         private Rigidbody _agentRigidbody;
         private List<Goal> _goodGoalsMultiSpawned;
+        private List<Goal> _badGoalsMultiSpawned;
         private int _totalObjectsSpawned;
         private AAI3EnvironmentManager _environmentManager;
 
@@ -60,6 +61,7 @@ int maxSpawnAttemptsForAgent
             _maxSpawnAttemptsForAgent = maxSpawnAttemptsForAgent;
             Spawnables = new List<Spawnable>();
             _goodGoalsMultiSpawned = new List<Goal>();
+            _badGoalsMultiSpawned = new List<Goal>();
         }
 
         public void Build()
@@ -97,6 +99,7 @@ public void Build()
             }
 
             updateGoodGoalsMulti();
+            updateBadGoalsMulti();
         }
 
         private void InstantiateSpawnables(GameObject spawnedObjectsHolder)
@@ -318,6 +321,10 @@ private void SpawnGameObject(
                 {
                     _goodGoalsMultiSpawned.Add(gameObjectInstance.GetComponent<Goal>());
                 }
+                if (gameObjectInstance.CompareTag("badGoalMulti"))
+                {
+                    _badGoalsMultiSpawned.Add(gameObjectInstance.GetComponent<Goal>());
+                }
                 if (optionals != null)
                 {
                     if (
@@ -634,6 +641,27 @@ private void updateGoodGoalsMulti()
             }
         }
 
+        private void updateBadGoalsMulti()
+        {
+            int numberOfGoals = _badGoalsMultiSpawned.Count;
+            foreach (Goal badGoalMulti in _badGoalsMultiSpawned)
+            {
+                badGoalMulti.numberOfGoals = numberOfGoals;
+            }
+        }
+
+        private void AddToBadGoalsMultiSpawned(Goal bgm)
+        {
+            _badGoalsMultiSpawned.Add(bgm);
+            updateBadGoalsMulti();
+        }
+
+        private void AddToBadGoalsMultiSpawned(GameObject bgm)
+        {
+            _badGoalsMultiSpawned.Add(bgm.GetComponent<Goal>());
+            updateBadGoalsMulti();
+        }
+
         public void AddToGoodGoalsMultiSpawned(Goal ggm)
         {
             _goodGoalsMultiSpawned.Add(ggm);