Traktor/source/tile.py

59 lines
1.6 KiB
Python

import random
import os
import numpy as np
import matplotlib.pyplot as plt
# path to plant images folder (used in randomize_photo function)
folder_path = "resources/images/plant_photos"
class Tile:
x = 0
y = 0
plant = None
ground = None
photo = None
image = None
# after checking a photo add information plant object ("potato"/ "wheat"/"none")
# add Ground instance
def __init__(self, x, y):
self.x = x
self.y = y
# add to init ground instance
# called on a created tile
# choose random photo from the folder:
def randomize_photo(self):
random_photo = random.choice(os.listdir(folder_path))
photo_path = os.path.join(folder_path, random_photo)
return photo_path
def randomizeContent(self):
photo_path = self.randomize_photo()
i = random.randint(1, 3) # szansa 1/3
if i == 1:
self.image = "resources/images/sampling.png"
self.photo = photo_path
# self.photo losuje dodatkowo zdjecie jakies rosliny do danego rzędu
else:
self.image = "resources/images/dirt.png"
self.photo = "resources/images/background.jpg"
ground = self.ground
if ground.obstacle and self.x != 170 and self.y != 100: #warunek po and do zmiany
self.image = "resources/images/rock_dirt.png"
def displayPhoto(self):
img = self.photo
img = img / 2 + 0.5 # unnormalize
npimg = img.numpy()
plt.imshow(np.transpose(npimg, (1, 2, 0)))
plt.show()