feat: add traffic control ui

Signed-off-by: Alptuğ Cırıt <[email protected]>

Assets/AWSIM/Scenes/Main/AutowareSimulation/AutowareSimulation.cs

@@ -42,6 +42,7 @@ public class Configuration
         }
 
         private EgoVehiclePositionManager egoVehiclePositionManager;
+        private TrafficControlManager trafficControlManager;
 
         void Awake()
         {
@@ -77,9 +78,29 @@ void Awake()
                 timeSourceSelector?.SetType(config.TimeSource);
             }
 
-            egoVehiclePositionManager = gameObject.AddComponent<EgoVehiclePositionManager>();
-            egoVehiclePositionManager.EgoTransform = egoTransform;
-            egoVehiclePositionManager.enabled = true;
+            // Add EgoVehiclePositionManager
+            if (!egoTransform)
+            {
+                Debug.LogWarning("Missing EgoTransform reference.");
+            }
+            else
+            {
+                egoVehiclePositionManager = gameObject.AddComponent<EgoVehiclePositionManager>();
+                egoVehiclePositionManager.EgoTransform = egoTransform;
+                egoVehiclePositionManager.enabled = true;
+            }
+
+            // Add TrafficControlManager
+            if (!trafficManager)
+            {
+                Debug.LogWarning("Missing TrafficManager reference. ");
+            }
+            else
+            {
+                trafficControlManager = gameObject.AddComponent<TrafficControlManager>();
+                trafficControlManager.TrafficManager = trafficManager;
+                trafficControlManager.enabled = true;
+            }
         }
     }
 }

Assets/AWSIM/Scripts/UI/TrafficControlManager.cs

@@ -0,0 +1,57 @@
+using System;
+using AWSIM.TrafficSimulation;
+using UnityEngine;
+
+namespace AWSIM.Scripts.UI
+{
+    public class TrafficControlManager : MonoBehaviour
+    {
+        public TrafficManager TrafficManager { private get; set; }
+
+        public void TrafficManagerPauseResumeToggle()
+        {
+            if (TrafficManager.gameObject.activeSelf)
+            {
+                TrafficManager.enabled = !TrafficManager.enabled;
+            }
+            else
+                Debug.Log("TrafficManager is not active. Can't pause/resume!");
+        }
+
+        public void TrafficManagerEnable()
+        {
+            TrafficManager.gameObject.SetActive(true);
+            TrafficManager.enabled = true;
+        }
+
+        public void TrafficManagerDisable()
+        {
+            TrafficManager.gameObject.SetActive(false);
+        }
+
+        public void TrafficManagerReset()
+        {
+            if (TrafficManager.gameObject.activeSelf)
+            {
+                TrafficManager.enabled = true;
+                TrafficManager.npcVehicleSimulator.ClearAll();
+            }
+            else
+            {
+                TrafficManager.gameObject.SetActive(true);
+                TrafficManager.enabled = true;
+                TrafficManager.npcVehicleSimulator.ClearAll();
+            }
+        }
+
+        public void TrafficMangerSetSeed(int seed)
+        {
+            if (!TrafficManager.gameObject.activeSelf)
+            {
+                Debug.LogWarning("TrafficManager is not active. Can't set seed!");
+            }
+
+            TrafficManager.seed = seed;
+        }
+    }
+}

Assets/AWSIM/Scripts/UI/UIBridge.cs

@@ -1,19 +1,56 @@
+using System;
 using UnityEngine;
+using UnityEngine.UI;
 
 namespace AWSIM.Scripts.UI
 {
     public class UIBridge : MonoBehaviour
     {
         private EgoVehiclePositionManager egoVehiclePositionManager;
+        private TrafficControlManager trafficControlManager;
+        [SerializeField] InputField trafficSeedInputField;
 
         private void Awake()
         {
             egoVehiclePositionManager = GetComponent<EgoVehiclePositionManager>();
+            trafficControlManager = GetComponent<TrafficControlManager>();
         }
 
         public void ResetEgoToSpawnPoint()
         {
             egoVehiclePositionManager.ResetEgoToSpawnPoint();
         }
+
+        public void TrafficManagerPauseResumeToggle()
+        {
+            trafficControlManager.TrafficManagerPauseResumeToggle();
+        }
+
+        public void TrafficManagerEnable()
+        {
+            trafficControlManager.TrafficManagerEnable();
+        }
+
+        public void TrafficManagerDisable()
+        {
+            trafficControlManager.TrafficManagerDisable();
+        }
+
+        public void TrafficManagerReset()
+        {
+            trafficControlManager.TrafficManagerReset();
+        }
+
+        public void TrafficMangerSetSeed()
+        {
+            if (!trafficSeedInputField)
+            {
+                Debug.LogWarning("TrafficSeedInputField reference is missing. Can't set seed without it!");
+                return;
+            }
+
+            var seed = Convert.ToInt32(trafficSeedInputField.text);
+            trafficControlManager.TrafficMangerSetSeed(seed);
+        }
     }
 }