30 lines
925 B
Python
30 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)
|