synergine2/synergine2_xyz/physics.py

# coding: utf-8
import typing

from dijkstar import Graph
from dijkstar import find_path

from synergine2.config import Config
from synergine2.share import shared
from synergine2_xyz.map import TMXMap, XYZTile
from synergine2_xyz.subjects import XYZSubject
from synergine2_xyz.tmx_utils import fill_matrix
from synergine2_xyz.utils import get_line_xy_path
from synergine2_xyz.xyz import get_neighbor_positions


class MoveCostComputer(object):
    def __init__(
        self,
        config: Config,
    ) -> None:
        self.config = config

    def for_subject(self, subject: XYZSubject):
        # TODO: Verifier ce que sont les parametres pour les nommer correctement
        def move_cost_func(previous_node: str, next_node: str, tile: XYZTile, unknown):
            return self.compute_move_cost(subject, tile, previous_node, next_node, unknown)
        return move_cost_func

    def compute_move_cost(
        self,
        subject: XYZSubject,
        tile: XYZTile,
        previous_node: str,
        next_node: str,
        unknown,
    ) -> float:
        return 1.0


class Matrixes(object):
    _matrixes = shared.create('matrixes', value=lambda: {})  # type: typing.Dict[str, typing.List[typing.List[tuple]]]
    _value_structures = shared.create('value_structures', value=lambda: {})  # type: typing.List[str]

    def initialize_empty_matrix(
        self,
        name: str,
        matrix_width: int,
        matrix_height: int,
        value_structure: typing.List[str],
    ) -> None:
        self._matrixes[name] = []
        self._value_structures[name] = value_structure

        for y in range(matrix_height):
            x_list = []
            for x in range(matrix_width):
                x_list.append(tuple([0.0] * len(value_structure)))
            self._matrixes[name].append(x_list)

    def get_matrix(self, name: str) -> typing.List[typing.List[tuple]]:
        return self._matrixes[name]

    def update_matrix(self, name: str, x: int, y: int, value: tuple) -> None:
        matrix = self.get_matrix(name)
        matrix[y][x] = value
        # TODO: Test if working and needed ? This is not perf friendly ...
        # Force shared data update
        self._matrixes = dict(self._matrixes)

    def get_path_positions(
        self,
        from_: typing.Tuple[int, int],
        to: typing.Tuple[int, int],
    ) -> typing.List[typing.Tuple[int, int]]:
        return get_line_xy_path(from_, to)

    def get_values_for_path(
        self,
        name: str,
        path_positions: typing.List[typing.Tuple[int, int]],
        value_name: str=None,
    ):
        values = []
        value_name_position = None

        if value_name:
            value_name_position = self._value_structures[name].index(value_name)

        matrix = self.get_matrix(name)
        for path_position in path_positions:
            x, y = path_position
            if value_name_position is None:
                values.append(matrix[y][x])
            else:
                values.append(matrix[y][x][value_name_position])
        return values

    def get_value(self, matrix_name: str, x: int, y: int, value_name: str) -> float:
        matrix = self.get_matrix(matrix_name)
        values = matrix[y][x]
        value_position = self._value_structures[matrix_name].index(value_name)
        return values[value_position]


class Physics(object):
    visibility_matrix = Matrixes
    move_cost_computer_class = MoveCostComputer

    def __init__(
        self,
        config: Config,
    ) -> None:
        self.config = config
        self.graph = Graph()  # Graph of possible movements for dijkstar algorithm lib
        self.visibility_matrix = self.visibility_matrix()
        self.move_cost_computer = self.move_cost_computer_class(config)

    def load(self) -> None:
        pass

    def position_to_key(self, position: typing.Tuple[int, int]) -> str:
        return '{}.{}'.format(*position)

    def found_path(
        self,
        start: typing.Tuple[int, int],
        end: typing.Tuple[int, int],
        subject: XYZSubject,
    ) -> typing.List[typing.Tuple[int, int]]:
        start_key = self.position_to_key(start)
        end_key = self.position_to_key(end)

        found_path = find_path(self.graph, start_key, end_key, cost_func=self.move_cost_computer.for_subject(subject))
        regular_path = []
        for position in found_path[0][1:]:
            x, y = map(int, position.split('.'))
            regular_path.append((x, y))

        return regular_path


class TMXPhysics(Physics):
    tmx_map_class = TMXMap
    matrixes_configuration = None  # type: typing.Dict[str, typing.List[str]]

    def __init__(
        self,
        config: Config,
        map_file_path: str,
    ) -> None:
        super().__init__(config)
        self.map_file_path = map_file_path
        self.tmx_map = self.tmx_map_class(map_file_path)
        self.matrixes = Matrixes()

    def load(self) -> None:
        self.load_graph_from_map(self.map_file_path)
        self.load_matrixes_from_map()

    def load_graph_from_map(self, map_file_path: str) -> None:
        # TODO: tmx_map contient tout en cache, faire le dessous en exploitant tmx_map.
        for y in range(self.tmx_map.height):
            for x in range(self.tmx_map.width):
                position = self.position_to_key((x, y))

                for neighbor_position in get_neighbor_positions((x, y)):
                    neighbor = '{}.{}'.format(*neighbor_position)
                    neighbor_x, neighbor_y = neighbor_position

                    if neighbor_x > self.tmx_map.width-1 or neighbor_x < 0:
                        continue

                    if neighbor_y > self.tmx_map.height-1 or neighbor_y < 0:
                        continue

                    # Note: movement consider future tile properties
                    to_tile = self.tmx_map.layer_tiles('terrain')[neighbor]
                    # Note: Voir https://pypi.python.org/pypi/Dijkstar/2.2
                    self.graph.add_edge(position, neighbor, to_tile)

    def load_matrixes_from_map(self) -> None:
        if not self.matrixes_configuration:
            return

        for matrix_name, properties in self.matrixes_configuration.items():
            self.matrixes.initialize_empty_matrix(
                matrix_name,
                matrix_width=self.tmx_map.width,
                matrix_height=self.tmx_map.height,
                value_structure=properties,
            )
            fill_matrix(self.tmx_map, self.matrixes, 'terrain', matrix_name, properties)