import pygame


class Agent:

    def __init__(self, screen, position_x, position_y, turn):
        self.screen = screen
        self.position_x = position_x
        self.position_y = position_y
        self.turn = turn
        self.x = (self.position_x - 50) // 100
        self.y = (self.position_y - 50) // 100

        self.image = pygame.image.load('img/' + turn + '.png')
        self.rect = self.image.get_rect()
        self.screen_rect = screen.get_rect()

        self.rect.center = (position_x, position_y)

    def blitme(self):
        self.screen.blit(self.image, self.rect)

    def turn_left(self):
        if self.turn == "Up":
            self.turn = "Left"
        elif self.turn == "Left":
            self.turn = "Down"
        elif self.turn == "Down":
            self.turn = "Right"
        elif self.turn == "Right":
            self.turn = "Up"
        self.image = pygame.image.load('img/' + self.turn + '.png')

    def turn_right(self):
        if self.turn == "Up":
            self.turn = "Right"
        elif self.turn == "Left":
            self.turn = "Up"
        elif self.turn == "Down":
            self.turn = "Left"
        elif self.turn == "Right":
            self.turn = "Down"
        self.image = pygame.image.load('img/' + self.turn + '.png')

    def move_forward(self, board):
        if self.turn == "Up" and self.position_y > 50 and board[(self.position_y - 100) // 100][self.position_x // 100]. \
                is_shelf == False:
            self.position_y -= 100
            self.y -= 1
        elif self.turn == "Left" and self.position_x > 50 and board[self.position_y // 100][(self.position_x - 100) // 100].\
                is_shelf == False:
            self.position_x -= 100
            self.x -= 1
        elif self.turn == "Down" and self.position_y < 950 and board[(self.position_y + 100) // 100][self.position_x // 100]. \
                is_shelf == False:
            self.position_y += 100
            self.y += 1
        elif self.turn == "Right" and self.position_x < 950 and board[self.position_y // 100][(self.position_x + 100) // 100].\
                is_shelf == False:
            self.position_x += 100
            self.x += 1
        self.rect.center = (self.position_x, self.position_y)