diff options
Diffstat (limited to 'dungeon/src/map.rs')
| -rw-r--r-- | dungeon/src/map.rs | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/dungeon/src/map.rs b/dungeon/src/map.rs index a2cdd9a..fb42a77 100644 --- a/dungeon/src/map.rs +++ b/dungeon/src/map.rs @@ -1,6 +1,9 @@ //! The `map` module contains structures of the dungeon game map //! including the current `Floor`, and map `Tile`. +use strum::IntoEnumIterator; +use strum_macros::EnumIter; + use crate::wfc::Wfc; use std::{ cell::RefCell, @@ -20,7 +23,7 @@ pub const TILE_COUNT: usize = MAP_SIZE_USIZE * MAP_SIZE_USIZE; /// The `Tile` enum represents what is (or is not) at /// any given spot in the dungeon grid. -#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)] +#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, EnumIter)] pub enum Tile { /// `Wall` represents an impassible wall Wall, @@ -29,13 +32,15 @@ pub enum Tile { /// `Stairs` represents stairs to another floor Stairs, } - impl Tile { /// Returns a list of all possible tiles - /// TODO! Use a crate for enum itterator - #[must_use] - pub fn all_tiles() -> Vec<Tile> { - vec![Tile::Wall, Tile::Air, Tile::Stairs] + pub fn values() -> impl Iterator<Item = Self> { + Self::iter() + } +} +impl Default for Tile { + fn default() -> Self { + Self::Air } } |