46 lines
2.0 KiB
Python
46 lines
2.0 KiB
Python
from Main.constants import *
|
|
from Main.images import *
|
|
import pygame
|
|
|
|
|
|
def drawUI(board, display, tractor_horizontal_index, tractor_vertical_index, direction):
|
|
display.fill(WHITE)
|
|
makeField(board, display)
|
|
drawTractor(display, tractor_horizontal_index, tractor_vertical_index, direction)
|
|
pygame.display.update()
|
|
|
|
|
|
def makeField(board, screen: pygame.Surface):
|
|
for i in range(int(HORIZONTAL_TILES_NUMBER)):
|
|
for j in range(int(VERTICAL_TILES_NUMBER)):
|
|
field = board[i][j]
|
|
|
|
pos_x = i * TILE_SIZE + TILE_SIZE // 2
|
|
pos_y = j * TILE_SIZE + TILE_SIZE // 2
|
|
|
|
if field.state == 0:
|
|
do_podlania_rect.center = (pos_x, pos_y)
|
|
screen.blit(do_podlania, do_podlania_rect)
|
|
elif field.state == 1:
|
|
do_zaorania_rect.center = (pos_x, pos_y)
|
|
screen.blit(do_zaorania, do_zaorania_rect)
|
|
elif field.state == 2:
|
|
do_zasiania_rect.center = (pos_x, pos_y)
|
|
screen.blit(do_zasiania, do_zasiania_rect)
|
|
elif field.state == 3:
|
|
do_zebrania_rect.center = (pos_x, pos_y)
|
|
screen.blit(do_zebrania, do_zebrania_rect)
|
|
|
|
|
|
def drawTractor(screen: pygame.Surface, tractor_horizontal_index, tractor_vertical_index, direction):
|
|
if direction == "UP":
|
|
screen.blit(tractor_up, (tractor_horizontal_index * TILE_SIZE, tractor_vertical_index * TILE_SIZE))
|
|
elif direction is "DOWN":
|
|
screen.blit(tractor_down, (tractor_horizontal_index * TILE_SIZE, tractor_vertical_index * TILE_SIZE))
|
|
elif direction == "LEFT":
|
|
screen.blit(tractor_left, (tractor_horizontal_index * TILE_SIZE, tractor_vertical_index * TILE_SIZE))
|
|
elif direction == "RIGHT":
|
|
screen.blit(tractor_right, (tractor_horizontal_index * TILE_SIZE, tractor_vertical_index * TILE_SIZE))
|
|
else:
|
|
screen.blit(tractor_right, (tractor_horizontal_index * TILE_SIZE, tractor_vertical_index * TILE_SIZE))
|