Spaces:
Runtime error
Runtime error
File size: 6,435 Bytes
32d7589 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 |
class BasicLevelTranslator:
def __init__(self):
self.player_direction_map = {
72: "facing left",
40: "facing left, climbing down ladder/rope",
24: "facing left, climbing up ladder/rope",
128: "facing right",
32: "facing right, climbing down ladder/rope",
16: "facing right, climbing up ladder/rope",
}
def translate(self, state):
(
room_number, player_x, player_y, player_direction, enemy_skull_x, enemy_skull_y,
key_monster_x, key_monster_y, level, num_lives, items_in_inventory_count,
room_state, score_0, score_1, score_2
) = state
player_dir = self.player_direction_map.get(player_direction, "unknown direction")
picked_up_items = "None"
if items_in_inventory_count > 0:
items = [
("Key", "Opens locked doors.", 1),
("Ankh", "Freeze enemies.", 2),
("Gem", "Extra bonus points.", 4),
("Torch", "Lights up dark rooms.", 8),
("Sword", "Vanquishes certain enemies.", 16),
]
picked_up_items = ""
for name, desc, val in items:
if items_in_inventory_count & val == val:
picked_up_items += f"{name} ({desc}), "
picked_up_items = picked_up_items[:-2]
res = f"""Room Number: {room_number}
Player Position: ({player_x}, {player_y})
Player Direction: {player_dir}
Enemy Skull Position: ({enemy_skull_x}, {enemy_skull_y})
Key Monster Position: ({key_monster_x}, {key_monster_y})
Level: {level}
Remaining Lives: {num_lives}
Items in Inventory: {picked_up_items if picked_up_items else "None"}
Room State (Mapped Based on Room Number): {room_state}
Current Score: {score_0}{score_1}{score_2}\n"""
return res
class GameDescriber:
def __init__(self, args):
self.is_only_local_obs = args.is_only_local_obs == 1
self.max_episode_len = args.max_episode_len
self.action_desc_dict = {
}
self.reward_desc_dict = {
}
def describe_goal(self):
return ("The goal is to guide PANAMA JOE safely to reach Montezuma's fantastic treasure. "
"Avoid danger, collect special tools and rewards, and navigate through the chambers of the emperor's fortress.")
def describe_game(self):
return ("""In Montezuma's Revenge, you control a fearless adventurer named PANAMA JOE who aims to navigate through a maze
of death-dealing chambers within Emperor Montezuma's fortress. PANAMA JOE can walk, climb, and jump in the game. In each room of the
maze, there are several dangers, including various creatures such as skulls, snakes, spiders, and bats, as well as several deadly room
fixtures like fire pits, conveyor belts, disappearing floors, laser gates, floor spikes, and laser walls.
PANAMA JOE can act on several elements within the game environment. Some items in the game are:
1. Keys: Essential to open locked doors, allowing access to other rooms and deeper exploration.
2. Ankhs: Freeze all Killer Creatures in the room for 6.5 seconds, during which they can't move or kill.
3. Gems: Extra bonus points when collected.
4. Torches: Light up dark rooms, making it easier to navigate through threats.
5. Swords: Used to defeat certain enemies, by contact with the tip of the sword.
The game's ultimate goal is to reach the fabulous Treasure Room containing Montezuma's treasure while amassing as many points as
possible and keeping PANAMA JOE alive through the challenges. The game ends when you lose all of your PANAMA JOEs, with a maximum
of 6 lives.""")
def translate_terminate_state(self, state, episode_len, max_episode_len):
return ""
def translate_potential_next_state(self, state, action):
return ""
def describe_action(self):
actions = {
1: "No Operation",
2: "Move Right",
3: "Move Left",
4: "Move Down",
5: "Move Up",
6: "Move Right + Climb Down",
7: "Move Left + Climb Down",
8: "Move Right + Climb Up",
9: "Move Left + Climb Up",
10: "Jump",
11: "Jump Right",
12: "Jump Left",
13: "Jump Down",
14: "Jump Up",
15: "Jump Right + Climb Down",
16: "Jump Left + Climb Down",
17: "Jump Right + Climb Up",
18: "Jump Left + Climb Up",
}
description = "Your Next Move:\n"
for action_number, action_name in actions.items():
description += f"{action_number}: {action_name}\n"
description += "Please choose an action from the list above."
return description
class BasicStateSequenceTranslator(BasicLevelTranslator):
def __init__(self):
super().__init__()
def translate(self, infos, is_current=False):
descriptions = []
if is_current:
state_desc = BasicLevelTranslator().translate(infos[-1]['state'])
return state_desc
for i, info in enumerate(infos):
assert 'state' in info, "info should contain state information"
state_desc = BasicLevelTranslator().translate(info['state'])
action_desc = f'Take Action: {["No Operation", "Move Right", "Move Left", "Move Down", "Move Up", "Move Right + Climb Down", "Move Left + Climb Down", "Move Right + Climb Up", "Move Left + Climb Up", "Jump","Jump Right", "Jump Left", "Jump Down", "Jump Up", "Jump Right + Climb Down", "Jump Left + Climb Down", "Jump Right + Climb Up", "Jump Left + Climb Up"][info["action"]]} ({info["action"]}).'
reward_desc = f"Result: Reward of {info['reward']}"
next_state_desc = BasicLevelTranslator().translate(info['next_state'])
descriptions.append(f"{state_desc}\n"
f"{action_desc}\n"
f"{reward_desc}\n"
f"Transit to {next_state_desc}\n")
return descriptions
|