68 lines
2.8 KiB
Python
68 lines
2.8 KiB
Python
import definitions
|
|
import field
|
|
import plant
|
|
import pygame
|
|
import random
|
|
import soil
|
|
import tractor
|
|
pygame.display.set_caption("Smart Tractor")
|
|
def draw_window(tractor1_rectangle):
|
|
definitions.WIN.fill(definitions.WHITE)
|
|
definitions.WIN.blit(definitions.TRACTOR, (tractor1_rectangle.x, tractor1_rectangle.y))
|
|
pygame.display.update()
|
|
def is_move_allowed(move, tractor1_rectangle):
|
|
if ((move == 1) and (tractor1_rectangle.y + definitions.VEL + definitions.TRACTOR_HEIGHT <= definitions.HEIGHT)):
|
|
return True
|
|
elif ((move == 2) and (tractor1_rectangle.x - definitions.VEL >= 0)):
|
|
return True
|
|
elif ((move == 3) and (tractor1_rectangle.x + definitions.VEL + definitions.TRACTOR_WIDTH <= definitions.WIDTH)):
|
|
return True
|
|
elif ((move == 4) and (tractor1_rectangle.y - definitions.VEL >= 0)):
|
|
return True
|
|
else:
|
|
return False
|
|
def tractor1_handle_movement(tractor1, tractor1_rectangle):
|
|
random1 = random.randint(1, 4)
|
|
if ((random1 == 1) and (is_move_allowed(1, tractor1_rectangle) == True)):
|
|
tractor1.move_down()
|
|
tractor1_rectangle.x = tractor1.get_x()
|
|
tractor1_rectangle.y = tractor1.get_y()
|
|
elif ((random1 == 2) and (is_move_allowed(2, tractor1_rectangle) == True)):
|
|
tractor1.move_left()
|
|
tractor1_rectangle.x = tractor1.get_x()
|
|
tractor1_rectangle.y = tractor1.get_y()
|
|
elif ((random1 == 3) and (is_move_allowed(3, tractor1_rectangle) == True)):
|
|
tractor1.move_right()
|
|
tractor1_rectangle.x = tractor1.get_x()
|
|
tractor1_rectangle.y = tractor1.get_y()
|
|
elif ((random1 == 4) and (is_move_allowed(4, tractor1_rectangle) == True)):
|
|
tractor1.move_up()
|
|
tractor1_rectangle.x = tractor1.get_x()
|
|
tractor1_rectangle.y = tractor1.get_y()
|
|
def main():
|
|
fields = []
|
|
a = 0
|
|
b = 0
|
|
for i in range(100):
|
|
plant1 = plant.Plant(1, definitions.WHEAT_MAXIMUM_STATE, "Wheat", "None", definitions.WHEAT_REQUIRED_WATER_LEVEL)
|
|
soil1 = soil.Soil(True, definitions.MAXIMUM_WATER_LEVEL)
|
|
if (a > definitions.WIDTH - definitions.VEL):
|
|
a = 0
|
|
b = b + definitions.VEL
|
|
field1 = field.Field(a, b, plant1, soil1)
|
|
a = a + definitions.VEL;
|
|
fields.append(field1)
|
|
tractor1 = tractor.Tractor(400, 400)
|
|
tractor1_rectangle = pygame.Rect(tractor1.get_x(), tractor1.get_y(), definitions.TRACTOR_WIDTH, definitions.TRACTOR_HEIGHT)
|
|
clock = pygame.time.Clock()
|
|
run = True
|
|
while run:
|
|
clock.tick(definitions.FPS)
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
run = False
|
|
tractor1_handle_movement(tractor1, tractor1_rectangle)
|
|
draw_window(tractor1_rectangle)
|
|
pygame.quit()
|
|
if __name__ == "__main__":
|
|
main() |