77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
from __future__ import annotations
|
|
from typing import Iterable, Iterator, Optional
|
|
|
|
import numpy as np
|
|
from tcod.console import Console
|
|
|
|
import tile_types
|
|
from entity import Entity, Actor
|
|
|
|
class FloorMap:
|
|
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(entities)
|
|
self.procgen_cache = {} #reserved for the procgen algorithm, otherwise ignored
|
|
|
|
#set externally
|
|
self.engine = None
|
|
self.player = None
|
|
|
|
@property
|
|
def actors(self) -> Iterator[Actor]:
|
|
yield from (
|
|
entity
|
|
for entity in self.entities
|
|
if isinstance(entity, Actor) and entity.is_alive()
|
|
)
|
|
|
|
def in_bounds(self, x: int, y: int) -> bool:
|
|
return 0 <= x < self.width and 0 <= y < self.height
|
|
|
|
def get_entity_at(self, x: int, y: int, unwalkable_only: bool = False) -> Optional[Entity]:
|
|
for entity in self.entities:
|
|
if entity.x == x and entity.y == y:
|
|
if unwalkable_only:
|
|
if not entity.walkable:
|
|
return entity
|
|
else:
|
|
return entity
|
|
|
|
return None
|
|
|
|
def get_actor_at(self, x: int, y: int) -> Optional[Actor]:
|
|
for actor in self.actors:
|
|
if actor.x == x and actor.y == y and actor.is_alive():
|
|
return actor
|
|
|
|
return None
|
|
|
|
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) #TODO: I didn't realize the render order would be fixed in the tutorial |