Kayne Ruse
f831019148
When more than one item can be picked up, and options window is shown. Stubs for "using" an item are in place.
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
from __future__ import annotations
|
|
from typing import Iterable, List, Optional, Set, TYPE_CHECKING
|
|
|
|
import numpy as np
|
|
from tcod.console import Console
|
|
|
|
import tile_types
|
|
|
|
if TYPE_CHECKING:
|
|
from engine import Engine
|
|
from entity import Entity
|
|
|
|
class FloorMap:
|
|
engine: Engine
|
|
player: Entity
|
|
|
|
def __init__(self, width: int, height: int, entities: Iterable[Entity] = ()):
|
|
#terrain stuff
|
|
self.width, self.height = width, height
|
|
self.tiles = np.full((width, height), fill_value=tile_types.wall, order="F")
|
|
|
|
self.visible = np.full((width, height), fill_value=False, order="F")
|
|
self.explored = np.full((width, height), fill_value=False, order="F")
|
|
|
|
#contents stuff
|
|
self.entities: Set[Entity] = set(entities)
|
|
self.procgen_cache = {} #reserved for the procgen algorithm, otherwise ignored
|
|
|
|
#set externally
|
|
self.engine = None
|
|
self.player = None
|
|
|
|
def in_bounds(self, x: int, y: int) -> bool:
|
|
return 0 <= x < self.width and 0 <= y < self.height
|
|
|
|
def get_all_entities_at(self, x: int, y: int, *, unwalkable_only: bool = False, items_only: bool = False) -> List[Entity]:
|
|
result: List[Entity] = []
|
|
|
|
for entity in self.entities:
|
|
if entity.x == x and entity.y == y:
|
|
if unwalkable_only and entity.walkable:
|
|
continue
|
|
if items_only and not entity.is_item():
|
|
continue
|
|
result.append(entity)
|
|
|
|
return result
|
|
|
|
def render(self, console: Console) -> None:
|
|
console.rgb[0:self.width, 0:self.height] = np.select(
|
|
condlist = [self.visible, self.explored],
|
|
choicelist = [self.tiles["light"], self.tiles["dark"]],
|
|
default = tile_types.SHROUD
|
|
)
|
|
|
|
#render the dead stuff below the alive stuff
|
|
alive = (entity for entity in self.entities if entity.is_alive())
|
|
dead = (entity for entity in self.entities if not entity.is_alive())
|
|
|
|
for entity in dead:
|
|
if self.visible[entity.x, entity.y]:
|
|
console.print(entity.x, entity.y, entity.char, fg=entity.color)
|
|
|
|
for entity in alive:
|
|
if self.visible[entity.x, entity.y]:
|
|
console.print(entity.x, entity.y, entity.char, fg=entity.color)
|
|
|
|
#print the player above everything else for clarity
|
|
if self.player:
|
|
console.print(self.player.x, self.player.y, self.player.char, fg=self.player.color)
|