gameboy_worlds.interface.harvest_moon.environments

 1from typing import Optional, Dict, Any, List, Tuple
 2
 3from gymnasium import spaces
 4
 5from gameboy_worlds.emulation.harvest_moon.base_metrics import CoreHarvestMoonMetrics
 6from gameboy_worlds.utils import load_parameters, log_dict, log_info
 7from gameboy_worlds.emulation.emulator import Emulator
 8from gameboy_worlds.emulation.harvest_moon.trackers import (
 9    CoreHarvestMoonTracker,
10    HarvestMoonOCRTracker,
11)
12from gameboy_worlds.interface.environment import (
13    DummyEnvironment,
14    Environment,
15    TestEnvironmentMixin,
16    TrainEnvironmentMixin,
17)
18from gameboy_worlds.interface.controller import Controller
19
20import gymnasium as gym
21import numpy as np
22
23
24class HarvestMoonEnvironment(DummyEnvironment):
25    """
26    A basic Harvest Moon Environment.
27    """
28
29    REQUIRED_EMULATOR = Emulator
30    REQUIRED_STATE_TRACKER = CoreHarvestMoonMetrics
31
32
33class HarvestMoonOCREnvironment(HarvestMoonEnvironment):
34    """
35    A Harvest Moon Environment that includes OCR region captures and agent state.
36    """
37
38    REQUIRED_STATE_TRACKER = HarvestMoonOCRTracker
39    REQUIRED_EMULATOR = Emulator
40
41    @staticmethod
42    def override_emulator_kwargs(emulator_kwargs: dict) -> dict:
43        Environment.override_state_tracker_class(
44            emulator_kwargs, HarvestMoonOCREnvironment.REQUIRED_STATE_TRACKER
45        )
46        return emulator_kwargs
47
48
49class HarvestMoonTestEnvironment(TestEnvironmentMixin, HarvestMoonOCREnvironment):
50    pass
51
52
53class HarvestMoonTrainEnvironment(TrainEnvironmentMixin, HarvestMoonOCREnvironment):
54    pass