165 lines
5.6 KiB
Python
165 lines
5.6 KiB
Python
import os
|
|
import random
|
|
|
|
from AI.decision_tree import *
|
|
from src.dimensions import *
|
|
from src.sprites import *
|
|
from src.colors import *
|
|
|
|
path = os.path.dirname(__file__) + "\\src\\test\\"
|
|
|
|
class Plant(pygame.sprite.Sprite):
|
|
def __init__(self, field, species):
|
|
self.species = species
|
|
if self.species == "tomato":
|
|
self.img0 = wheat_img_0
|
|
self.img1 = wheat_img_1
|
|
self.img2 = wheat_img_2
|
|
self.img3 = wheat_img_3
|
|
elif self.species == "potato":
|
|
self.img0 = potato_img_0
|
|
self.img1 = potato_img_1
|
|
self.img2 = potato_img_2
|
|
self.img3 = potato_img_3
|
|
elif self.species == "strawberry":
|
|
self.img0 = strawberry_img_0
|
|
self.img1 = strawberry_img_1
|
|
self.img2 = strawberry_img_2
|
|
self.img3 = strawberry_img_3
|
|
elif self.species == "pepper":
|
|
self.img0 = strawberry_img_0
|
|
self.img1 = strawberry_img_1
|
|
self.img2 = strawberry_img_2
|
|
self.img3 = strawberry_img_3
|
|
self.surf = self.img0
|
|
self.position = field.position
|
|
self.field = field
|
|
self.rect = self.surf.get_rect(
|
|
topleft=((MARGIN + WIDTH) * self.position[0] + MARGIN, (MARGIN + HEIGHT) * self.position[1] + MARGIN))
|
|
self.growth = 0
|
|
self.is_healthy = True
|
|
field.planted = True
|
|
self.tickscount = 0
|
|
self.ticks = 0
|
|
self.path = path + self.species + "\\"
|
|
self.testimage = self.path + random.choice(os.listdir(self.path))
|
|
|
|
def dtree(self):
|
|
if self.field.hydration == 4:
|
|
if self.is_healthy == 1:
|
|
if self.field.tractor_there == 0:
|
|
if self.ticks == 0:
|
|
return 0
|
|
elif self.ticks == 1:
|
|
return 1
|
|
elif self.field.tractor_there == 1:
|
|
return 0
|
|
elif self.is_healthy == 0:
|
|
return 0
|
|
elif self.field.hydration == 2:
|
|
if self.species == "pepper":
|
|
if self.ticks == 1:
|
|
if self.is_healthy == 1:
|
|
return 1
|
|
elif self.is_healthy == 0:
|
|
return 0
|
|
elif self.ticks == 0:
|
|
return 0
|
|
elif self.species == "potato":
|
|
return 0
|
|
elif self.species == "tomato":
|
|
return 0
|
|
elif self.species == "strawberry":
|
|
return 0
|
|
elif self.field.hydration == 1:
|
|
if self.species == "potato":
|
|
return 0
|
|
elif self.species == "strawberry":
|
|
if self.ticks == 1:
|
|
return -1
|
|
elif self.ticks == 0:
|
|
return 0
|
|
elif self.species == "tomato":
|
|
return 0
|
|
elif self.species == "pepper":
|
|
if self.is_healthy == 0:
|
|
return 0
|
|
elif self.is_healthy == 1:
|
|
if self.field.tractor_there == 0:
|
|
if self.ticks == 0:
|
|
return 0
|
|
elif self.ticks == 1:
|
|
return 1
|
|
elif self.field.tractor_there == 1:
|
|
return 0
|
|
elif self.field.hydration == 3:
|
|
if self.ticks == 1:
|
|
if self.field.tractor_there == 0:
|
|
if self.is_healthy == 1:
|
|
if self.species == "potato":
|
|
if self.field.fertility == 1:
|
|
return 1
|
|
elif self.field.fertility == 0:
|
|
return 0
|
|
elif self.species == "strawberry":
|
|
return 1
|
|
elif self.species == "pepper":
|
|
return 1
|
|
elif self.species == "tomato":
|
|
return 1
|
|
elif self.is_healthy == 0:
|
|
return 0
|
|
elif self.field.tractor_there == 1:
|
|
return 0
|
|
elif self.ticks == 0:
|
|
return 0
|
|
elif self.field.hydration == 5:
|
|
if self.field.tractor_there == 1:
|
|
return 0
|
|
elif self.field.tractor_there == 0:
|
|
if self.is_healthy == 0:
|
|
return 0
|
|
elif self.is_healthy == 1:
|
|
if self.ticks == 1:
|
|
return 1
|
|
elif self.ticks == 0:
|
|
return 0
|
|
elif self.field.hydration == 0:
|
|
if self.ticks == 0:
|
|
return 0
|
|
elif self.ticks == 1:
|
|
return -1
|
|
|
|
def update(self):
|
|
if self.growth == 0:
|
|
self.surf = self.img0
|
|
if self.growth == 1:
|
|
self.surf = self.img1
|
|
if self.growth == 2:
|
|
self.surf = self.img2
|
|
if self.growth == 3:
|
|
self.surf = self.img3
|
|
|
|
def grow(self):
|
|
if self.dtree() == 1:
|
|
self.growth += 1
|
|
self.ticks = 0
|
|
elif self.dtree() == -1:
|
|
self.growth -= 1
|
|
self.ticks = 0
|
|
if self.growth > 4:
|
|
self.growth = 4
|
|
if self.growth < 0:
|
|
self.growth = 0
|
|
|
|
self.update()
|
|
|
|
def tick(self):
|
|
self.tickscount += 1
|
|
if self.tickscount >= 25:
|
|
self.tickscount = 0
|
|
self.ticks = 1
|
|
|
|
def remove(self):
|
|
self.field.planted = False
|