64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
|
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)
|
||
|
|
|