AI_PRO/main.py

import Board
import FindPath
import TractorAction
import WeatherConditions
import convertToPrediction
import drawUI
import Graphsearch as graph
from Tractor import Tractor
from TractorLoad import TillageUnit
from constants import *
from manualSteering import manualSteeringDriver
import joblib
import pygame


pygame.init()


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

working = True
autoAction = True

animationSpeed = ANIMATION_PART

hitchCounter = 0
loadCounter = 0
toolCounter = - 1

board = Board.generate()

tractor = Tractor(horizontal_index=-0, vertical_index=0, hitch="nothing", header=False, autodrive=True,
                  direction="RIGHT")

tillageUnit = TillageUnit("Nothing")

tractor.turnOnEngine()
move_list = []

clock = pygame.time.Clock()

weather, day_time, temperature, wind, humidy = WeatherConditions.checkConditions()

weatherTimer = 0

while working:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            working = False
        if event.type == pygame.KEYDOWN:
            hitchCounter, loadCounter = \
                manualSteeringDriver(event, board, tractor, hitchCounter, tillageUnit, loadCounter)

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

    if weatherTimer == 2:
        weatherTimer = 0
        weather, day_time, temperature, wind, humidy = WeatherConditions.checkConditions()

    question = [convertToPrediction.convert(field.state, weather, day_time, temperature, wind, humidy)]

    dtree = joblib.load('treedata\\DecisionTree.pkl')

    decision = dtree.predict(question)

    if tractor.autodrive:
        if not move_list and decision == 1:
            field.state = TractorAction.changeFieldState(field, tractor)
            istate = graph.Istate(tractor.direction, tractor.horizontal_index, tractor.vertical_index)
            move_list = graph.graphsearch([], [], istate, FindPath.nearestLookingField(board, tractor, TillageUnit), board)
            print(move_list)

        elif move_list:
            tractor.auto_movement(move_list.pop(0))
            # TractorAction.changeFieldState(field, tractor)

        if field.horizontal_index == 0 and field.vertical_index == 0 and not move_list:
            tractor, tillageUnit, toolCounter = TractorAction.autoToolsChange(tractor, tillageUnit, toolCounter)

    drawUI.drawUI(board, display, tractor, tractor.direction, tillageUnit, field, animationSpeed, weather, day_time,
                  temperature, wind, humidy, decision)

    clock.tick(FPS)
    weatherTimer = weatherTimer + 1

pygame.quit()
quit()