logic.py

#!/usr/bin/python3

from point import Direction, Point
from collections import defaultdict


CELL_SIZE = 16
EXPLOSION_LIVE = 10
TIME_EXPLOSION_DELAY = 100
START_EXPLOSION_RADIUS = 2
CORNER_SIZE = 8
BUFF_TIME = 1000


class OutOfMapRangeException(Exception):
    pass


class Map:

    def __init__(self):
        self._objects = defaultdict(list)
        self._sorted_points_by_x = []
        self._sorted_points_by_y = []

    def add_map_object(self, map_object: 'MapObject', coordinates: 'Point'):
        self._objects[coordinates].append(map_object)
        self._sorted_points_by_x = None
        self._sorted_points_by_y = None

    def get_map_objects(self, coordinates: Point):
        if coordinates not in self._objects:
            return []
        return self._objects[coordinates]

    @staticmethod
    def round(number: int, cell_width: int) -> int:
        modulo = number % cell_width
        if modulo < cell_width / 2:
            return number - modulo
        else:
            return number + cell_width - modulo

    @staticmethod
    def round_point(point: Point, cell_width: int):
        return Point(Map.round(point.x, cell_width),
                     Map.round(point.y, cell_width))

    def add_bomb(self, bomb: 'Bomb', coordinates: 'Point'):
        self.add_map_object(bomb, Map.round_point(coordinates, CELL_SIZE))

    def enumerate_all_objects(self):
        for point in self._objects.keys():
            for game_object in self.get_map_objects(point):
                yield point, game_object

    @staticmethod
    def are_intersected(first_coordinates: Point,
                        second_coordinates: Point):

        return abs(first_coordinates.x - second_coordinates.x) < \
               CELL_SIZE and \
               abs(first_coordinates.y - second_coordinates.y) < \
               CELL_SIZE

    def get_collisions(self, coordinates: 'Point'):
        self._check_sorted_points()
        set_x = self._find_intersections(coordinates, self._sorted_points_by_x,
                                         lambda point: point.x, CELL_SIZE)
        set_y = self._find_intersections(coordinates, self._sorted_points_by_y,
                                         lambda point: point.y, CELL_SIZE)
        collisions = []
        for point in set_x.intersection(set_y):
            for object in self.get_map_objects(point):
                collisions.append(object)
        return collisions

    def _check_sorted_points(self):
        if self._sorted_points_by_x is None:
            self._sorted_points_by_x = list(self._objects.keys())
            self._sorted_points_by_x.sort(key=lambda point: point.x)
        if self._sorted_points_by_y is None:
            self._sorted_points_by_y = list(self._objects.keys())
            self._sorted_points_by_y.sort(key=lambda point: point.y)

    def _find_intersections(self, point, sorted_list, coordinates_func,
                            cell_width):

        def compare_func(point1, point2):
            return coordinates_func(point1) - coordinates_func(point2)

        point_index = self._binary_search(point, sorted_list, compare_func)
        collisions_set = set()
        for i in range(point_index, len(sorted_list)):
            if abs(compare_func(point, sorted_list[i])) >= cell_width:
                break
            collisions_set.add(sorted_list[i])

        for i in range(point_index - 1, -1, -1):
            if abs(compare_func(point, sorted_list[i])) >= cell_width:
                break
            collisions_set.add(sorted_list[i])
        return collisions_set

    def _binary_search(self, point, sorted_list, compare_function):
        first = 0
        last = len(sorted_list) - 1
        while first != last:
            middle = (first + last) // 2
            if compare_function(point, sorted_list[middle]) <= 0:
                last = middle
            else:
                first = middle + 1
        return first

    @property
    def occupied_cells(self) -> set:
        return set(self._objects.keys())


class Game:

    def __init__(self, game_map: 'Map', player: 'Player'):
        self._map = game_map
        self._player = player

    @property
    def monster_count(self):
        monster_count = 0
        for cell in self._map.occupied_cells:
            for obj in self._map.get_map_objects(cell):
                if isinstance(obj, Monster):
                    monster_count += 1
        return monster_count

    @property
    def player(self) -> 'Player':
        return self._player

    @property
    def map(self):
        return self._map

    def set_map(self, map):
        self._map = map

    def make_turn(self):
        animations = []
        intermediate_map = Map()
        for point in self._map.occupied_cells:
            for game_object in self._map.get_map_objects(point):
                action = game_object.move(point, self._map)
                new_animations = action.change_game_state(
                    game_object, point, self._map, intermediate_map
                )
                for animation in new_animations:
                    animations.append(animation)

        game_map = Map()
        for point in intermediate_map.occupied_cells:
            for game_object in intermediate_map.get_map_objects(point):
                other_objects = intermediate_map.get_collisions(point)
                game_object.solve_collision(other_objects)
                if not game_object.is_dead:
                    game_map.add_map_object(game_object, point)

        self._map = game_map
        return animations


class Animation:

    def __init__(self, object, location, direction):
        self.object = object
        self.location = location
        self.direction = direction


class Move:

    def __init__(self, direction):
        self._direction = direction

    @property
    def direction(self):
        return self._direction

    def change_game_state(self, game_object, coordinates, old_map, game_map):
        collisions = old_map.get_collisions(coordinates +
                                            self.direction.value)
        old_collisions = old_map.get_collisions(coordinates)
        if game_object.can_move(collisions, old_collisions):
            game_map.add_map_object(game_object,
                                    coordinates + self.direction.value)
            return Animation(game_object, coordinates, self.direction),
        else:
            directions = (Direction.Up, Direction.Down) \
                    if self.direction in (Direction.Right, Direction.Left) \
                    else (Direction.Right, Direction.Left)
            for new_direction in directions:
                for i in range(1, CORNER_SIZE + 1):
                    point = i * new_direction.value + self.direction.value
                    collisions = old_map.get_collisions(coordinates + point)
                    if game_object.can_move(collisions, old_collisions):
                        game_map.add_map_object(game_object,
                                                coordinates +
                                                new_direction.value)
                        return Animation(game_object, coordinates,
                                         new_direction),
        game_map.add_map_object(game_object, coordinates)
        return Animation(game_object, coordinates, Direction.Stand),


class PutBomb:

    def __init__(self, bomb):
        self._bomb = bomb

    @property
    def bomb(self):
        return self._bomb

    def change_game_state(self, game_object, coordinates, old_map, game_map):
        animations = []
        point = Map.round_point(coordinates, CELL_SIZE)
        collisions = old_map.get_collisions(point)
        if collisions == [game_object]:
            game_map.add_bomb(self.bomb, coordinates)
            game_object.set_last_bomb(self.bomb)
            animations.append(Animation(self.bomb, coordinates,
                                        Direction.Stand))
        game_map.add_map_object(game_object, coordinates)
        animations.append(Animation(game_object, coordinates,
                                    Direction.Stand))
        return animations


class Explose:

    def __init__(self, explosion_type, center: 'Point', radius: int):
        self._explosion_type = explosion_type
        self._center = center
        self._radius = radius

    def change_game_state(self, game_object, coordinates, old_map, game_map):
        animations = []
        stop_explosion_classes = (Block, Bomb, Bonus)
        game_map.add_map_object(ExplosionBlock(EXPLOSION_LIVE), coordinates)
        for direction in (Direction.Right, Direction.Left,
                          Direction.Up, Direction.Down):
            for i in range(1, self._radius):
                point = Point(coordinates.x +
                              i * direction.value.x * CELL_SIZE,
                              coordinates.y +
                              i * direction.value.y * CELL_SIZE)
                explosion = self._explosion_type(EXPLOSION_LIVE, direction)
                animations.append(Animation(explosion, point, Direction.Stand))
                game_map.add_map_object(explosion, point)
                collisions = old_map.get_collisions(point)
                if any(isinstance(game_object, class_)
                       for game_object in collisions
                       for class_ in stop_explosion_classes):
                    break
        return animations


class MapObject:

    def __init__(self):
        self._is_dead = False

    def can_move(self, collisions, old_collisions):
        return True

    def move(self, coordinates: 'Point', old_map: 'Map'):
        return Move(Direction.Stand)

    def solve_collision(self, other_objects):
        pass

    @property
    def is_dead(self):
        return self._is_dead

    @is_dead.setter
    def is_dead(self, is_dead):
        self._is_dead = is_dead


class BombCreator:

    def __init__(self):
        self._radius = START_EXPLOSION_RADIUS
        self._bomb_type = Bomb

    def set_radius(self, radius):
        self._radius = radius

    def set_bomb_type(self, bomb_type):
        self._bomb_type = bomb_type

    def get_bomb(self):
        return self._bomb_type(TIME_EXPLOSION_DELAY, self._radius)


class Player(MapObject):

    def __init__(self, controller):
        super().__init__()
        self._controller = controller
        controller.set_player(self)
        self._last_bomb = None
        self._bomb_creator = BombCreator()
        self._buffs = []
        self.immune = False
        self._ghost_mode = False

    def set_ghost_mode(self, value):
        self._ghost_mode = value

    def set_controller(self, controller):
        self._controller = controller
        controller.set_player(self)

    def get_bomb(self):
        return self._bomb_creator.get_bomb()

    def set_bomb_type(self, bomb_type):
        self._bomb_creator.set_bomb_type(bomb_type)

    def set_bomb_radius(self, radius):
        self._bomb_creator.set_radius(radius)

    def set_last_bomb(self, bomb):
        self._last_bomb = bomb

    def add_buff(self, buff):
        buff.start(self)
        self._buffs.append(buff)

    def move(self, coordinates: 'Point', old_map: 'Map'):

        for buff in self._buffs:
            buff.time -= 1
            if buff.time == 0:
                buff.end(self)

        self._buffs = [buff for buff in self._buffs if buff.time > 0]
        return self._controller.select_action()

    def can_move(self, collisions, old_collisions):
        if self._ghost_mode:
            return True
        object_types = [Block, Bomb]
        return all(not isinstance(object_, type_)
                   for object_ in collisions
                   for type_ in object_types
                   if not (object_ is self._last_bomb and
                           self._last_bomb in old_collisions))

    def solve_collision(self, other_objects: list):
        object_types = [Monster, ExplosionBlock]
        if any(isinstance(object_, type_)
               for object_ in other_objects
               for type_ in object_types) and not self.immune:
            self._is_dead = True


class Monster(MapObject):

    def solve_collision(self, other_objects):
        if any(isinstance(object_, ExplosionBlock)
               for object_ in other_objects):
            self._is_dead = True


class Block(MapObject):
    pass


class Bomb(MapObject):

    def __init__(self, tick_delay: int, explosion_radius: int):
        super().__init__()
        self._tick_delay = tick_delay
        self._explosion_radius = explosion_radius
        self._explosion_type = ExplosionBlock

    def move(self, coordinates: 'Point', old_map: 'Map'):
        if ExplosionBlock in map(type, old_map.get_collisions(coordinates)):
            return Explose(self._explosion_type, coordinates,
                           self._explosion_radius)
        self._tick_delay -= 1
        if self._tick_delay == 0:
            self._is_dead = True
            return Explose(self._explosion_type, coordinates,
                           self._explosion_radius)
        return Move(Direction.Stand)


class ExplosionBlock(MapObject):

    def __init__(self, life_time: int, direction=Direction.Stand):
        super().__init__()
        self._life_time = life_time
        self._direction = direction

    @property
    def life_time(self):
        return self._life_time

    @property
    def direction(self):
        return self._direction

    def solve_collision(self, other_objects):
        self._life_time -= 1
        if self._life_time == 0:
            self._is_dead = True


class Bonus(MapObject):

    def solve_collision(self, other_objects):
        if any(isinstance(object_, Player) for object_ in other_objects):
            self._is_dead = True
            player = next(obj for obj in other_objects
                          if isinstance(obj, Player))
            self.add_bonus(player)

    def add_bonus(self, player):
        pass


class Buff:

    def __init__(self):
        self.time = BUFF_TIME

    def start(self, player):
        pass

    def end(self, player):
        pass