gameboy_worlds.interface.deja_vu.controllers

 1from gameboy_worlds.utils import log_error
 2from gameboy_worlds.interface.deja_vu.actions import (
 3    TickUntilStable,
 4    MoveCursor,
 5    MoveGrid,
 6    OpenButtonMenu,
 7    CloseButtonMenu,
 8    SelectMenuOption,
 9    AdvanceDialogue,
10)
11from gameboy_worlds.interface.controller import Controller
12from gameboy_worlds.interface.action import HighLevelAction
13from gameboy_worlds.emulation.deja_vu.parsers import AgentState
14from typing import Dict
15
16
17class DejaVuStateWiseController(Controller):
18    """
19    High-level controller for Deja Vu. Only supports:
20    - TickUntilStable: tick until screen is stable
21    - MoveCursor: move menu cursor
22    - MoveGrid: grid movement
23    - OpenButtonMenu: open bottom menu
24    - CloseButtonMenu: close bottom menu
25    - SelectMenuOption: select menu option
26    - AdvanceDialogue: advance dialogue
27    """
28
29    ACTIONS = [
30        TickUntilStable,
31        MoveCursor,
32        MoveGrid,
33        OpenButtonMenu,
34        CloseButtonMenu,
35        SelectMenuOption,
36        AdvanceDialogue,
37    ]
38
39    def string_to_high_level_action(self, input_str):
40        input_str = input_str.lower().strip()
41        if "(" not in input_str or ")" not in input_str:
42            return None, None
43        action_name = input_str.split("(")[0].strip()
44        action_args_str = input_str.split("(")[1].split(")")[0].strip()
45
46        if action_name == "tickuntilstable":
47            return TickUntilStable, {}
48        if action_name == "movecursor":
49            direction = action_args_str.strip()
50            if direction in ["up", "down", "left", "right"]:
51                return MoveCursor, {"direction": direction}
52            return None, None
53        if action_name == "movegrid":
54            try:
55                x, y = map(int, action_args_str.split(","))
56                return MoveGrid, {"x_steps": x, "y_steps": y}
57            except Exception:
58                return None, None
59        if action_name == "openbuttonmenu":
60            return OpenButtonMenu, {}
61        if action_name == "closebuttonmenu":
62            return CloseButtonMenu, {}
63        if action_name == "selectmenuoption":
64            return SelectMenuOption, {}
65        if action_name == "advancedialogue":
66            return AdvanceDialogue, {}
67        return None, None
68
69    def get_action_strings(self, return_all: bool = False):
70        return {
71            TickUntilStable: "tickuntilstable(): Tick until screen is stable.",
72            MoveCursor: "movecursor(<up,down,left,right>): Move menu cursor.",
73            MoveGrid: "movegrid(<x:int>,<y:int>): Grid movement.",
74            OpenButtonMenu: "openbuttonmenu(): Open bottom menu.",
75            CloseButtonMenu: "closebuttonmenu(): Close bottom menu.",
76            SelectMenuOption: "selectmenuoption(): Select current menu option.",
77            AdvanceDialogue: "advancedialogue(): Advance dialogue.",
78        }