main.py

from time import time
from json import load
from functools import cmp_to_key
from src.spellcast import Spellcast
from src.searchnode import SearchNode

start_time = time()

config = load(open("config.json"))

game = Spellcast()
game.load_file("board.txt")

print("searching for moves...")

def compare_moves(a: SearchNode, b: SearchNode):
    a_score = a.estimated_long_term_score(game)
    b_score = b.estimated_long_term_score(game)

    difference = a_score - b_score
    if difference == 0:
        return a.gem_count() - b.gem_count()
    else:
        return difference


best_moves = game.legal_moves(
    cmp_to_key(compare_moves)
    if config["gemManagement"]
    else SearchNode.score
)

shuffle_score, shuffle_recommended = game.evaluate_shuffle(best_moves[0])

if shuffle_recommended:
    print(f"engine recommends shuffling - {shuffle_score} estimated value")

print("\n".join([
    f"{i + 1} > {node.to_string(game)}"
    for i, node in enumerate(
        best_moves[:config["movesShown"]]
    )
]))

print(
    f"found {len(best_moves)} words"
    + f" in {round(time() - start_time, 2)} seconds."
)