use crate::behaviors::maze::MazeGeneration;
use crate::grid::Grid;
use crate::cell::Coordinates;
use crate::error::Error;
use std::collections::HashSet;
pub struct RecursiveBacktracker;
impl MazeGeneration for RecursiveBacktracker {
fn generate(&self, grid: &mut Grid) -> Result<(), Error> {
let mut stack: Vec<Coordinates> = Vec::new();
let mut visited: HashSet<Coordinates> = HashSet::new();
stack.push(grid.start_coords);
visited.insert(grid.start_coords);
if grid.capture_steps {
let changed_cells = HashSet::new();
self.capture_step(grid, &changed_cells);
}
while let Some(current_coords) = stack.last().cloned() {
let neighbors: Vec<Coordinates> = grid
.get(current_coords)?
.neighbors()
.into_iter()
.filter(|neighbor| !visited.contains(neighbor))
.collect();
if neighbors.is_empty() {
stack.pop();
} else {
let random_index = {
let upper_bound = neighbors.len();
grid.bounded_random_usize(upper_bound)
};
let next_coords = neighbors[random_index];
grid.link(current_coords, next_coords)?;
if grid.capture_steps {
let mut changed_cells = HashSet::new();
changed_cells.insert(current_coords);
changed_cells.insert(next_coords);
self.capture_step(grid, &changed_cells);
}
visited.insert(next_coords);
stack.push(next_coords);
}
}
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::cell::{ MazeType, Coordinates };
#[test]
fn generate_and_print_5_x_5_orthogonal_maze() {
match Grid::new(MazeType::Orthogonal, 4, 4, Coordinates { x: 0, y: 0 }, Coordinates { x: 3, y: 3 }, false) {
Ok(mut grid) => {
assert!(!grid.is_perfect_maze().unwrap());
RecursiveBacktracker.generate(&mut grid).expect("RecursiveBacktracker maze generation failed");
println!("\n\nRecursive Backtracker\n\n{}\n\n", grid.to_asci());
assert!(grid.is_perfect_maze().unwrap());
}
Err(e) => panic!("Unexpected error occurred running test: {:?}", e),
}
}
#[test]
fn generate_and_print_12_x_6_orthogonal_maze() {
match Grid::new(MazeType::Orthogonal, 12, 6, Coordinates { x: 0, y: 0 }, Coordinates { x: 11, y: 5 }, false) {
Ok(mut grid) => {
assert!(!grid.is_perfect_maze().unwrap());
RecursiveBacktracker.generate(&mut grid).expect("RecursiveBacktracker maze generation failed");
println!("\n\nRecursive Backtracker\n\n{}\n\n", grid.to_asci());
assert!(grid.is_perfect_maze().unwrap());
}
Err(e) => panic!("Unexpected error running test: {:?}", e),
}
}
#[test]
fn generate_5_x_5_delta_maze() {
match Grid::new(MazeType::Delta, 4, 4, Coordinates { x: 0, y: 0 }, Coordinates { x: 3, y: 3 }, false) {
Ok(mut grid) => {
assert!(!grid.is_perfect_maze().unwrap());
RecursiveBacktracker.generate(&mut grid).expect("RecursiveBacktracker maze generation failed");
assert!(grid.is_perfect_maze().unwrap());
}
Err(e) => panic!("Unexpected error occurred running test: {:?}", e),
}
}
#[test]
fn generate_12_x_6_delta_maze() {
match Grid::new(MazeType::Delta, 12, 6, Coordinates { x: 0, y: 0 }, Coordinates { x: 11, y: 5 }, false) {
Ok(mut grid) => {
assert!(!grid.is_perfect_maze().unwrap());
RecursiveBacktracker.generate(&mut grid).expect("RecursiveBacktracker maze generation failed");
assert!(grid.is_perfect_maze().unwrap());
}
Err(e) => panic!("Unexpected error running test: {:?}", e),
}
}
#[test]
fn generate_5_x_5_sigma_maze() {
match Grid::new(MazeType::Sigma, 4, 4, Coordinates { x: 0, y: 0 }, Coordinates { x: 3, y: 3 }, false) {
Ok(mut grid) => {
assert!(!grid.is_perfect_maze().unwrap());
RecursiveBacktracker.generate(&mut grid).expect("RecursiveBacktracker maze generation failed");
assert!(grid.is_perfect_maze().unwrap());
}
Err(e) => panic!("Unexpected error occurred running test: {:?}", e),
}
}
#[test]
fn generate_12_x_6_sigma_maze() {
match Grid::new(MazeType::Sigma, 12, 6, Coordinates { x: 0, y: 0 }, Coordinates { x: 11, y: 5 }, false) {
Ok(mut grid) => {
assert!(!grid.is_perfect_maze().unwrap());
RecursiveBacktracker.generate(&mut grid).expect("RecursiveBacktracker maze generation failed");
assert!(grid.is_perfect_maze().unwrap());
}
Err(e) => panic!("Unexpected error running test: {:?}", e),
}
}
#[test]
fn generate_12_x_6_rhombic_maze_recursive_backtracker() {
match Grid::new(MazeType::Rhombic, 12, 6, Coordinates { x: 0, y: 0 }, Coordinates { x: 11, y: 5 }, false) {
Ok(mut grid) => {
assert!(!grid.is_perfect_maze().unwrap());
RecursiveBacktracker.generate(&mut grid).expect("RecursiveBacktracker maze generation failed");
assert!(grid.is_perfect_maze().unwrap());
}
Err(e) => panic!("Unexpected error running test: {:?}", e),
}
}
#[test]
fn test_recursive_backtracker_with_capture_steps() {
let start = Coordinates { x: 0, y: 0 };
let goal = Coordinates { x: 19, y: 19 };
match Grid::new(MazeType::Orthogonal, 20, 20, start, goal, true) {
Ok(mut grid) => {
assert!(!grid.is_perfect_maze().unwrap());
RecursiveBacktracker.generate(&mut grid).expect("Maze generation failed");
assert!(grid.is_perfect_maze().unwrap());
assert!(grid.generation_steps.is_some());
let steps = grid.generation_steps.as_ref().unwrap(); assert!(!steps.is_empty());
let has_linked_cells = steps.iter().any(|step| {
step.cells.iter().filter_map(|opt| opt.as_ref()).any(|cell| !cell.linked.is_empty())
});
assert!(has_linked_cells, "No cells were linked during maze generation");
let has_open_walls = steps.iter().any(|step| {
step.cells.iter().filter_map(|opt| opt.as_ref()).any(|cell| !cell.open_walls.is_empty())
});
assert!(has_open_walls, "No cells have open walls in generation steps");
}
Err(e) => panic!("Unexpected error generating grid: {:?}", e),
}
}
}