lab6

Assets/Code/UI/Lab6Controller.cs

@@ -0,0 +1,244 @@
+using System.Collections.Generic;
+using Code.Base;
+using Code.Lab6;
+using TMPro;
+using UnityEngine;
+using UnityEngine.SceneManagement;
+using UnityEngine.UI;
+
+namespace Code.UI
+{
+    public class Lab6Controller : MonoBehaviour
+    {
+        public Graph graph;
+        public Agent agent;
+        public Transform targetMarker;
+        
+        public Button toMenuButton;
+        public Button resetButton;
+        public Button findPathButton;
+        public Button dfsButton;
+        public Button bfsButton;
+        public Button stopButton;
+        
+        public LineRenderer pathLine;
+        public Color pathColor = Color.green;
+        public TextMeshProUGUI statusText;
+
+        private List<Vertex> _currentPath;
+        private VertexPathFollower _pathFollower;
+        private bool _useBfs = true;
+
+        private void Awake()
+        {
+            if (toMenuButton != null)
+                toMenuButton.onClick.AddListener(LoadMainMenu);
+            if (resetButton != null)
+                resetButton.onClick.AddListener(ResetScene);
+            if (findPathButton != null)
+                findPathButton.onClick.AddListener(FindPathToTarget);
+            if (dfsButton != null)
+                dfsButton.onClick.AddListener(() => SetAlgorithm(false));
+            if (bfsButton != null)
+                bfsButton.onClick.AddListener(() => SetAlgorithm(true));
+            if (stopButton != null)
+                stopButton.onClick.AddListener(StopAgent);
+
+            _pathFollower = agent?.GetComponent<VertexPathFollower>();
+            if (_pathFollower == null && agent != null)
+            {
+                _pathFollower = agent.gameObject.AddComponent<VertexPathFollower>();
+            }
+            
+            if (pathLine != null)
+            {
+                pathLine.startColor = pathColor;
+                pathLine.endColor = pathColor;
+                pathLine.startWidth = 0.2f;
+                pathLine.endWidth = 0.2f;
+            }
+            
+            _currentPath = new List<Vertex>();
+        }
+
+        private void Start()
+        {
+            if (graph == null)
+                graph = FindFirstObjectByType<Graph>();
+        
+            if (agent == null)
+                agent = FindFirstObjectByType<Agent>();
+        
+            if (targetMarker == null)
+            {
+                var marker = GameObject.CreatePrimitive(PrimitiveType.Sphere);
+                marker.name = "Target Marker";
+                marker.transform.position = new Vector3(5, 0.5f, 5);
+                marker.transform.localScale = new Vector3(0.5f, 0.5f, 0.5f);
+        
+                var renderer1 = marker.GetComponent<Renderer>();
+                renderer1.material.color = Color.red;
+
+                Destroy(marker.GetComponent<Collider>());
+        
+                targetMarker = marker.transform;
+            }
+
+            if (agent == null) return;
+            
+            var pos = agent.transform.position;
+            pos.y = 0.5f;
+            agent.transform.position = pos;
+        }
+
+        private void Update()
+        {
+            if (Input.GetMouseButtonDown(1))
+            {
+                if (Camera.main != null)
+                {
+                    var ray = Camera.main.ScreenPointToRay(Input.mousePosition);
+                    RaycastHit hit;
+                
+                    if (Physics.Raycast(ray, out hit))
+                    {
+                        var targetPos = hit.point;
+                        targetPos.y = 0.5f;
+                        targetMarker.position = targetPos;
+                        FindPathToTarget();
+                    }
+                }
+            }
+            
+            DrawPath();
+            UpdateStatusText();
+        }
+
+        private void UpdateStatusText()
+        {
+            if (!statusText) return;
+            
+            var status = $"Algorithm: {(_useBfs ? "BFS" : "DFS")}\n";
+            
+            if (_pathFollower && _pathFollower.IsFollowing())
+            {
+                status += $"Status: Moving to waypoint {_pathFollower.GetRemainingWaypoints()}/{_currentPath.Count}\n";
+                var currentTarget = _pathFollower.GetCurrentTarget();
+                if (currentTarget)
+                {
+                    status += $"Target pos: {currentTarget.transform.position}\n";
+                }
+            }
+            else
+            {
+                status += "Status: Idle\n";
+            }
+            
+            status += $"Path nodes: {_currentPath.Count}";
+            
+            statusText.text = status;
+        }
+
+        public void SetAlgorithm(bool useBfs)
+        {
+            _useBfs = useBfs;
+            Debug.Log(useBfs ? "Using BFS algorithm" : "Using DFS algorithm");
+
+            if (targetMarker != null)
+            {
+                FindPathToTarget();
+            }
+        }
+
+        public void StopAgent()
+        {
+            if (_pathFollower != null)
+            {
+                _pathFollower.StopFollowing();
+            }
+            _currentPath.Clear();
+        }
+
+        public void FindPathToTarget()
+        {
+            if (!graph || !agent || !targetMarker)
+            {
+                Debug.LogError("Missing references!");
+                return;
+            }
+            
+            var targetPos = targetMarker.position;
+            targetPos.y = 0.5f;
+            targetMarker.position = targetPos;
+
+            var path = _useBfs ? graph.GetPathBfs(agent.gameObject, targetMarker.gameObject) : graph.GetPathDfs(agent.gameObject, targetMarker.gameObject);
+                
+            if (path is { Count: > 0 })
+            {
+                _currentPath = path;
+                Debug.Log($"Path found! Length: {path.Count} nodes");
+
+                if (_pathFollower)
+                {
+                    _pathFollower.SetPath(path);
+                }
+            }
+            else
+            {
+                Debug.LogWarning("No path found!");
+                _currentPath.Clear();
+                if (_pathFollower)
+                {
+                    _pathFollower.StopFollowing();
+                }
+            }
+        }
+
+        private void DrawPath()
+        {
+            if (!pathLine || _currentPath == null || _currentPath.Count < 2)
+            {
+                if (pathLine)
+                    pathLine.positionCount = 0;
+                return;
+            }
+            
+            pathLine.positionCount = _currentPath.Count;
+            
+            for (var i = 0; i < _currentPath.Count; i++)
+            {
+                if (!_currentPath[i]) continue;
+                
+                var pos = _currentPath[i].transform.position;
+                pos.y += 0.2f;
+                pathLine.SetPosition(i, pos);
+            }
+        }
+
+        private static void ResetScene()
+        {
+            SceneManager.LoadScene(SceneManager.GetActiveScene().name);
+        }
+
+        private static void LoadMainMenu()
+        {
+            SceneManager.LoadScene("MainMenu");
+        }
+
+        private void OnDestroy()
+        {
+            if (toMenuButton != null)
+                toMenuButton.onClick.RemoveListener(LoadMainMenu);
+            if (resetButton != null)
+                resetButton.onClick.RemoveListener(ResetScene);
+            if (findPathButton != null)
+                findPathButton.onClick.RemoveListener(FindPathToTarget);
+            if (dfsButton != null)
+                dfsButton.onClick.RemoveAllListeners();
+            if (bfsButton != null)
+                bfsButton.onClick.RemoveAllListeners();
+            if (stopButton != null)
+                stopButton.onClick.RemoveListener(StopAgent);
+        }
+    }
+}

Assets/Code/UI/Lab6Controller.cs.meta

@@ -0,0 +1,3 @@
+fileFormatVersion: 2
+guid: 58f148e1e6954b1fb160c3279addc512
+timeCreated: 1771392756