diff --git a/plant.py b/plant.py new file mode 100644 index 0000000..d297b5d --- /dev/null +++ b/plant.py @@ -0,0 +1,44 @@ +import pygame +from colors import * +from dimensions import * +from sprites import * + +class Plant(pygame.sprite.Sprite): + def __init__(self, field, species): + super(Plant, self).__init__() + self.surf = plant_img_0 + 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.isHealthy = True + self.species = species + if self.species == "beetroot": + self.growth_speed = 1.5 + self.humidity_needed = 2 + elif self.species == "wheat": + self.growth_speed = 1 + self.humidity_needed = 1 + elif self.species == "cotton": + self.growth_speed = 0.8 + self.humidity_needed = 1 + + + def grow(self): + if self.field.hydration >= self.humidity_needed: + self.growth += self.growth_speed + if self.field.hydration == 0: + self.growth -= self.growth_speed + if self.growth > 4: + self.growth = 4 + if self.growth < 0: + self.growth = 0 + if self.growth == 0: + self.surf = plant_img_0 + elif self.growth == 1: + self.surf = plant_img_1 + elif self.growth == 2: + self.surf = plant_img_2 + elif self.growth == 3: + self.surf = plant_img_3