gameboy_worlds.interface.pokemon.registry

 1from typing import Dict, Type
 2from gameboy_worlds.interface.controller import Controller
 3from gameboy_worlds.interface.environment import Environment, DummyEnvironment
 4from gameboy_worlds.interface.pokemon.environments import (
 5    PokemonEnvironment,
 6    PokemonRedChooseCharmanderEnvironment,
 7    PokemonRedChooseCharmanderEasyEnvironment,
 8    PokemonRedChooseCharmanderHardEnvironment,
 9    PokemonOCREnvironment,
10    PokemonTestEnvironment,
11    PokemonTrainEnvironment,
12)
13from gameboy_worlds.interface.pokemon.controllers import PokemonStateWiseController
14
15AVAILABLE_ENVIRONMENTS: Dict[str, Dict[str, Type[Environment]]] = {
16    "pokemon_red": {
17        "dummy": DummyEnvironment,
18        "default": PokemonOCREnvironment,
19        "basic": PokemonEnvironment,
20        "charmander": PokemonRedChooseCharmanderEnvironment,
21        "charmander_easy": PokemonRedChooseCharmanderEasyEnvironment,
22        "charmander_hard": PokemonRedChooseCharmanderHardEnvironment,
23        "train": PokemonTrainEnvironment,
24        "test": PokemonTestEnvironment,
25    },
26    "pokemon_brown": {
27        "default": PokemonOCREnvironment,
28        "basic": PokemonEnvironment,
29        "train": PokemonTrainEnvironment,
30        "test": PokemonTestEnvironment,
31    },
32    "pokemon_starbeasts": {
33        "default": PokemonOCREnvironment,
34        "basic": PokemonEnvironment,
35        "train": PokemonTrainEnvironment,
36        "test": PokemonTestEnvironment,
37    },
38    "pokemon_starbeasts_comet": {
39        "default": PokemonOCREnvironment,
40        "basic": PokemonEnvironment,
41        "train": PokemonTrainEnvironment,
42        "test": PokemonTestEnvironment,
43    },
44    "pokemon_crystal": {
45        "default": PokemonOCREnvironment,
46        "basic": PokemonEnvironment,
47        "train": PokemonTrainEnvironment,
48        "test": PokemonTestEnvironment,
49    },
50    "pokemon_prism": {
51        "default": PokemonOCREnvironment,
52        "basic": PokemonEnvironment,
53        "train": PokemonTrainEnvironment,
54        "test": PokemonTestEnvironment,
55    },
56    "pokemon_fools_gold": {
57        "default": PokemonOCREnvironment,
58        "basic": PokemonEnvironment,
59        "train": PokemonTrainEnvironment,
60        "test": PokemonTestEnvironment,
61    },
62}
63
64AVAILABLE_CONTROLLERS: Dict[str, Dict[str, Type[Controller]]] = {
65    "pokemon_red": {
66        "state_wise": PokemonStateWiseController,
67    },
68    "pokemon_brown": {
69        "state_wise": PokemonStateWiseController,
70    },
71    "pokemon_crystal": {
72        "state_wise": PokemonStateWiseController,
73    },
74    "pokemon_starbeasts": {
75        "state_wise": PokemonStateWiseController,
76    },
77    "pokemon_starbeasts_comet": {
78        "state_wise": PokemonStateWiseController,
79    },
80    "pokemon_prism": {
81        "state_wise": PokemonStateWiseController,
82    },
83    "pokemon_fools_gold": {
84        "state_wise": PokemonStateWiseController,
85    },
86}