Traktor/source/area/tractor.py

132 lines
5.0 KiB
Python
Raw Normal View History

from crop_protection_product import CropProtectionProduct
2024-04-14 14:28:40 +02:00
from area.constants import TILE_SIZE, DIRECTION_EAST, DIRECTION_SOUTH, DIRECTION_WEST, DIRECTION_NORTH, FIELD_WIDTH, FIELD_HEIGHT
2024-04-09 15:02:58 +02:00
from area.field import fieldX,fieldY
import pygame
2024-03-07 18:01:12 +01:00
class Tractor:
x = None
y = None
2024-04-10 01:52:13 +02:00
direction = None #direction takes values in the range of 1 to 4 (1->North, 2->East etc...)
2024-03-07 18:01:12 +01:00
image = None
cypermetryna = CropProtectionProduct("pests", "cereal")
diflufenikan = CropProtectionProduct("weeds", "cereal")
spirotetramat = CropProtectionProduct("pests", "fruit")
oksadiargyl = CropProtectionProduct("weeds", "fruit")
spinosad = CropProtectionProduct("pests", "vegetable")
metazachlor = CropProtectionProduct("weeds", "vegetable")
2024-03-07 18:01:12 +01:00
# etc
2024-04-09 15:02:58 +02:00
def __init__(self, x, y, direction):
2024-03-07 18:01:12 +01:00
self.x = x
self.y = y
2024-04-10 01:52:13 +02:00
self.rect = pygame.Rect(x, y, TILE_SIZE, TILE_SIZE)
2024-04-09 15:02:58 +02:00
self.direction = direction
2024-04-10 01:52:13 +02:00
self.image = pygame.image.load('resources/images/tractor_right.png').convert_alpha()
if (self.direction==1):
self.image = pygame.image.load('resources/images/tractor_up.png').convert_alpha()
elif (self.direction==3):
self.image = pygame.image.load('resources/images/tractor_down.png').convert_alpha()
elif (self.direction==4):
self.image = pygame.image.load('resources/images/tractor_left.png').convert_alpha()
2024-04-09 15:02:58 +02:00
def work_on_field(self, tile, ground, plant1):
if plant1 is None:
tile.randomizeContent()
# sprobuj zasadzic cos
2024-03-25 01:02:03 +01:00
print("Tarctor planted something")
elif plant1.growth_level == 100:
tile.plant = None
2024-03-25 01:02:03 +01:00
ground.nutrients_level -= 40
ground.water_level -= 40
print("Tractor collected something")
else:
2024-03-25 01:02:03 +01:00
plant1.try_to_grow(50,50) #mozna dostosowac jeszcze
ground.nutrients_level -= 11
ground.water_level -= 11
if ground.pest:
# traktor pozbywa sie szkodnikow
if plant1.plant_type == self.cypermetryna.plant_type:
t = "Tractor used Cypermetryna"
elif plant1.plant_type == self.spirotetramat.plant_type:
t = "Tractor used Spirotetramat"
elif plant1.plant_type == self.spinosad.plant_type:
t = "Tractor used Spinosad"
print(t)
ground.pest = False
if ground.weed:
# traktor pozbywa się chwastow
if plant1.plant_type == self.diflufenikan.plant_type:
t = "Tractor used Diflufenikan"
elif plant1.plant_type == self.oksadiargyl.plant_type:
t = "Tractor used Oksadiargyl"
elif plant1.plant_type == self.metazachlor.plant_type:
t = "Tractor used Metazachlor"
print(t)
ground.weed = False
2024-03-25 01:02:03 +01:00
if ground.water_level < plant1.water_requirements:
ground.water_level += 20
print("Tractor watered the plant")
if ground.nutrients_level < plant1.nutrients_requirements:
ground.nutrients_level += 20
print("Tractor added some nutrients")
2024-04-09 15:02:58 +02:00
def move(self):
if self.direction == DIRECTION_EAST:
2024-04-10 01:52:13 +02:00
#self.x += TILE_SIZE
self.rect.x += TILE_SIZE
2024-04-09 15:02:58 +02:00
elif self.direction == DIRECTION_WEST:
2024-04-10 01:52:13 +02:00
#self.x -= TILE_SIZE
self.rect.x -= TILE_SIZE
2024-04-09 15:02:58 +02:00
elif self.direction == DIRECTION_NORTH:
2024-04-10 01:52:13 +02:00
#self.y -= TILE_SIZE
self.rect.y -= TILE_SIZE
2024-04-09 15:02:58 +02:00
elif self.direction == DIRECTION_SOUTH:
2024-04-10 01:52:13 +02:00
#self.y += TILE_SIZE
self.rect.y += TILE_SIZE
2024-04-09 15:02:58 +02:00
def rotate_to_right(self):
if self.direction == 4:
self.direction = 1
else:
self.direction += 1
self.image = pygame.transform.rotate(self.image, -90)
def rotate_to_left(self):
if self.direction == 1:
self.direction = 4
else:
self.direction -= 1
self.image = pygame.transform.rotate(self.image, 90)
2024-04-14 14:28:40 +02:00
#sprawdza czy mozna sie dalej poruszyc, czy nie wyjdziemy poza mape - do funkcji succ w bfs
def can_it_move_node(node):
if node.get_direction() == DIRECTION_EAST and node.get_x() + TILE_SIZE < FIELD_WIDTH:
return "move east"
elif node.get_direction() == DIRECTION_WEST and node.get_x() - TILE_SIZE >=0:
return "move west"
elif node.get_direction() == DIRECTION_NORTH and node.get_y() - TILE_SIZE >=0:
return "move north"
elif node.get_direction() == DIRECTION_SOUTH and node.get_y() + TILE_SIZE < FIELD_HEIGHT:
return "move south"
else:
return False
2024-04-09 15:02:58 +02:00
def draw_tractor(self, win):
2024-04-10 01:52:13 +02:00
2024-04-09 15:02:58 +02:00
imageTractor = pygame.transform.scale(self.image, (TILE_SIZE, TILE_SIZE))
2024-04-10 01:52:13 +02:00
win.blit(imageTractor, (self.rect.x, self.rect.y))
pygame.display.flip()