47 lines
1.4 KiB
Rust
47 lines
1.4 KiB
Rust
//! Connect nearest rooms on the map with corridors
|
|
//!
|
|
use rand::prelude::StdRng;
|
|
use crate::MapFilter;
|
|
use crate::Map;
|
|
use std::collections::HashSet;
|
|
|
|
|
|
pub struct NearestCorridors {}
|
|
|
|
impl MapFilter for NearestCorridors {
|
|
fn modify_map(&self, _: &mut StdRng, map: &Map) -> Map {
|
|
self.corridors(map)
|
|
}
|
|
}
|
|
|
|
impl NearestCorridors {
|
|
|
|
pub fn new() -> Box<NearestCorridors> {
|
|
Box::new(NearestCorridors{})
|
|
}
|
|
|
|
fn corridors(&self, map: &Map) -> Map {
|
|
let mut new_map = map.clone();
|
|
|
|
let mut connected : HashSet<usize> = HashSet::new();
|
|
for (i,room) in map.rooms.iter().enumerate() {
|
|
let mut room_distance : Vec<(usize, f32)> = Vec::new();
|
|
let room_center = room.center();
|
|
for (j,other_room) in new_map.rooms.iter().enumerate() {
|
|
if i != j && !connected.contains(&j) {
|
|
let other_center = other_room.center();
|
|
let distance = room_center.distance_to(&other_center);
|
|
room_distance.push((j, distance));
|
|
}
|
|
}
|
|
|
|
if !room_distance.is_empty() {
|
|
room_distance.sort_by(|a,b| a.1.partial_cmp(&b.1).unwrap() );
|
|
let dest_center = new_map.rooms[room_distance[0].0].center();
|
|
new_map.add_corridor(room_center, dest_center);
|
|
connected.insert(i);
|
|
}
|
|
}
|
|
new_map
|
|
}
|
|
} |