Pass the whole Map into MapGenerator.generate

This commit is contained in:
Eryn Wells 2023-02-18 22:55:20 -08:00
parent 7428d95126
commit 37ffa423b6
3 changed files with 27 additions and 16 deletions

View file

@ -20,7 +20,8 @@ class MapGenerator:
'''The location of any routes to a lower floor of the dungeon.'''
raise NotImplementedError()
def generate(self, tiles: np.ndarray):
# pylint: disable=redefined-builtin
def generate(self, map: 'Map'):
'''Generate a map and place it in `tiles`'''
raise NotImplementedError()
@ -42,9 +43,10 @@ class RoomsAndCorridorsGenerator(MapGenerator):
def down_stairs(self) -> List[Point]:
return self.room_generator.down_stairs
def generate(self, tiles: np.ndarray):
# pylint: disable=redefined-builtin
def generate(self, map: 'Map'):
self.room_generator.generate()
self.room_generator.apply(tiles)
self.room_generator.apply(map)
self.corridor_generator.generate(self.room_generator.rooms)
self.corridor_generator.apply(tiles)
self.corridor_generator.apply(map.tiles)