Eryn Wells
6780b0495c
Draw three windows with frames: - map window - info window (hit point bar; turn count) - message window Clean up the UI code in the Engine.
69 lines
2.6 KiB
Python
69 lines
2.6 KiB
Python
# Eryn Wells <eryn@erynwells.me>
|
|
|
|
'''
|
|
This module defines the level map, a number of basic building blocks (Rooms, etc), and objects that generate various
|
|
parts of a map.
|
|
'''
|
|
|
|
import random
|
|
from typing import Iterable
|
|
|
|
import numpy as np
|
|
import numpy.typing as npt
|
|
import tcod
|
|
|
|
from ..geometry import Point, Rect, Size
|
|
from .generator import MapGenerator
|
|
from .tile import Empty, Shroud
|
|
|
|
|
|
class Map:
|
|
def __init__(self, size: Size, generator: MapGenerator):
|
|
self.size = size
|
|
|
|
self.tiles = np.full(tuple(size), fill_value=Empty, order='F')
|
|
generator.generate(self.tiles)
|
|
|
|
self.up_stairs = generator.up_stairs
|
|
self.down_stairs = generator.down_stairs
|
|
|
|
self.highlighted = np.full(tuple(self.size), fill_value=False, order='F')
|
|
# Map tiles that are currently visible to the player
|
|
self.visible = np.full(tuple(self.size), fill_value=False, order='F')
|
|
# Map tiles that the player has explored
|
|
self.explored = np.full(tuple(self.size), fill_value=False, order='F')
|
|
|
|
self.__walkable_points = None
|
|
|
|
def random_walkable_position(self) -> Point:
|
|
'''Return a random walkable point on the map.'''
|
|
if not self.__walkable_points:
|
|
self.__walkable_points = [Point(x, y) for x, y in np.ndindex(
|
|
self.tiles.shape) if self.tiles[x, y]['walkable']]
|
|
return random.choice(self.__walkable_points)
|
|
|
|
def tile_is_in_bounds(self, point: Point) -> bool:
|
|
'''Return True if the given point is inside the bounds of the map'''
|
|
return 0 <= point.x < self.size.width and 0 <= point.y < self.size.height
|
|
|
|
def tile_is_walkable(self, point: Point) -> bool:
|
|
'''Return True if the tile at the given point is walkable'''
|
|
return self.tile_is_in_bounds(point) and self.tiles[point.x, point.y]['walkable']
|
|
|
|
def highlight_points(self, points: Iterable[Point]):
|
|
'''Update the highlight graph with the list of points to highlight.'''
|
|
self.highlighted.fill(False)
|
|
|
|
for pt in points if points:
|
|
self.highlighted[pt.x, pt.y] = True
|
|
|
|
def print_to_console(self, console: tcod.Console, bounds: Rect) -> None:
|
|
'''Render the map to the console.'''
|
|
size = self.size
|
|
|
|
# If a tile is in the visible array, draw it with the "light" color. If it's not, but it's in the explored
|
|
# array, draw it with the "dark" color. Otherwise, draw it as Empty.
|
|
console.tiles_rgb[0:size.width, 0:size.height] = np.select(
|
|
condlist=[self.highlighted, self.visible, self.explored],
|
|
choicelist=[self.tiles['highlighted'], self.tiles['light'], self.tiles['dark']],
|
|
default=Shroud)
|