squirrel-battle/squirrelbattle/entities/player.py

# Copyright (C) 2020 by ÿnérant, eichhornchen, nicomarg, charlse
# SPDX-License-Identifier: GPL-3.0-or-later

from functools import reduce
from queue import PriorityQueue
from random import randint
from typing import Dict, Optional, Tuple

from .items import Item
from ..interfaces import FightingEntity, InventoryHolder


class Player(InventoryHolder, FightingEntity):
    """
    The class of the player
    """
    current_xp: int = 0
    max_xp: int = 10
    paths: Dict[Tuple[int, int], Tuple[int, int]]
    equipped_main: Optional[Item]
    equipped_secondary: Optional[Item]
    equipped_helmet: Optional[Item]
    equipped_armor: Optional[Item]

    def __init__(self, name: str = "player", maxhealth: int = 20,
                 strength: int = 5, intelligence: int = 1, charisma: int = 1,
                 dexterity: int = 1, constitution: int = 1, level: int = 1,
                 current_xp: int = 0, max_xp: int = 10, inventory: list = None,
                 hazel: int = 42, equipped_main: Optional[Item] = None,
                 equipped_armor: Optional[Item] = None, critical: int = 5,\
                 equipped_secondary: Optional[Item] = None, \
                 equipped_helmet: Optional[Item] = None, \
                 *args, **kwargs) -> None:
        super().__init__(name=name, maxhealth=maxhealth, strength=strength,
                         intelligence=intelligence, charisma=charisma,
                         dexterity=dexterity, constitution=constitution,
                         level=level, critical=critical, *args, **kwargs)
        self.current_xp = current_xp
        self.max_xp = max_xp
        self.inventory = self.translate_inventory(inventory or [])
        self.paths = dict()
        self.hazel = hazel
        if isinstance(equipped_main, dict):
            equipped_main = self.dict_to_item(equipped_main)
        if isinstance(equipped_armor, dict):
            equipped_armor = self.dict_to_item(equipped_armor)
        if isinstance(equipped_secondary, dict):
            equipped_secondary = self.dict_to_item(equipped_secondary)
        if isinstance(equipped_helmet, dict):
            equipped_helmet = self.dict_to_item(equipped_helmet)
        self.equipped_main = equipped_main
        self.equipped_armor = equipped_armor
        self.equipped_secondary = equipped_secondary
        self.equipped_helmet = equipped_helmet

    def move(self, y: int, x: int) -> None:
        """
        Moves the view of the map (the point on which the camera is centered)
        according to the moves of the player.
        """
        super().move(y, x)
        self.map.currenty = y
        self.map.currentx = x
        self.recalculate_paths()

    def level_up(self) -> None:
        """
        Add levels to the player as much as it is possible.
        """
        while self.current_xp > self.max_xp:
            self.level += 1
            self.current_xp -= self.max_xp
            self.max_xp = self.level * 10
            self.health = self.maxhealth
            self.strength = self.strength + 1
            # TODO Remove it, that's only fun
            self.map.spawn_random_entities(randint(3 * self.level,
                                                   10 * self.level))

    def add_xp(self, xp: int) -> None:
        """
        Add some experience to the player.
        If the required amount is reached, level up.
        """
        self.current_xp += xp
        self.level_up()

    def remove_from_inventory(self, obj: Item) -> None:
        """
        Remove the given item from the inventory, even if the item is equipped.
        """
        if obj == self.equipped_main:
            self.equipped_main = None
        elif obj == self.equipped_armor:
            self.equipped_armor = None
        elif obj == self.equipped_secondary:
            self.equipped_secondary = None
        elif obj == self.equipped_helmet:
            self.equipped_helmet = None
        else:
            return super().remove_from_inventory(obj)

    # noinspection PyTypeChecker,PyUnresolvedReferences
    def check_move(self, y: int, x: int, move_if_possible: bool = False) \
            -> bool:
        """
        If the player tries to move but a fighting entity is there,
        the player fights this entity.
        If the entity dies, the player is rewarded with some XP
        """
        # Don't move if we are dead
        if self.dead:
            return False
        for entity in self.map.entities:
            if entity.y == y and entity.x == x:
                if entity.is_fighting_entity():
                    self.map.logs.add_message(self.hit(entity))
                    if entity.dead:
                        self.add_xp(randint(3, 7))
                    return True
                elif entity.is_item():
                    entity.hold(self)
        return super().check_move(y, x, move_if_possible)

    def recalculate_paths(self, max_distance: int = 8) -> None:
        """
        Use Dijkstra algorithm to calculate best paths for monsters to go to
        the player. Actually, the paths are computed for each tile adjacent to
        the player then for each step the monsters use the best path avaliable.
        """
        distances = []
        predecessors = []
        # four Dijkstras, one for each adjacent tile
        for dir_y, dir_x in [(1, 0), (-1, 0), (0, 1), (0, -1)]:
            queue = PriorityQueue()
            new_y, new_x = self.y + dir_y, self.x + dir_x
            if not 0 <= new_y < self.map.height or \
                    not 0 <= new_x < self.map.width or \
                    not self.map.tiles[new_y][new_x].can_walk():
                continue
            queue.put(((1, 0), (new_y, new_x)))
            visited = [(self.y, self.x)]
            distances.append({(self.y, self.x): (0, 0), (new_y, new_x): (1, 0)})
            predecessors.append({(new_y, new_x): (self.y, self.x)})
            while not queue.empty():
                dist, (y, x) = queue.get()
                if dist[0] >= max_distance or (y, x) in visited:
                    continue
                visited.append((y, x))
                for diff_y, diff_x in [(1, 0), (-1, 0), (0, 1), (0, -1)]:
                    new_y, new_x = y + diff_y, x + diff_x
                    if not 0 <= new_y < self.map.height or \
                            not 0 <= new_x < self.map.width or \
                            not self.map.tiles[new_y][new_x].can_walk():
                        continue
                    new_distance = (dist[0] + 1,
                                    dist[1] + (not self.map.is_free(y, x)))
                    if not (new_y, new_x) in distances[-1] or \
                            distances[-1][(new_y, new_x)] > new_distance:
                        predecessors[-1][(new_y, new_x)] = (y, x)
                        distances[-1][(new_y, new_x)] = new_distance
                        queue.put((new_distance, (new_y, new_x)))
        # For each tile that is reached by at least one Dijkstra, sort the
        # different paths by distance to the player. For the technical bits :
        # The reduce function is a fold starting on the first element of the
        # iterable, and we associate the points to their distance, sort
        # along the distance, then only keep the points.
        self.paths = {}
        for y, x in reduce(set.union,
                           [set(p.keys()) for p in predecessors], set()):
            self.paths[(y, x)] = [p for d, p in sorted(
                [(distances[i][(y, x)], predecessors[i][(y, x)])
                 for i in range(len(distances)) if (y, x) in predecessors[i]])]

    def save_state(self) -> dict:
        """
        Saves the state of the entity into a dictionary
        """
        d = super().save_state()
        d["current_xp"] = self.current_xp
        d["max_xp"] = self.max_xp
        d["equipped_main"] = self.equipped_main.save_state()\
            if self.equipped_main else None
        d["equipped_armor"] = self.equipped_armor.save_state()\
            if self.equipped_armor else None
        d["equipped_secondary"] = self.equipped_secondary.save_state()\
            if self.equipped_secondary else None
        d["equipped_helmet"] = self.equipped_helmet.save_state()\
            if self.equipped_helmet else None
        return d