AI_PRO/main.py
2021-03-17 01:02:20 +01:00

82 lines
2.7 KiB
Python

import pygame
# wersja 1.02
from constants import *
from Main import Board
from Main.constants import *
pygame.init()
# display size in pixels
display = pygame.display.set_mode((DISPLAY_SIZE_HORIZONTAL, DISPLAY_SIZE_VERTICAL))
# program name
pygame.display.set_caption('Tryryryry')
working = True
tractor_horizontal_location = TRACTOR_HORIZONTAL_LOCATION
tractor_vertical_location = TRACTOR_VERTICAL_LOCATION
horizontal_change = 0
vertical_change = 0
change_tile = False
board = Board.generate()
color = BLACK
clock = pygame.time.Clock()
while working:
for event in pygame.event.get():
if event.type == pygame.QUIT:
working = False
if event.type == pygame.KEYDOWN:
print(tractor_horizontal_location, " ", tractor_vertical_location)
if event.key == pygame.K_LEFT and tractor_horizontal_location > 0:
horizontal_change = -TRACTOR_WIDTH
vertical_change = 0
elif event.key == pygame.K_RIGHT and tractor_horizontal_location < DISPLAY_SIZE_HORIZONTAL - TRACTOR_WIDTH:
horizontal_change = TRACTOR_WIDTH
vertical_change = 0
elif event.key == pygame.K_UP and tractor_vertical_location > 0:
vertical_change = -TRACTOR_HEIGHT
horizontal_change = 0
elif event.key == pygame.K_DOWN and tractor_vertical_location < DISPLAY_SIZE_VERTICAL - TRACTOR_HEIGHT:
vertical_change = TRACTOR_HEIGHT
horizontal_change = 0
elif event.key == pygame.K_SPACE:
change_tile = True
tractor_horizontal_location += horizontal_change
tractor_vertical_location += vertical_change
display.fill(WHITE)
for i in range(int(HORIZONTAL_TILES_NUMBER)):
for j in range(int(VERTICAL_TILES_NUMBER)):
if change_tile and i * TILE_SIZE == tractor_horizontal_location and j * TILE_SIZE == tractor_vertical_location:
board[i][j] = 4
change_tile = False
if board[i][j] == 0:
color = WHITE
elif board[i][j] == 1:
color = RED
elif board[i][j] == 2:
color = YELLOW
elif board[i][j] == 3:
color = GREEN
elif board[i][j] == 4:
color = BLACK
pygame.draw.rect(display, color,
[i * TILE_SIZE, j * TILE_SIZE, TILE_SIZE, TILE_SIZE])
pygame.draw.rect(display, BLACK,
[tractor_horizontal_location, tractor_vertical_location, TRACTOR_WIDTH, TRACTOR_HEIGHT])
pygame.display.update()
clock.tick(FPS)
horizontal_change = 0
vertical_change = 0
# commit test 2
pygame.quit()
quit()