squirrel-battle/dungeonbattle/interfaces.py

143 lines
3.5 KiB
Python
Raw Normal View History

#!/usr/bin/env python
2020-11-06 16:43:30 +00:00
from enum import Enum, auto
from dungeonbattle.display.texturepack import TexturePack
class Map:
2020-10-16 16:05:49 +00:00
"""
Object that represents a Map with its width, height
and the whole tiles, with their custom properties.
"""
width: int
height: int
tiles: list
currentx : int #coordinates of the point that should be on the topleft corner of the screen
currenty : int
2020-11-06 16:59:19 +00:00
def __init__(self, width: int, height: int, tiles: list):
self.width = width
self.height = height
self.tiles = tiles
2020-11-06 16:59:19 +00:00
self.entities = []
def add_entity(self, entity: "Entity") -> None:
"""
Register a new entity in the map.
"""
self.entities.append(entity)
entity.map = self
@staticmethod
2020-10-09 16:24:13 +00:00
def load(filename: str):
2020-10-16 16:05:49 +00:00
"""
Read a file that contains the content of a map, and build a Map object.
"""
with open(filename, "r") as f:
file = f.read()
return Map.load_from_string(file)
@staticmethod
2020-10-09 16:24:13 +00:00
def load_from_string(content: str):
2020-10-16 16:05:49 +00:00
"""
Load a map represented by its characters and build a Map object.
"""
lines = content.split("\n")
2020-10-09 16:24:13 +00:00
lines = [line for line in lines if line]
height = len(lines)
2020-10-09 16:24:13 +00:00
width = len(lines[0])
2020-11-06 19:18:27 +00:00
tiles = [[Tile.from_ascii_char(c)
2020-10-16 13:41:25 +00:00
for x, c in enumerate(line)] for y, line in enumerate(lines)]
2020-10-23 13:55:30 +00:00
2020-11-06 16:59:19 +00:00
return Map(width, height, tiles)
2020-10-16 13:41:25 +00:00
2020-11-06 16:43:30 +00:00
def draw_string(self, pack: TexturePack) -> str:
2020-10-16 16:05:49 +00:00
"""
Draw the current map as a string object that can be rendered
in the window.
"""
2020-11-06 16:43:30 +00:00
return "\n".join("".join(tile.char(pack) for tile in line)
2020-10-16 14:41:38 +00:00
for line in self.tiles)
2020-10-16 13:41:25 +00:00
class Tile(Enum):
2020-11-06 16:43:30 +00:00
EMPTY = auto()
WALL = auto()
FLOOR = auto()
2020-11-06 19:18:27 +00:00
@classmethod
def from_ascii_char(cls, ch: str) -> "Tile":
for tile in Tile:
if tile.char(TexturePack.ASCII_PACK) == ch:
return tile
raise ValueError(ch)
2020-11-06 16:43:30 +00:00
def char(self, pack: TexturePack) -> str:
return getattr(pack, self.name)
2020-10-16 13:41:25 +00:00
def is_wall(self) -> bool:
return self == Tile.WALL
def can_walk(self) -> bool:
2020-10-16 16:05:49 +00:00
"""
Check if an entity (player or not) can move in this tile.
"""
return not self.is_wall()
2020-10-11 13:24:51 +00:00
class Entity:
2020-10-16 13:41:25 +00:00
y: int
2020-11-06 15:13:28 +00:00
x: int
2020-11-06 15:05:01 +00:00
name : str
2020-11-06 16:59:19 +00:00
map: Map
2020-11-06 15:13:28 +00:00
def __init__(self):
self.y = 0
self.x = 0
2020-11-06 16:59:19 +00:00
def check_move(self, y: int, x: int, move_if_possible: bool = False)\
-> bool:
tile = self.map.tiles[y][x]
if tile.can_walk() and move_if_possible:
self.move(y, x)
return tile.can_walk()
2020-11-06 15:13:28 +00:00
def move(self, y: int, x: int) -> None:
2020-10-11 13:24:51 +00:00
self.y = y
2020-11-06 15:13:28 +00:00
self.x = x
2020-11-06 14:33:26 +00:00
def act(self, m: Map) -> None:
"""
Define the action of the entity that is ran each tick.
By default, does nothing.
"""
2020-10-23 16:02:57 +00:00
pass
2020-11-06 14:33:26 +00:00
class FightingEntity(Entity):
maxhealth: int
health: int
strength: int
2020-11-06 15:13:28 +00:00
dead: bool
2020-11-06 17:08:10 +00:00
intelligence: int
charisma: int
dexterity: int
constitution: int
level: int
def __init__(self):
2020-11-06 15:13:28 +00:00
super().__init__()
self.health = self.maxhealth
2020-11-06 15:13:28 +00:00
self.dead = False
2020-11-06 14:33:26 +00:00
def hit(self, opponent: "FightingEntity") -> None:
opponent.take_damage(self, self.strength)
2020-11-06 14:33:26 +00:00
def take_damage(self, attacker: "Entity", amount: int) -> None:
self.health -= amount
if self.health <= 0:
self.die()
2020-11-06 14:33:26 +00:00
def die(self) -> None:
2020-11-06 15:13:28 +00:00
self.dead = True