forked from s452751/AI_PRO
28 lines
891 B
Python
28 lines
891 B
Python
import random
|
|
|
|
from Field import Field
|
|
from constants import *
|
|
|
|
states = ['toPlow', 'toSeed', 'toFertilize', 'toWater', 'toCut']
|
|
|
|
|
|
def generate():
|
|
board = []
|
|
for i in range(0, int(HORIZONTAL_TILES_NUMBER)):
|
|
board.append([])
|
|
for j in range(0, int(VERTICAL_TILES_NUMBER)):
|
|
current_state = random.choice(states)
|
|
current_cost = getCost(current_state)
|
|
board[i].append(Field(int(i * TILE_SIZE), int(j * TILE_SIZE), current_state, current_cost))
|
|
|
|
board[0][0] = Field(int(0 * TILE_SIZE), int(0 * TILE_SIZE), "TOOLS_FIELD", cost=0)
|
|
board[1][0] = Field(int(1 * TILE_SIZE), int(0 * TILE_SIZE), "FUEL_FIELD", cost=0)
|
|
return board
|
|
|
|
def getCost(state):
|
|
if state == 'toPlow': return 1
|
|
if state == 'toSeed': return 2
|
|
if state == 'toFertilize': return 3
|
|
if state == 'toWater': return 4
|
|
if state == 'toCut': return 5
|