roguelike-game/src/map_builders/cellular_automata.rs
2021-12-15 12:08:23 -05:00

92 lines
3.2 KiB
Rust

use rltk::RandomNumberGenerator;
use crate::map_builders::{BuilderMap, InitialMapBuilder};
use crate::TileType;
pub struct CellularAutomataBuilder {}
impl InitialMapBuilder for CellularAutomataBuilder {
#[allow(dead_code)]
fn build_map(&mut self, rng: &mut RandomNumberGenerator, build_data: &mut BuilderMap) {
self.build(rng, build_data);
}
}
impl CellularAutomataBuilder {
pub fn new() -> Box<CellularAutomataBuilder> {
Box::new(CellularAutomataBuilder {})
}
fn build(&mut self, rng: &mut RandomNumberGenerator, build_data: &mut BuilderMap) {
// First we completely randomize the map, setting 55% of it to be floor.
for y in 1..build_data.map.height - 1 {
for x in 1..build_data.map.width - 1 {
let roll = rng.roll_dice(1, 100);
let idx = build_data.map.xy_idx(x, y);
if roll > 55 {
build_data.map.tiles[idx] = TileType::Floor
} else {
build_data.map.tiles[idx] = TileType::Wall
}
}
}
build_data.take_snapshot();
// Now we iteratively apply cellular automata rules
for _i in 0..15 {
let mut newtiles = build_data.map.tiles.clone();
for y in 1..build_data.map.height - 1 {
for x in 1..build_data.map.width - 1 {
let idx = build_data.map.xy_idx(x, y);
let mut neighbors = 0;
if build_data.map.tiles[idx - 1] == TileType::Wall {
neighbors += 1;
}
if build_data.map.tiles[idx + 1] == TileType::Wall {
neighbors += 1;
}
if build_data.map.tiles[idx - build_data.map.width as usize] == TileType::Wall {
neighbors += 1;
}
if build_data.map.tiles[idx + build_data.map.width as usize] == TileType::Wall {
neighbors += 1;
}
if build_data.map.tiles[idx - (build_data.map.width as usize - 1)]
== TileType::Wall
{
neighbors += 1;
}
if build_data.map.tiles[idx - (build_data.map.width as usize + 1)]
== TileType::Wall
{
neighbors += 1;
}
if build_data.map.tiles[idx + (build_data.map.width as usize - 1)]
== TileType::Wall
{
neighbors += 1;
}
if build_data.map.tiles[idx + (build_data.map.width as usize + 1)]
== TileType::Wall
{
neighbors += 1;
}
if neighbors > 4 || neighbors == 0 {
newtiles[idx] = TileType::Wall;
} else {
newtiles[idx] = TileType::Floor;
}
}
}
build_data.map.tiles = newtiles.clone();
build_data.take_snapshot();
}
}
}