Eryn Wells
9a04692539
- 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.
29 lines
925 B
Python
29 lines
925 B
Python
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)
|