diff --git a/drawUI.py b/drawUI.py index 1505326..a2d6644 100644 --- a/drawUI.py +++ b/drawUI.py @@ -5,12 +5,12 @@ import pygame def drawUI(board, display, tractor, direction, tillageUnit, field): - display.fill(WHITE) - makeField(board, display) - drawTractor(display, tractor.horizontal_index, tractor.vertical_index, direction) - drawInfo(display, tractor, tillageUnit, field) - - pygame.display.update() + for i in range(1, 11): + display.fill(WHITE) + makeField(board, display) + drawTractor(display, tractor.horizontal_index, tractor.vertical_index, direction, i) + drawInfo(display, tractor, tillageUnit, field) + pygame.display.update() def drawInfo(display, tractor, tillageUnit, field): @@ -54,14 +54,21 @@ def makeField(board, screen: pygame.Surface): screen.blit(do_zebrania, do_zebrania_rect) -def drawTractor(screen: pygame.Surface, tractor_horizontal_index, tractor_vertical_index, direction): +def drawTractor(screen: pygame.Surface, tractor_horizontal_index, tractor_vertical_index, direction, i): + tractor_pic = tractor_up + horizontal = tractor_horizontal_index * TILE_SIZE + vertical = tractor_vertical_index * TILE_SIZE if direction == "UP": - screen.blit(tractor_up, (tractor_horizontal_index * TILE_SIZE, tractor_vertical_index * TILE_SIZE)) - elif direction == "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)) + tractor_pic = tractor_up + vertical = vertical + (TILE_SIZE / i) + if direction == "DOWN": + tractor_pic = tractor_down + vertical = vertical - (TILE_SIZE / i) + if direction == "LEFT": + tractor_pic = tractor_left + horizontal = horizontal + (TILE_SIZE/i) + if direction == "RIGHT": + tractor_pic = tractor_right + horizontal = horizontal - (TILE_SIZE/i) + + screen.blit(tractor_pic, (horizontal, vertical))