import csv
import time
from typing import List, Dict
from models import Maze
from builders import TextFileMazeBuilder
from strategies import BFSStrategy, DFSStrategy, AStarStrategy
from solver import MazeSolver


def run_experiment(maze: Maze, strategy_name: str, strategy, repeats: int = 5) -> Dict:
    times = []
    visited_counts = []
    path_lengths = []
    path_found = True
    
    for _ in range(repeats):
        solver = MazeSolver(maze, strategy)
        path, stats = solver.solve()
        
        times.append(stats.time_ms)
        visited_counts.append(stats.visited_cells)
        path_lengths.append(stats.path_length)
        path_found = stats.path_found
    
    return {
        'strategy': strategy_name,
        'time_mean': sum(times) / len(times),
        'time_min': min(times),
        'time_max': max(times),
        'visited_mean': sum(visited_counts) / len(visited_counts),
        'path_length_mean': sum(path_lengths) / len(path_lengths) if path_found else 0,
        'path_found': path_found
    }


def run_all_experiments(maze_files: List[str], repeats: int = 5) -> List[Dict]:
    builder = TextFileMazeBuilder()
    strategies = [
        ('BFS', BFSStrategy()),
        ('DFS', DFSStrategy()),
        ('A*', AStarStrategy())
    ]
    
    results = []
    
    for maze_file in maze_files:
        try:
            maze = builder.build_from_file(maze_file)
        except (ValueError, FileNotFoundError) as e:
            print(f" Ошибка: {e}")
            continue
        
        print(f"    Размер: {maze.width}×{maze.height}")
        print(f"    Старт: ({maze.start.x}, {maze.start.y})")
        print(f"    Выход: ({maze.exit.x}, {maze.exit.y})")
        
        for strategy_name, strategy in strategies:
            print(f"    Тестирование: {strategy_name}")
            result = run_experiment(maze, strategy_name, strategy, repeats)
            result['maze_file'] = maze_file.split('/')[-1]
            result['maze_size'] = f"{maze.width}×{maze.height}"
            results.append(result)
            
            status = "ok" if result['path_found'] else "ne ok"
            print(f"        {status} Время: {result['time_mean']:.2f} мс, "
                  f"Посещено: {result['visited_mean']:.0f}, "
                  f"Путь: {result['path_length_mean']:.0f}")
    
    return results


def save_results_to_csv(results: List[Dict], filename: str = "experiment_results.csv") -> None:
    with open(filename, 'w', newline='', encoding='utf-8') as f:
        writer = csv.DictWriter(f, fieldnames=[
            'maze_file', 'maze_size', 'strategy',
            'time_mean', 'time_min', 'time_max',
            'visited_mean', 'path_length_mean', 'path_found'
        ])
        writer.writeheader()
        writer.writerows(results)



def print_results_table(results: List[Dict]) -> None:
    print("\n" + "=" * 80)
    print("РЕЗУЛЬТАТЫ ЭКСПЕРИМЕНТОВ")
    print("=" * 80)
    
    for res in results:
        print(f"\nЛабиринт: {res['maze_file']}")
        print(f"  Стратегия: {res['strategy']}")
        print(f"    Время (ср):   {res['time_mean']:.2f} мс")
        print(f"    Посещено:     {res['visited_mean']:.0f} клеток")
        print(f"    Длина пути:   {res['path_length_mean']:.0f}")