2021-12-10 20:16:48 -05:00
|
|
|
use rltk::RandomNumberGenerator;
|
|
|
|
|
2021-12-15 12:08:23 -05:00
|
|
|
use super::common::{paint, Symmetry};
|
2021-12-10 20:16:48 -05:00
|
|
|
use crate::components::Position;
|
2021-12-15 13:58:47 -05:00
|
|
|
use crate::map_builders::{BuilderMap, InitialMapBuilder, MetaMapBuilder};
|
2021-12-15 12:08:23 -05:00
|
|
|
use crate::TileType;
|
2021-12-03 19:33:56 -05:00
|
|
|
|
2021-12-06 14:23:35 -05:00
|
|
|
#[derive(PartialEq, Copy, Clone)]
|
|
|
|
pub enum DrunkSpawnMode {
|
|
|
|
StaringPoint,
|
|
|
|
Random,
|
|
|
|
}
|
|
|
|
|
|
|
|
pub struct DrunkardSettings {
|
|
|
|
pub spawn_mode: DrunkSpawnMode,
|
|
|
|
pub drunken_lifetime: i32,
|
|
|
|
pub floor_percent: f32,
|
2021-12-07 10:36:41 -05:00
|
|
|
pub brush_size: i32,
|
|
|
|
pub symmetry: Symmetry,
|
2021-12-06 14:23:35 -05:00
|
|
|
}
|
|
|
|
|
2021-12-03 19:33:56 -05:00
|
|
|
pub struct DrunkardsWalkBuilder {
|
2021-12-06 14:23:35 -05:00
|
|
|
settings: DrunkardSettings,
|
2021-12-03 19:33:56 -05:00
|
|
|
}
|
|
|
|
|
2021-12-15 10:45:58 -05:00
|
|
|
impl InitialMapBuilder for DrunkardsWalkBuilder {
|
|
|
|
fn build_map(&mut self, rng: &mut RandomNumberGenerator, build_data: &mut BuilderMap) {
|
|
|
|
self.build(rng, build_data);
|
2021-12-10 16:34:11 -05:00
|
|
|
}
|
2021-12-03 19:33:56 -05:00
|
|
|
}
|
|
|
|
|
2021-12-15 13:58:47 -05:00
|
|
|
impl MetaMapBuilder for DrunkardsWalkBuilder {
|
|
|
|
#[allow(dead_code)]
|
|
|
|
fn build_map(&mut self, rng: &mut RandomNumberGenerator, build_data: &mut BuilderMap) {
|
|
|
|
self.build(rng, build_data);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-03 19:33:56 -05:00
|
|
|
impl DrunkardsWalkBuilder {
|
2021-12-15 10:45:58 -05:00
|
|
|
pub fn new(settings: DrunkardSettings) -> DrunkardsWalkBuilder {
|
|
|
|
DrunkardsWalkBuilder { settings }
|
2021-12-03 19:33:56 -05:00
|
|
|
}
|
|
|
|
|
2021-12-15 10:45:58 -05:00
|
|
|
pub fn open_area() -> Box<DrunkardsWalkBuilder> {
|
|
|
|
Box::new(DrunkardsWalkBuilder::new(DrunkardSettings {
|
|
|
|
spawn_mode: DrunkSpawnMode::StaringPoint,
|
|
|
|
drunken_lifetime: 400,
|
|
|
|
floor_percent: 0.5,
|
|
|
|
brush_size: 1,
|
|
|
|
symmetry: Symmetry::None,
|
|
|
|
}))
|
2021-12-06 14:23:35 -05:00
|
|
|
}
|
|
|
|
|
2021-12-15 10:45:58 -05:00
|
|
|
pub fn open_halls() -> Box<DrunkardsWalkBuilder> {
|
|
|
|
Box::new(DrunkardsWalkBuilder::new(DrunkardSettings {
|
|
|
|
spawn_mode: DrunkSpawnMode::Random,
|
|
|
|
drunken_lifetime: 400,
|
|
|
|
floor_percent: 0.5,
|
|
|
|
brush_size: 1,
|
|
|
|
symmetry: Symmetry::None,
|
|
|
|
}))
|
2021-12-06 14:23:35 -05:00
|
|
|
}
|
|
|
|
|
2021-12-15 10:45:58 -05:00
|
|
|
pub fn winding_passages() -> Box<DrunkardsWalkBuilder> {
|
|
|
|
Box::new(DrunkardsWalkBuilder::new(DrunkardSettings {
|
|
|
|
spawn_mode: DrunkSpawnMode::Random,
|
|
|
|
drunken_lifetime: 100,
|
|
|
|
floor_percent: 0.4,
|
|
|
|
brush_size: 1,
|
|
|
|
symmetry: Symmetry::None,
|
|
|
|
}))
|
2021-12-07 10:36:41 -05:00
|
|
|
}
|
|
|
|
|
2021-12-15 10:45:58 -05:00
|
|
|
pub fn fat_passages() -> Box<DrunkardsWalkBuilder> {
|
|
|
|
Box::new(DrunkardsWalkBuilder::new(DrunkardSettings {
|
|
|
|
spawn_mode: DrunkSpawnMode::Random,
|
|
|
|
drunken_lifetime: 100,
|
|
|
|
floor_percent: 0.4,
|
|
|
|
brush_size: 2,
|
|
|
|
symmetry: Symmetry::None,
|
|
|
|
}))
|
2021-12-07 10:36:41 -05:00
|
|
|
}
|
|
|
|
|
2021-12-15 10:45:58 -05:00
|
|
|
pub fn fearful_symmetry() -> Box<DrunkardsWalkBuilder> {
|
|
|
|
Box::new(DrunkardsWalkBuilder::new(DrunkardSettings {
|
|
|
|
spawn_mode: DrunkSpawnMode::Random,
|
|
|
|
drunken_lifetime: 100,
|
|
|
|
floor_percent: 0.4,
|
|
|
|
brush_size: 1,
|
|
|
|
symmetry: Symmetry::Both,
|
|
|
|
}))
|
2021-12-06 14:23:35 -05:00
|
|
|
}
|
|
|
|
|
2021-12-03 19:33:56 -05:00
|
|
|
#[allow(clippy::map_entry)]
|
2021-12-15 10:45:58 -05:00
|
|
|
fn build(&mut self, rng: &mut RandomNumberGenerator, build_data: &mut BuilderMap) {
|
2021-12-03 19:33:56 -05:00
|
|
|
// Set a central starting point
|
2021-12-15 10:45:58 -05:00
|
|
|
let starting_position = Position {
|
|
|
|
x: build_data.map.width / 2,
|
|
|
|
y: build_data.map.height / 2,
|
2021-12-03 19:33:56 -05:00
|
|
|
};
|
2021-12-15 10:45:58 -05:00
|
|
|
let start_idx = build_data
|
2021-12-03 19:33:56 -05:00
|
|
|
.map
|
2021-12-15 10:45:58 -05:00
|
|
|
.xy_idx(starting_position.x, starting_position.y);
|
|
|
|
build_data.map.tiles[start_idx] = TileType::Floor;
|
2021-12-03 19:33:56 -05:00
|
|
|
|
2021-12-15 10:45:58 -05:00
|
|
|
let total_tiles = build_data.map.width * build_data.map.height;
|
2021-12-06 14:23:35 -05:00
|
|
|
let desired_floor_tiles = (self.settings.floor_percent * total_tiles as f32) as usize;
|
2021-12-15 10:45:58 -05:00
|
|
|
let mut floor_tile_count = build_data
|
2021-12-06 14:23:35 -05:00
|
|
|
.map
|
|
|
|
.tiles
|
|
|
|
.iter()
|
|
|
|
.filter(|a| **a == TileType::Floor)
|
|
|
|
.count();
|
|
|
|
let mut digger_count = 0;
|
|
|
|
|
|
|
|
while floor_tile_count < desired_floor_tiles {
|
|
|
|
let mut did_something = false;
|
|
|
|
let mut drunk_x;
|
|
|
|
let mut drunk_y;
|
|
|
|
match self.settings.spawn_mode {
|
|
|
|
DrunkSpawnMode::StaringPoint => {
|
2021-12-15 10:45:58 -05:00
|
|
|
drunk_x = starting_position.x;
|
|
|
|
drunk_y = starting_position.y;
|
2021-12-06 14:23:35 -05:00
|
|
|
}
|
|
|
|
DrunkSpawnMode::Random => {
|
|
|
|
if digger_count == 0 {
|
2021-12-15 10:45:58 -05:00
|
|
|
drunk_x = starting_position.x;
|
|
|
|
drunk_y = starting_position.y;
|
2021-12-06 14:23:35 -05:00
|
|
|
} else {
|
2021-12-15 10:45:58 -05:00
|
|
|
drunk_x = rng.roll_dice(1, build_data.map.width - 3) + 1;
|
|
|
|
drunk_y = rng.roll_dice(1, build_data.map.height - 3) + 1;
|
2021-12-06 14:23:35 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
let mut drunk_life = self.settings.drunken_lifetime;
|
|
|
|
|
|
|
|
while drunk_life > 0 {
|
2021-12-15 10:45:58 -05:00
|
|
|
let drunk_idx = build_data.map.xy_idx(drunk_x, drunk_y);
|
|
|
|
if build_data.map.tiles[drunk_idx] == TileType::Wall {
|
2021-12-06 14:23:35 -05:00
|
|
|
did_something = true;
|
|
|
|
}
|
|
|
|
|
2021-12-07 10:36:41 -05:00
|
|
|
paint(
|
2021-12-15 10:45:58 -05:00
|
|
|
&mut build_data.map,
|
2021-12-07 10:36:41 -05:00
|
|
|
self.settings.symmetry,
|
|
|
|
self.settings.brush_size,
|
|
|
|
drunk_x,
|
|
|
|
drunk_y,
|
|
|
|
);
|
2021-12-15 10:45:58 -05:00
|
|
|
build_data.map.tiles[drunk_idx] = TileType::DownStairs;
|
2021-12-06 14:23:35 -05:00
|
|
|
|
|
|
|
let stagger_direction = rng.roll_dice(1, 4);
|
|
|
|
match stagger_direction {
|
|
|
|
1 => {
|
|
|
|
if drunk_x > 2 {
|
|
|
|
drunk_x -= 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
2 => {
|
2021-12-15 10:45:58 -05:00
|
|
|
if drunk_x < build_data.map.width - 2 {
|
2021-12-06 14:23:35 -05:00
|
|
|
drunk_x += 1
|
|
|
|
}
|
|
|
|
}
|
|
|
|
3 => {
|
|
|
|
if drunk_y > 2 {
|
|
|
|
drunk_y -= 1
|
|
|
|
}
|
|
|
|
}
|
|
|
|
_ => {
|
2021-12-15 10:45:58 -05:00
|
|
|
if drunk_y < build_data.map.height - 2 {
|
2021-12-06 14:23:35 -05:00
|
|
|
drunk_y += 1
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
drunk_life -= 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if did_something {
|
2021-12-15 10:45:58 -05:00
|
|
|
build_data.take_snapshot();
|
2021-12-06 14:23:35 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
digger_count += 1;
|
2021-12-15 10:45:58 -05:00
|
|
|
for t in build_data.map.tiles.iter_mut() {
|
2021-12-06 14:23:35 -05:00
|
|
|
if *t == TileType::DownStairs {
|
|
|
|
*t = TileType::Floor;
|
|
|
|
}
|
|
|
|
}
|
2021-12-15 10:45:58 -05:00
|
|
|
floor_tile_count = build_data
|
2021-12-06 14:23:35 -05:00
|
|
|
.map
|
|
|
|
.tiles
|
|
|
|
.iter()
|
|
|
|
.filter(|a| **a == TileType::Floor)
|
|
|
|
.count();
|
|
|
|
}
|
2021-12-03 19:33:56 -05:00
|
|
|
}
|
|
|
|
}
|