52 lines
2.4 KiB
Python
52 lines
2.4 KiB
Python
import pygame, random
|
|
from request import Request
|
|
|
|
DIRT_IMG = pygame.image.load("dirt.jpg")
|
|
DIRT = pygame.transform.scale(DIRT_IMG, (50, 50))
|
|
GRASS_IMG = pygame.image.load("grass.png")
|
|
GRASS = pygame.transform.scale(GRASS_IMG, (50, 50))
|
|
SAND_IMG = pygame.image.load("sand.jpeg")
|
|
SAND = pygame.transform.scale(SAND_IMG, (50, 50))
|
|
COBBLE_IMG = pygame.image.load("cobble.jpeg")
|
|
COBBLE = pygame.transform.scale(COBBLE_IMG, (50, 50))
|
|
|
|
def randomize_map(): # tworzenie mapy z losowymi polami
|
|
request_list = []
|
|
field_array_1 = []
|
|
field_array_2 = []
|
|
imgpath_array = [[0 for x in range(16)] for x in range(16)]
|
|
field_priority = []
|
|
for i in range(16):
|
|
temp_priority = []
|
|
for j in range(16):
|
|
if i in (0, 1) and j in (0, 1):
|
|
field_array_2.append(GRASS)
|
|
temp_priority.append(1)
|
|
else:
|
|
prob = random.uniform(0, 100)
|
|
if 0 <= prob <= 12:
|
|
garbage_type = random.choice(['glass', 'mixed', 'paper', 'plastic'])
|
|
garbage_image_number = random.randrange(1, 100)
|
|
GARBAGE_IMG = pygame.image.load(f"./model_training/test_dataset/{garbage_type}/{garbage_type} ({str(garbage_image_number)}).jpg")
|
|
GARBAGE = pygame.transform.scale(GARBAGE_IMG, (50, 50))
|
|
field_array_2.append(GARBAGE)
|
|
#field_array_2.append(COBBLE)
|
|
imgpath_array[i][j] = (f"./model_training/test_dataset/{garbage_type}/{garbage_type} ({str(garbage_image_number)}).jpg")
|
|
|
|
temp_priority.append(100)
|
|
request_list.append(Request(
|
|
i*50,j*50, #lokacja
|
|
random.randint(0,3), #typ śmieci
|
|
random.random(), #objętość śmieci
|
|
random.randint(0,30), #ostatni odbiór
|
|
random.randint(0,1), #czy opłacone w terminie
|
|
random.random() * 10, #intensywność odoru
|
|
random.random() * 50 #waga śmieci
|
|
))
|
|
else:
|
|
field_array_2.append(GRASS)
|
|
temp_priority.append(1)
|
|
field_array_1.append(field_array_2)
|
|
field_array_2 = []
|
|
field_priority.append(temp_priority)
|
|
return field_array_1, field_priority, request_list, imgpath_array |