AI_PRO/main.py

import pygame
# wersja 1.05

from Main import Board, driving, drawUI
from Main.Tractor import Tractor
from Main.TractorAction import action
from Main.TractorLoad import TillageUnit
from Main.constants import *

pygame.init()

display = pygame.display.set_mode((DISPLAY_SIZE_HORIZONTAL, DISPLAY_SIZE_VERTICAL))
pygame.display.set_caption('Tractor')

working = True
cruiseControl = True
autoAction = True
autonomousDrive = False
lastDirection = "RIGHT"

horizontal_change = 1
vertical_change = 0

direction = "RIGHT"

hitchCounter = 0
loadCounter = 0

board = Board.generate()

tractor = Tractor(horizontal_index=0, vertical_index=0, hitch="nothing", header=False)
tillageUnit = TillageUnit("Nothing")

tractor.turnOnEngine()

clock = pygame.time.Clock()

while working:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            working = False
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_SPACE:
                field = board[tractor.horizontal_index][tractor.vertical_index]
                field.state = action(field, tractor, tillageUnit)
            if event.key == pygame.K_q:
                hitchCounter = (hitchCounter + 1) % 3
                if hitchCounter == 0:
                    tractor.hitch = "Crop Trailer"
                if hitchCounter == 1:
                    print(type(tillageUnit))
                    print(isinstance(tillageUnit, TillageUnit))
                    tractor.hitch = tillageUnit
                if hitchCounter == 2:
                    tractor.hitch = "Nothing"
            if event.key == pygame.K_w:
                if tractor.header:
                    tractor.header = False
                else:
                    tractor.header = True
            if event.key == pygame.K_e:
                loadCounter = (loadCounter + 1) % 4
                if loadCounter == 0:
                    tillageUnit.load = "Nothing"
                elif loadCounter == 1:
                    tillageUnit.load = "Seeds"
                elif loadCounter == 2:
                    tillageUnit.load = "Water"
                elif loadCounter == 3:
                    tillageUnit.load = "Fertilizer"

            direction = driving.manualTurning(event, tractor)

    if direction == lastDirection:
        tractor.turn(driving.cruiseControl(tractor, direction, cruiseControl))

    print(tractor.horizontal_index, " ", tractor.vertical_index)

    field = board[tractor.horizontal_index][tractor.vertical_index]

    if autoAction:
        field.state = action(field, tractor, tillageUnit)

    if direction != "STOP":
        lastDirection = direction
        drawUI.drawUI(board, display, tractor, direction, tillageUnit, field)
    else:
        drawUI.drawUI(board, display, tractor, lastDirection, tillageUnit, field)

    if autonomousDrive:
        if vertical_change == 1:
            vertical_change = 0
            if tractor.vertical_index % 2 == 0:
                horizontal_change = 1
            else:
                horizontal_change = -1
        elif tractor.horizontal_index == HORIZONTAL_TILES_NUMBER - 1:
            vertical_change = 1
        elif tractor.horizontal_index == 0 and tractor.vertical_index != 0:
            vertical_change = 1


    clock.tick(FPS)
    tractor.reduce_fuel()

pygame.quit()
quit()