AI_PRO/Board.py

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