Refactor map generator package

- Move room generators to map.generators.room
- Move corridor generators to map.generators.corridor

Generators have a generate() method that generates the things they place,
and an apply() method that applies their objects to a grid of tiles.
This commit is contained in:
Eryn Wells 2023-02-09 16:07:29 -08:00
parent 843aa2823f
commit 9a04692539
7 changed files with 391 additions and 262 deletions

View file

@ -0,0 +1,29 @@
import numpy as np
from ..tile import Empty
from .corridor import CorridorGenerator
from .room import RoomGenerator
class MapGenerator:
'''Abstract base class defining an interface for generating a map and applying it to a set of tiles.'''
def generate(self, tiles: np.ndarray):
raise NotImplementedError()
class RoomsAndCorridorsGenerator(MapGenerator):
'''
Generates a classic "rooms and corridors" style map with the given room and corridor generators.
'''
def __init__(self, room_generator: RoomGenerator, corridor_generator: CorridorGenerator):
self.room_generator = room_generator
self.corridor_generator = corridor_generator
def generate(self, tiles: np.ndarray):
self.room_generator.generate()
self.room_generator.apply(tiles)
self.corridor_generator.generate(self.room_generator.rooms)
self.corridor_generator.apply(tiles)