SI_InteligentnyWozekWidlowy/ForkliftAgent.py

from typing import Tuple, List

from AgentBase import AgentBase
from data.enum.Direction import Direction
from decision.ActionType import ActionType


class ForkliftAgent(AgentBase):

    def __init__(self, model):
        super().__init__(model)
        self.movement_queue = []
        self.current_position = Tuple[int, int]
        self.current_rotation = Direction.right

    def queue_movement_actions(self, movement_actions: List[ActionType]):
        self.movement_queue.extend(movement_actions)

    def move(self):
        if len(self.movement_queue) > 0:
            action = self.movement_queue.pop(0)

            if action == ActionType.ROTATE_UP:
                print("rotate {} --> {}".format(self.current_rotation, action))
                self.current_rotation = Direction.top

            elif action == ActionType.ROTATE_RIGHT:
                print("rotate {} --> {}".format(self.current_rotation, action))
                self.current_rotation = Direction.right

            elif action == ActionType.ROTATE_DOWN:
                print("rotate {} --> {}".format(self.current_rotation, action))
                self.current_rotation = Direction.down

            elif action == ActionType.ROTATE_LEFT:
                print("rotate {} --> {}".format(self.current_rotation, action))
                self.current_rotation = Direction.left

            elif action == ActionType.MOVE:
                if self.current_rotation == Direction.top:
                    print("move {} --> {}".format(self.current_position, action))
                    self.current_position = (self.current_position[0], self.current_position[1] + 1)

                elif self.current_rotation == Direction.down:
                    print("move {} --> {}".format(self.current_position, action))
                    self.current_position = (self.current_position[0], self.current_position[1] - 1)

                elif self.current_rotation == Direction.right:
                    print("move {} --> {}".format(self.current_position, action))
                    self.current_position = (self.current_position[0] + 1, self.current_position[1])

                elif self.current_rotation == Direction.left:
                    print("move {} --> {}".format(self.current_position, action))
                    self.current_position = (self.current_position[0] - 1, self.current_position[1])

    def step(self) -> None:
        print("forklift step")
        self.move()

    def creation_log(self):
        print("Created Forklift Agent [id: {}]".format(self.unique_id))