spirecomm/main.py at main · matthewReff/spirecomm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
import random
import time
import os
import logging
from pathlib import Path

from spirecomm.ai.nnAgent import NnAgent
from spirecomm.communication.coordinator import Coordinator
from spirecomm.ai.agent import Agent
from spirecomm.spire.character import PlayerClass


def main():
    logging.basicConfig(filename="neuralNet.log", level=logging.INFO)

    # Add a checkpoint directory to resume training or load a trained model
    checkpoint_directory = None
    # checkpoint_directory = Path("checkpoints") / "{Timestamp}/{Snapshot}"

    agent: Agent = NnAgent(
        chosen_class=PlayerClass.IRONCLAD, checkpoint_directory=checkpoint_directory
    )
    coordinator = Coordinator()
    coordinator.signal_ready()
    coordinator.register_command_error_callback(agent.handle_error)
    coordinator.register_state_change_callback(agent.get_next_action_in_game)
    coordinator.register_out_of_game_callback(agent.get_next_action_out_of_game)

    # We're running an AI, it doesn't make sense to play anything other than defect
    chosenClass = PlayerClass.IRONCLAD
    agent.change_class(chosenClass)
    while True:
        seed = str(int(time.time()))
        save_name = "SlayAI"
        random.seed(seed)

        logging.info("Starting run playing " + str(chosenClass) + " with seed " + seed)
        results = coordinator.climb_till_defeat(chosenClass, seed)

        folder_name = save_name + "-" + str(chosenClass) + "-" + seed
        copy_run_files(results, chosenClass, folder_name)


def get_class_folder_name(chosenClass: PlayerClass) -> str:
    if chosenClass == PlayerClass.IRONCLAD:
        return "1_IRONCLAD"
    elif chosenClass == PlayerClass.THE_SILENT:
        return "1_THE_SILENT"
    elif chosenClass == PlayerClass.DEFECT:
        return "1_DEFECT"


def copy_run_files(results, chosenClass, folder_name):
    # Quick and dirty grab of run files for a given ascension streak
    # We assume that the CWD is the SlayTheSpire game folder
    repo_base = os.path.join(os.getcwd(), "..")
    game_path = os.path.join(repo_base, "SlayTheSpire")
    mod_path = os.path.join(repo_base, "Mods", "spirecomm")

    # We are located in the SlayTheSpire directory by default
    game_runs_path = os.path.join(game_path, "runs", get_class_folder_name(chosenClass))

    mod_runs_path = os.path.join(mod_path, "runs")
    mod_specific_runs_path = os.path.join(mod_runs_path, folder_name)

    logging.info("Creating runs folder in mod folder")
    try:
        Path(mod_runs_path).mkdir(parents=True, exist_ok=True)
        logging.debug("Created path: " + mod_runs_path, "continuing...")
    except FileExistsError:
        logging.debug("Path already exists: " + mod_runs_path, "continuing...")

    logging.info("Creating specific run folder in mod runs folder")
    try:
        Path(mod_specific_runs_path).mkdir(parents=True, exist_ok=True)
        logging.debug("Created path: " + mod_specific_runs_path, "continuing...")
    except FileExistsError:
        logging.debug("Path already exists: " + mod_specific_runs_path, "continuing...")

    try:
        logging.info("Copying from game runs folder to mod runs folder")
        run_files = os.listdir(game_runs_path)

        run_files.sort()
        run_file_names = run_files[-len(results) :]

        for file in run_file_names:
            with open(os.path.join(game_runs_path, file), "r") as source_data:
                with open(os.path.join(mod_specific_runs_path, file), "w") as dest_data:
                    dest_data.write(source_data.read())
    except OSError as e:
        logging.error(
            "Ran into a problem while copying run information from game to mod folder: "
            + e
        )


if __name__ == "__main__":
    main()