2020-04-02 15:25:15 +02:00
|
|
|
import pygame
|
|
|
|
import warehouse
|
2020-04-04 22:03:15 +02:00
|
|
|
import agent
|
|
|
|
import random
|
2020-04-06 15:18:00 +02:00
|
|
|
import sys
|
2020-04-30 16:36:55 +02:00
|
|
|
from attributes import PackSize, PackStatus, COLORS, DIRECTION_ANGLES
|
2020-04-02 15:25:15 +02:00
|
|
|
|
2020-04-30 16:36:55 +02:00
|
|
|
WINDOW_SIZE = (640, 640)
|
2020-04-04 22:03:15 +02:00
|
|
|
COLOR_OF_FIELD = {
|
|
|
|
'Floor': 'gray',
|
|
|
|
'Rack': 'white',
|
2020-04-28 22:03:48 +02:00
|
|
|
'Pack': 'yellow',
|
|
|
|
'path': 'orange'
|
2020-04-04 22:03:15 +02:00
|
|
|
}
|
2020-04-30 16:36:55 +02:00
|
|
|
TILE_WIDTH = 32
|
|
|
|
TILE_HEIGHT = 32
|
2020-04-04 22:03:15 +02:00
|
|
|
CIRCLE_CENTER_X, CIRCLE_CENTER_Y = int(TILE_WIDTH/2), int(TILE_HEIGHT/2)
|
2020-04-02 15:25:15 +02:00
|
|
|
|
2020-04-04 22:03:15 +02:00
|
|
|
class MainGameFrame:
|
|
|
|
def __init__(self):
|
|
|
|
self.display = pygame.display.set_mode(WINDOW_SIZE)
|
2020-04-06 16:11:57 +02:00
|
|
|
agent_radius = int(TILE_WIDTH/2)
|
2020-04-30 16:36:55 +02:00
|
|
|
self.agent_tex = pygame.image.load('forklift.png')
|
2020-05-07 17:39:00 +02:00
|
|
|
self.warehouse_map = warehouse.Warehouse(20, 20, 150, 10)
|
2020-04-05 15:04:38 +02:00
|
|
|
starting_x, starting_y = self.set_starting_agent_position()
|
2020-04-06 16:11:57 +02:00
|
|
|
self.agent = agent.Agent(starting_x, starting_y, self.warehouse_map, agent_radius)
|
2020-04-07 01:18:04 +02:00
|
|
|
self.clock = pygame.time.Clock()
|
2020-04-07 18:50:25 +02:00
|
|
|
|
2020-04-04 22:03:15 +02:00
|
|
|
def run(self):
|
|
|
|
while True:
|
2020-04-06 15:18:00 +02:00
|
|
|
for event in pygame.event.get():
|
|
|
|
if event.type == pygame.QUIT:
|
|
|
|
pygame.quit()
|
|
|
|
sys.exit()
|
|
|
|
self.draw_floor()
|
|
|
|
self.draw_packages()
|
|
|
|
self.draw_agent()
|
2020-04-27 22:14:07 +02:00
|
|
|
self.agent.move()
|
2020-04-04 22:03:15 +02:00
|
|
|
pygame.display.update()
|
2020-04-28 23:12:57 +02:00
|
|
|
self.clock.tick(5)
|
2020-04-02 15:25:15 +02:00
|
|
|
|
2020-04-04 22:03:15 +02:00
|
|
|
def draw_floor(self):
|
|
|
|
for x in range(self.warehouse_map.width):
|
|
|
|
for y in range(self.warehouse_map.height):
|
2020-04-05 15:04:38 +02:00
|
|
|
self.draw_field(x, y)
|
2020-04-28 01:18:23 +02:00
|
|
|
self.draw_target(1, 1)
|
|
|
|
|
|
|
|
def draw_target(self, x, y):
|
|
|
|
target_screen_position = (
|
|
|
|
(x * TILE_WIDTH) + CIRCLE_CENTER_X, (y * TILE_HEIGHT) + CIRCLE_CENTER_Y)
|
|
|
|
pygame.draw.circle(self.display, COLORS['lightblue'], target_screen_position, 6)
|
2020-04-04 22:03:15 +02:00
|
|
|
|
|
|
|
def draw_field(self, x, y):
|
|
|
|
current_tile = self.warehouse_map.tiles[x][y]
|
2020-04-28 22:03:48 +02:00
|
|
|
# if not isinstance(current_tile, warehouse.Tile):
|
|
|
|
# current_tile = current_tile.lays_on_field if isinstance(current_tile, warehouse.Pack) else None
|
2020-04-04 22:03:15 +02:00
|
|
|
color = COLOR_OF_FIELD.get(current_tile.category.name, 'white')
|
|
|
|
color = COLORS[color]
|
2020-04-28 22:03:48 +02:00
|
|
|
if (current_tile.x_position,current_tile.y_position) in [(a.x, a.y) for a in self.agent.path]:
|
|
|
|
color = COLORS.get('orange')
|
2020-04-04 22:03:15 +02:00
|
|
|
pygame.draw.rect(self.display, COLORS['black'],
|
|
|
|
(x * TILE_WIDTH, y * TILE_HEIGHT, TILE_WIDTH, TILE_HEIGHT))
|
|
|
|
pygame.draw.rect(self.display, color,
|
|
|
|
((x * TILE_WIDTH) + 1, (y * TILE_HEIGHT) + 1, TILE_WIDTH - 1, TILE_HEIGHT - 1))
|
|
|
|
|
2020-04-05 15:04:38 +02:00
|
|
|
def draw_packages(self):
|
2020-04-05 15:26:28 +02:00
|
|
|
def get_package_color(pack):
|
|
|
|
colors = {
|
|
|
|
PackStatus.LOOSE: COLORS['yellow'],
|
|
|
|
PackStatus.STORED: COLORS['lightgreen'],
|
|
|
|
PackStatus.STORED_BAD_LOCATION: COLORS['red']
|
|
|
|
}
|
|
|
|
return colors[pack.status]
|
|
|
|
|
2020-04-05 15:04:38 +02:00
|
|
|
for pack in self.warehouse_map.packages:
|
|
|
|
pack_x, pack_y = pack.lays_on_field.x_position, pack.lays_on_field.y_position
|
2020-04-05 15:26:28 +02:00
|
|
|
package_color = get_package_color(pack)
|
2020-04-05 15:04:38 +02:00
|
|
|
pygame.draw.rect(self.display, package_color,
|
|
|
|
((pack_x * TILE_WIDTH) + 3, (pack_y * TILE_HEIGHT) + 3, TILE_WIDTH - 5, TILE_HEIGHT - 5))
|
|
|
|
|
2020-04-04 22:03:15 +02:00
|
|
|
def draw_agent(self):
|
2020-04-30 16:36:55 +02:00
|
|
|
rotated = pygame.transform.rotate(self.agent_tex, DIRECTION_ANGLES.get(self.agent.direction))
|
|
|
|
self.display.blit(rotated, (self.agent.x*TILE_WIDTH, self.agent.y*TILE_WIDTH))
|
2020-04-02 15:25:15 +02:00
|
|
|
|
2020-04-05 15:04:38 +02:00
|
|
|
def set_starting_agent_position(self):
|
|
|
|
starting_x, starting_y = random.randrange(self.warehouse_map.width), random.randrange(self.warehouse_map.height)
|
2020-04-05 15:26:28 +02:00
|
|
|
while not isinstance(self.warehouse_map.tiles[starting_x][starting_y], warehouse.Tile) or self.warehouse_map.tiles[starting_x][starting_y].category.name != 'Floor':
|
2020-04-05 15:04:38 +02:00
|
|
|
starting_x, starting_y = random.randrange(self.warehouse_map.width), random.randrange(
|
|
|
|
self.warehouse_map.height)
|
|
|
|
return starting_x, starting_y
|
2020-04-23 19:43:51 +02:00
|
|
|
|
2020-04-02 15:25:15 +02:00
|
|
|
if __name__ == '__main__':
|
2020-04-04 22:03:15 +02:00
|
|
|
maingame = MainGameFrame()
|
2020-04-27 22:14:07 +02:00
|
|
|
maingame.run()
|