1from typing import List
 2
 3from hcraft.elements import Item, Zone
 4from hcraft.task import GetItemTask
 5from hcraft.transformation import Transformation, Use, Yield, PLAYER, CURRENT_ZONE
 6
 7from hcraft.examples.minicraft.minicraft import MiniCraftEnv
 8
 9MINICRAFT_NAME = "MultiRoom"
10__doc__ = MiniCraftEnv.description(MINICRAFT_NAME, for_module_header=True)
11
12
13class MiniHCraftMultiRoom(MiniCraftEnv):
14    MINICRAFT_NAME = MINICRAFT_NAME
15    __doc__ = MiniCraftEnv.description(MINICRAFT_NAME)
16
17    GOAL = Item("goal")
18    """Goal to reach."""
19
20    def __init__(self, n_rooms: int = 6, **kwargs) -> None:
21        self.rooms = [Zone(f"Room {i + 1}") for i in range(n_rooms)]
22        self.task = GetItemTask(self.GOAL)
23        super().__init__(
24            self.MINICRAFT_NAME,
25            purpose=self.task,
26            start_zone=self.rooms[0],
27            **kwargs,
28        )
29
30    def build_transformations(self) -> List[Transformation]:
31        transformations = []
32        find_goal = Transformation(
33            "Find goal",
34            inventory_changes=[Yield(CURRENT_ZONE, self.GOAL, max=0)],
35            zone=self.rooms[-1],
36        )
37        transformations.append(find_goal)
38
39        reach_goal = Transformation(
40            "Reach goal",
41            inventory_changes=[
42                Use(CURRENT_ZONE, self.GOAL, consume=1),
43                Yield(PLAYER, self.GOAL),
44            ],
45        )
46        transformations.append(reach_goal)
47
48        for i, room in enumerate(self.rooms):
49            connected_rooms: List[Zone] = []
50            if i > 0:
51                connected_rooms.append(self.rooms[i - 1])
52            if i < len(self.rooms) - 1:
53                connected_rooms.append(self.rooms[i + 1])
54            for connected_room in connected_rooms:
55                transformations.append(
56                    Transformation(
57                        f"Go to {room.name} from {connected_room.name}",
58                        destination=room,
59                        zone=connected_room,
60                    )
61                )
62
63        return transformations