import csv
from pathlib import Path

import matplotlib.pyplot as plt


def create_plot(csv_path="docs/data/maze_results_summary.csv", output_path="docs/data/maze_performance.png"):
    with Path(csv_path).open(encoding="utf-8-sig") as file:
        rows = list(csv.DictReader(file))
    mazes = sorted({row["maze"] for row in rows})
    strategies = ["BFS", "DFS", "A*"]
    colors = {"BFS": "#4472C4", "DFS": "#ED7D31", "A*": "#70AD47"}
    figure, axes = plt.subplots(2, 1, figsize=(12, 9))
    positions = range(len(mazes))
    width = 0.24
    for index, strategy in enumerate(strategies):
        selected = [next(row for row in rows if row["maze"] == maze and row["strategy"] == strategy) for maze in mazes]
        offsets = [position + (index - 1) * width for position in positions]
        axes[0].bar(offsets, [float(row["mean_time_ms"]) for row in selected], width, label=strategy, color=colors[strategy])
        axes[1].bar(offsets, [float(row["mean_visited_cells"]) for row in selected], width, label=strategy, color=colors[strategy])
    for axis, ylabel, title in zip(axes, ["Время, мс", "Посещено клеток"], ["Среднее время поиска", "Объём исследования лабиринта"]):
        axis.set_xticks(list(positions), mazes, rotation=20, ha="right")
        axis.set_ylabel(ylabel)
        axis.set_title(title)
        axis.grid(axis="y", alpha=0.25)
        axis.legend()
    figure.suptitle("Сравнение стратегий поиска (средние значения)")
    figure.tight_layout()
    figure.savefig(output_path, dpi=180)
    plt.close(figure)
    return Path(output_path)


if __name__ == "__main__":
    print(f"График сохранён: {create_plot()}")