forked from s452751/AI_PRO
89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
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()
|