Projekt_Sztuczna_Inteligencja/tile.py

23 lines
728 B
Python

from project_constants import Terrain as t
#Assume_cost function assumes colour as an argument (colour that is already given to a tile) and depending
#on what it is returns value of Terrein Enum
#It is used in Tile.cost (giving the value to the tile)
def assume_cost(terrain_type):
if (terrain_type == "CONCRETE"):
return t.CONCRETE
elif (terrain_type == "GRASS"):
return t.GRASS
elif (terrain_type == "MUD"):
return t.MUD
class Tile:
def __init__(self, position, terrain_type=None, mine=None):
self.position = position
self.terrain_type = terrain_type
self.cost = assume_cost(terrain_type)
# mine is an instance of Mine class
self.mine = mine