intelligine/socialintengine/synergy/event/move/MoveAction.py

from synergine.synergy.event.Action import Action
from socialintengine.synergy.event.move.MoveEvent import MoveEvent
from random import randint, choice
from xyzworld.cst import POSITION, POSITIONS
from socialintengine.cst import IMPENETRABLE, PREVIOUS_DIRECTION, BLOCKED_SINCE
from synergine.synergy.Simulation import Simulation
from socialintengine.synergy.event.move.direction import directions_same_level, directions_modifiers, directions_slighty


class MoveAction(Action):

    _listen = MoveEvent

    def __init__(self, object_id, parameters):
      super().__init__(object_id, parameters)
      self._move_to_point = None
      self._move_to_direction = None

    def prepare(self, context):
      object_point = context.metas.value.get(POSITION, self._object_id)
      choosed_direction_name, choosed_direction_point = self._get_random_direction_point(context, object_point)
      if self._direction_point_is_possible(context, choosed_direction_point):
        self._move_to_point = choosed_direction_point
        self._move_to_direction = choosed_direction_name

    def _get_random_direction_point(self, context, reference_point):
        z, x, y = reference_point
        direction_name = self._get_random_direction_name(context)
        directions_modifier = directions_modifiers[direction_name]
        new_position = (z + directions_modifier[0], x + directions_modifier[1], y + directions_modifier[2])
        return (direction_name, new_position)

    def _get_random_direction_name(self, context):
        try:
            blocked_since = context.metas.value.get(BLOCKED_SINCE, self._object_id)
        except KeyError:
            blocked_since = 0
        direction_name = None
        if blocked_since <= 3:  #TODO: config
            try:
                previous_direction = context.metas.value.get(PREVIOUS_DIRECTION, self._object_id)
                directions_list = directions_slighty[previous_direction]
                direction_name = choice(directions_list)
            except KeyError:
                pass

        if not direction_name:
            direction_name = randint(directions_same_level[0], directions_same_level[1])

        return direction_name

    def _direction_point_is_possible(self, context, direction_point):
        objects_ids_on_this_point = context.metas.list.get(POSITIONS, direction_point, allow_empty=True)
        for object_id_on_this_point in objects_ids_on_this_point:
          if context.metas.list.have(Simulation.STATE, object_id_on_this_point, IMPENETRABLE):
            return False
        return True

    def run(self, obj, collection, context):
        if self._move_to_point is not None:
            obj.add_trace(self._move_to_point)
            context.metas.value.set(PREVIOUS_DIRECTION, self._object_id, self._move_to_direction)
            context.metas.value.set(BLOCKED_SINCE, self._object_id, 0)
        else:
            try:
                blocked_since = context.metas.value.get(BLOCKED_SINCE, self._object_id)
            except:
                blocked_since = 0
            context.metas.value.set(BLOCKED_SINCE, self._object_id, blocked_since+1)