SI_2020/agent.py

from warehouse import Coordinates, Tile


class Agent:
    def __init__(self, start_x, start_y, assigned_warehouse, radius=5):
        self.x = start_x
        self.y = start_y
        self.direction = 'down'
        self.radius = radius
        self.assigned_warehouse = assigned_warehouse
        self.is_loaded = False


    def move(self):
        next_coords = self.get_next_move_coordinates()
        can_move = self.check_if_can_move(next_coords)
        if can_move:
            self.x = next_coords.x
            self.y = next_coords.y

    def get_next_move_coordinates(self):
        direction_moves = {
            'up': self.move_up(),
            'down': self.move_down(),
            'left': self.move_left(),
            'right': self.move_right()
        }
        next_coords = direction_moves.get(self.direction, Coordinates(self.x, self.y))
        return next_coords

    def check_if_can_move(self, next_coords: Coordinates):
        # import pdb
        # pdb.set_trace()
        next_tile = self.assigned_warehouse.tiles[next_coords.x][next_coords.y]
        tile_passable = isinstance(next_tile, Tile) and next_tile.category.passable
        return tile_passable

    def move_right(self):
        pos_x = self.x + 1
        if pos_x > self.assigned_warehouse.width - 1:
            pos_x = self.assigned_warehouse.width - 1
        return Coordinates(x=pos_x, y=self.y)

    def move_left(self):
        pos_x = self.x - 1
        if pos_x < 0:
            pos_x = 0
        return Coordinates(x=pos_x, y=self.y)

    def move_down(self):
        pos_y = self.y + 1
        if pos_y > self.assigned_warehouse.height - 1:
            pos_y = self.assigned_warehouse.height - 1
        return Coordinates(x=self.x, y=pos_y)

    def move_up(self):
        pos_y = self.y - 1
        if pos_y < 0:
            pos_y = 0
        return Coordinates(x=self.x, y=pos_y)