1
0
python-roguelike/actions.py

183 lines
5.6 KiB
Python
Raw Normal View History

2022-01-06 11:56:08 -05:00
from __future__ import annotations
2022-01-11 16:08:57 -05:00
from typing import overload, Optional, Tuple, TYPE_CHECKING
import color
import exceptions
2022-01-06 11:56:08 -05:00
if TYPE_CHECKING:
from engine import Engine
from entity import Actor, Entity, Item
2022-01-06 11:56:08 -05:00
2022-01-07 14:18:47 -05:00
2022-01-06 11:09:43 -05:00
class Action:
def __init__(self, entity: Actor) -> None:
super().__init__()
self.entity = entity
@property
def engine(self) -> Engine:
"""Return the engine this action belongs to."""
return self.entity.gamemap.engine
2022-01-07 16:48:49 -05:00
@overload
def perform(self) -> None:
2022-01-06 11:56:08 -05:00
"""Perform this action with the objects needed to determine its scope.
`self.engine` is the scope this action is being performed in.
2022-01-06 11:56:08 -05:00
`self.entity` is the object performing the action.
2022-01-06 11:56:08 -05:00
This method must be overwritten by Action subclasses.
"""
2022-01-06 11:09:43 -05:00
class PickupAction(Action):
"""Pickup an item and add it to the inventory, if there is room for it."""
def __init__(self, entity: Actor):
super().__init__(entity)
def perform(self) -> None:
actor_location_x = self.entity.x
actor_location_y = self.entity.y
inventory = self.entity.inventory
for item in self.engine.game_map.items:
if actor_location_x == item.x and actor_location_y == item.y:
if len(inventory.items) >= inventory.capacity:
raise exceptions.Impossible("Your inventory is full.")
self.engine.game_map.entities.remove(item)
item.parent = self.entity.inventory
inventory.items.append(item)
self.engine.message_log.add_message(f"You picked up the {item.name}!")
return
raise exceptions.Impossible("There is nothing here to pick up.")
class ItemAction(Action):
def __init__(
self,
entity: Actor,
item: Item,
target_xy: Optional[Tuple[int, int]] = None
):
super().__init__(entity)
self.item = item
if not target_xy:
target_xy = entity.x, entity.y
self.target_xy = target_xy
@property
def target_actor(self) -> Optional[Actor]:
"""Return the actor at this action's destination."""
return self.engine.game_map.get_actor_at_location(*self.target_xy)
def perform(self) -> None:
"""Invoke the item's ability, this action will be given to provide context."""
self.item.consumable.activate(self)
class DropItem(ItemAction):
def perform(self) -> None:
self.entity.inventory.drop(self.item)
2022-01-10 13:21:17 -05:00
class WaitAction(Action):
def perform(self) -> None:
pass
class TakeStairsAction(Action):
def perform(self) -> None:
"""
Take the stairs, if any exist at the entity's location.
"""
if (self.entity.x, self.entity.y) == self.engine.game_map.downstairs_location:
self.engine.game_world.generate_floor()
self.engine.message_log.add_message(
"You descent the staircase.",
color.descend
)
else:
raise exceptions.Impossible("There are no stairs here.")
2022-01-07 15:52:53 -05:00
class ActionWithDirection(Action):
def __init__(self, entity: Actor, dx: int, dy: int):
super().__init__(entity)
2022-01-06 11:09:43 -05:00
self.dx = dx
self.dy = dy
2022-01-06 11:56:08 -05:00
@property
def dest_xy(self) -> Tuple[int, int]:
"""Returns this action's destination."""
return self.entity.x + self.dx, self.entity.y + self.dy
@property
def blocking_entity(self) -> Optional[Entity]:
"""Return the blocking entity at this action's destination."""
return self.engine.game_map.get_blocking_entity_at_location(*self.dest_xy)
@property
def target_actor(self) -> Optional[Actor]:
"""Return the actor at this action's destination."""
return self.engine.game_map.get_actor_at_location(*self.dest_xy)
2022-01-07 15:52:53 -05:00
class MeleeAction(ActionWithDirection):
def perform(self) -> None:
target = self.target_actor
2022-01-07 15:52:53 -05:00
if not target:
raise exceptions.Impossible("Nothing to attack.")
2022-01-07 15:52:53 -05:00
damage = self.entity.fighter.power - target.fighter.defense
attack_desc = f"{self.entity.name.capitalize()} attacks {target.name}"
2022-01-11 16:08:57 -05:00
if self.entity is self.engine.player:
attack_color = color.player_atk
else:
attack_color = color.enemy_atk
if damage > 0:
2022-01-11 16:08:57 -05:00
self.engine.message_log.add_message(
f"{attack_desc} for {damage} hit points.",
attack_color,
)
target.fighter.hp -= damage
else:
2022-01-11 16:08:57 -05:00
self.engine.message_log.add_message(
f"{attack_desc} but does no damage.",
attack_color,
)
2022-01-07 15:52:53 -05:00
class MovementAction(ActionWithDirection):
def perform(self) -> None:
dest_x, dest_y = self.dest_xy
2022-01-06 11:56:08 -05:00
if not self.engine.game_map.in_bounds(dest_x, dest_y):
# Destination is out of bounds
raise exceptions.Impossible("That way is blocked.")
if not self.engine.game_map.tiles["walkable"][dest_x, dest_y]:
# Destination is blocked by a tile.
raise exceptions.Impossible("That way is blocked.")
if self.engine.game_map.get_blocking_entity_at_location(dest_x, dest_y):
# Destination is blocked by an entity.
raise exceptions.Impossible("That way is blocked.")
2022-01-06 11:56:08 -05:00
self.entity.move(self.dx, self.dy)
2022-01-07 15:52:53 -05:00
class BumpAction(ActionWithDirection):
def perform(self) -> None:
if self.target_actor:
return MeleeAction(self.entity, self.dx, self.dy).perform()
2022-01-07 15:52:53 -05:00
else:
return MovementAction(self.entity, self.dx, self.dy).perform()