IntelligentTractor/main2.py

46 lines
1.1 KiB
Python
Raw Normal View History

import pygame
import sys
import tractor
import src.screen as screen
import src.plant as plant
class Tractor2(pygame.sprite.Sprite):
def __init__(self,engine,transmission,pos_x,pos_y):
super.__init__()
self.image=pygame.image.load("../assets/tractor/tractor.png")
self.image=pygame.transform.scale(self,(36,36))
self.rect = self.image.get_rect()
self.engine=engine
self.transmission=transmission
self.pos_x=pos_x
self.pos_y=pos_y
self.fuel=100
def update(self):
self.rect.center=pygame.mouse.get_pos()
# self.pos_x=pos_x
# self.pos_y=pos_y
# pygame initialization
pygame.init()
pygame.mouse.set_visible(False)
myTractor = tractor.Tractor
if __name__ == "__main__":
running = True
while running:
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
pygame.quit()
sys.exit()
# defines agent movement
tractor.movement(myTractor)
# screen visualisation
screen.set_screen(myTractor)