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:
parent
843aa2823f
commit
9a04692539
7 changed files with 391 additions and 262 deletions
29
erynrl/map/generator/__init__.py
Normal file
29
erynrl/map/generator/__init__.py
Normal 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)
|
||||
Loading…
Add table
Add a link
Reference in a new issue