gameboy_worlds.interface.bomberman.registry

 1from typing import Dict, Type
 2
 3from gameboy_worlds.interface.bomberman.controllers import (
 4    BombermanMaxStateWiseController,
 5    BombermanPocketStateWiseController,
 6    BombermanQuestStateWiseController,
 7)
 8from gameboy_worlds.interface.bomberman.environments import (
 9    BombermanMaxEnvironment,
10    BombermanMaxTestEnvironment,
11    BombermanPocketEnvironment,
12    BombermanPocketTestEnvironment,
13    BombermanQuestEnvironment,
14    BombermanQuestTestEnvironment,
15)
16from gameboy_worlds.interface.controller import Controller
17from gameboy_worlds.interface.environment import DummyEnvironment, Environment
18
19
20AVAILABLE_ENVIRONMENTS: Dict[str, Dict[str, Type[Environment]]] = {
21    "bomberman_max": {
22        "dummy": DummyEnvironment,
23        "default": BombermanMaxEnvironment,
24        "test": BombermanMaxTestEnvironment,
25    },
26    "bomberman_pocket": {
27        "dummy": DummyEnvironment,
28        "default": BombermanPocketEnvironment,
29        "test": BombermanPocketTestEnvironment,
30    },
31    "bomberman_quest": {
32        "dummy": DummyEnvironment,
33        "default": BombermanQuestEnvironment,
34        "test": BombermanQuestTestEnvironment,
35    },
36}
37
38AVAILABLE_CONTROLLERS: Dict[str, Dict[str, Type[Controller]]] = {
39    "bomberman_max": {
40        "state_wise": BombermanMaxStateWiseController,
41    },
42    "bomberman_pocket": {
43        "state_wise": BombermanPocketStateWiseController,
44    },
45    "bomberman_quest": {
46        "state_wise": BombermanQuestStateWiseController,
47    },
48}