GraphTraversalControl.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace WinFormsApp1
{
    public class GraphTraversalControl : BaseGraphControl
    {
        private Button btnGenerateGraph;
        private Button btnDFS;
        private Button btnBFS;
        private Button btnReset;
        private Button btnSave;
        private Button btnLoad;
        private Button btnAddNode;
        private Button btnAddEdge;
        private Button btnDeleteNode;
        private Button btnMoveNode;
        private Button btnCancelMode;
        private NumericUpDown numNodes;
        private NumericUpDown numEdges;
        private Label lblMode;

        private WeightedGraph.Node edgeStartNode = null;
        private EditMode currentMode = EditMode.None;

        private enum EditMode { None, AddNode, AddEdge, DeleteNode, MoveNode }

        public GraphTraversalControl() : base()
        {
            graph = new WeightedGraph(false);
            InitializeControls();
        }

        private void InitializeControls()
        {
            var flowPanel = new FlowLayoutPanel
            {
                Dock = DockStyle.Fill,
                FlowDirection = FlowDirection.TopDown,
                WrapContents = false,
                AutoScroll = true,
                Padding = new Padding(5)
            };

            // === Генерация ===
            flowPanel.Controls.Add(CreateLabel("Генерация графа:", true, 10));

            var genPanel = new FlowLayoutPanel { FlowDirection = FlowDirection.LeftToRight, AutoSize = true };
            genPanel.Controls.Add(CreateLabel("Узлы:"));
            numNodes = new NumericUpDown { Width = 55, Minimum = 3, Maximum = 15, Value = 6 };
            genPanel.Controls.Add(numNodes);
            genPanel.Controls.Add(CreateLabel("Рёбра:"));
            numEdges = new NumericUpDown { Width = 55, Minimum = 3, Maximum = 30, Value = 8 };
            genPanel.Controls.Add(numEdges);
            flowPanel.Controls.Add(genPanel);

            btnGenerateGraph = CreateButton("🎲 Генерировать", Color.LightBlue, (s, e) => GenerateGraph());
            flowPanel.Controls.Add(btnGenerateGraph);

            flowPanel.Controls.Add(new Label { Height = 10 });

            // === Редактирование ===
            flowPanel.Controls.Add(CreateLabel("Редактирование:", true, 10));

            var editPanel1 = new FlowLayoutPanel { FlowDirection = FlowDirection.LeftToRight, AutoSize = true };
            btnAddNode = CreateButton("+ Узел", Color.LightGreen, (s, e) => SetMode(EditMode.AddNode));
            btnAddNode.Width = 80;
            btnAddEdge = CreateButton("+ Ребро", Color.LightYellow, (s, e) => SetMode(EditMode.AddEdge));
            btnAddEdge.Width = 80;
            btnDeleteNode = CreateButton("✕ Удалить", Color.LightCoral, (s, e) => SetMode(EditMode.DeleteNode));
            btnDeleteNode.Width = 80;
            editPanel1.Controls.AddRange(new Control[] { btnAddNode, btnAddEdge, btnDeleteNode });
            flowPanel.Controls.Add(editPanel1);

            var editPanel2 = new FlowLayoutPanel { FlowDirection = FlowDirection.LeftToRight, AutoSize = true };
            btnMoveNode = CreateButton("✋ Переместить", Color.LightSkyBlue, (s, e) => SetMode(EditMode.MoveNode));
            btnMoveNode.Width = 120;
            editPanel2.Controls.Add(btnMoveNode);
            flowPanel.Controls.Add(editPanel2);

            lblMode = new Label
            {
                Text = "Режим: просмотр",
                AutoSize = true,
                ForeColor = Color.DarkBlue,
                Font = new Font("Segoe UI", 9, FontStyle.Italic),
                MaximumSize = new Size(250, 0)
            };
            flowPanel.Controls.Add(lblMode);

            btnCancelMode = CreateButton("↩ Отмена режима", Color.Gainsboro, (s, e) => SetMode(EditMode.None));
            flowPanel.Controls.Add(btnCancelMode);

            flowPanel.Controls.Add(new Label { Height = 10 });

            // === Скорость ===
            var lblSpeed = CreateLabel("Скорость анимации: 500мс");
            flowPanel.Controls.Add(lblSpeed);
            flowPanel.Controls.Add(CreateSpeedTrackBar(lblSpeed));

            flowPanel.Controls.Add(new Label { Height = 10 });

            // === Алгоритмы ===
            flowPanel.Controls.Add(CreateLabel("Алгоритмы обхода:", true, 10));

            btnDFS = CreateButton("▶ DFS (Глубина)", Color.MediumSeaGreen, async (s, e) => await RunTraversal(true));
            btnDFS.ForeColor = Color.White;
            btnDFS.Width = 145;
            flowPanel.Controls.Add(btnDFS);

            btnBFS = CreateButton("▶ BFS (Ширина)", Color.RoyalBlue, async (s, e) => await RunTraversal(false));
            btnBFS.ForeColor = Color.White;
            btnBFS.Width = 145;
            flowPanel.Controls.Add(btnBFS);

            flowPanel.Controls.Add(new Label { Height = 10 });

            btnReset = CreateButton("🔄 Сбросить", Color.Silver, (s, e) => ResetGraph());
            flowPanel.Controls.Add(btnReset);

            flowPanel.Controls.Add(new Label { Height = 15 });

            // === Файлы ===
            flowPanel.Controls.Add(CreateLabel("Файлы:", true, 10));

            var filePanel = new FlowLayoutPanel { FlowDirection = FlowDirection.LeftToRight, AutoSize = true };
            btnSave = CreateButton("💾 Сохранить", Color.LightSteelBlue, (s, e) => SaveGraph());
            btnSave.Width = 115;
            btnLoad = CreateButton("📂 Загрузить", Color.LightSteelBlue, (s, e) => LoadGraph());
            btnLoad.Width = 115;
            filePanel.Controls.AddRange(new Control[] { btnSave, btnLoad });
            flowPanel.Controls.Add(filePanel);

            flowPanel.Controls.Add(new Label { Height = 15 });
            
            var infoLabel = new Label
            {
                Text = "ℹ DFS использует СТЕК\n" +
                       "  BFS использует ОЧЕРЕДЬ\n" +
                       "  Сложность: O(V + E)",
                AutoSize = true,
                ForeColor = Color.DarkSlateGray,
                Font = new Font("Segoe UI", 8.5f)
            };
            flowPanel.Controls.Add(infoLabel);

            controlPanel.Controls.Add(flowPanel);
        }

        private void SetMode(EditMode mode)
        {
            currentMode = mode;
            edgeStartNode = null;
            selectedNode = null;
            isDragging = false;
            draggingNode = null;

            lblMode.Text = mode switch
            {
                EditMode.AddNode => "Режим: добавление узла\n(кликните на пустое место)",
                EditMode.AddEdge => "Режим: добавление ребра\n(выберите 2 узла)",
                EditMode.DeleteNode => "Режим: удаление\n(кликните на узел)",
                EditMode.MoveNode => "Режим: перемещение\n(перетащите узел)",
                _ => "Режим: просмотр"
            };
            lblMode.ForeColor = mode == EditMode.None ? Color.DarkBlue : Color.DarkRed;

            // Меняем курсор в зависимости от режима
            drawPanel.Cursor = mode == EditMode.MoveNode ? Cursors.SizeAll : Cursors.Default;

            RefreshGraph();
        }

        private void GenerateGraph()
        {
            graph = new WeightedGraph(false);
            graph.GenerateRandom((int)numNodes.Value, (int)numEdges.Value,
                drawPanel.Width, drawPanel.Height, 1, 10);
            Log($"Сгенерирован граф: {graph.Nodes.Count} узлов, {graph.Edges.Count} рёбер");
            UpdateMatrix();
            RefreshGraph();
        }

        private void ResetGraph()
        {
            graph.ResetState();
            SetMode(EditMode.None);
            RefreshGraph();
            Log("Граф сброшен");
        }

        protected override void DrawPanel_Paint(object sender, PaintEventArgs e)
        {
            e.Graphics.Clear(Color.White);
            GraphRenderer.DrawGraph(e.Graphics, graph, selectedNode, showWeights: true);

            var legend = new Dictionary<Color, string>
            {
                { Color.LightBlue, "Не посещён" },
                { Color.Yellow, "В очереди/стеке" },
                { Color.Red, "Обрабатывается" },
                { Color.Green, "Обработан" }
            };
            GraphRenderer.DrawLegend(e.Graphics, legend);
        }

        protected override void OnMouseDownHandler(MouseEventArgs e)
        {
            Point clickPos = e.Location;
            WeightedGraph.Node clickedNode = FindNodeAtPosition(clickPos);

            switch (currentMode)
            {
                case EditMode.AddNode:
                    if (clickedNode == null)
                    {
                        graph.AddNode(clickPos);
                        Log($"Добавлен узел {graph.Nodes.Count - 1}");
                        UpdateMatrix();
                        RefreshGraph();
                    }
                    break;

                case EditMode.AddEdge:
                    if (clickedNode != null)
                    {
                        if (edgeStartNode == null)
                        {
                            edgeStartNode = clickedNode;
                            selectedNode = clickedNode;
                            lblMode.Text = $"Выбран узел {clickedNode.Id}\nКликните на второй узел";
                            RefreshGraph();
                        }
                        else if (clickedNode != edgeStartNode)
                        {
                            string input = ShowInputDialog("Введите вес ребра:", "Вес ребра", "1");
                            if (!string.IsNullOrEmpty(input) && int.TryParse(input, out int weight) && weight > 0)
                            {
                                graph.AddEdge(edgeStartNode, clickedNode, weight);
                                Log($"Добавлено ребро {edgeStartNode.Id} — {clickedNode.Id} (вес: {weight})");
                                UpdateMatrix();
                            }
                            edgeStartNode = null;
                            selectedNode = null;
                            lblMode.Text = "Режим: добавление ребра\n(выберите 2 узла)";
                            RefreshGraph();
                        }
                    }
                    break;

                case EditMode.DeleteNode:
                    if (clickedNode != null)
                    {
                        int id = clickedNode.Id;
                        graph.RemoveNode(clickedNode);
                        Log($"Узел {id} удалён");
                        UpdateMatrix();
                        RefreshGraph();
                    }
                    break;

                case EditMode.MoveNode:
                    if (clickedNode != null)
                    {
                        StartDragging(clickedNode, e.Location);
                    }
                    break;

                default:
                    // В режиме просмотра - выбираем узел
                    selectedNode = clickedNode;
                    RefreshGraph();
                    break;
            }
        }

        protected override void DrawPanel_MouseMove(object sender, MouseEventArgs e)
        {
            if (isRunning) return;

            // Если мы в режиме перетаскивания
            if (currentMode == EditMode.MoveNode && isDragging && draggingNode != null)
            {
                int newX = e.X - dragOffset.X;
                int newY = e.Y - dragOffset.Y;
                
                int margin = 30;
                newX = Math.Max(margin, Math.Min(drawPanel.Width - margin, newX));
                newY = Math.Max(margin, Math.Min(drawPanel.Height - margin, newY));
                
                draggingNode.Position = new Point(newX, newY);
                RefreshGraph();
            }
            else if (currentMode == EditMode.MoveNode)
            {
                // Показываем курсор руки если над узлом
                var node = FindNodeAtPosition(e.Location);
                drawPanel.Cursor = node != null ? Cursors.SizeAll : Cursors.Hand;
            }
        }

        protected override void DrawPanel_MouseUp(object sender, MouseEventArgs e)
        {
            if (currentMode == EditMode.MoveNode && isDragging && draggingNode != null)
            {
                Log($"Узел {draggingNode.Id} перемещён");
                isDragging = false;
                draggingNode = null;
                RefreshGraph();
            }
        }

        private async Task RunTraversal(bool isDFS)
        {
            if (graph.Nodes.Count == 0)
            {
                MessageBox.Show("Сначала создайте граф!", "Ошибка", MessageBoxButtons.OK, MessageBoxIcon.Warning);
                return;
            }

            isRunning = true;
            SetButtonsEnabled(false);
            graph.ResetState();

            string algoName = isDFS ? "DFS" : "BFS";
            Log($"\n=== Начало {algoName} ===");

            var startNode = graph.Nodes[0];
            List<int> visitOrder = new List<int>();

            if (isDFS)
            {
                Stack<WeightedGraph.Node> stack = new Stack<WeightedGraph.Node>();
                stack.Push(startNode);

                while (stack.Count > 0)
                {
                    var current = stack.Pop();
                    if (current.Visited) continue;

                    current.Visited = true;
                    current.Color = Color.Red;
                    visitOrder.Add(current.Id);
                    Log($"DFS: Посещён узел {current.Id}. Стек: [{GetCollectionContent(stack)}]");
                    RefreshGraph();
                    await Task.Delay(animationDelay);

                    foreach (var neighbor in graph.GetNeighbors(current))
                    {
                        if (!neighbor.Visited)
                        {
                            stack.Push(neighbor);
                            if (neighbor.Color == Color.LightBlue)
                                neighbor.Color = Color.Yellow;
                        }
                    }
                    RefreshGraph();
                    await Task.Delay(animationDelay / 2);

                    current.Color = Color.Green;
                    RefreshGraph();
                }
            }
            else
            {
                Queue<WeightedGraph.Node> queue = new Queue<WeightedGraph.Node>();
                queue.Enqueue(startNode);
                startNode.Visited = true;
                startNode.Color = Color.Yellow;

                while (queue.Count > 0)
                {
                    var current = queue.Dequeue();
                    current.Color = Color.Red;
                    visitOrder.Add(current.Id);
                    Log($"BFS: Обрабатывается узел {current.Id}. Очередь: [{GetCollectionContent(queue)}]");
                    RefreshGraph();
                    await Task.Delay(animationDelay);

                    foreach (var neighbor in graph.GetNeighbors(current))
                    {
                        if (!neighbor.Visited)
                        {
                            neighbor.Visited = true;
                            neighbor.Color = Color.Yellow;
                            queue.Enqueue(neighbor);
                        }
                    }
                    RefreshGraph();
                    await Task.Delay(animationDelay / 2);

                    current.Color = Color.Green;
                    RefreshGraph();
                }
            }

            Log($"=== {algoName} завершён. Порядок: {string.Join(" → ", visitOrder)} ===\n");
            isRunning = false;
            SetButtonsEnabled(true);
        }

        private string GetCollectionContent<T>(IEnumerable<T> collection) where T : WeightedGraph.Node
        {
            var ids = new List<int>();
            foreach (var node in collection)
                ids.Add(node.Id);
            return string.Join(", ", ids);
        }

        private void SetButtonsEnabled(bool enabled)
        {
            btnDFS.Enabled = enabled;
            btnBFS.Enabled = enabled;
            btnGenerateGraph.Enabled = enabled;
            btnAddNode.Enabled = enabled;
            btnAddEdge.Enabled = enabled;
            btnDeleteNode.Enabled = enabled;
            btnMoveNode.Enabled = enabled;
            btnSave.Enabled = enabled;
            btnLoad.Enabled = enabled;
        }
    }
}