211 lines
6.1 KiB
Python
211 lines
6.1 KiB
Python
|
import pygame as pg
|
||
|
|
import sys
|
||
|
|
import numpy as np
|
||
|
|
from tile import Tile
|
||
|
|
from config import *
|
||
|
|
from agent import Agent
|
||
|
|
from Astar import *
|
||
|
|
|
||
|
|
done = False
|
||
|
|
|
||
|
|
clock = pg.time.Clock()
|
||
|
|
pg.display.set_caption('Intelligent Tractor')
|
||
|
|
pg.init()
|
||
|
|
pg.font.init()
|
||
|
|
|
||
|
|
|
||
|
|
class Game:
|
||
|
|
def __init__(self):
|
||
|
|
self.tiles = []
|
||
|
|
self.water_lvl = []
|
||
|
|
self.day = 1
|
||
|
|
|
||
|
|
for y in range(0, 9):
|
||
|
|
for x in range(0, 9):
|
||
|
|
pos = (x, y)
|
||
|
|
t = Tile(background[x][y], objects[x][y], pos)
|
||
|
|
t.water_usage()
|
||
|
|
self.tiles.append(t)
|
||
|
|
|
||
|
|
def display_background(self):
|
||
|
|
for tile in self.tiles:
|
||
|
|
screen.blit(bg_textures[tile.ground], (tile.x * tile_size, tile.y * tile_size))
|
||
|
|
|
||
|
|
def display_objects(self):
|
||
|
|
for tile in self.tiles:
|
||
|
|
screen.blit(ob_textures[tile.object], (tile.x * tile_size, tile.y * tile_size))
|
||
|
|
|
||
|
|
def display_legend(self):
|
||
|
|
screen.blit(black_surface, (0, 720))
|
||
|
|
self.water_lvl = []
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.pos == (1, 1):
|
||
|
|
rose_w = tile.w
|
||
|
|
rose_f = tile.f
|
||
|
|
self.water_lvl.append((rose_w, (0, 3)))
|
||
|
|
if tile.pos == (7, 1):
|
||
|
|
orchid_w = tile.w
|
||
|
|
orchid_f = tile.f
|
||
|
|
self.water_lvl.append((orchid_w, (5, 0)))
|
||
|
|
if tile.pos == (1, 7):
|
||
|
|
tulip_w = tile.w
|
||
|
|
tulip_f = tile.f
|
||
|
|
self.water_lvl.append((tulip_w, (3, 8)))
|
||
|
|
if tile.pos == (7, 7):
|
||
|
|
sunflower_w = tile.w
|
||
|
|
sunflower_f = tile.f
|
||
|
|
self.water_lvl.append((sunflower_w, (8, 5)))
|
||
|
|
font = pg.font.SysFont('ubuntumono', 40)
|
||
|
|
text = font.render("water levels", 1, (255, 255, 255))
|
||
|
|
text1 = font.render("roses: " + str(rose_w) + "% " + str(rose_f), 1, (255, 255, 255))
|
||
|
|
text2 = font.render("orchids: " + str(orchid_w) + "% " + str(orchid_f), 1, (255, 255, 255))
|
||
|
|
text3 = font.render("tulips: " + str(tulip_w) + "% " + str(tulip_f), 1, (255, 255, 255))
|
||
|
|
text4 = font.render("sunflowers: " + str(sunflower_w) + "%" + str(sunflower_f), 1, (255, 255, 255))
|
||
|
|
text5 = font.render("day: " + str(self.day), 1, (255, 255, 255))
|
||
|
|
screen.blit(text, (10, 720))
|
||
|
|
screen.blit(text1, (10, 760))
|
||
|
|
screen.blit(text2, (360, 760))
|
||
|
|
screen.blit(text3, (10, 800))
|
||
|
|
screen.blit(text4, (360, 800))
|
||
|
|
screen.blit(text5, (10, 840))
|
||
|
|
|
||
|
|
def water_target(self):
|
||
|
|
o = None
|
||
|
|
for lvl in self.water_lvl:
|
||
|
|
if lvl[0] < 10:
|
||
|
|
o = lvl[1]
|
||
|
|
break
|
||
|
|
return o
|
||
|
|
|
||
|
|
def watering(self, plants):
|
||
|
|
if plants == (0, 3):
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == rose:
|
||
|
|
tile.w = 100
|
||
|
|
if plants == (5, 0):
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == orchid:
|
||
|
|
tile.w = 100
|
||
|
|
if plants == (3, 8):
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == tulip:
|
||
|
|
tile.w = 100
|
||
|
|
if plants == (8, 5):
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == sunflower:
|
||
|
|
tile.w = 100
|
||
|
|
|
||
|
|
def fertilizing(self, soil):
|
||
|
|
if soil == (0, 3):
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == rose:
|
||
|
|
if tile.w < 50:
|
||
|
|
tile.w = 100
|
||
|
|
tile.f = "OK"
|
||
|
|
if soil == (5, 0):
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == orchid:
|
||
|
|
if tile.w < 50:
|
||
|
|
tile.w = 100
|
||
|
|
tile.f = "OK"
|
||
|
|
if soil == (3, 8):
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == tulip:
|
||
|
|
if tile.w < 50:
|
||
|
|
tile.w = 100
|
||
|
|
tile.f = "OK"
|
||
|
|
if soil == (8, 5):
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == sunflower:
|
||
|
|
if tile.w < 50:
|
||
|
|
tile.w = 100
|
||
|
|
tile.f = "OK"
|
||
|
|
|
||
|
|
def fertilization_target(self):
|
||
|
|
f = []
|
||
|
|
if self.day % 2 == 0:
|
||
|
|
fert_sunflowers = (8, 5)
|
||
|
|
f.append(fert_sunflowers)
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == sunflower:
|
||
|
|
tile.f = "NO"
|
||
|
|
if self.day % 5 == 0:
|
||
|
|
fert_orchids = (5, 0)
|
||
|
|
f.append(fert_orchids)
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == orchid:
|
||
|
|
tile.f = "NO"
|
||
|
|
if self.day % 10 == 0:
|
||
|
|
fert_roses = (0, 3)
|
||
|
|
f.append(fert_roses)
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == rose:
|
||
|
|
tile.f = "NO"
|
||
|
|
if self.day % 20 == 0:
|
||
|
|
fert_tulip = (3, 8)
|
||
|
|
f.append(fert_tulip)
|
||
|
|
for tile in self.tiles:
|
||
|
|
if tile.object == tulip:
|
||
|
|
tile.f = "NO"
|
||
|
|
return f
|
||
|
|
|
||
|
|
def add_day(self):
|
||
|
|
pg.time.wait(100)
|
||
|
|
self.day = self.day + 1
|
||
|
|
for tile in self.tiles:
|
||
|
|
tile.w = tile.w - tile.wu
|
||
|
|
if tile.w < 0:
|
||
|
|
tile.w = 0
|
||
|
|
display_farm()
|
||
|
|
|
||
|
|
|
||
|
|
def watering_paths():
|
||
|
|
path = astar(background, a.pos, g.water_target())
|
||
|
|
while len(path) > 1:
|
||
|
|
display_farm()
|
||
|
|
a.display(path[1])
|
||
|
|
del(path[1])
|
||
|
|
else:
|
||
|
|
g.watering(path[0])
|
||
|
|
|
||
|
|
|
||
|
|
def fertilization_ways():
|
||
|
|
targets = g.fertilization_target()
|
||
|
|
for one in targets:
|
||
|
|
way = astar(background, a.pos, one)
|
||
|
|
while len(way) > 1:
|
||
|
|
display_farm()
|
||
|
|
a.display(way[1])
|
||
|
|
del(way[1])
|
||
|
|
else:
|
||
|
|
g.fertilizing(one)
|
||
|
|
g.add_day()
|
||
|
|
|
||
|
|
|
||
|
|
def display_farm():
|
||
|
|
g.display_background()
|
||
|
|
g.display_objects()
|
||
|
|
g.display_legend()
|
||
|
|
|
||
|
|
def game_loop():
|
||
|
|
display_farm()
|
||
|
|
if g.water_target() is None:
|
||
|
|
if g.fertilization_target():
|
||
|
|
fertilization_ways()
|
||
|
|
else:
|
||
|
|
g.add_day()
|
||
|
|
else:
|
||
|
|
watering_paths()
|
||
|
|
clock.tick(60)
|
||
|
|
|
||
|
|
|
||
|
|
g = Game()
|
||
|
|
a = Agent()
|
||
|
|
while not done:
|
||
|
|
for event in pg.event.get():
|
||
|
|
if event.type == pg.QUIT:
|
||
|
|
done = True
|
||
|
|
game_loop()
|
||
|
|
|
||
|
|
pg.quit()
|