94 lines
3.3 KiB
Python
94 lines
3.3 KiB
Python
|
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}")
|