Fertilization Decision Tree
This commit is contained in:
Szymon Komosinski
parent
cab97a1370
commit
b9c8793a16
210
fertilization_decision_tree.py
Normal file
210
fertilization_decision_tree.py
Normal file
@ -0,0 +1,210 @@
|
||||
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()
|
||||
Loading…
Reference in New Issue
Block a user