AI-Project/survival/systems/pathfinding_movement_system.py

from survival import esper
from survival.components.direction_component import DirectionChangeComponent
from survival.components.movement_component import MovementComponent
from survival.components.moving_component import MovingComponent
from survival.components.position_component import PositionComponent
from survival.enums import Direction
from survival.graph_search import graph_search, Action
from survival.systems.input_system import PathfindingComponent


class CollectingResourceComponent:
    def __init__(self, action):
        self.action = action


class PathfindingMovementSystem(esper.Processor):
    def __init__(self, game_map):
        self.game_map = game_map

    def process(self, dt):
        for ent, (pos, pathfinding, movement) in self.world.get_components(PositionComponent, PathfindingComponent,
                                                                           MovementComponent):
            if pathfinding.path is None:
                pathfinding.path = graph_search(self.game_map, pos, pathfinding.target_grid_pos, self.world)

            if len(pathfinding.path) < 1:
                self.world.remove_component(ent, PathfindingComponent)
                continue

            if pathfinding.searching_for_resource and len(pathfinding.path) == 1:
                self.world.add_component(ent, CollectingResourceComponent(pathfinding.path.pop(0)))
                self.world.remove_component(ent, PathfindingComponent)
                continue

            if self.world.has_component(ent, MovingComponent) or self.world.has_component(ent, DirectionChangeComponent):
                continue

            action = pathfinding.path.pop(0)

            if action == Action.ROTATE_LEFT:
                self.world.add_component(ent, DirectionChangeComponent(Direction.rotate_left(pos.direction)))
            elif action == Action.ROTATE_RIGHT:
                self.world.add_component(ent, DirectionChangeComponent(Direction.rotate_right(pos.direction)))
            else:
                self.world.add_component(ent, MovingComponent())

            # if pathfinding.path is None:
            #     pathfinding.path = breadth_first_search(self.game_map, pos.grid_position, pathfinding.target_grid_pos)
            #
            # if len(pathfinding.path) < 1 and pathfinding.current_target is None:
            #     self.world.remove_component(ent, PathfindingComponent)
            #     continue
            #
            # if self.world.has_component(ent, MovingComponent):
            #     continue
            #
            # if pathfinding.current_target is None:
            #     target = pathfinding.path.pop(0)
            # else:
            #     target = pathfinding.current_target
            #
            # vector = (target[0] - pos.grid_position[0], target[1] - pos.grid_position[1])
            # direction = Direction.from_vector(vector)
            # if direction != pos.direction:
            #     pathfinding.current_target = target
            #     self.world.add_component(ent, DirectionChangeComponent(direction))
            #     continue
            #
            # pathfinding.current_target = None
            # self.world.add_component(ent, MovingComponent())