going-rogue/roguebasin/object.py

53 lines
2 KiB
Python

#!/usr/bin/env python3
# Eryn Wells <eryn@erynwells.me>
from typing import TYPE_CHECKING, Optional, Tuple, Type
import tcod
from .components import Fighter
from .geometry import Point
from .monsters import Species
if TYPE_CHECKING:
from .ai import AI
class Entity:
'''A single-tile drawable entity with a symbol and position.'''
def __init__(self, symbol: str, *,
position: Optional[Point] = None,
ai: Optional[Type['AI']] = None,
fg: Optional[Tuple[int, int, int]] = None,
bg: Optional[Tuple[int, int, int]] = None):
self.position = position if position else Point()
self.foreground = fg if fg else (255, 255, 255)
self.background = bg
self.symbol = symbol
self.ai = ai
def print_to_console(self, console: tcod.Console) -> None:
console.print(x=self.position.x, y=self.position.y, string=self.symbol, fg=self.foreground, bg=self.background)
def __str__(self):
return f'{self.symbol}[{self.position}]'
def __repr__(self):
return f'{self.__class__.__name__}({self.symbol}, position={self.position}, fg={self.foreground}, bg={self.background})'
class Hero(Entity):
def __init__(self, position: Point):
super().__init__('@', position=position, fg=tuple(tcod.white))
class Monster(Entity):
'''An instance of a Species.'''
def __init__(self, species: Species, ai_class: Type['AI'], position: Point = None):
super().__init__(species.symbol, ai=ai_class(self), position=position, fg=species.foreground_color, bg=species.background_color)
self.species = species
self.fighter = Fighter(maximum_hit_points=species.maximum_hit_points,
attack_power=species.attack_power,
defense=species.defense)
def __str__(self) -> str:
return f'{self.symbol}[{self.species.name}][{self.position}][{self.fighter.hit_points}/{self.fighter.maximum_hit_points}]'