refactoring
This commit is contained in:
parent
278b549094
commit
f661da36ce
|
@ -21,10 +21,10 @@
|
||||||
//!
|
//!
|
||||||
|
|
||||||
use rand::prelude::*;
|
use rand::prelude::*;
|
||||||
use super::MapFilter;
|
use crate::MapFilter;
|
||||||
use crate::geometry::{Point, Rect};
|
use crate::geometry::{Point, Rect};
|
||||||
use crate::random::Rng;
|
use crate::random::Rng;
|
||||||
use crate::map::Map;
|
use crate::Map;
|
||||||
|
|
||||||
|
|
||||||
pub struct BspInterior {
|
pub struct BspInterior {
|
||||||
|
|
|
@ -19,10 +19,10 @@
|
||||||
//!
|
//!
|
||||||
|
|
||||||
use rand::prelude::*;
|
use rand::prelude::*;
|
||||||
use super::MapFilter;
|
use crate::MapFilter;
|
||||||
use crate::geometry::Rect;
|
use crate::geometry::Rect;
|
||||||
use crate::random::Rng;
|
use crate::random::Rng;
|
||||||
use crate::map::{Map, TileType};
|
use crate::{Map, TileType};
|
||||||
|
|
||||||
|
|
||||||
pub struct BspRooms {
|
pub struct BspRooms {
|
||||||
|
|
|
@ -22,8 +22,8 @@
|
||||||
//!
|
//!
|
||||||
|
|
||||||
use rand::prelude::*;
|
use rand::prelude::*;
|
||||||
use super::MapFilter;
|
use crate::MapFilter;
|
||||||
use crate::map::{Map, TileType};
|
use crate::{Map, TileType};
|
||||||
|
|
||||||
|
|
||||||
/// Map generator and modifier
|
/// Map generator and modifier
|
||||||
|
|
|
@ -5,8 +5,8 @@
|
||||||
//!
|
//!
|
||||||
|
|
||||||
use rand::prelude::StdRng;
|
use rand::prelude::StdRng;
|
||||||
use super::MapFilter;
|
use crate::MapFilter;
|
||||||
use crate::map::{Map, TileType};
|
use crate::{Map, TileType};
|
||||||
use crate::dijkstra::DijkstraMap;
|
use crate::dijkstra::DijkstraMap;
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -7,8 +7,8 @@
|
||||||
use std::f32;
|
use std::f32;
|
||||||
use rand::prelude::StdRng;
|
use rand::prelude::StdRng;
|
||||||
use crate::geometry::Point;
|
use crate::geometry::Point;
|
||||||
use super::MapFilter;
|
use crate::MapFilter;
|
||||||
use crate::map::Map;
|
use crate::Map;
|
||||||
use crate::dijkstra::DijkstraMap;
|
use crate::dijkstra::DijkstraMap;
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -14,7 +14,7 @@
|
||||||
//!
|
//!
|
||||||
|
|
||||||
use rand::prelude::*;
|
use rand::prelude::*;
|
||||||
use super::MapFilter;
|
use crate::MapFilter;
|
||||||
use crate::{
|
use crate::{
|
||||||
map::{Map, Symmetry, TileType},
|
map::{Map, Symmetry, TileType},
|
||||||
geometry::Point,
|
geometry::Point,
|
||||||
|
|
|
@ -44,77 +44,5 @@ pub use distant_exit::DistantExit;
|
||||||
pub use drunkard::DrunkardsWalk;
|
pub use drunkard::DrunkardsWalk;
|
||||||
pub use simple_rooms::SimpleRooms;
|
pub use simple_rooms::SimpleRooms;
|
||||||
pub use rooms_corridors_nearest::NearestCorridors;
|
pub use rooms_corridors_nearest::NearestCorridors;
|
||||||
pub use starting_point::AreaStartingPosition;
|
pub use starting_point::{AreaStartingPosition, XStart, YStart};
|
||||||
|
|
||||||
use std::time::{SystemTime, UNIX_EPOCH};
|
|
||||||
use rand::prelude::*;
|
|
||||||
use crate::map::Map;
|
|
||||||
|
|
||||||
|
|
||||||
/// Trait which should be implemented by map modifier.
|
|
||||||
/// Modifier takes initiall map and apply changes to it.
|
|
||||||
pub trait MapFilter {
|
|
||||||
fn modify_map(&self, rng: &mut StdRng, map: &Map) -> Map;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Used to chain MapBuilder and MapModifiers to create the final map.
|
|
||||||
pub struct MapBuilder {
|
|
||||||
modifiers: Vec<Box<dyn MapFilter>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl MapBuilder {
|
|
||||||
/// Create Map Builder with initial map generator
|
|
||||||
pub fn new() -> MapBuilder {
|
|
||||||
MapBuilder {
|
|
||||||
modifiers: Vec::new(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn with(&mut self, modifier : Box<dyn MapFilter>) -> &mut MapBuilder {
|
|
||||||
self.modifiers.push(modifier);
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Build map using random number seeded with system time
|
|
||||||
pub fn build_map(&mut self, width: usize, height: usize) -> Map {
|
|
||||||
let system_time = SystemTime::now().duration_since(UNIX_EPOCH).expect("Can't access system time");
|
|
||||||
let mut rng = StdRng::seed_from_u64(system_time.as_millis() as u64);
|
|
||||||
self.build_map_with_rng(width, height, &mut rng)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Build map using provided random number generator
|
|
||||||
pub fn build_map_with_rng(&mut self, width: usize, height: usize, rng: &mut StdRng) -> Map {
|
|
||||||
let mut map = Map::new(width, height);
|
|
||||||
|
|
||||||
// Build additional layers in turn
|
|
||||||
for modifier in self.modifiers.iter() {
|
|
||||||
map = modifier.modify_map(rng, &map);
|
|
||||||
}
|
|
||||||
|
|
||||||
map
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
/// ------------------------------------------------------------------------------------------------
|
|
||||||
/// Module unit tests
|
|
||||||
/// ------------------------------------------------------------------------------------------------
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::*;
|
|
||||||
use cellular_automata::CellularAutomata;
|
|
||||||
use starting_point::{AreaStartingPosition, XStart, YStart};
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_ca_map() {
|
|
||||||
let map = MapBuilder::new()
|
|
||||||
.with(CellularAutomata::new())
|
|
||||||
.with(AreaStartingPosition::new(XStart::CENTER, YStart::CENTER))
|
|
||||||
.build_map(80, 50);
|
|
||||||
|
|
||||||
assert_eq!(map.width, 80);
|
|
||||||
assert_eq!(map.height, 50);
|
|
||||||
assert_eq!(map.starting_point.is_some(), true);
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
|
@ -1,8 +1,8 @@
|
||||||
//! Connect nearest rooms on the map with corridors
|
//! Connect nearest rooms on the map with corridors
|
||||||
//!
|
//!
|
||||||
use rand::prelude::StdRng;
|
use rand::prelude::StdRng;
|
||||||
use super::MapFilter;
|
use crate::MapFilter;
|
||||||
use crate::map::Map;
|
use crate::Map;
|
||||||
use std::collections::HashSet;
|
use std::collections::HashSet;
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -19,10 +19,10 @@
|
||||||
//!
|
//!
|
||||||
|
|
||||||
use rand::prelude::*;
|
use rand::prelude::*;
|
||||||
use super::MapFilter;
|
use crate::MapFilter;
|
||||||
use crate::geometry::Rect;
|
use crate::geometry::Rect;
|
||||||
use crate::random::Rng;
|
use crate::random::Rng;
|
||||||
use crate::map::{Map};
|
use crate::Map;
|
||||||
|
|
||||||
|
|
||||||
pub struct SimpleRooms {
|
pub struct SimpleRooms {
|
||||||
|
|
|
@ -21,9 +21,9 @@
|
||||||
//!
|
//!
|
||||||
|
|
||||||
use rand::prelude::StdRng;
|
use rand::prelude::StdRng;
|
||||||
use super::MapFilter;
|
use crate::MapFilter;
|
||||||
use crate::geometry::Point;
|
use crate::geometry::Point;
|
||||||
use crate::map::{Map, TileType};
|
use crate::{Map, TileType};
|
||||||
|
|
||||||
|
|
||||||
/// Initial x region position
|
/// Initial x region position
|
||||||
|
|
105
src/lib.rs
105
src/lib.rs
|
@ -1,15 +1,108 @@
|
||||||
//! A collection of map generators.
|
//! Generators for dungeon type maps.
|
||||||
//!
|
//!
|
||||||
//! The generators are implemented for the following types of maps:
|
//! Generators can bu used directly or they can be combined with
|
||||||
//! * Dungeon maps
|
//! `MapGenerator`s and `MapModifier`s
|
||||||
//!
|
//!
|
||||||
|
//! * MapGenerators are use to create initial map.
|
||||||
|
//! * MapModifiers modify existing map.
|
||||||
|
//!
|
||||||
|
//! Example
|
||||||
|
//! ```
|
||||||
|
//! use mapgen::{MapFilter, MapBuilder, Map, TileType};
|
||||||
|
//! use mapgen::filter::{
|
||||||
|
//! CellularAutomata,
|
||||||
|
//! starting_point::{AreaStartingPosition, XStart, YStart}
|
||||||
|
//! };
|
||||||
|
//! use mapgen::geometry::Point;
|
||||||
|
//!
|
||||||
|
//! let map = MapBuilder::new()
|
||||||
|
//! .with(CellularAutomata::new())
|
||||||
|
//! .with(AreaStartingPosition::new(XStart::CENTER, YStart::CENTER))
|
||||||
|
//! .build_map(80, 50);
|
||||||
|
//!
|
||||||
|
//! assert_eq!(map.width, 80);
|
||||||
|
//! assert_eq!(map.height, 50);
|
||||||
|
//! assert_eq!(map.starting_point.is_some(), true);
|
||||||
|
//! ```
|
||||||
|
//!
|
||||||
|
|
||||||
pub mod filter;
|
pub mod filter;
|
||||||
pub mod geometry;
|
pub mod geometry;
|
||||||
pub mod map;
|
pub mod map;
|
||||||
|
|
||||||
pub use map::{Map, TileType};
|
pub use map::{Map, Symmetry, TileType};
|
||||||
pub use filter::{MapFilter, MapBuilder};
|
|
||||||
|
|
||||||
pub (crate) mod dijkstra;
|
pub (crate) mod dijkstra;
|
||||||
pub (crate) mod random;
|
pub (crate) mod random;
|
||||||
|
|
||||||
|
use std::time::{SystemTime, UNIX_EPOCH};
|
||||||
|
use rand::prelude::*;
|
||||||
|
|
||||||
|
|
||||||
|
/// Trait which should be implemented by map modifier.
|
||||||
|
/// Modifier takes initiall map and apply changes to it.
|
||||||
|
pub trait MapFilter {
|
||||||
|
fn modify_map(&self, rng: &mut StdRng, map: &Map) -> Map;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Used to chain MapBuilder and MapModifiers to create the final map.
|
||||||
|
pub struct MapBuilder {
|
||||||
|
modifiers: Vec<Box<dyn MapFilter>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl MapBuilder {
|
||||||
|
/// Create Map Builder with initial map generator
|
||||||
|
pub fn new() -> MapBuilder {
|
||||||
|
MapBuilder {
|
||||||
|
modifiers: Vec::new(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn with(&mut self, modifier : Box<dyn MapFilter>) -> &mut MapBuilder {
|
||||||
|
self.modifiers.push(modifier);
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Build map using random number seeded with system time
|
||||||
|
pub fn build_map(&mut self, width: usize, height: usize) -> Map {
|
||||||
|
let system_time = SystemTime::now().duration_since(UNIX_EPOCH).expect("Can't access system time");
|
||||||
|
let mut rng = StdRng::seed_from_u64(system_time.as_millis() as u64);
|
||||||
|
self.build_map_with_rng(width, height, &mut rng)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Build map using provided random number generator
|
||||||
|
pub fn build_map_with_rng(&mut self, width: usize, height: usize, rng: &mut StdRng) -> Map {
|
||||||
|
let mut map = Map::new(width, height);
|
||||||
|
|
||||||
|
// Build additional layers in turn
|
||||||
|
for modifier in self.modifiers.iter() {
|
||||||
|
map = modifier.modify_map(rng, &map);
|
||||||
|
}
|
||||||
|
|
||||||
|
map
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/// ------------------------------------------------------------------------------------------------
|
||||||
|
/// Module unit tests
|
||||||
|
/// ------------------------------------------------------------------------------------------------
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
use filter::CellularAutomata;
|
||||||
|
use filter::{AreaStartingPosition, XStart, YStart};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_ca_map() {
|
||||||
|
let map = MapBuilder::new()
|
||||||
|
.with(CellularAutomata::new())
|
||||||
|
.with(AreaStartingPosition::new(XStart::CENTER, YStart::CENTER))
|
||||||
|
.build_map(80, 50);
|
||||||
|
|
||||||
|
assert_eq!(map.width, 80);
|
||||||
|
assert_eq!(map.height, 50);
|
||||||
|
assert_eq!(map.starting_point.is_some(), true);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user